package middleware

import (
	"github.com/gin-gonic/gin"
	"srm/pkg/contextx"
	"srm/pkg/ecode"
	"srm/request"
)

func VerifyResetPwd() gin.HandlerFunc {
	return func(c *gin.Context) {
		ctx := new(contextx.Context).SetCtx(c)
		params, ok := c.Get("claims")
		if !ok {
			c.Abort()
			return
		}
		claims := params.(*request.CustomClaims)
		if claims.ModifiedPwd == nil { //兼容没有ModifiedPwd值的token
			ctx.Fail(ecode.JWTExpire)
			c.Abort()
			return
		}

		if !*claims.ModifiedPwd {
			ctx.Fail(ecode.ResetPwd)
			c.Abort()
			return
		}
	}
}