🎨 新增用户验证相关工具类

utils/user_jwt/claims.go

@@ -0,0 +1,104 @@
+package user_jwt
+
+import (
+	"git.echol.cn/loser/lckt/model/user"
+	"git.echol.cn/loser/lckt/model/user/request"
+	"net"
+	"time"
+
+	"git.echol.cn/loser/lckt/global"
+	systemReq "git.echol.cn/loser/lckt/model/user/request"
+	"github.com/gin-gonic/gin"
+)
+
+func ClearToken(c *gin.Context) {
+	// 增加cookie app-token 向来源的web添加
+	host, _, err := net.SplitHostPort(c.Request.Host)
+	if err != nil {
+		host = c.Request.Host
+	}
+
+	if net.ParseIP(host) != nil {
+		c.SetCookie("app-token", "", -1, "/", "", false, false)
+	} else {
+		c.SetCookie("app-token", "", -1, "/", host, false, false)
+	}
+}
+
+func SetToken(c *gin.Context, token string, maxAge int) {
+	// 增加cookie app-token 向来源的web添加
+	host, _, err := net.SplitHostPort(c.Request.Host)
+	if err != nil {
+		host = c.Request.Host
+	}
+
+	if net.ParseIP(host) != nil {
+		c.SetCookie("app-token", token, maxAge, "/", "", false, false)
+	} else {
+		c.SetCookie("app-token", token, maxAge, "/", host, false, false)
+	}
+}
+
+func GetToken(c *gin.Context) string {
+	token := c.Request.Header.Get("app-token")
+	if token == "" {
+		j := NewUserJWT()
+		token, _ = c.Cookie("app-token")
+		claims, err := j.ParseToken(token)
+		if err != nil {
+			global.GVA_LOG.Error("重新写入cookie token失败,未能成功解析token,请检查请求头是否存在app-token且claims是否为规定结构")
+			return token
+		}
+		SetToken(c, token, int((claims.ExpiresAt.Unix()-time.Now().Unix())/60))
+	}
+	return token
+}
+
+func GetClaims(c *gin.Context) (*systemReq.CustomClaims, error) {
+	token := GetToken(c)
+	j := NewUserJWT()
+	claims, err := j.ParseToken(token)
+	if err != nil {
+		global.GVA_LOG.Error("从Gin的Context中获取从jwt解析信息失败, 请检查请求头是否存在app-token且claims是否为规定结构")
+	}
+	return claims, err
+}
+
+// GetUserID 从Gin的Context中获取从jwt解析出来的用户ID
+func GetUserID(c *gin.Context) uint {
+	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
+	}
+}
+
+// GetNickName 从Gin的Context中获取从jwt解析出来的用户名
+func GetNickName(c *gin.Context) string {
+	if claims, exists := c.Get("claims"); !exists {
+		if cl, err := GetClaims(c); err != nil {
+			return ""
+		} else {
+			return cl.NickName
+		}
+	} else {
+		waitUse := claims.(*systemReq.CustomClaims)
+		return waitUse.NickName
+	}
+}
+
+func LoginToken(user user.User) (token string, claims request.CustomClaims, err error) {
+	j := NewUserJWT()
+	claims = j.CreateClaims(request.BaseClaims{
+		ID:       user.ID,
+		NickName: user.NickName,
+		Phone:    user.Phone,
+	})
+	token, err = j.CreateToken(claims)
+	return
+}