package middleware

import (
	"github.com/gin-gonic/gin"
	"net/http"
	"psi/database"
	"psi/models"
	"psi/utils"
	"strings"
)

// JWTAuth JWT认证中间件
func JWTAuth() gin.HandlerFunc {
	// 处理JWT认证
	return func(c *gin.Context) {
		// 检查认证头
		authHeader := c.GetHeader("Authorization")
		// 检查认证头是否为空
		if authHeader == "" {
			// 返回错误
			c.JSON(http.StatusUnauthorized, gin.H{"error": "未提供认证令牌"})
			// 中断处理
			c.Abort()
			// 返回
			return
		}

		// 提取令牌
		parts := strings.SplitN(authHeader, " ", 2)
		// 处理错误
		if !(len(parts) == 2 && parts[0] == "Bearer") {
			// 返回错误
			c.JSON(http.StatusUnauthorized, gin.H{"error": "认证格式错误"})
			// 中断处理
			c.Abort()
			// 返回
			return
		}
		// 解析JWT
		claims, err := utils.ParseJWT(parts[1])
		// 处理错误
		if err != nil {
			// 返回错误
			c.JSON(http.StatusUnauthorized, gin.H{"error": "无效的认证令牌"})
			// 中断处理
			c.Abort()
			// 返回
			return
		}
		// 判断是员工还是用户
		if claims.ID > 0 {
			// 检查员工状态
			var employee models.Employee
			// 查询员工
			if err := database.DB.Where("id = ? AND status = ? AND deleted_at = ?", claims.ID, 1, 0).First(&employee).Error;
			// 处理错误
			err != nil {
				// 返回错误
				c.JSON(http.StatusUnauthorized, gin.H{"error": "账号不存在或已被禁用"})
				// 中断处理
				c.Abort()
				// 返回
				return
			}

			// 将员工信息存入上下文
			c.Set("id", employee.ID)             // 设置员工ID
			c.Set("role", employee.Role)         // 设置角色
			c.Set("username", employee.Username) // 设置用户名
			c.Set("employee", employee)          // 添加员工信息
			c.Set("about_id", claims.AboutID)    // 租户ID
			c.Set("fid", employee.Fid)           // 父级ID

			// 如果有租户ID，获取租户数据库连接并存入上下文
			if claims.AboutID > 0 {
				tenantDB, err := database.GetTenantDB(claims.AboutID)
				if err != nil {
					c.JSON(http.StatusInternalServerError, gin.H{"error": "租户数据库连接失败"})
					c.Abort()
					return
				}
				c.Set("tenant_db", tenantDB)
			}
		} else {
			// 无效的身份
			c.JSON(http.StatusUnauthorized, gin.H{"error": "无效的身份标识"})
			// 中断处理
			c.Abort()
			// 返回
			return
		}

		// 继续处理
		c.Next()
	}
}

// AdminRequired 管理员权限
func AdminRequired() gin.HandlerFunc {
	// 处理管理员权限
	return func(c *gin.Context) {
		// 检查角色
		role, exists := c.Get("role")
		// 处理错误
		if !exists || role.(int64) != 255 {
			// 返回错误
			c.JSON(http.StatusForbidden, gin.H{"error": "需要管理员权限"})
			// 中断处理
			c.Abort()
			// 返回
			return
		}
		// 继续处理
		c.Next()
	}
}