44 lines
1.0 KiB
Go
44 lines
1.0 KiB
Go
package router
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"senlinai-agent/backend/internal/config"
|
|
"senlinai-agent/backend/internal/logic/auth"
|
|
)
|
|
|
|
type RouteRegistrar interface {
|
|
Register(router gin.IRouter)
|
|
}
|
|
|
|
func NewRouter(cfg config.Config, registrars ...RouteRegistrar) *gin.Engine {
|
|
return newRouter(cfg, nil, registrars...)
|
|
}
|
|
|
|
func NewProtectedRouter(cfg config.Config, tokenVerifier func(string) (uint, error), registrars ...RouteRegistrar) *gin.Engine {
|
|
return newRouter(cfg, tokenVerifier, registrars...)
|
|
}
|
|
|
|
func newRouter(cfg config.Config, tokenVerifier func(string) (uint, error), registrars ...RouteRegistrar) *gin.Engine {
|
|
if cfg.Env == "test" {
|
|
gin.SetMode(gin.TestMode)
|
|
}
|
|
|
|
router := gin.New()
|
|
router.Use(gin.Recovery())
|
|
|
|
router.GET("/healthz", func(c *gin.Context) {
|
|
c.JSON(http.StatusOK, gin.H{"status": "ok"})
|
|
})
|
|
api := router.Group("/api")
|
|
if tokenVerifier != nil {
|
|
api.Use(auth.RequireUser(tokenVerifier))
|
|
}
|
|
for _, registrar := range registrars {
|
|
registrar.Register(api)
|
|
}
|
|
|
|
return router
|
|
}
|