33 lines
827 B
Go
33 lines
827 B
Go
package middleware
|
|
|
|
import (
|
|
"git.echol.cn/loser/ai_proxy/server/global"
|
|
"git.echol.cn/loser/ai_proxy/server/model/common/response"
|
|
"git.echol.cn/loser/ai_proxy/server/utils"
|
|
"github.com/gin-gonic/gin"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// CasbinHandler 拦截器
|
|
func CasbinHandler() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
waitUse, _ := utils.GetClaims(c)
|
|
//获取请求的PATH
|
|
path := c.Request.URL.Path
|
|
obj := strings.TrimPrefix(path, global.GVA_CONFIG.System.RouterPrefix)
|
|
// 获取请求方法
|
|
act := c.Request.Method
|
|
// 获取用户的角色
|
|
sub := strconv.Itoa(int(waitUse.AuthorityId))
|
|
e := utils.GetCasbin() // 判断策略中是否存在
|
|
success, _ := e.Enforce(sub, obj, act)
|
|
if !success {
|
|
response.FailWithDetailed(gin.H{}, "权限不足", c)
|
|
c.Abort()
|
|
return
|
|
}
|
|
c.Next()
|
|
}
|
|
}
|