mirror of https://github.com/Xhofe/alist
376 lines
10 KiB
Go
376 lines
10 KiB
Go
package pikpak
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"github.com/alist-org/alist/v3/drivers/base"
|
||
"github.com/alist-org/alist/v3/internal/driver"
|
||
"github.com/alist-org/alist/v3/internal/model"
|
||
"github.com/alist-org/alist/v3/internal/op"
|
||
"github.com/alist-org/alist/v3/pkg/utils"
|
||
hash_extend "github.com/alist-org/alist/v3/pkg/utils/hash"
|
||
"github.com/go-resty/resty/v2"
|
||
log "github.com/sirupsen/logrus"
|
||
"golang.org/x/oauth2"
|
||
"net/http"
|
||
"strconv"
|
||
"strings"
|
||
)
|
||
|
||
type PikPak struct {
|
||
model.Storage
|
||
Addition
|
||
*Common
|
||
RefreshToken string
|
||
AccessToken string
|
||
oauth2Token oauth2.TokenSource
|
||
}
|
||
|
||
func (d *PikPak) Config() driver.Config {
|
||
return config
|
||
}
|
||
|
||
func (d *PikPak) GetAddition() driver.Additional {
|
||
return &d.Addition
|
||
}
|
||
|
||
func (d *PikPak) Init(ctx context.Context) (err error) {
|
||
|
||
if d.Common == nil {
|
||
d.Common = &Common{
|
||
client: base.NewRestyClient(),
|
||
CaptchaToken: "",
|
||
UserID: "",
|
||
DeviceID: utils.GetMD5EncodeStr(d.Username + d.Password),
|
||
UserAgent: "",
|
||
RefreshCTokenCk: func(token string) {
|
||
d.Common.CaptchaToken = token
|
||
op.MustSaveDriverStorage(d)
|
||
},
|
||
}
|
||
}
|
||
|
||
if d.Platform == "android" {
|
||
d.ClientID = AndroidClientID
|
||
d.ClientSecret = AndroidClientSecret
|
||
d.ClientVersion = AndroidClientVersion
|
||
d.PackageName = AndroidPackageName
|
||
d.Algorithms = AndroidAlgorithms
|
||
d.UserAgent = BuildCustomUserAgent(utils.GetMD5EncodeStr(d.Username+d.Password), AndroidClientID, AndroidPackageName, AndroidSdkVersion, AndroidClientVersion, AndroidPackageName, "")
|
||
} else if d.Platform == "web" {
|
||
d.ClientID = WebClientID
|
||
d.ClientSecret = WebClientSecret
|
||
d.ClientVersion = WebClientVersion
|
||
d.PackageName = WebPackageName
|
||
d.Algorithms = WebAlgorithms
|
||
d.UserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/117.0.0.0 Safari/537.36"
|
||
}
|
||
|
||
if d.Addition.CaptchaToken != "" && d.Addition.RefreshToken == "" {
|
||
d.SetCaptchaToken(d.Addition.CaptchaToken)
|
||
}
|
||
|
||
if d.Addition.DeviceID != "" {
|
||
d.SetDeviceID(d.Addition.DeviceID)
|
||
} else {
|
||
d.Addition.DeviceID = d.Common.DeviceID
|
||
op.MustSaveDriverStorage(d)
|
||
}
|
||
// 初始化 oauth2Config
|
||
oauth2Config := &oauth2.Config{
|
||
ClientID: d.ClientID,
|
||
ClientSecret: d.ClientSecret,
|
||
Endpoint: oauth2.Endpoint{
|
||
AuthURL: "https://user.mypikpak.com/v1/auth/signin",
|
||
TokenURL: "https://user.mypikpak.com/v1/auth/token",
|
||
AuthStyle: oauth2.AuthStyleInParams,
|
||
},
|
||
}
|
||
|
||
// 如果已经有RefreshToken，直接获取AccessToken
|
||
if d.Addition.RefreshToken != "" {
|
||
// 使用 oauth2 刷新令牌
|
||
// 初始化 oauth2Token
|
||
d.oauth2Token = oauth2.ReuseTokenSource(nil, utils.TokenSource(func() (*oauth2.Token, error) {
|
||
return oauth2Config.TokenSource(ctx, &oauth2.Token{
|
||
RefreshToken: d.Addition.RefreshToken,
|
||
}).Token()
|
||
}))
|
||
} else {
|
||
// 如果没有填写RefreshToken，尝试登录 获取 refreshToken
|
||
if err := d.login(); err != nil {
|
||
return err
|
||
}
|
||
d.oauth2Token = oauth2.ReuseTokenSource(nil, utils.TokenSource(func() (*oauth2.Token, error) {
|
||
return oauth2Config.TokenSource(ctx, &oauth2.Token{
|
||
RefreshToken: d.RefreshToken,
|
||
}).Token()
|
||
}))
|
||
}
|
||
|
||
token, err := d.oauth2Token.Token()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
d.RefreshToken = token.RefreshToken
|
||
d.AccessToken = token.AccessToken
|
||
|
||
// 获取CaptchaToken
|
||
err = d.RefreshCaptchaTokenAtLogin(GetAction(http.MethodGet, "https://api-drive.mypikpak.com/drive/v1/files"), d.Username)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 获取用户ID
|
||
userID := token.Extra("sub").(string)
|
||
if userID != "" {
|
||
d.Common.SetUserID(userID)
|
||
}
|
||
// 更新UserAgent
|
||
if d.Platform == "android" {
|
||
d.Common.UserAgent = BuildCustomUserAgent(utils.GetMD5EncodeStr(d.Username+d.Password), AndroidClientID, AndroidPackageName, AndroidSdkVersion, AndroidClientVersion, AndroidPackageName, d.Common.UserID)
|
||
}
|
||
|
||
// 保存 有效的 RefreshToken
|
||
d.Addition.RefreshToken = d.RefreshToken
|
||
op.MustSaveDriverStorage(d)
|
||
return nil
|
||
}
|
||
|
||
func (d *PikPak) Drop(ctx context.Context) error {
|
||
return nil
|
||
}
|
||
|
||
func (d *PikPak) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||
files, err := d.getFiles(dir.GetID())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return utils.SliceConvert(files, func(src File) (model.Obj, error) {
|
||
return fileToObj(src), nil
|
||
})
|
||
}
|
||
|
||
func (d *PikPak) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||
var resp File
|
||
queryParams := map[string]string{
|
||
"_magic": "2021",
|
||
"usage": "FETCH",
|
||
"thumbnail_size": "SIZE_LARGE",
|
||
}
|
||
if !d.DisableMediaLink {
|
||
queryParams["usage"] = "CACHE"
|
||
}
|
||
_, err := d.request(fmt.Sprintf("https://api-drive.mypikpak.com/drive/v1/files/%s", file.GetID()),
|
||
http.MethodGet, func(req *resty.Request) {
|
||
req.SetQueryParams(queryParams)
|
||
}, &resp)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
link := model.Link{
|
||
URL: resp.WebContentLink,
|
||
}
|
||
if !d.DisableMediaLink && len(resp.Medias) > 0 && resp.Medias[0].Link.Url != "" {
|
||
log.Debugln("use media link")
|
||
link.URL = resp.Medias[0].Link.Url
|
||
}
|
||
return &link, nil
|
||
}
|
||
|
||
func (d *PikPak) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||
_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files", http.MethodPost, func(req *resty.Request) {
|
||
req.SetBody(base.Json{
|
||
"kind": "drive#folder",
|
||
"parent_id": parentDir.GetID(),
|
||
"name": dirName,
|
||
})
|
||
}, nil)
|
||
return err
|
||
}
|
||
|
||
func (d *PikPak) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||
_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files:batchMove", http.MethodPost, func(req *resty.Request) {
|
||
req.SetBody(base.Json{
|
||
"ids": []string{srcObj.GetID()},
|
||
"to": base.Json{
|
||
"parent_id": dstDir.GetID(),
|
||
},
|
||
})
|
||
}, nil)
|
||
return err
|
||
}
|
||
|
||
func (d *PikPak) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||
_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files/"+srcObj.GetID(), http.MethodPatch, func(req *resty.Request) {
|
||
req.SetBody(base.Json{
|
||
"name": newName,
|
||
})
|
||
}, nil)
|
||
return err
|
||
}
|
||
|
||
func (d *PikPak) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||
_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files:batchCopy", http.MethodPost, func(req *resty.Request) {
|
||
req.SetBody(base.Json{
|
||
"ids": []string{srcObj.GetID()},
|
||
"to": base.Json{
|
||
"parent_id": dstDir.GetID(),
|
||
},
|
||
})
|
||
}, nil)
|
||
return err
|
||
}
|
||
|
||
func (d *PikPak) Remove(ctx context.Context, obj model.Obj) error {
|
||
_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files:batchTrash", http.MethodPost, func(req *resty.Request) {
|
||
req.SetBody(base.Json{
|
||
"ids": []string{obj.GetID()},
|
||
})
|
||
}, nil)
|
||
return err
|
||
}
|
||
|
||
func (d *PikPak) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
||
hi := stream.GetHash()
|
||
sha1Str := hi.GetHash(hash_extend.GCID)
|
||
if len(sha1Str) < hash_extend.GCID.Width {
|
||
tFile, err := stream.CacheFullInTempFile()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
sha1Str, err = utils.HashFile(hash_extend.GCID, tFile, stream.GetSize())
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
var resp UploadTaskData
|
||
res, err := d.request("https://api-drive.mypikpak.com/drive/v1/files", http.MethodPost, func(req *resty.Request) {
|
||
req.SetBody(base.Json{
|
||
"kind": "drive#file",
|
||
"name": stream.GetName(),
|
||
"size": stream.GetSize(),
|
||
"hash": strings.ToUpper(sha1Str),
|
||
"upload_type": "UPLOAD_TYPE_RESUMABLE",
|
||
"objProvider": base.Json{"provider": "UPLOAD_TYPE_UNKNOWN"},
|
||
"parent_id": dstDir.GetID(),
|
||
"folder_type": "NORMAL",
|
||
})
|
||
}, &resp)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 秒传成功
|
||
if resp.Resumable == nil {
|
||
log.Debugln(string(res))
|
||
return nil
|
||
}
|
||
|
||
params := resp.Resumable.Params
|
||
//endpoint := strings.Join(strings.Split(params.Endpoint, ".")[1:], ".")
|
||
// web 端上传 返回的endpoint 为 `mypikpak.com` | android 端上传 返回的endpoint 为 `vip-lixian-07.mypikpak.com`·
|
||
if d.Addition.Platform == "android" {
|
||
params.Endpoint = "mypikpak.com"
|
||
}
|
||
|
||
if stream.GetSize() <= 10*utils.MB { // 文件大小 小于10MB，改用普通模式上传
|
||
return d.UploadByOSS(¶ms, stream, up)
|
||
}
|
||
// 分片上传
|
||
return d.UploadByMultipart(¶ms, stream.GetSize(), stream, up)
|
||
}
|
||
|
||
// 离线下载文件
|
||
func (d *PikPak) OfflineDownload(ctx context.Context, fileUrl string, parentDir model.Obj, fileName string) (*OfflineTask, error) {
|
||
requestBody := base.Json{
|
||
"kind": "drive#file",
|
||
"name": fileName,
|
||
"upload_type": "UPLOAD_TYPE_URL",
|
||
"url": base.Json{
|
||
"url": fileUrl,
|
||
},
|
||
"parent_id": parentDir.GetID(),
|
||
"folder_type": "",
|
||
}
|
||
|
||
var resp OfflineDownloadResp
|
||
_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files", http.MethodPost, func(req *resty.Request) {
|
||
req.SetBody(requestBody)
|
||
}, &resp)
|
||
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return &resp.Task, err
|
||
}
|
||
|
||
/*
|
||
获取离线下载任务列表
|
||
phase 可能的取值：
|
||
PHASE_TYPE_RUNNING, PHASE_TYPE_ERROR, PHASE_TYPE_COMPLETE, PHASE_TYPE_PENDING
|
||
*/
|
||
func (d *PikPak) OfflineList(ctx context.Context, nextPageToken string, phase []string) ([]OfflineTask, error) {
|
||
res := make([]OfflineTask, 0)
|
||
url := "https://api-drive.mypikpak.com/drive/v1/tasks"
|
||
|
||
if len(phase) == 0 {
|
||
phase = []string{"PHASE_TYPE_RUNNING", "PHASE_TYPE_ERROR", "PHASE_TYPE_COMPLETE", "PHASE_TYPE_PENDING"}
|
||
}
|
||
params := map[string]string{
|
||
"type": "offline",
|
||
"thumbnail_size": "SIZE_SMALL",
|
||
"limit": "10000",
|
||
"page_token": nextPageToken,
|
||
"with": "reference_resource",
|
||
}
|
||
|
||
// 处理 phase 参数
|
||
if len(phase) > 0 {
|
||
filters := base.Json{
|
||
"phase": map[string]string{
|
||
"in": strings.Join(phase, ","),
|
||
},
|
||
}
|
||
filtersJSON, err := json.Marshal(filters)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("failed to marshal filters: %w", err)
|
||
}
|
||
params["filters"] = string(filtersJSON)
|
||
}
|
||
|
||
var resp OfflineListResp
|
||
_, err := d.request(url, http.MethodGet, func(req *resty.Request) {
|
||
req.SetContext(ctx).
|
||
SetQueryParams(params)
|
||
}, &resp)
|
||
|
||
if err != nil {
|
||
return nil, fmt.Errorf("failed to get offline list: %w", err)
|
||
}
|
||
res = append(res, resp.Tasks...)
|
||
return res, nil
|
||
}
|
||
|
||
func (d *PikPak) DeleteOfflineTasks(ctx context.Context, taskIDs []string, deleteFiles bool) error {
|
||
url := "https://api-drive.mypikpak.com/drive/v1/tasks"
|
||
params := map[string]string{
|
||
"task_ids": strings.Join(taskIDs, ","),
|
||
"delete_files": strconv.FormatBool(deleteFiles),
|
||
}
|
||
_, err := d.request(url, http.MethodDelete, func(req *resty.Request) {
|
||
req.SetContext(ctx).
|
||
SetQueryParams(params)
|
||
}, nil)
|
||
if err != nil {
|
||
return fmt.Errorf("failed to delete tasks %v: %w", taskIDs, err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
var _ driver.Driver = (*PikPak)(nil)
|