feat: user migration

middleware/jwt_for_user.go

@@ -1,6 +1,8 @@
 package middleware
 
 import (
+	"hr_receiver/config"
+	"hr_receiver/models"
 	"hr_receiver/util"
 	"net/http"
 	"strings"
@@ -33,12 +35,33 @@ func JWTAuth() gin.HandlerFunc {
 			return
 		}
 
+		role := claims.Role
+		flavorType := claims.FlavorType
+		regionIDs := claims.RegionIDs
+
+		if role == "" || flavorType == "" {
+			var user models.User
+			if err := config.DB.Preload("Regions").First(&user, claims.UserID).Error; err != nil {
+				c.JSON(http.StatusUnauthorized, gin.H{"error": "User not found"})
+				c.Abort()
+				return
+			}
+			if !user.IsActive {
+				c.JSON(http.StatusForbidden, gin.H{"error": "User is disabled"})
+				c.Abort()
+				return
+			}
+			role = user.Role
+			flavorType = user.FlavorType
+			regionIDs = user.RegionIDs()
+		}
+
 		// 将用户信息存入上下文
 		c.Set("userID", claims.UserID)
 		c.Set("username", claims.Username)
-		c.Set("role", claims.Role)
-		c.Set("flavorType", claims.FlavorType)
-		c.Set("regionIDs", claims.RegionIDs)
+		c.Set("role", role)
+		c.Set("flavorType", flavorType)
+		c.Set("regionIDs", regionIDs)
 
 		c.Next()
 	}