/**
 * package middleware
 * 作用：JWT认证中间件
 */
package middleware

import (
	"net/http"
	"xk-websocket-v2/internal/utils"

	"github.com/gin-gonic/gin"
)

/**
 * JWTAuthMiddleware
 * 
 * 功能：JWT认证中间件
 * 
 * 作用：
 * 1. 从HTTP请求头或查询参数中提取JWT Token
 * 2. 验证Token的有效性和过期时间
 * 3. 从Token中解析出用户ID
 * 4. 将用户ID注入到Gin Context中，供后续处理器使用
 * 5. 如果Token无效或缺失，返回401未授权错误
 * 
 * 使用场景：
 * - 需要用户登录才能访问的API接口
 * - 需要在处理器中获取当前用户信息的接口
 * 
 * @returns gin.HandlerFunc 中间件处理函数
 */
func JWTAuthMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 步骤1: 从HTTP请求头中获取Token（标准方式）
		token := c.GetHeader("Authorization")
		if token == "" {
			// 步骤2: 如果请求头中没有，尝试从查询参数获取（兼容旧代码）
			token = c.Query("token")
		}

		// 步骤3: 如果仍然没有Token，返回401未授权错误
		if token == "" {
			c.JSON(http.StatusUnauthorized, gin.H{"error": "缺少认证Token"})
			c.Abort() // 终止请求处理
			return
		}

		// 步骤4: 移除 "Bearer " 前缀（如果存在）
		// 标准JWT Token格式：Bearer <token>
		if len(token) > 7 && token[:7] == "Bearer " {
			token = token[7:]
		}

		// 步骤5: 验证Token的有效性和过期时间
		userID, err := utils.ValidateToken(token)
		if err != nil {
			// Token无效或已过期，返回401错误
			c.JSON(http.StatusUnauthorized, gin.H{"error": "Token无效或已过期: " + err.Error()})
			c.Abort()
			return
		}

		// 步骤6: 将解析出的用户ID注入到Context中
		// 后续处理器可以通过 c.Get("user_id") 获取当前用户ID
		c.Set("user_id", userID)
		c.Next() // 继续执行下一个中间件或处理器
	}
}

/**
 * OptionalJWTAuthMiddleware
 * 
 * 功能：可选的JWT认证中间件（不强制要求认证）
 * 
 * 作用：
 * 1. 如果请求中提供了Token，则验证Token并注入用户ID
 * 2. 如果没有提供Token，则继续执行，不返回错误
 * 3. 适用于既支持登录用户访问，也支持匿名用户访问的接口
 * 
 * 使用场景：
 * - 公开接口，但登录用户可以获取更多信息
 * - 兼容旧代码，不强制要求认证
 * 
 * @returns gin.HandlerFunc 中间件处理函数
 */
func OptionalJWTAuthMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 步骤1: 尝试从请求头或查询参数获取Token
		token := c.GetHeader("Authorization")
		if token == "" {
			token = c.Query("token")
		}

		// 步骤2: 如果提供了Token，则验证并注入用户ID
		if token != "" {
			// 移除 "Bearer " 前缀
			if len(token) > 7 && token[:7] == "Bearer " {
				token = token[7:]
			}

			// 验证Token，如果有效则注入用户ID
			// 如果Token无效，不返回错误，继续执行（允许匿名访问）
			if userID, err := utils.ValidateToken(token); err == nil {
				c.Set("user_id", userID)
			}
		}

		// 步骤3: 继续执行，无论是否有Token
		c.Next()
	}
}