feat: user migration

This commit is contained in:
2026-04-28 19:06:42 +08:00
parent aa90b10f06
commit f9077dafcf
6 changed files with 114 additions and 4 deletions
+55
View File
@@ -0,0 +1,55 @@
package middleware
import (
"hr_receiver/models"
"net/http"
"github.com/gin-gonic/gin"
)
func RequireStepTrainingAccess() gin.HandlerFunc {
return func(c *gin.Context) {
roleValue, exists := c.Get("role")
if !exists {
c.JSON(http.StatusForbidden, gin.H{"error": "missing user role"})
c.Abort()
return
}
role, ok := roleValue.(models.UserRole)
if !ok {
c.JSON(http.StatusForbidden, gin.H{"error": "invalid user role"})
c.Abort()
return
}
if role != models.UserRoleSuperAdmin &&
role != models.UserRoleRegionAdmin &&
role != models.UserRoleOperator {
c.JSON(http.StatusForbidden, gin.H{"error": "insufficient role permissions"})
c.Abort()
return
}
flavorValue, exists := c.Get("flavorType")
if !exists {
c.JSON(http.StatusForbidden, gin.H{"error": "missing user flavor"})
c.Abort()
return
}
flavorType, ok := flavorValue.(models.UserFlavorType)
if !ok {
c.JSON(http.StatusForbidden, gin.H{"error": "invalid user flavor"})
c.Abort()
return
}
if flavorType != models.UserFlavorAll &&
flavorType != models.UserFlavorFlink {
c.JSON(http.StatusForbidden, gin.H{"error": "insufficient flavor permissions"})
c.Abort()
return
}
c.Next()
}
}