diff --git a/lib/routes/routes.go b/lib/routes/routes.go index 88903786..a543d0bc 100644 --- a/lib/routes/routes.go +++ b/lib/routes/routes.go @@ -18,6 +18,9 @@ import ( "ems.agt/features/pm" "ems.agt/features/security" "ems.agt/features/state" + sysmenu "ems.agt/features/sys_menu" + sysrole "ems.agt/features/sys_role" + sysuser "ems.agt/features/sys_user" "ems.agt/features/trace" "ems.agt/lib/midware" "ems.agt/lib/services" @@ -249,6 +252,24 @@ func init() { Register("POST", lm.ExtBackupDataUri, lm.ExtDatabaseBackupData, nil) Register("POST", lm.CustomExtBackupDataUri, lm.ExtDatabaseBackupData, nil) + // 登录用户路由信息 + Register("GET", security.UriRouters, security.Routers, midware.Authorize(nil)) + Register("GET", security.CustomUriRouters, security.Routers, midware.Authorize(nil)) + + // 菜单接口添加到路由 + for _, v := range sysmenu.Routers() { + Register(v.Method, v.Pattern, v.Handler, v.Middleware) + } + + // 角色接口添加到路由 + for _, v := range sysrole.Routers() { + Register(v.Method, v.Pattern, v.Handler, v.Middleware) + } + + // 用户接口添加到路由 + for _, v := range sysuser.Routers() { + Register(v.Method, v.Pattern, v.Handler, v.Middleware) + } } // To resolv rest POST/PUT/DELETE/PATCH cross domain @@ -268,22 +289,12 @@ func NewRouter() *mux.Router { r.Use(midware.ArrowIPAddr) for _, router := range routers { - // r.Methods(router.Method). - // Path(router.Pattern). - // Handler(router.Handler) - // if router.Middleware != nil { - // r.Use(router.Middleware) - // } - // if router.Middleware != nil { - // r.HandleFunc(router.Pattern, router.Handler).Methods(router.Method).Subrouter().Use(router.Middleware) - // } else { - // r.HandleFunc(router.Pattern, router.Handler).Methods(router.Method) - // } rt := r.Methods(router.Method).Subrouter() rt.HandleFunc(router.Pattern, router.Handler) if router.Middleware != nil { rt.Use(router.Middleware) } + } return r