go_process_manager/internal/app/middle/permission.go

package middle

import (
	"reflect"
	"strconv"

	"github.com/lzh-1625/go_process_manager/internal/app/eum"
	"github.com/lzh-1625/go_process_manager/internal/app/repository"

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

func RolePermission(needPermission eum.Role) func(ctx *gin.Context) {
	return func(ctx *gin.Context) {
		if v, ok := ctx.Get(eum.CtxRole); !ok || v.(eum.Role) > needPermission {
			rErr(ctx, -1, "Insufficient permissions; please check your access rights!", nil)
			ctx.Abort()
			return
		}
		ctx.Next()
	}
}

func OprPermission(op eum.OprPermission) func(ctx *gin.Context) {
	return func(ctx *gin.Context) {
		uuid, err := strconv.Atoi(ctx.Query("uuid"))
		if err != nil {
			rErr(ctx, -1, "Invalid parameters!", nil)
			ctx.Abort()
			return
		}
		if v, ok := ctx.Get(eum.CtxRole); !ok || v.(eum.Role) <= eum.RoleAdmin {
			ctx.Next()
			return
		}
		if !reflect.ValueOf(repository.PermissionRepository.GetPermission(ctx.GetString(eum.CtxUserName), uuid)).FieldByName(string(op)).Bool() {
			rErr(ctx, -1, "Insufficient permissions; please check your access rights!", nil)
			ctx.Abort()
			return
		}
		ctx.Next()
	}
}