mirror of
https://github.com/twofas/2fas-server.git
synced 2024-11-22 02:10:38 +01:00
8c1bac3d90
* feat: remove admin ep from public server --------- Co-authored-by: Tobiasz Heller <14020794+tobiaszheller@users.noreply.github.com>
89 lines
2.5 KiB
Go
89 lines
2.5 KiB
Go
package api
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-playground/validator/v10"
|
|
"github.com/twofas/2fas-server/config"
|
|
extension "github.com/twofas/2fas-server/internal/api/browser_extension/service"
|
|
health "github.com/twofas/2fas-server/internal/api/health/service"
|
|
icons "github.com/twofas/2fas-server/internal/api/icons/service"
|
|
mobile "github.com/twofas/2fas-server/internal/api/mobile/service"
|
|
support "github.com/twofas/2fas-server/internal/api/support/service"
|
|
"github.com/twofas/2fas-server/internal/common/api"
|
|
"github.com/twofas/2fas-server/internal/common/db"
|
|
"github.com/twofas/2fas-server/internal/common/redis"
|
|
"github.com/twofas/2fas-server/internal/common/validation"
|
|
)
|
|
|
|
var validate *validator.Validate
|
|
|
|
type Module interface {
|
|
RegisterPublicRoutes(router *gin.Engine)
|
|
RegisterAdminRoutes(g *gin.RouterGroup)
|
|
}
|
|
|
|
type Application struct {
|
|
Addr string
|
|
Router *gin.Engine
|
|
Config config.Configuration
|
|
Modules []Module
|
|
HealthModule *health.HealthModule
|
|
}
|
|
|
|
func NewApplication(applicationName string, config config.Configuration) *Application {
|
|
validate = validator.New()
|
|
|
|
gorm := db.NewGormConnection(config)
|
|
database := db.NewDbConnection(config)
|
|
redisClient := redis.New(config.Redis.ServiceUrl, config.Redis.Port)
|
|
|
|
validate.RegisterValidation("not_blank", validation.NotBlank)
|
|
|
|
h := health.NewHealthModule(applicationName, config, redisClient)
|
|
|
|
modules := []Module{
|
|
h,
|
|
support.NewSupportModule(config, gorm, database, validate),
|
|
icons.NewIconsModule(config, gorm, database, validate),
|
|
extension.NewBrowserExtensionModule(config, gorm, database, redisClient, validate),
|
|
mobile.NewMobileModule(config, gorm, database, validate, redisClient),
|
|
}
|
|
|
|
app := &Application{
|
|
Addr: config.App.ListenAddr,
|
|
Config: config,
|
|
Modules: modules,
|
|
HealthModule: h,
|
|
}
|
|
|
|
return app
|
|
}
|
|
|
|
func (a *Application) RegisterRoutes(router *gin.Engine) {
|
|
router.NoRoute(func(c *gin.Context) {
|
|
c.JSON(404, api.NotFoundError(errors.New("URI not found")))
|
|
})
|
|
|
|
for _, module := range a.Modules {
|
|
module.RegisterPublicRoutes(router)
|
|
}
|
|
}
|
|
|
|
func (a *Application) RegisterAdminRoutes(router *gin.Engine) {
|
|
router.NoRoute(func(c *gin.Context) {
|
|
c.JSON(404, api.NotFoundError(errors.New("URI not found")))
|
|
})
|
|
|
|
// The only route method is /health. Everything else
|
|
// is nested under /admin so that oAuth proxy can route to it.
|
|
a.HealthModule.RegisterHealth(router)
|
|
|
|
g := router.Group("/admin")
|
|
|
|
for _, module := range a.Modules {
|
|
module.RegisterAdminRoutes(g)
|
|
}
|
|
}
|