be.ems/src/framework/middleware/pre_authorize.go

package middleware

import (
	"strings"

	"be.ems/src/framework/config"
	AdminConstants "be.ems/src/framework/constants/admin"
	commonConstants "be.ems/src/framework/constants/common"
	"be.ems/src/framework/i18n"
	ctxUtils "be.ems/src/framework/utils/ctx"
	tokenUtils "be.ems/src/framework/utils/token"
	"be.ems/src/framework/vo/result"

	"github.com/gin-gonic/gin"
)

/**无Token可访问白名单 */
var URL_WHITE_LIST = []string{
	"/performanceManagement",
	"/faultManagement",
	"/systemState",
	"/omcNeConfig",
	"/cdrEvent",
	"/ueEvent",
	"/upload-ue",
	"/oauth/token",
}

// PreAuthorize 用户身份授权认证校验
//
// 只需含有其中角色 "hasRoles": {"xxx"},
//
// 只需含有其中权限 "hasPerms": {"xxx"},
//
// 同时匹配其中角色 "matchRoles": {"xxx"},
//
// 同时匹配其中权限 "matchPerms": {"xxx"},
func PreAuthorize(options map[string][]string) gin.HandlerFunc {
	return func(c *gin.Context) {
		// 登录认证，默认打开
		enable := true
		if v := config.Get("user.loginAuth"); v != nil {
			enable = v.(bool)
		}
		if !enable {
			loginUser, _ := ctxUtils.LoginUser(c)
			loginUser.UserID = "2"
			loginUser.User.UserID = "2"
			loginUser.User.UserName = "admin"
			loginUser.User.NickName = "admin"
			c.Set(commonConstants.CTX_LOGIN_USER, loginUser)
			c.Next()
			return
		}

		language := ctxUtils.AcceptLanguage(c)

		requestURI := c.Request.RequestURI

		// 判断白名单
		isWhite := false
		for _, w := range URL_WHITE_LIST {
			if strings.Contains(requestURI, w) {
				isWhite = true
				break
			}
		}
		if isWhite {
			c.Next()
			return
		}

		// 获取请求头标识信息
		tokenStr := ctxUtils.Authorization(c)
		if tokenStr == "" {
			c.JSON(401, result.CodeMsg(401, i18n.TKey(language, "app.common.err401")))
			c.Abort() // 停止执行后续的处理函数
			return
		}

		// 验证令牌
		claims, err := tokenUtils.Verify(tokenStr)
		if err != nil {
			c.JSON(401, result.CodeMsg(401, err.Error()))
			c.Abort() // 停止执行后续的处理函数
			return
		}

		// 获取缓存的用户信息
		loginUser := tokenUtils.LoginUser(claims)
		if loginUser.UserID == "" {
			c.JSON(401, result.CodeMsg(401, i18n.TKey(language, "app.common.err401")))
			c.Abort() // 停止执行后续的处理函数
			return
		}

		// 检查刷新有效期后存入上下文
		tokenUtils.RefreshIn(&loginUser)
		c.Set(commonConstants.CTX_LOGIN_USER, loginUser)

		// 登录用户角色权限校验
		if options != nil {
			var roles []string
			for _, item := range loginUser.User.Roles {
				roles = append(roles, item.RoleKey)
			}
			perms := loginUser.Permissions
			verifyOk := verifyRolePermission(roles, perms, options)
			if !verifyOk {
				msg := i18n.TTemplate(language, "app.common.err403", map[string]any{"method": c.Request.Method, "requestURI": requestURI})
				c.JSON(403, result.CodeMsg(403, msg))
				c.Abort() // 停止执行后续的处理函数
				return
			}
		}

		// 调用下一个处理程序
		c.Next()
	}
}

// verifyRolePermission 校验角色权限是否满足
//
// roles 角色字符数组
//
// perms 权限字符数组
//
// options 参数
func verifyRolePermission(roles, perms []string, options map[string][]string) bool {
	// 直接放行 管理员角色或任意权限
	if contains(roles, AdminConstants.ROLE_KEY) || contains(perms, AdminConstants.PERMISSION) {
		return true
	}
	opts := make([]bool, 4)

	// 只需含有其中角色
	hasRole := false
	if arr, ok := options["hasRoles"]; ok && len(arr) > 0 {
		hasRole = some(roles, arr)
		opts[0] = true
	}

	// 只需含有其中权限
	hasPerms := false
	if arr, ok := options["hasPerms"]; ok && len(arr) > 0 {
		hasPerms = some(perms, arr)
		opts[1] = true
	}

	// 同时匹配其中角色
	matchRoles := false
	if arr, ok := options["matchRoles"]; ok && len(arr) > 0 {
		matchRoles = every(roles, arr)
		opts[2] = true
	}

	// 同时匹配其中权限
	matchPerms := false
	if arr, ok := options["matchPerms"]; ok && len(arr) > 0 {
		matchPerms = every(perms, arr)
		opts[3] = true
	}

	// 同时判断 含有其中
	if opts[0] && opts[1] {
		return hasRole || hasPerms
	}
	// 同时判断 匹配其中
	if opts[2] && opts[3] {
		return matchRoles && matchPerms
	}
	// 同时判断 含有其中且匹配其中
	if opts[0] && opts[3] {
		return hasRole && matchPerms
	}
	if opts[1] && opts[2] {
		return hasPerms && matchRoles
	}

	return hasRole || hasPerms || matchRoles || matchPerms
}

// contains 检查字符串数组中是否包含指定的字符串
func contains(arr []string, target string) bool {
	for _, str := range arr {
		if str == target {
			return true
		}
	}
	return false
}

// some 检查字符串数组中含有其中一项
func some(origin []string, target []string) bool {
	has := false
	for _, t := range target {
		for _, o := range origin {
			if t == o {
				has = true
				break
			}
		}
		if has {
			break
		}
	}
	return has
}

// every 检查字符串数组中同时包含所有项
func every(origin []string, target []string) bool {
	match := true
	for _, t := range target {
		found := false
		for _, o := range origin {
			if t == o {
				found = true
				break
			}
		}
		if !found {
			match = false
			break
		}
	}
	return match
}