mirror of https://github.com/Xhofe/alist
326 lines
8.6 KiB
Go
326 lines
8.6 KiB
Go
package pikpak
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/Xhofe/alist/conf"
|
|
"github.com/Xhofe/alist/drivers/base"
|
|
"github.com/Xhofe/alist/model"
|
|
"github.com/Xhofe/alist/utils"
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/s3/s3manager"
|
|
jsoniter "github.com/json-iterator/go"
|
|
log "github.com/sirupsen/logrus"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type PikPak struct{}
|
|
|
|
func (driver PikPak) Config() base.DriverConfig {
|
|
return base.DriverConfig{
|
|
Name: "PikPak",
|
|
ApiProxy: true,
|
|
LocalSort: true,
|
|
}
|
|
}
|
|
|
|
func (driver PikPak) Items() []base.Item {
|
|
return []base.Item{
|
|
{
|
|
Name: "username",
|
|
Label: "username",
|
|
Type: base.TypeString,
|
|
Required: true,
|
|
},
|
|
{
|
|
Name: "password",
|
|
Label: "password",
|
|
Type: base.TypeString,
|
|
Required: true,
|
|
},
|
|
{
|
|
Name: "root_folder",
|
|
Label: "root folder id",
|
|
Type: base.TypeString,
|
|
Required: false,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (driver PikPak) Save(account *model.Account, old *model.Account) error {
|
|
if account == nil {
|
|
return nil
|
|
}
|
|
err := driver.Login(account)
|
|
return err
|
|
}
|
|
|
|
func (driver PikPak) File(path string, account *model.Account) (*model.File, error) {
|
|
path = utils.ParsePath(path)
|
|
if path == "/" {
|
|
return &model.File{
|
|
Id: account.RootFolder,
|
|
Name: account.Name,
|
|
Size: 0,
|
|
Type: conf.FOLDER,
|
|
Driver: driver.Config().Name,
|
|
UpdatedAt: account.UpdatedAt,
|
|
}, nil
|
|
}
|
|
dir, name := filepath.Split(path)
|
|
files, err := driver.Files(dir, account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, file := range files {
|
|
if file.Name == name {
|
|
return &file, nil
|
|
}
|
|
}
|
|
return nil, base.ErrPathNotFound
|
|
}
|
|
|
|
func (driver PikPak) Files(path string, account *model.Account) ([]model.File, error) {
|
|
path = utils.ParsePath(path)
|
|
var files []model.File
|
|
cache, err := base.GetCache(path, account)
|
|
if err == nil {
|
|
files, _ = cache.([]model.File)
|
|
} else {
|
|
file, err := driver.File(path, account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rawFiles, err := driver.GetFiles(file.Id, account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
files = make([]model.File, 0)
|
|
for _, file := range rawFiles {
|
|
files = append(files, *driver.FormatFile(&file))
|
|
}
|
|
if len(files) > 0 {
|
|
_ = base.SetCache(path, files, account)
|
|
}
|
|
}
|
|
return files, nil
|
|
}
|
|
|
|
func (driver PikPak) Link(args base.Args, account *model.Account) (*base.Link, error) {
|
|
file, err := driver.File(args.Path, account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var resp File
|
|
_, err = driver.Request(fmt.Sprintf("https://api-drive.mypikpak.com/drive/v1/files/%s?_magic=2021&thumbnail_size=SIZE_LARGE", file.Id),
|
|
base.Get, nil, nil, &resp, account)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &base.Link{
|
|
Url: resp.WebContentLink,
|
|
}, nil
|
|
}
|
|
|
|
func (driver PikPak) Path(path string, account *model.Account) (*model.File, []model.File, error) {
|
|
path = utils.ParsePath(path)
|
|
log.Debugf("pikpak path: %s", path)
|
|
file, err := driver.File(path, account)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
if !file.IsDir() {
|
|
return file, nil, nil
|
|
}
|
|
files, err := driver.Files(path, account)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return nil, files, nil
|
|
}
|
|
|
|
//func (driver PikPak) Proxy(r *http.Request, account *model.Account) {
|
|
//
|
|
//}
|
|
|
|
func (driver PikPak) Preview(path string, account *model.Account) (interface{}, error) {
|
|
return nil, base.ErrNotSupport
|
|
}
|
|
|
|
func (driver PikPak) MakeDir(path string, account *model.Account) error {
|
|
path = utils.ParsePath(path)
|
|
dir, name := filepath.Split(path)
|
|
parentFile, err := driver.File(dir, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !parentFile.IsDir() {
|
|
return base.ErrNotFolder
|
|
}
|
|
_, err = driver.Request("https://api-drive.mypikpak.com/drive/v1/files", base.Post, nil, &base.Json{
|
|
"kind": "drive#folder",
|
|
"parent_id": parentFile.Id,
|
|
"name": name,
|
|
}, nil, account)
|
|
return err
|
|
}
|
|
|
|
func (driver PikPak) Move(src string, dst string, account *model.Account) error {
|
|
dstDir, _ := filepath.Split(dst)
|
|
srcFile, err := driver.File(src, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
dstDirFile, err := driver.File(dstDir, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = driver.Request("https://api-drive.mypikpak.com/drive/v1/files:batchMove", base.Post, nil, &base.Json{
|
|
"ids": []string{srcFile.Id},
|
|
"to": base.Json{
|
|
"parent_id": dstDirFile.Id,
|
|
},
|
|
}, nil, account)
|
|
return err
|
|
}
|
|
|
|
func (driver PikPak) Rename(src string, dst string, account *model.Account) error {
|
|
_, dstName := filepath.Split(dst)
|
|
srcFile, err := driver.File(src, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = driver.Request("https://api-drive.mypikpak.com/drive/v1/files/"+srcFile.Id, base.Patch, nil, &base.Json{
|
|
"name": dstName,
|
|
}, nil, account)
|
|
return err
|
|
}
|
|
|
|
func (driver PikPak) Copy(src string, dst string, account *model.Account) error {
|
|
srcFile, err := driver.File(src, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
dstDirFile, err := driver.File(utils.Dir(dst), account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = driver.Request("https://api-drive.mypikpak.com/drive/v1/files:batchCopy", base.Post, nil, &base.Json{
|
|
"ids": []string{srcFile.Id},
|
|
"to": base.Json{
|
|
"parent_id": dstDirFile.Id,
|
|
},
|
|
}, nil, account)
|
|
return err
|
|
}
|
|
|
|
func (driver PikPak) Delete(path string, account *model.Account) error {
|
|
file, err := driver.File(path, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = driver.Request("https://api-drive.mypikpak.com/drive/v1/files:batchTrash", base.Post, nil, &base.Json{
|
|
"ids": []string{file.Id},
|
|
}, nil, account)
|
|
return err
|
|
}
|
|
|
|
func (driver PikPak) Upload(file *model.FileStream, account *model.Account) error {
|
|
if file == nil {
|
|
return base.ErrEmptyFile
|
|
}
|
|
parentFile, err := driver.File(file.ParentPath, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
data := base.Json{
|
|
"kind": "drive#file",
|
|
"name": file.GetFileName(),
|
|
"size": file.GetSize(),
|
|
"hash": "1CF254FBC456E1B012CD45C546636AA62CF8350E",
|
|
"upload_type": "UPLOAD_TYPE_RESUMABLE",
|
|
"objProvider": base.Json{"provider": "UPLOAD_TYPE_UNKNOWN"},
|
|
"parent_id": parentFile.Id,
|
|
}
|
|
res, err := driver.Request("https://api-drive.mypikpak.com/drive/v1/files", base.Post, nil, &data, nil, account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
params := jsoniter.Get(res, "resumable").Get("params")
|
|
endpoint := params.Get("endpoint").ToString()
|
|
endpointS := strings.Split(endpoint, ".")
|
|
endpoint = strings.Join(endpointS[1:], ".")
|
|
accessKeyId := params.Get("access_key_id").ToString()
|
|
accessKeySecret := params.Get("access_key_secret").ToString()
|
|
securityToken := params.Get("security_token").ToString()
|
|
key := params.Get("key").ToString()
|
|
bucket := params.Get("bucket").ToString()
|
|
cfg := &aws.Config{
|
|
Credentials: credentials.NewStaticCredentials(accessKeyId, accessKeySecret, securityToken),
|
|
Region: aws.String("pikpak"),
|
|
Endpoint: &endpoint,
|
|
}
|
|
s, err := session.NewSession(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
uploader := s3manager.NewUploader(s)
|
|
input := &s3manager.UploadInput{
|
|
Bucket: &bucket,
|
|
Key: &key,
|
|
Body: file,
|
|
}
|
|
_, err = uploader.Upload(input)
|
|
return err
|
|
}
|
|
|
|
// use aliyun-oss-sdk
|
|
//func (driver PikPak) Upload(file *model.FileStream, account *model.Account) error {
|
|
// if file == nil {
|
|
// return base.ErrEmptyFile
|
|
// }
|
|
// parentFile, err := driver.File(file.ParentPath, account)
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// data := base.Json{
|
|
// "kind": "drive#file",
|
|
// "name": file.GetFileName(),
|
|
// "size": file.GetSize(),
|
|
// "hash": "1CF254FBC456E1B012CD45C546636AA62CF8350E",
|
|
// "upload_type": "UPLOAD_TYPE_RESUMABLE",
|
|
// "objProvider": base.Json{"provider": "UPLOAD_TYPE_UNKNOWN"},
|
|
// "parent_id": parentFile.Id,
|
|
// }
|
|
// res, err := driver.Request("https://api-drive.mypikpak.com/drive/v1/files", base.Post, nil, &data, nil, account)
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// params := jsoniter.Get(res, "resumable").Get("params")
|
|
// endpoint := params.Get("endpoint").ToString()
|
|
// endpointS := strings.Split(endpoint, ".")
|
|
// endpoint = strings.Join(endpointS[1:], ".")
|
|
// accessKeyId := params.Get("access_key_id").ToString()
|
|
// accessKeySecret := params.Get("access_key_secret").ToString()
|
|
// securityToken := params.Get("security_token").ToString()
|
|
// client, err := oss.New("https://"+endpoint, accessKeyId,
|
|
// accessKeySecret, oss.SecurityToken(securityToken))
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// bucket, err := client.Bucket(params.Get("bucket").ToString())
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// signedURL, err := bucket.SignURL(params.Get("key").ToString(), oss.HTTPPut, 60)
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// err = bucket.PutObjectWithURL(signedURL, file)
|
|
// return err
|
|
//}
|
|
|
|
var _ base.Driver = (*PikPak)(nil)
|