mirror of
https://github.com/YFGaia/dify-plus.git
synced 2026-06-04 10:14:00 +08:00
npc0-hue
42 lines
1.1 KiB
Go
42 lines
1.1 KiB
Go
package middleware
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/flipped-aurora/gin-vue-admin/server/global"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/model/common/response"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/service"
|
|
"github.com/flipped-aurora/gin-vue-admin/server/utils"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
var casbinService = service.ServiceGroupApp.SystemServiceGroup.CasbinService
|
|
|
|
// 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 := casbinService.Casbin() // 判断策略中是否存在
|
|
if e == nil {
|
|
global.GVA_LOG.Warn("Casbin enforcer is nil, skipping permission check")
|
|
c.Next()
|
|
return
|
|
}
|
|
success, _ := e.Enforce(sub, obj, act)
|
|
if !success {
|
|
response.FailWithDetailed(gin.H{}, "权限不足", c)
|
|
c.Abort()
|
|
return
|
|
}
|
|
c.Next()
|
|
}
|
|
}
|