nginx-ui/server/routers/router.go

package routers

import (
	"fmt"
	"github.com/astaxie/beego"
	"github.com/astaxie/beego/context"
	"github.com/astaxie/beego/logs"
	"nginx-ui/server/base"
	config2 "nginx-ui/server/config"
	"nginx-ui/server/constants"
	"nginx-ui/server/middleware"
	"nginx-ui/server/modules/agent_server"
	"nginx-ui/server/modules/auth_token"
	"nginx-ui/server/modules/ldap"
	"nginx-ui/server/modules/nginx/nginx_controller"
	"nginx-ui/server/modules/oauth2"
	"nginx-ui/server/modules/proxy"
	"nginx-ui/server/modules/settings"
	"nginx-ui/server/modules/user"
	"nginx-ui/server/modules/wechat"
	"nginx-ui/server/utils"
	"regexp"
	"strings"
)

var NginxR = "/nginx"
var NginxGetR = "/nginx/:id"
var NginxRefreshR = "/nginx/:id/http/refresh"
var NginxStartR = "/nginx/:id/start"
var NginxStopR = "/nginx/:id/stop"
var NginxStatusR = "/nginx/:id/status"

func init() {
	config := config2.Config

	userController := user.NewUserController()

	logs.Info("baseApi", config.BaseApi)

	agentServer := agent_server.NewServer()

	ns := beego.NewNamespace(config.BaseApi,
		beego.NSRouter(NginxR, &nginx_controller.NginxController{}),
		beego.NSRouter(NginxGetR, &nginx_controller.NginxController{}, "post:Update"),
		beego.NSRouter(NginxGetR, &nginx_controller.NginxController{}, "get:GetNginx"),
		beego.NSRouter(NginxGetR, &nginx_controller.NginxController{}, "delete:DelNginx"),
		beego.NSRouter(NginxRefreshR, &nginx_controller.NginxController{}, "post:RefreshHttp"),
		beego.NSRouter(NginxStartR, &nginx_controller.NginxController{}, "post:StartNginx"),
		beego.NSRouter(NginxStopR, &nginx_controller.NginxController{}, "post:StopNginx"),
		beego.NSRouter(NginxStatusR, &nginx_controller.NginxController{}, "post:StatusNginx"),
		// certs
		beego.NSRouter("/nginx/:id/certs", &nginx_controller.CertController{}),
		beego.NSRouter("/nginx/:id/certs/sync", &nginx_controller.CertController{}, "post:Sync"),
		// nginx server apis
		beego.NSRouter("/nginx/:id/server", &nginx_controller.ServerController{}),
		beego.NSRouter("/nginx/:id/server/refresh", &nginx_controller.ServerController{}, "post:Refresh"),
		// file upload download
		beego.NSRouter("/nginx/:id/file/deploy", &nginx_controller.FileController{}, "post:Deploy"),
		beego.NSRouter("/file", &nginx_controller.FileController{}),
		beego.NSRouter(constants.DownloadFileUrl, &nginx_controller.FileController{}, "get:Download"),
		beego.NSRouter("/logger", &nginx_controller.LoggerController{}),
		//	角色
		beego.NSRouter("/role/list", &user.RoleController{}, "post:List"),
		beego.NSRouter("/role/save", &user.RoleController{}, "post:Save"),
		beego.NSRouter("/role/delete", &user.RoleController{}, "post:Delete"),
		// 用户
		beego.NSRouter("/user/login", userController, "post:Login"),
		beego.NSRouter("/user/logout", userController, "post:Logout"),
		beego.NSRouter("/user/info", userController, "get:User"),
		beego.NSRouter("/user/info", userController, "post:UpdateUserInfo"),
		beego.NSRouter("/user/register", userController, "post:Register"),
		beego.NSRouter("/user/list", userController, "post:Users"),
		beego.NSRouter("/user/detail", userController, "get:GetDetail"),
		beego.NSRouter("/user/save", userController, "post:Update"),
		beego.NSRouter("/user/modifyPassword", userController, "post:UpdatePassword"),
		beego.NSRouter("/user/resetPassword", userController, "post:ResetPassword"),
		beego.NSRouter("/oauth2", &oauth2.Controller{}),
		beego.NSRouter("/oauth2/callback", &oauth2.Controller{}, "post:Callback"),
		//	settings
		beego.NSRouter("/settings/list", &settings.SettingController{}, "post:List"),
		beego.NSRouter("/settings/save", &settings.SettingController{}, "post:Save"),
		beego.NSRouter("/settings/delete", &settings.SettingController{}, "post:Delete"),
		//	路由
		beego.NSRouter("/settings/route/list", &settings.RouteController{}, "post:List"),
		beego.NSRouter("/settings/route/save", &settings.RouteController{}, "post:Save"),
		beego.NSRouter("/settings/route/delete", &settings.RouteController{}, "post:Delete"),
		//	其它设置
		beego.NSRouter("/wechat/webhook/gitlab", &wechat.Controller{}, "post:WebhookFromGitlab"),
		// 代理设置
		beego.NSRouter("/proxy/list", proxy.Instance, "post:List"),
		beego.NSRouter("/proxy/save", proxy.Instance, "post:Save"),
		beego.NSRouter("/proxy/remove", proxy.Instance, "post:Remove"),
		// auth-token
		beego.NSRouter("/auth-token/list", auth_token.AuthController, "post:List"),
		beego.NSRouter("/auth-token/create", auth_token.AuthController, "post:Create"),
		beego.NSRouter("/auth-token/update", auth_token.AuthController, "post:Update"),
		beego.NSRouter("/auth-token/remove", auth_token.AuthController, "post:DeleteById"),
		beego.NSRouter("/auth-token/detail", auth_token.AuthController, "get:GetDetail"),
		// agent
		beego.NSHandler(constants.AgentConnectUrl, agentServer),
	)
	beego.AddNamespace(ns)
	// LDAP路由
	beego.AddNamespace(ldap.InitRouter(config.BaseApi))

	proxy.Proxy.AddNoProxy(regexp.MustCompile(fmt.Sprintf("^(%s)", config.BaseApi)))

	beego.InsertFilter("/**", beego.BeforeRouter, proxy.Proxy.Filter)
	beego.InsertFilter(fmt.Sprintf("%s/**", config.BaseApi), beego.BeforeRouter, middleware.AuthFilter)

	beego.Router(fmt.Sprintf("%s/config.js", config.ContextPath), &nginx_controller.ConfigController{})
	// portal static assets
	beego.SetStaticPath(config.ContextPath, utils.GetStaticDir())
	//
	webPrefix := config.ContextPath
	if !strings.HasSuffix(webPrefix, "/") {
		webPrefix = webPrefix + "/"
	}
	beego.Get("/", func(ctx *context.Context) {
		ctx.Redirect(301, webPrefix)
	})

	beego.ErrorController(&base.ErrorController{})
}