🎨 添加图库和客服插件(有问题-待修改)

plugin/customerservice/middleware/jwt.go

@@ -0,0 +1,57 @@
+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()
+		}
+
+	}
+}