update: 优化网盘操作，移除特殊操作

common/alipan.go

@@ -9,6 +9,7 @@ import (
 	"time"
 
 	"github.com/ctwj/urldb/db/entity"
+	"github.com/ctwj/urldb/db/repo"
 	"github.com/ctwj/urldb/utils"
 )
 
@@ -405,6 +406,9 @@ func (a *AlipanService) getAlipan4(shareData map[string]interface{}) (*AlipanSha
 	return &result, nil
 }
 
+func (u *AlipanService) SetCKSRepository(cksRepo repo.CksRepository, entity entity.Cks) {
+}
+
 // manageAccessToken 管理access token
 func (a *AlipanService) manageAccessToken() (string, error) {
 	if a.accessToken != "" {

common/baidu_pan.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 
 	"github.com/ctwj/urldb/db/entity"
+	"github.com/ctwj/urldb/db/repo"
 )
 
 // BaiduPanService 百度网盘服务
@@ -108,3 +109,6 @@ func (b *BaiduPanService) GetUserInfo(cookie *string) (*UserInfo, error) {
 func (b *BaiduPanService) GetUserInfoByEntity(cks entity.Cks) (*UserInfo, error) {
 	return nil, nil
 }
+
+func (u *BaiduPanService) SetCKSRepository(cksRepo repo.CksRepository, entity entity.Cks) {
+}

common/pan_factory.go

@@ -7,6 +7,7 @@ import (
 	"sync"
 
 	"github.com/ctwj/urldb/db/entity"
+	"github.com/ctwj/urldb/db/repo"
 )
 
 // ServiceType 定义网盘服务类型
@@ -93,10 +94,10 @@ type PanService interface {
 	// GetUserInfo 获取用户信息
 	GetUserInfo(ck *string) (*UserInfo, error)
 
-	GetUserInfoByEntity(entity entity.Cks) (*UserInfo, error)
-
 	// GetServiceType 获取服务类型
 	GetServiceType() ServiceType
+
+	SetCKSRepository(cksRepo repo.CksRepository, entity entity.Cks)
 }
 
 // PanFactory 网盘工厂

common/quark_pan.go

@@ -1024,9 +1024,7 @@ func (q *QuarkPanService) GetUserInfo(cookie *string) (*UserInfo, error) {
 	}, nil
 }
 
-// GetUserInfoByEntity 根据 entity.Cks 获取用户信息（待实现）
+func (xq *QuarkPanService) SetCKSRepository(cksRepo repo.CksRepository, entity entity.Cks) {
-func (q *QuarkPanService) GetUserInfoByEntity(cks entity.Cks) (*UserInfo, error) {
-	return nil, nil
 }
 
 // formatBytes 格式化字节数为可读格式

common/uc_pan.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 
 	"github.com/ctwj/urldb/db/entity"
+	"github.com/ctwj/urldb/db/repo"
 )
 
 // UCService UC网盘服务
@@ -106,3 +107,6 @@ func (u *UCService) GetUserInfo(cookie *string) (*UserInfo, error) {
 func (u *UCService) GetUserInfoByEntity(cks entity.Cks) (*UserInfo, error) {
 	return nil, nil
 }
+
+func (u *UCService) SetCKSRepository(cksRepo repo.CksRepository, entity entity.Cks) {
+}

common/xunlei_pan.go

@@ -323,20 +323,8 @@ func (x *XunleiPanService) Transfer(shareID string) (*TransferResult, error) {
 		return ErrorResult(fmt.Sprintf("获取captchaToken失败: %v", err)), nil
 	}
 
-	// 检查是否为检验模式
-	if config.IsType == 1 {
-		// 检验模式：直接获取分享信息
-		urls := map[string]interface{}{
-			"title":     "",
-			"share_url": config.URL,
-			"stoken":    "",
-		}
-		return SuccessResult("检验成功", urls), nil
-	}
-
 	// 转存模式：实现完整的转存流程
-	shareID = strings.TrimRight(shareID, "#/")
+	thisCode := strings.TrimLeft(config.URL, "#")
-	thisCode := strings.TrimLeft(x.config.Code, "#")
 
 	// 获取分享详情
 	shareDetail, err := x.getShare(shareID, thisCode, accessToken, captchaToken)
@@ -352,6 +340,17 @@ func (x *XunleiPanService) Transfer(shareID string) (*TransferResult, error) {
 		return ErrorResult(message), nil
 	}
 
+	// 检查是否为检验模式
+	if config.IsType == 1 {
+		// 检验模式：直接获取分享信息
+		urls := map[string]interface{}{
+			"title":     "",
+			"share_url": config.URL,
+			"stoken":    "",
+		}
+		return SuccessResult("检验成功", urls), nil
+	}
+
 	shareData := shareDetail["data"].(map[string]interface{})
 	files := shareData["files"].([]interface{})
 

handlers/cks_handler.go

@@ -345,13 +345,15 @@ func RefreshCapacity(c *gin.Context) {
 	}
 
 	var userInfo *panutils.UserInfo
-	switch s := service.(type) {
+	service.SetCKSRepository(repoManager.CksRepository, *cks) // 迅雷需要初始化 token 后才能获取，
-	case *panutils.XunleiPanService:
+	userInfo, err = service.GetUserInfo(&cks.Ck)
-		s.SetCKSRepository(repoManager.CksRepository, *cks) // 迅雷需要初始化 token 后才能获取，
+	// switch s := service.(type) {
-		userInfo, err = s.GetUserInfo(nil)
+	// case *panutils.XunleiPanService:
-	default:
+
-		userInfo, err = service.GetUserInfo(&cks.Ck)
+	// 	userInfo, err = s.GetUserInfo(nil)
-	}
+	// default:
+	// 	userInfo, err = service.GetUserInfo(&cks.Ck)
+	// }
 	if err != nil {
 		ErrorResponse(c, "无法获取用户信息，刷新失败: "+err.Error(), http.StatusBadRequest)
 		return

handlers/resource_handler.go

@@ -386,8 +386,8 @@ func GetResourceLink(c *gin.Context) {
 	}
 
 	// 如果不是夸克网盘，直接返回原链接
-	if panInfo.Name != "quark" {
+	if panInfo.Name != "quark" && panInfo.Name != "xunlei" {
-		utils.Info("非夸克资源，直接返回原链接")
+		utils.Info("非夸克和迅雷资源，直接返回原链接")
 		SuccessResponse(c, gin.H{
 			"url":         resource.URL,
 			"type":        "original",
@@ -397,9 +397,6 @@ func GetResourceLink(c *gin.Context) {
 		return
 	}
 
-	// 夸克资源处理逻辑
-	utils.Info("夸克资源处理开始")
-
 	// 如果已存在转存链接，直接返回
 	if resource.SaveURL != "" {
 		utils.Info("已存在转存链接，直接返回: %s", resource.SaveURL)
@@ -475,18 +472,11 @@ type TransferResult struct {
 func performAutoTransfer(resource *entity.Resource) TransferResult {
 	utils.Info("开始执行资源转存 - ID: %d, URL: %s", resource.ID, resource.URL)
 
-	// 获取夸克平台ID
+	// 平台ID
-	quarkPanID, err := getQuarkPanID()
+	panID := resource.PanID
-	if err != nil {
-		utils.Error("获取夸克平台ID失败: %v", err)
-		return TransferResult{
-			Success:  false,
-			ErrorMsg: fmt.Sprintf("获取夸克平台ID失败: %v", err),
-		}
-	}
 
 	// 获取可用的夸克账号
-	accounts, err := repoManager.CksRepository.FindAll()
+	accounts, err := repoManager.CksRepository.FindByPanID(*panID)
 	if err != nil {
 		utils.Error("获取网盘账号失败: %v", err)
 		return TransferResult{
@@ -495,34 +485,36 @@ func performAutoTransfer(resource *entity.Resource) TransferResult {
 		}
 	}
 
+	// 测试阶段，移除最小限制
 	// 获取最小存储空间配置
-	autoTransferMinSpace, err := repoManager.SystemConfigRepository.GetConfigInt(entity.ConfigKeyAutoTransferMinSpace)
+	// autoTransferMinSpace, err := repoManager.SystemConfigRepository.GetConfigInt(entity.ConfigKeyAutoTransferMinSpace)
-	if err != nil {
+	// if err != nil {
-		utils.Error("获取最小存储空间配置失败: %v", err)
+	// 	utils.Error("获取最小存储空间配置失败: %v", err)
-		autoTransferMinSpace = 5 // 默认5GB
+	// 	autoTransferMinSpace = 5 // 默认5GB
-	}
+	// }
 
-	// 过滤：只保留已激活、夸克平台、剩余空间足够的账号
+	// // 过滤：只保留已激活、夸克平台、剩余空间足够的账号
-	minSpaceBytes := int64(autoTransferMinSpace) * 1024 * 1024 * 1024
+	// minSpaceBytes := int64(autoTransferMinSpace) * 1024 * 1024 * 1024
-	var validAccounts []entity.Cks
+	// var validAccounts []entity.Cks
-	for _, acc := range accounts {
+	// for _, acc := range accounts {
-		if acc.IsValid && acc.PanID == quarkPanID && acc.LeftSpace >= minSpaceBytes {
+	// 	if acc.IsValid && acc.PanID == *panID && acc.LeftSpace >= minSpaceBytes {
-			validAccounts = append(validAccounts, acc)
+	// 		validAccounts = append(validAccounts, acc)
-		}
+	// 	}
-	}
+	// }
 
-	if len(validAccounts) == 0 {
+	// if len(validAccounts) == 0 {
-		utils.Info("没有可用的夸克网盘账号")
+	// 	utils.Info("没有可用的网盘账号")
-		return TransferResult{
+	// 	return TransferResult{
-			Success:  false,
+	// 		Success:  false,
-			ErrorMsg: "没有可用的夸克网盘账号",
+	// 		ErrorMsg: "没有可用的网盘账号",
-		}
+	// 	}
-	}
+	// }
 
-	utils.Info("找到 %d 个可用夸克网盘账号，开始转存处理...", len(validAccounts))
+	// utils.Info("找到 %d 个可用网盘账号，开始转存处理...", len(validAccounts))
 
 	// 使用第一个可用账号进行转存
-	account := validAccounts[0]
+	// account := validAccounts[0]
+	account := accounts[0]
 
 	// 创建网盘服务工厂
 	factory := pan.NewPanFactory()
@@ -566,6 +558,9 @@ func transferSingleResource(resource *entity.Resource, account entity.Cks, facto
 		}
 	}
 
+	// 设置账号信息
+	service.SetCKSRepository(repoManager.CksRepository, account)
+
 	// 提取分享ID
 	shareID, _ := commonutils.ExtractShareIdString(resource.URL)
 	if shareID == "" {
@@ -576,7 +571,7 @@ func transferSingleResource(resource *entity.Resource, account entity.Cks, facto
 	}
 
 	// 执行转存
-	transferResult, err := service.Transfer(shareID)
+	transferResult, err := service.Transfer(shareID) // 有些链接还需要其他信息从 url 中自行解析
 	if err != nil {
 		utils.Error("转存失败: %v", err)
 		return TransferResult{