95 lines
2.2 KiB
Go
Executable File
95 lines
2.2 KiB
Go
Executable File
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
|
|
}
|
|
|
|
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)
|
|
public := router.Group("/")
|
|
public.Handle(http.MethodGet, "/_health", hc.GetHealth)
|
|
|
|
public.Handle("LOGIN", "/users", hc.ConnectUser)
|
|
// start: user routes
|
|
public.Handle(http.MethodGet, "/users", hc.GetAllUsers)
|
|
public.Handle(http.MethodPost, "/users", hc.CreateUser)
|
|
public.Handle(http.MethodGet, "/users/:id", hc.GetUser)
|
|
public.Handle(http.MethodPut, "/users/:id", hc.UpdateUser)
|
|
public.Handle(http.MethodDelete, "/users/:id", hc.DeleteUser)
|
|
// end: user routes
|
|
|
|
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
|
|
}
|