clean up

handlers_v2/handler.go

@@ -0,0 +1,126 @@
+package handlers
+
+import (
+	"hamster-tycoon/middlewares"
+	"hamster-tycoon/service"
+	"hamster-tycoon/storage/dao"
+	"hamster-tycoon/storage/dao/fake"
+	"hamster-tycoon/storage/dao/postgresql"
+	"hamster-tycoon/storage/validators"
+	"net/http"
+	"time"
+
+	"reflect"
+	"strings"
+
+	"github.com/gin-contrib/cors"
+	"github.com/gin-gonic/gin"
+	"gopkg.in/go-playground/validator.v9"
+)
+
+type Config struct {
+	Mock            bool
+	DBConnectionURI string
+	Port            int
+	LogLevel        string
+	LogFormat       string
+}
+
+type handlersContext struct {
+	db             dao.Database
+	validator      *validator.Validate
+	userService    *service.UserService
+	gameService    *service.GameService
+	cageService    *service.CageService
+	hamsterService *service.HamsterService
+}
+
+func NewRouter(config *Config) *gin.Engine {
+	gin.SetMode(gin.ReleaseMode)
+
+	router := gin.New()
+	router.HandleMethodNotAllowed = true
+
+	router.Use(cors.New(cors.Config{
+		AllowOrigins:     []string{"*"},
+		AllowMethods:     []string{"*"},
+		AllowHeaders:     []string{"*"},
+		ExposeHeaders:    []string{"*"},
+		AllowCredentials: true,
+		MaxAge:           12 * time.Hour,
+	}))
+	router.Use(gin.Recovery())
+	router.Use(middlewares.GetLoggerMiddleware())
+	router.Use(middlewares.GetHTTPLoggerMiddleware())
+
+	hc := &handlersContext{}
+	if config.Mock {
+		hc.db = fake.NewDatabaseFake()
+	} else {
+		hc.db = postgresql.NewDatabasePostgreSQL(config.DBConnectionURI)
+	}
+	hc.validator = newValidator()
+	hc.userService = service.NewUserService(hc.db)
+	hc.gameService = service.NewGameService(hc.db)
+	hc.cageService = service.NewCageService(hc.db)
+	hc.hamsterService = service.NewHamsterService(hc.db)
+
+	public := router.Group("/")
+	public.Handle(http.MethodGet, "/_health", hc.GetHealth)
+
+	userRoute := public.Group("/users")
+	userRoute.Handle("LOGIN", "", hc.connectUser)
+	userRoute.Handle(http.MethodPost, "", hc.createUser)
+
+	securedUserRoute := userRoute.Group("")
+	//TODO add secure auth
+	securedUserRoute.Handle(http.MethodGet, "", hc.getAllUsers)
+	securedUserRoute.Handle(http.MethodGet, "/:userID", hc.getUser)
+	securedUserRoute.Handle(http.MethodPut, "/:userID", hc.updateUser)
+	securedUserRoute.Handle(http.MethodDelete, "/:userID", hc.deleteUser)
+	// end: user routes
+
+	gameRoute := securedUserRoute.Group("/:userID/games")
+	gameRoute.Handle(http.MethodGet, "", hc.getAllGames)
+	gameRoute.Handle(http.MethodGet, "/:gameID", hc.getAGame)
+	gameRoute.Handle(http.MethodPost, "/:gameID", hc.createAGame)
+	gameRoute.Handle(http.MethodPut, "/:gameID", hc.updateAGame)
+	gameRoute.Handle(http.MethodGet, "/:gameID/hamsters", hc.getHamstersOfGame)
+	gameRoute.Handle(http.MethodDelete, "/:gameID", hc.deleteAGame)
+
+	cageRoute := gameRoute.Group("/:gameID/cages")
+	cageRoute.Handle(http.MethodGet, "", hc.getAllCages)
+	cageRoute.Handle(http.MethodGet, "/:cageID", hc.getACage)
+	cageRoute.Handle(http.MethodPost, "/:cageID", hc.createACage)
+	cageRoute.Handle(http.MethodPut, "/:cageID", hc.updateACage)
+	cageRoute.Handle(http.MethodDelete, "/:cageID", hc.deleteACage)
+
+	hamsterRoute := cageRoute.Group("/:cageID/hamsters")
+	hamsterRoute.Handle(http.MethodGet, "", hc.getAllHamsters)
+	hamsterRoute.Handle(http.MethodGet, "/:hamsterID", hc.getAHamster)
+	hamsterRoute.Handle(http.MethodPost, "/:hamsterID", hc.createAHamster)
+	hamsterRoute.Handle(http.MethodPut, "/:hamsterID", hc.updateAHamster)
+	hamsterRoute.Handle(http.MethodDelete, "/:hamsterID", hc.deleteAHamster)
+
+	return router
+}
+
+func newValidator() *validator.Validate {
+	va := validator.New()
+
+	va.RegisterTagNameFunc(func(fld reflect.StructField) string {
+		name := strings.SplitN(fld.Tag.Get("json"), ",", 2)
+		if len(name) < 1 {
+			return ""
+		}
+		return name[0]
+	})
+
+	for k, v := range validators.CustomValidators {
+		if v.Validator != nil {
+			va.RegisterValidationCtx(k, v.Validator)
+		}
+	}
+
+	return va
+}