99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package server
|
|
|
|
import (
|
|
"net/http"
|
|
"path"
|
|
"reCoreD-UI/controllers"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"github.com/sirupsen/logrus"
|
|
swaggerfiles "github.com/swaggo/files"
|
|
ginSwagger "github.com/swaggo/gin-swagger"
|
|
)
|
|
|
|
const (
|
|
apiPrefix = "/api"
|
|
metricPrefix = "/metrics"
|
|
swaggerPrefix = "/swagger"
|
|
)
|
|
|
|
func (s *Server) setupRoute() {
|
|
username, password, err := controllers.GetAdmin()
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
logrus.Debugf("got %s:%s", username, password)
|
|
|
|
server := s.webServer.Group(s.prefix)
|
|
|
|
swaggerHandler := server
|
|
if s.debug {
|
|
swaggerHandler.GET(path.Join(swaggerPrefix, "*any"), ginSwagger.WrapHandler(swaggerfiles.Handler))
|
|
} else {
|
|
swaggerHandler.GET(path.Join(swaggerPrefix, "*any"), func(ctx *gin.Context) {
|
|
ctx.HTML(http.StatusNotFound, "", nil)
|
|
})
|
|
}
|
|
|
|
controllers.RegisterMetrics()
|
|
metricHandler := server
|
|
metricHandler.GET(metricPrefix, func(ctx *gin.Context) {
|
|
if err := controllers.RefreshMetrics(); err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
promhttp.Handler().ServeHTTP(ctx.Writer, ctx.Request)
|
|
})
|
|
|
|
apiHandler := server
|
|
|
|
groupV1 := apiHandler.Group(apiPrefix, gin.BasicAuth(gin.Accounts{
|
|
username: password,
|
|
}), func(ctx *gin.Context) {
|
|
_, ok := ctx.Get(gin.AuthUserKey)
|
|
if !ok {
|
|
ctx.AbortWithStatusJSON(http.StatusUnauthorized, Response{
|
|
Succeed: false,
|
|
})
|
|
}
|
|
}).Group("/v1")
|
|
|
|
domains := groupV1.Group("/domains")
|
|
domains.
|
|
GET("/", getDomains).
|
|
POST("/", createDomain).
|
|
PUT("/", updateDomain).
|
|
DELETE("/:id", deleteDomain)
|
|
|
|
records := groupV1.Group("/records")
|
|
records.
|
|
GET("/:domain", getRecords).
|
|
POST("/:domain", createRecord).
|
|
POST("/:domain/bulk", createRecords).
|
|
PUT("/:domain", updateRecord).
|
|
DELETE("/:domain/:id", deleteRecord)
|
|
|
|
/*server := s.webServer.Group(s.prefix)
|
|
server.Use(apiHandler.HandleContext, metricHandler.HandleContext, func(ctx *gin.Context) {
|
|
uri := ctx.Request.RequestURI
|
|
logrus.Debug(uri)
|
|
switch {
|
|
case strings.HasPrefix(uri, path.Join(s.prefix, apiPrefix)):
|
|
//apiHandler.HandleContext(ctx)
|
|
case strings.HasPrefix(uri, path.Join(s.prefix, metricPrefix)):
|
|
//metricHandler.HandleContext(ctx)
|
|
case strings.HasPrefix(uri, path.Join(s.prefix, swaggerPrefix)):
|
|
if s.debug {
|
|
swaggerHandler.HandleContext(ctx)
|
|
} else {
|
|
ctx.HTML(http.StatusNotFound, "", nil)
|
|
}
|
|
default:
|
|
staticFileHandler()(ctx)
|
|
}
|
|
})*/
|
|
|
|
server.GET("/", staticFileHandler())
|
|
server.GET("/assets/*any", staticFileHandler())
|
|
}
|