package middleware

import (
	"git.echol.cn/loser/lckt/model/common/response"
	"git.echol.cn/loser/lckt/plugin/customerservice/tools"
	"git.echol.cn/loser/lckt/utils"
	"github.com/gin-gonic/gin"
	"strings"
)

func JWTAuthMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 从请求头获取 token
		authHeader := c.GetHeader("chat-token")
		userHeader := c.GetHeader("x-token")
		if userHeader == "" && authHeader == "" {
			response.FailWithMessage("参数错误:"+"Authorization header is missing", c)
			c.Abort()
			return
		}

		if authHeader != "" {
			// 按照格式 "Bearer <token>" 提取 token
			tokenString := strings.TrimSpace(strings.TrimPrefix(authHeader, "Bearer "))
			if tokenString == "" {
				response.FailWithMessage("参数错误:"+"Token is missing", c)
				c.Abort()
				return
			}
			// 验证 token
			claims, err := tools.ValidateToken(tokenString)
			if err != nil {
				response.FailWithMessage("Invalid token："+err.Error(), c)
				c.Abort()
				return
			}

			// 将用户信息存储在上下文中，便于后续处理
			c.Set("service_id", claims.ServiceId)
			//c.Request.URL.Query().Add("service_id", strconv.FormatInt(claims.ServiceId, 10))
			c.Next() // 继续处理请求

		} else {
			//为了方便客服后台和前端客服聊天界面公用方法，共用同一套jwt，前端的jwt的值取的还是gva-shop的x-token
			j := utils.NewJWT()
			claims, err := j.ParseToken(userHeader)
			if err != nil {
				response.FailWithMessage("参数错误:"+"Token is error", c)
				c.Abort()
				return
			}
			c.Set("jwt_user_id", claims.BaseClaims.ID)
			c.Next()
		}

	}
}