x

internal/logic/public/user/queryUserInfoLogic.go

@@ -2,13 +2,12 @@ package user
 
 import (
 	"context"
-	"encoding/json"
 	"fmt"
 	"sort"
 	"strings"
 
+	logicCommon "github.com/perfect-panel/server/internal/logic/common"
 	"github.com/perfect-panel/server/pkg/constant"
-	"github.com/perfect-panel/server/pkg/kutt"
 	"github.com/perfect-panel/server/pkg/uuidx"
 	"github.com/perfect-panel/server/pkg/xerr"
 	"github.com/pkg/errors"
@@ -99,8 +98,8 @@ func (l *QueryUserInfoLogic) QueryUserInfo() (resp *types.User, err error) {
 	resp.AuthMethods = userMethods
 
 	// 生成邀请短链接
-	if l.svcCtx.Config.Kutt.Enable && resp.ReferCode != "" {
-		shortLink := l.generateInviteShortLink(resp.ReferCode)
+	if resp.ReferCode != "" {
+		shortLink := logicCommon.NewInviteLinkResolver(l.ctx, l.svcCtx).ResolveInviteLink(resp.ReferCode)
 		if shortLink != "" {
 			resp.ShareLink = shortLink
 		}
@@ -212,112 +211,6 @@ func getFamilyStatusName(status uint8) string {
 	return "disabled"
 }
 
-// customData 用于解析 SiteConfig.CustomData JSON 字段
-// 包含从自定义数据中提取所需的配置项
-type customData struct {
-	ShareUrl string `json:"shareUrl"` // 分享链接前缀 URL（目标落地页）
-	Domain   string `json:"domain"`   // 短链接域名
-}
-
-// getShareUrl 从 SiteConfig.CustomData 中获取 shareUrl
-//
-// 返回:
-//   - string: 分享链接前缀 URL，如果获取失败则返回 Kutt.TargetURL 作为 fallback
-func (l *QueryUserInfoLogic) getShareUrl() string {
-	siteConfig := l.svcCtx.Config.Site
-	if siteConfig.CustomData != "" {
-		var data customData
-		if err := json.Unmarshal([]byte(siteConfig.CustomData), &data); err == nil {
-			if data.ShareUrl != "" {
-				return data.ShareUrl
-			}
-		}
-	}
-	// fallback 到 Kutt.TargetURL
-	return l.svcCtx.Config.Kutt.TargetURL
-}
-
-// getDomain 从 SiteConfig.CustomData 中获取短链接域名
-//
-// 返回:
-//   - string: 短链接域名，如果获取失败则返回 Kutt.Domain 作为 fallback
-func (l *QueryUserInfoLogic) getDomain() string {
-	siteConfig := l.svcCtx.Config.Site
-	if siteConfig.CustomData != "" {
-		var data customData
-		if err := json.Unmarshal([]byte(siteConfig.CustomData), &data); err == nil {
-			if data.Domain != "" {
-				return data.Domain
-			}
-		}
-	}
-	// fallback 到 Kutt.Domain
-	return l.svcCtx.Config.Kutt.Domain
-}
-
-// generateInviteShortLink 生成邀请短链接（带 Redis 缓存）
-//
-// 参数:
-//   - inviteCode: 邀请码
-//
-// 返回:
-//   - string: 短链接 URL，失败时返回空字符串
-func (l *QueryUserInfoLogic) generateInviteShortLink(inviteCode string) string {
-	cfg := l.svcCtx.Config.Kutt
-	shareUrl := l.getShareUrl()
-	domain := l.getDomain()
-
-	// 检查必要配置
-	if cfg.ApiURL == "" || cfg.ApiKey == "" {
-		l.Sloww("Kutt config incomplete",
-			logger.Field("api_url", cfg.ApiURL != ""),
-			logger.Field("api_key", cfg.ApiKey != ""))
-		return ""
-	}
-	if shareUrl == "" {
-		l.Sloww("ShareUrl not configured in CustomData or Kutt.TargetURL")
-		return ""
-	}
-
-	// Redis 缓存 key
-	cacheKey := "cache:invite:short_link:" + inviteCode
-
-	// 1. 尝试从 Redis 缓存读取
-	cachedLink, err := l.svcCtx.Redis.Get(l.ctx, cacheKey).Result()
-	if err == nil && cachedLink != "" {
-		l.Debugw("Hit cache for invite short link",
-			logger.Field("invite_code", inviteCode),
-			logger.Field("short_link", cachedLink))
-		return cachedLink
-	}
-
-	// 2. 缓存未命中，调用 Kutt API 创建短链接
-	client := kutt.NewClient(cfg.ApiURL, cfg.ApiKey)
-	shortLink, err := client.CreateInviteShortLink(l.ctx, shareUrl, inviteCode, domain)
-	if err != nil {
-		l.Errorw("Failed to create short link",
-			logger.Field("error", err.Error()),
-			logger.Field("invite_code", inviteCode),
-			logger.Field("share_url", shareUrl))
-		return ""
-	}
-
-	// 3. 写入 Redis 缓存（永不过期，因为邀请码不变短链接也不会变）
-	if err := l.svcCtx.Redis.Set(l.ctx, cacheKey, shortLink, 0).Err(); err != nil {
-		l.Errorw("Failed to cache short link",
-			logger.Field("error", err.Error()),
-			logger.Field("invite_code", inviteCode))
-		// 缓存失败不影响返回
-	}
-
-	l.Infow("Created and cached invite short link",
-		logger.Field("invite_code", inviteCode),
-		logger.Field("short_link", shortLink),
-		logger.Field("share_url", shareUrl))
-
-	return shortLink
-}
-
 // getAuthTypePriority 获取认证类型的排序优先级
 // email: 1 (第一位)
 // mobile: 2 (第二位)