package middlewares

import (
	"net/http"

	"gitee.com/fkil555/gin-extend/app/result"
	"gitee.com/fkil555/gin-extend/helpers/tokenhelper"
	"github.com/gin-gonic/gin"
)

func CheckToken() gin.HandlerFunc {
	return func(ctx *gin.Context) {

		header := ctx.Request.Header
		token := header.Get("Authorization")
		if len(token) <= 0 {
			ctx.Abort()
			ctx.JSON(http.StatusOK, result.Error(101, "need login token"))
			return
		}
		jwtToken, err := tokenhelper.ParseJwtToken(token)
		if err != nil || jwtToken == nil || jwtToken.Uid <= 0 {
			ctx.Abort()
			ctx.JSON(http.StatusOK, result.Error(102, "token parse fail"))
			return
		}
		ctx.Set("loginUserId", jwtToken.Uid)

		ctx.Next()
	}
}