client.go

package baidudisk

import (
	"fmt"
	"log/slog"
	"net/http"
	"path/filepath"
	"strconv"
	"strings"
	"time"

	"github.com/imroc/req/v3"
	"github.com/spf13/cast"

	"github.com/fghwett/BaiduNetdiskTransfer/config"
)

const (
	appid   = "38824127" // 固定值 网页版为 "250528"
	channel = "chunlei"  // 固定值
)

type Client struct {
	appid    string
	channel  string
	provider config.Provider

	client *req.Client
	logger *slog.Logger
}

func NewClient(provider config.Provider, logger *slog.Logger) (*Client, error) {
	conf := provider.GetConfig()

	client := req.C().
		SetBaseURL("https://pan.baidu.com").
		SetCookieJar(provider).
		SetCommonHeaders(map[string]string{
			"Host":                      "pan.baidu.com",
			"Connection":                "keep-alive",
			"Upgrade-Insecure-Requests": "1",
			"Sec-Fetch-Dest":            "document",
			"Accept":                    "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9",
			"Sec-Fetch-Site":            "same-site",
			"Sec-Fetch-Mode":            "navigate",
			"Referer":                   "https://pan.baidu.com",
			"Accept-Language":           "zh-CN,zh;q=0.9,en;q=0.8,en-US;q=0.7,en-GB;q=0.6,ru;q=0.5",
			"User-Agent":                conf.UserAgent,
			//"Accept-Encoding":         "gzip, deflate, br",
		}).
		SetCommonQueryParams(map[string]string{
			"web":        "1",
			"clienttype": "0",
		})

	c := &Client{
		appid:    appid,
		channel:  channel,
		provider: provider,
		client:   client,
		logger:   logger,
	}
	return c, nil
}

// CheckValid 检查百度网盘链接有效性
func (client *Client) CheckValid(link string, password string) (bool, error) {
	logger := client.logger.With(
		slog.String("action", "检查百度网盘链接有效性"),
		slog.String("link", link),
		slog.String("password", password),
	)

	if !strings.Contains(link, "?pwd=") {
		link = fmt.Sprintf("%s?pwd=%s", link, password)
	}

	resp, err := req.C().
		SetRedirectPolicy(req.NoRedirectPolicy()).
		R().
		SetHeader("user-agent", client.provider.GetConfig().UserAgent).
		Get(link)
	if err != nil {
		logger.Error("检查失败", slog.Any("err", err))
		return false, err
	}
	if _, err = resp.Location(); err != nil {
		logger.Info("链接已失效", slog.Any("err", err))
		return false, nil
	}
	logger.Debug("链接有效")

	return true, nil
}

// GetQuota 获取空间配额信息
func (client *Client) GetQuota() (*QuotaResponse, error) {
	resp, err := client.client.R().Get("/api/quota")
	if err != nil {
		return nil, err
	}
	result := new(QuotaResponse)
	if err = resp.Unmarshal(&result); err != nil {
		return nil, err
	}
	return result, nil
}

// GetFileList 获取文件列表
func (client *Client) GetFileList(folder string) (*FileListResponse, error) {
	resp, err := client.client.R().
		SetQueryParams(map[string]string{
			"order":     "time",
			"desc":      "1",
			"showempty": "0",
			"page":      "1",
			"num":       "1000",
			"dir":       folder,
			"bdstoken":  client.provider.GetConfig().BdsToken,
		}).
		Get("/api/list")
	if err != nil {
		return nil, err
	}
	result := new(FileListResponse)
	if err = resp.Unmarshal(&result); err != nil {
		return nil, err
	}
	return result, nil
}

// GetTemplateVariable 获取变量信息
func (client *Client) GetTemplateVariable() (*GettemplatevariableResponse, error) {
	resp, err := client.client.R().
		SetQueryParams(map[string]string{
			"appid":  client.appid,
			"fields": `["bdstoken","token","uk","isdocuser","servertime"]`,
		}).
		Get("/api/gettemplatevariable")
	if err != nil {
		return nil, err
	}
	result := new(GettemplatevariableResponse)
	if err = resp.Unmarshal(&result); err != nil {
		return nil, err
	}
	return result, nil
}

// CreateFolder 创建文件夹
func (client *Client) CreateFolder(folder string) (*CreateFolderResponse, error) {
	resp, err := client.client.R().
		SetQueryParams(map[string]string{
			"a":        "commit",
			"bdstoken": client.provider.GetConfig().BdsToken,
		}).
		SetFormData(map[string]string{
			"path":       folder,
			"isdir":      "1",
			"block_list": "[]",
		}).
		Post("/api/create")
	if err != nil {
		return nil, err
	}
	result := new(CreateFolderResponse)
	if err = resp.Unmarshal(&result); err != nil {
		return nil, err
	}
	return result, nil
}

// ShareVerify 验证分享链接
func (client *Client) ShareVerify(link, password string) (*ShareVerifyResponse, error) {
	surl, pass := client.parseLink(link, password)

	resp, err := client.client.R().
		SetQueryParams(map[string]string{
			"surl":     surl,
			"bdstoken": client.provider.GetConfig().BdsToken,
			"t":        strconv.FormatInt(time.Now().UnixMilli(), 10),
			"channel":  client.channel,
		}).
		SetFormData(map[string]string{
			"pwd":       pass,
			"vcode":     "",
			"vcode_str": "",
		}).
		Post("/share/verify")
	if err != nil {
		return nil, err
	}
	result := new(ShareVerifyResponse)
	if err = resp.Unmarshal(&result); err != nil {
		return nil, err
	}
	return result, nil
}

func (client *Client) parseLink(link, password string) (string, string) {
	if strings.Contains(link, "?pwd=") {
		arr := strings.Split(link, "?pwd=")
		link = arr[0]
		if len(arr) >= 2 {
			password = arr[1]
		}
	}
	surl := filepath.Base(link) // 取短链
	if len(surl) == 23 {
		surl = surl[1:] // 取最终链接 22位短链，个人端内分享出来的短链需要去掉前缀 1
	}
	return surl, password
}

// GetShareList 获取分享列表
func (client *Client) GetShareList(link, password string) (*ShareListResponse, error) {

	resp, err := req.C().
		SetBaseURL("https://pan.baidu.com").
		SetCommonHeader("User-Agent", "netdisk").
		SetCommonCookies(
			&http.Cookie{Name: "BDUSS", Value: client.provider.GetCookieValue("BDUSS")},
			&http.Cookie{Name: "ndut_fmt"},
		).
		R().
		SetFormData(map[string]string{
			"pwd":      password,
			"root":     "1",
			"shorturl": filepath.Base(link),
		}).
		Post("share/wxlist?channel=weixin&version=2.2.2&clienttype=25&web=1")
	if err != nil {
		return nil, err
	}
	result := new(ShareListResponse)
	if err = resp.Unmarshal(&result); err != nil {
		return nil, err
	}
	return result, nil
}

// ShareTransfer 分享链接转存
func (client *Client) ShareTransfer(randsk, shareId, from string, fsIdList []string, path string) (*ShareTransferResponse, error) {
	resp, err := client.client.R().
		SetCookies(&http.Cookie{
			Name:  "BDCLND",
			Value: randsk,
		}).
		SetQueryParams(map[string]string{
			"shareid":  shareId, // shareid 是文件 id
			"from":     from,    // share_uk 猜是分享者的 id
			"bdstoken": client.provider.GetConfig().BdsToken,
			"channel":  client.channel,
			"ondup":    "newcopy",
		}).
		SetFormData(map[string]string{
			// 针对一个分享链接带有多个分享文件的情况，转换一下列表格式
			"fsidlist": fmt.Sprintf("[%s]", strings.Join(fsIdList, ",")),
			"path":     path,
		}).
		Post("/share/transfer")
	if err != nil {
		return nil, err
	}
	result := new(ShareTransferResponse)
	if err = resp.Unmarshal(&result); err != nil {
		return nil, err
	}
	return result, nil
}

// ShareTransferWorkflow 分享链接转存工作流
func (client *Client) ShareTransferWorkflow(link, password, path string) (*ShareTransferResponse, error) {
	logger := client.logger.With(
		slog.String("action", "百度网盘分享链接转存工作流"),
		slog.String("link", link),
		slog.String("password", password),
		slog.String("path", path),
	)

	shareListResp, err := client.GetShareList(link, password)
	if err != nil {
		logger.Error("获取分享列表请求失败", slog.Any("err", err))
		return nil, err
	}
	if shareListResp.Errno != 0 {
		logger.Error("获取分享列表失败", slog.Any("resp", shareListResp))
		return nil, err
	}

	var verifyResp *ShareVerifyResponse
	if verifyResp, err = client.ShareVerify(link, password); err != nil {
		logger.Error("验证密码请求失败", slog.Any("err", err))
		return nil, err
	}
	if verifyResp.Errno != 0 {
		logger.Error("验证密码失败", slog.Any("resp", verifyResp))
		return nil, err
	}

	var listResp *FileListResponse
	if listResp, err = client.GetFileList(path); err != nil {
		logger.Error("获取文件列表请求失败", slog.Any("err", err))
		return nil, err
	}
	if listResp.Errno == ErrCodeNoFileOrDir {
		logger.Info("文件夹不存在 即将创建文件夹", slog.Any("resp", listResp))

		var createFolderResp *CreateFolderResponse
		if createFolderResp, err = client.CreateFolder(path); err != nil {
			logger.Error("创建文件夹请求失败", slog.Any("err", err))
			return nil, err
		}
		if createFolderResp.Errno != 0 {
			logger.Error("创建文件夹失败", slog.Any("resp", createFolderResp))
			return nil, err
		}

	} else if listResp.Errno != 0 {
		logger.Error("获取文件列表失败", slog.Any("resp", listResp))
		return nil, err
	}

	randSk := verifyResp.Randsk
	shareId := cast.ToString(shareListResp.Data.Shareid)
	uk := cast.ToString(shareListResp.Data.Uk)

	var list []string
	for _, v := range shareListResp.Data.List {
		list = append(list, v.FsId.String())
	}

	var transferResp *ShareTransferResponse
	if transferResp, err = client.ShareTransfer(randSk, shareId, uk, list, path); err != nil {
		logger.Error("文件转存请求失败", slog.Any("err", err))
		return nil, err
	}
	if transferResp.Errno != 0 {
		logger.Error("文件转存失败", slog.Any("resp", transferResp))
		return nil, err
	}

	return transferResp, nil
}