nginx-ui/server/router/routers.go

package router

import (
    "encoding/base64"
    "github.com/0xJacky/Nginx-UI/server/api"
    "github.com/0xJacky/Nginx-UI/server/model"
    "github.com/gin-gonic/gin"
    "net/http"
)

func authRequired() gin.HandlerFunc {
    return func(c *gin.Context) {
        token := c.GetHeader("Authorization")
        if token == "" {
            tmp, _ := base64.StdEncoding.DecodeString(c.Query("token"))
            token = string(tmp)
            if token == "" {
                c.JSON(http.StatusForbidden, gin.H{
                    "message": "auth fail",
                })
                c.Abort()
                return
            }
        }

        n := model.CheckToken(token)

        if n < 1 {
            c.JSON(http.StatusForbidden, gin.H{
                "message": "auth fail",
            })
            c.Abort()
            return
        }
        c.Next()
    }
}

func InitRouter() *gin.Engine {
	r := gin.New()
	r.Use(gin.Logger())

	r.Use(gin.Recovery())

	r.GET("/", func(c *gin.Context) {
		c.JSON(http.StatusOK, gin.H{
			"message": "Hello World",
		})
	})
    r.GET("/analytic", api.Analytic)
	r.POST("/login", api.Login)
    r.DELETE("/logout", api.Logout)

	g := r.Group("/", authRequired())
	{
	    g.GET("/users", api.GetUsers)
	    g.GET("/user/:id", api.GetUser)
	    g.POST("/user", api.AddUser)
	    g.POST("/user/:id", api.EditUser)
        g.DELETE("/user/:id", api.DeleteUser)

		g.GET("domains", api.GetDomains)
		g.GET("domain/:name", api.GetDomain)
		g.POST("domain/:name", api.EditDomain)
		g.POST("domain/:name/enable", api.EnableDomain)
		g.POST("domain/:name/disable", api.DisableDomain)
		g.DELETE("domain/:name", api.DeleteDomain)

		g.GET("configs", api.GetConfigs)
		g.GET("config/:name", api.GetConfig)
		g.POST("config", api.AddConfig)
		g.POST("config/:name", api.EditConfig)

		g.GET("backups", api.GetFileBackupList)
		g.GET("backup/:id", api.GetFileBackup)

        g.GET("template/:name", api.GetTemplate)

		g.GET("cert/issue/:domain", api.IssueCert)
        g.GET("cert/:domain/info", api.CertInfo)
	}

	return r
}