实现 jwt、密码加密、casbin 的接入

utility/jwt/jwt.go

@@ -0,0 +1,115 @@
+package jwt
+
+import (
+	"server/utility/ecode"
+	"strings"
+	"time"
+
+	"github.com/golang-jwt/jwt/v5"
+	"github.com/google/uuid"
+)
+
+var (
+	// secretKey 是用于签名 JWT 的密钥。应替换为更加安全的密钥并妥善保管。
+	secretKey = []byte("1a40c1d5b7a1b0f0f835e0b24ca292a6")
+
+	// issuer 表示签发者信息，作为 JWT 标准字段的一部分。
+	issuer = "arenax.com"
+)
+
+type (
+	// TokenIn 表示生成 JWT 所需的输入参数。
+	//
+	// 字段:
+	//   - UserId: 用户 ID；
+	//   - Permission: 权限标识；
+	//   - ExpireTime: token 过期时间（可选）。
+	TokenIn struct {
+		UserId     int           // 用户 ID
+		Permission string        // 权限标识
+		ExpireTime time.Duration // 令牌有效期
+	}
+
+	// TokenOut 表示从 JWT 中解析出的用户信息。
+	//
+	// 字段:
+	//   - UserId: 用户 ID；
+	//   - Permission: 权限标识；
+	//   - JTI: JWT 的唯一标识。
+	TokenOut struct {
+		UserId     int    // 用户 ID
+		Permission string // 权限标识
+		JTI        string // JWT 唯一标识
+	}
+
+	// jwtClaims 自定义 JWT 的声明体结构，嵌入标准声明字段。
+	jwtClaims struct {
+		UserId     int    `json:"user_id"`    // 用户 ID
+		Permission string `json:"permission"` // 权限标识
+		JTI        string `json:"jti"`        // 唯一标识
+		jwt.RegisteredClaims
+	}
+)
+
+// GenerateToken 生成带有自定义权限和过期时间的 JWT 字符串。
+//
+// 参数:
+//   - in: 包含用户 ID、权限、过期时间等信息的 TokenIn 对象。
+//
+// 返回:
+//   - 生成的 JWT 字符串；
+//   - 若出错则返回错误信息。
+func GenerateToken(in *TokenIn) (string, error) {
+	expire := in.ExpireTime
+	if expire <= 0 {
+		expire = 2 * time.Hour
+	}
+
+	claims := jwtClaims{
+		UserId:     in.UserId,
+		Permission: in.Permission,
+		JTI:        uuid.NewString(),
+		RegisteredClaims: jwt.RegisteredClaims{
+			ExpiresAt: jwt.NewNumericDate(time.Now().Add(expire)),
+			IssuedAt:  jwt.NewNumericDate(time.Now()),
+			NotBefore: jwt.NewNumericDate(time.Now()),
+			Issuer:    issuer,
+		},
+	}
+
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
+	return token.SignedString(secretKey)
+}
+
+// ParseToken 解析并验证 JWT 字符串并返回对应的用户信息。
+// 自动去除开头的 "Bearer " 前缀。
+//
+// 参数:
+//   - tokenString: 需要解析的 JWT 字符串。
+//
+// 返回:
+//   - 解析后的 TokenOut 对象；
+//   - 若失败则返回相应的错误信息。
+func ParseToken(tokenString string) (*TokenOut, error) {
+	if strings.HasPrefix(tokenString, "Bearer ") {
+		tokenString = strings.TrimPrefix(tokenString, "Bearer ")
+	}
+
+	token, err := jwt.ParseWithClaims(tokenString, &jwtClaims{}, func(token *jwt.Token) (interface{}, error) {
+		return secretKey, nil
+	})
+	if err != nil {
+		return nil, ecode.Fail.Sub("解析 token 出现异常")
+	}
+
+	claims, ok := token.Claims.(*jwtClaims)
+	if !ok || !token.Valid {
+		return nil, ecode.InvalidOperation.Sub("无效的 Token")
+	}
+
+	return &TokenOut{
+		UserId:     claims.UserId,
+		Permission: claims.Permission,
+		JTI:        claims.JTI,
+	}, nil
+}