package middleware

import (
	"github.com/gin-gonic/gin"
	"hy2xs-admin/model/constant"
	"hy2xs-admin/model/vo"
	"hy2xs-admin/service"
	"hy2xs-admin/util"
)

func AdminHandler() gin.HandlerFunc {
	return func(c *gin.Context) {
		myClaims, err := service.ParseToken(service.GetToken(c))
		if err != nil {
			vo.Fail(err.Error(), c)
			c.Abort()
			return
		}
		if !util.ArrContain(myClaims.AccountBo.Roles, "admin") {
			vo.Fail(constant.ForbiddenError, c)
			c.Abort()
			return
		}
		c.Next()
	}
}