mirror of
https://github.com/yhl452493373/frps-panel.git
synced 2026-04-04 14:27:00 +08:00
55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
package controller
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
type HandleController struct {
|
|
CommonInfo CommonInfo
|
|
Tokens map[string]TokenInfo
|
|
Version string
|
|
ConfigFile string
|
|
TokensFile string
|
|
}
|
|
|
|
func NewHandleController(config *HandleController) *HandleController {
|
|
return config
|
|
}
|
|
|
|
func (c *HandleController) Register(rootDir string, engine *gin.Engine) {
|
|
assets := filepath.Join(rootDir, "assets")
|
|
_, err := os.Stat(assets)
|
|
if err != nil && !os.IsExist(err) {
|
|
assets = "./assets"
|
|
}
|
|
|
|
engine.Delims("${", "}")
|
|
engine.LoadHTMLGlob(filepath.Join(assets, "templates/*"))
|
|
engine.POST("/handler", c.MakeHandlerFunc())
|
|
engine.Static("/static", filepath.Join(assets, "static"))
|
|
engine.GET("/lang.json", c.MakeLangFunc())
|
|
engine.GET(LoginUrl, c.MakeLoginFunc())
|
|
engine.POST(LoginUrl, c.MakeLoginFunc())
|
|
engine.GET(LogoutUrl, c.MakeLogoutFunc())
|
|
|
|
var group *gin.RouterGroup
|
|
if len(c.CommonInfo.AdminUser) != 0 {
|
|
//group = engine.Group("/", gin.BasicAuthForRealm(gin.Accounts{
|
|
// c.CommonInfo.User: c.CommonInfo.Pwd,
|
|
//}, "Restricted"))
|
|
group = engine.Group("/", c.BasicAuth())
|
|
} else {
|
|
group = engine.Group("/")
|
|
}
|
|
group.GET("/", c.MakeIndexFunc())
|
|
group.GET("/tokens", c.MakeQueryTokensFunc())
|
|
group.POST("/add", c.MakeAddTokenFunc())
|
|
group.POST("/update", c.MakeUpdateTokensFunc())
|
|
group.POST("/remove", c.MakeRemoveTokensFunc())
|
|
group.POST("/disable", c.MakeDisableTokensFunc())
|
|
group.POST("/enable", c.MakeEnableTokensFunc())
|
|
group.GET("/proxy/*serverApi", c.MakeProxyFunc())
|
|
}
|