opencatd-open/backend/middleware/auth.go

package middleware

import (
	"net/http"
	"opencatd-open/internal/auth"
	"opencatd-open/internal/store"
	"opencatd-open/internal/pkg/jwt"

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

func Auth(db *gorm.DB) gin.HandlerFunc {
	return func(c *gin.Context) {
		authToken := c.GetHeader("Authorization")
		if authToken == "" {
			c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{
				"code":  http.StatusUnauthorized,
				"error": "未提供认证信息",
			})
			return
		}
		if len(authToken) > 7 {
			authToken = authToken[7:]
		}
		claim, err := jwt.ValidateToken(authToken, auth.GetSecretKey())
		if err != nil {
			c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{
				"code":  http.StatusUnauthorized,
				"error": "无效的认证信息",
			})
			return
		}
		var user store.User
		if err := db.First(&user, claim.UserID).Error; err != nil {
			c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{
				"code":  http.StatusUnauthorized,
				"error": "无效的认证信息",
			})
			return
		}
		c.Set("user", &user)
		c.Set("user_id", claim.UserID)
		c.Set("user_role", user.Role)
		c.Next()
	}
}

func CheckRole(role string) gin.HandlerFunc {
	return func(c *gin.Context) {
		userRole, _ := c.Get("user_role")
		if roleStr, ok := userRole.(string); !ok || roleStr != role {
			c.AbortWithStatusJSON(http.StatusForbidden, gin.H{"error": "权限不足"})
			return
		}
		c.Next()
	}
}

// AdminOnly 管理后台中间件：要求 user_role 为 admin。
// 由 middleware.Auth 先行设置 user_role；缺失时拒绝。
func AdminOnly() gin.HandlerFunc {
	return func(c *gin.Context) {
		role, _ := c.Get("user_role")
		roleStr, _ := role.(string)
		if roleStr != store.RoleAdmin {
			c.AbortWithStatusJSON(http.StatusForbidden, gin.H{"error": "需要管理员权限"})
			return
		}
		c.Next()
	}
}