1

business/auth2/auth2.go

@@ -335,7 +335,7 @@ func Login(ctx *Context, authType, authID, authIDType, authSecret string) (authI
 	return LoginInternal(ctx, authType, authID, authIDType, authSecret)
 }
 
-// 抖音用户信息解密
+// TikTokDecryptInfo 抖音用户信息解密
 type TikTokDecryptInfo struct {
 	CountryCode     string `json:"countryCode"`
 	PhoneNumber     string `json:"phoneNumber"`
@@ -346,7 +346,13 @@ type TikTokDecryptInfo struct {
 	} `json:"watermark"`
 }
 
-func DecryptUserMsg(sessionKey, iv, msg string) (string, error) {
+// KuaiShouInfo 抖音用户信息解密
+type KuaiShouInfo struct {
+	CountryCode string `json:"countryCode"`
+	PhoneNumber string `json:"phoneNumber"`
+}
+
+func DecryptUserMsg(sessionKey, iv, msg, loginType string) (string, error) {
 	decodeMsg, err := base64.StdEncoding.DecodeString(msg)
 	if err != nil {
 		return "", err
@@ -364,11 +370,22 @@ func DecryptUserMsg(sessionKey, iv, msg string) (string, error) {
 	if err != nil {
 		return "", err
 	}
-	result := &TikTokDecryptInfo{}
-	if err := json.Unmarshal(userInfo, result); err != nil {
-		return "", err
+
+	switch loginType {
+	case "tiktokmini":
+		result := &TikTokDecryptInfo{}
+		if err := json.Unmarshal(userInfo, result); err != nil {
+			return "", err
+		}
+		return result.PhoneNumber, nil
+	case "kuaishoumini":
+		result := &KuaiShouInfo{}
+		if err := json.Unmarshal(userInfo, result); err != nil {
+			return "", err
+		}
+		return result.PhoneNumber, nil
 	}
-	return result.PhoneNumber, nil
+	return "", ErrIllegalAuthType
 }
 
 // 通过临时TOKEN绑定新创建的用户