89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
|
package router
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"go-admin/app/admin/apis"
|
|||
|
|
"mime"
|
|||
|
|
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk/config"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
jwt "github.com/go-admin-team/go-admin-core/sdk/pkg/jwtauth"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk/pkg/ws"
|
|||
|
|
ginSwagger "github.com/swaggo/gin-swagger"
|
|||
|
|
|
|||
|
|
swaggerfiles "github.com/swaggo/files"
|
|||
|
|
|
|||
|
|
"go-admin/common/middleware"
|
|||
|
|
"go-admin/common/middleware/handler"
|
|||
|
|
_ "go-admin/docs/admin"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func InitSysRouter(r *gin.Engine, authMiddleware *jwt.GinJWTMiddleware) *gin.RouterGroup {
|
|||
|
|
g := r.Group("")
|
|||
|
|
sysBaseRouter(g)
|
|||
|
|
// 静态文件
|
|||
|
|
sysStaticFileRouter(g)
|
|||
|
|
// swagger；注意：生产环境可以注释掉
|
|||
|
|
if config.ApplicationConfig.Mode != "prod" {
|
|||
|
|
sysSwaggerRouter(g)
|
|||
|
|
}
|
|||
|
|
// 需要认证
|
|||
|
|
sysCheckRoleRouterInit(g, authMiddleware)
|
|||
|
|
return g
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func sysBaseRouter(r *gin.RouterGroup) {
|
|||
|
|
|
|||
|
|
go ws.WebsocketManager.Start()
|
|||
|
|
go ws.WebsocketManager.SendService()
|
|||
|
|
go ws.WebsocketManager.SendAllService()
|
|||
|
|
|
|||
|
|
if config.ApplicationConfig.Mode != "prod" {
|
|||
|
|
r.GET("/", apis.GoAdmin)
|
|||
|
|
}
|
|||
|
|
r.GET("/info", handler.Ping)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func sysStaticFileRouter(r *gin.RouterGroup) {
|
|||
|
|
err := mime.AddExtensionType(".js", "application/javascript")
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
r.Static("/static", "./static")
|
|||
|
|
if config.ApplicationConfig.Mode != "prod" {
|
|||
|
|
r.Static("/form-generator", "./static/form-generator")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func sysSwaggerRouter(r *gin.RouterGroup) {
|
|||
|
|
r.GET("/swagger/admin/*any", ginSwagger.WrapHandler(swaggerfiles.NewHandler(), ginSwagger.InstanceName("admin")))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func sysCheckRoleRouterInit(r *gin.RouterGroup, authMiddleware *jwt.GinJWTMiddleware) {
|
|||
|
|
wss := r.Group("").Use(authMiddleware.MiddlewareFunc())
|
|||
|
|
{
|
|||
|
|
wss.GET("/ws/:id/:channel", ws.WebsocketManager.WsClient)
|
|||
|
|
wss.GET("/wslogout/:id/:channel", ws.WebsocketManager.UnWsClient)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
v1 := r.Group("/api/v1")
|
|||
|
|
{
|
|||
|
|
v1.POST("/login", authMiddleware.LoginHandler)
|
|||
|
|
// Refresh time can be longer than token timeout
|
|||
|
|
v1.GET("/refresh_token", authMiddleware.RefreshHandler)
|
|||
|
|
}
|
|||
|
|
registerBaseRouter(v1, authMiddleware)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func registerBaseRouter(v1 *gin.RouterGroup, authMiddleware *jwt.GinJWTMiddleware) {
|
|||
|
|
api := apis.SysMenu{}
|
|||
|
|
api2 := apis.SysDept{}
|
|||
|
|
v1auth := v1.Group("").Use(authMiddleware.MiddlewareFunc()).Use(middleware.AuthCheckRole())
|
|||
|
|
{
|
|||
|
|
v1auth.GET("/roleMenuTreeselect/:roleId", api.GetMenuTreeSelect)
|
|||
|
|
//v1.GET("/menuTreeselect", api.GetMenuTreeSelect)
|
|||
|
|
v1auth.GET("/roleDeptTreeselect/:roleId", api2.GetDeptTreeRoleSelect)
|
|||
|
|
v1auth.POST("/logout", handler.LogOut)
|
|||
|
|
}
|
|||
|
|
}
|