ProjectHentai/alist
6
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🎇 123pan upload

Browse Source
This commit is contained in:
微凉
2022-01-03 22:29:32 +08:00
parent beb06f2f7f
commit 84fc0ab1bd
5 changed files with 182 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
drivers/pikpak/driver.go
View File

@ -6,7 +6,10 @@ import (
"github.com/Xhofe/alist/drivers/base"
"github.com/Xhofe/alist/model"
"github.com/Xhofe/alist/utils"
"github.com/aliyun/aliyun-oss-go-sdk/oss"
"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"
"github.com/gin-gonic/gin"
jsoniter "github.com/json-iterator/go"
log "github.com/sirupsen/logrus"
@ -250,21 +253,71 @@ func (driver PikPak) Upload(file *model.FileStream, account *model.Account) erro
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))
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
}
bucket, err := client.Bucket(params.Get("bucket").ToString())
if err != nil {
return err
uploader := s3manager.NewUploader(s)
input := &s3manager.UploadInput{
Bucket: &bucket,
Key: &key,
Body: file,
}
signedURL, err := bucket.SignURL(params.Get("key").ToString(), oss.HTTPPut, 60)
if err != nil {
return err
}
err = bucket.PutObjectWithURL(signedURL, 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)