first commit

utils/auth/claims.go

@@ -0,0 +1,130 @@
+package auth
+
+import (
+	"net"
+	"strings"
+	"sundynix-go/global"
+	"sundynix-go/model/system"
+	systemReq "sundynix-go/model/system/request"
+
+	"github.com/gin-gonic/gin"
+)
+
+// GetLoginToken 获取登录token
+func GetLoginToken(user system.Login) (token string, claims systemReq.CustomClaims, err error) {
+	j := NewJWT()
+	claims = j.CreateClaims(systemReq.BaseClaims{
+		Account: user.GetAccount(),
+		ID:      user.GetUserId(),
+	})
+	token, err = j.CreateToken(claims)
+	return
+}
+
+// GetToken 从请求头中获取JWT token，并确保其有效性
+//
+// 参数:
+//   - c: *gin.Context, Gin框架的上下文对象，用于获取请求信息和设置响应。
+//
+// 返回值:
+//   - string: 获取到的JWT token，如果获取失败则返回空字符串。
+func GetToken(c *gin.Context) string {
+	// 从请求头中获取Authorization字段的值
+	token := c.Request.Header.Get("Authorization")
+	prefix := strings.HasPrefix(token, "Bearer ")
+	if prefix {
+		token = strings.TrimPrefix(token, "Bearer ")
+	}
+	// 返回获取到的token
+	return token
+}
+
+// ClearToken 清除Cookie中的token
+func ClearToken(c *gin.Context) {
+	host, _, err := net.SplitHostPort(c.Request.Host)
+	if err != nil {
+		host = c.Request.Host
+	}
+	if net.ParseIP(host) != nil {
+		c.SetCookie("sundynix-token", "", -1, "/", "", false, false)
+	} else {
+		c.SetCookie("sundynix-token", "", -1, "/", host, false, false)
+	}
+}
+
+// GetUserInfo 从 gin.Context 中获取用户信息，并返回 CustomClaims 类型的指针。
+// 该函数首先尝试从上下文中获取已存在的 claims，如果不存在，则调用 GetClaims 函数获取 claims。
+// 如果获取 claims 失败，则返回 nil。
+//
+// 参数:
+//   - c: *gin.Context, gin 框架的上下文对象，用于获取请求相关的信息。
+//
+// 返回值:
+//   - *systemReq.CustomClaims: 返回用户的自定义 claims 信息，如果获取失败则返回 nil。
+func GetUserInfo(c *gin.Context) *systemReq.CustomClaims {
+	// 尝试从上下文中获取已存在的 claims
+	if claims, exists := c.Get("claims"); !exists {
+		// 如果 claims 不存在，则调用 GetClaims 函数获取 claims
+		if cl, err := GetClaims(c); err != nil {
+			// 如果获取 claims 失败，返回 nil
+			return nil
+		} else {
+			// 成功获取 claims，返回 claims
+			return cl
+		}
+	} else {
+		// 如果 claims 存在，将其转换为 CustomClaims 类型并返回
+		waitUse := claims.(*systemReq.CustomClaims)
+		return waitUse
+	}
+}
+
+// GetUserId 从 gin.Context 中获取用户 ID，并返回 uint 类型的 ID。
+// 该函数首先尝试从上下文中获取已存在的 claims，如果不存在，则调用 GetClaims 函数获取 claims。
+// 如果获取 claims 失败，则返回 0。
+//
+// 参数:
+//   - c: *gin.Context, gin 框架的上下文对象，用于获取请求相关的信息。
+//
+// 返回值:
+//   - uint: 返回用户的 ID，如果获取失败则返回 0。
+func GetUserId(c *gin.Context) string {
+	if claims, exists := c.Get("claims"); !exists {
+		if cl, err := GetClaims(c); err != nil {
+			return "0"
+		} else {
+			return cl.BaseClaims.ID
+		}
+	} else {
+		waitUse := claims.(*systemReq.CustomClaims)
+		return waitUse.BaseClaims.ID
+	}
+}
+
+// GetClaims 从 Gin 上下文中提取并解析 JWT 令牌，返回自定义的 claims 信息。
+// 该函数首先从请求头中获取 JWT 令牌，然后使用 JWT 解析器解析令牌并返回 claims。
+// 如果解析过程中发生错误，函数会记录错误日志并返回错误信息。
+//
+// 参数:
+//   - c: *gin.Context, Gin 上下文对象，用于获取请求头中的 JWT 令牌。
+//
+// 返回值:
+//   - *systemReq.CustomClaims: 解析后的自定义 claims 信息。
+//   - error: 解析过程中发生的错误，如果解析成功则为 nil。
+func GetClaims(c *gin.Context) (*systemReq.CustomClaims, error) {
+	// 从 Gin 上下文中获取 JWT 令牌
+	token := GetToken(c)
+
+	// 创建新的 JWT 解析器
+	j := NewJWT()
+
+	// 解析 JWT 令牌并获取 claims 信息
+	claims, err := j.ParseToken(token)
+	if err != nil {
+		// 如果解析失败，记录错误日志
+		global.Logger.Error("获取用户信息失败,请检查请求头是否存在x-token且claims是否为规定结构")
+	}
+
+	// 返回解析后的 claims 信息和可能的错误
+	return claims, err
+}