sail:jwt新增加解密方法

Author: keepchen

sail/jwt.go

@@ -3,6 +3,7 @@ package sail
 import (
 	"github.com/keepchen/go-sail/v3/lib/jwt"
 	"github.com/keepchen/go-sail/v3/sail/config"
+	"github.com/keepchen/go-sail/v3/utils"
 )
 
 // IJwt Jwt接口
@@ -16,7 +17,28 @@ type IJwt interface {
 	// otherFields 其他可扩展字段
 	MakeToken(uid string, exp int64, otherFields ...map[string]interface{}) (string, error)
 	// ValidToken 验证JWT令牌
+	//
+	// # ⚠️注意⚠️
+	//
+	// 此方法继承 jwt.Validator 的 Validate 方法，因此只验证了存在且格式正确的字段，
+	// 如果调用方有其他的验证规则，需要自行处理验证逻辑。
 	ValidToken(token string) (bool, jwt.MapClaims, error)
+	//Encrypt 加密字符
+	//
+	// 此方法可以与 Decrypt 方法对应进行加解密
+	//
+	// # Note
+	//
+	// 此方法调用 utils.RSA 方法并使用 jwt.Conf 配置中的公钥PublicKey进行加密
+	Encrypt(plaintext string) (string, error)
+	//Decrypt 解密字符
+	//
+	// 此方法可以与 Encrypt 方法对应进行加解密
+	//
+	// # Note
+	//
+	// 此方法调用 utils.RSA 方法并使用 jwt.Conf 配置中的私钥PrivateKey进行解密
+	Decrypt(encryptedStr string) (string, error)
 }
 
 type jwtImpl struct{}
@@ -68,3 +90,27 @@ func (jwtImpl) ValidToken(token string) (bool, jwt.MapClaims, error) {
 
 	return err == nil, mp, err
 }
+
+// Encrypt 加密字符
+//
+// 此方法可以与 Decrypt 方法对应进行加解密
+//
+// # Note
+//
+// 此方法调用 utils.RSA 方法并使用 jwt.Conf 配置中的公钥PublicKey进行加密
+func (jwtImpl) Encrypt(plaintext string) (string, error) {
+	conf := config.Get()
+	return utils.RSA().Encrypt(plaintext, []byte(conf.JwtConf.PublicKey))
+}
+
+// Decrypt 解密字符
+//
+// 此方法可以与 Encrypt 方法对应进行加解密
+//
+// # Note
+//
+// 此方法调用 utils.RSA 方法并使用 jwt.Conf 配置中的私钥PrivateKey进行解密
+func (jwtImpl) Decrypt(encryptedStr string) (string, error) {
+	conf := config.Get()
+	return utils.RSA().Decrypt(encryptedStr, []byte(conf.JwtConf.PrivateKey))
+}

sail/jwt_test.go

@@ -121,3 +121,42 @@ func TestValidTokenExpired(t *testing.T) {
 		t.Log(claims)
 	})
 }
+
+func TestEncrypt(t *testing.T) {
+	t.Run("Encrypt", func(t *testing.T) {
+		conf := &config.Config{
+			JwtConf: &jwt.Conf{
+				Enable:     true,
+				Algorithm:  string(jwt.SigningMethodRS256),
+				PublicKey:  string(publicKey),
+				PrivateKey: string(privateKey),
+			},
+		}
+		conf.JwtConf.MustLoad()
+		config.Set(conf)
+		encryptedStr, err := JWT().Encrypt(uid)
+		t.Log("Encrypt:", encryptedStr)
+		assert.NoError(t, err)
+	})
+}
+
+func TestDecrypt(t *testing.T) {
+	t.Run("Decrypt", func(t *testing.T) {
+		conf := &config.Config{
+			JwtConf: &jwt.Conf{
+				Enable:     true,
+				Algorithm:  string(jwt.SigningMethodRS256),
+				PublicKey:  string(publicKey),
+				PrivateKey: string(privateKey),
+			},
+		}
+		conf.JwtConf.MustLoad()
+		config.Set(conf)
+		encryptedStr, err := JWT().Encrypt(uid)
+		t.Log("Encrypt:", encryptedStr)
+		plaintext, err := JWT().Decrypt(encryptedStr)
+		t.Log("Decrypt:", plaintext)
+		assert.NoError(t, err)
+		assert.Equal(t, plaintext, uid)
+	})
+}