mirror of
https://github.com/OpenListTeam/OpenList.git
synced 2025-07-18 17:38:07 +08:00
3838ef0663
* feat(traffic): update progress when caching file * 调整参数位置和命名 --------- Co-authored-by: j2rong4cn <j2rong@qq.com>
131 lines
3.3 KiB
Go
131 lines
3.3 KiB
Go
package _123_open
|
||
|
||
import (
|
||
"context"
|
||
"strconv"
|
||
|
||
"github.com/OpenListTeam/OpenList/v4/internal/driver"
|
||
"github.com/OpenListTeam/OpenList/v4/internal/errs"
|
||
"github.com/OpenListTeam/OpenList/v4/internal/model"
|
||
"github.com/OpenListTeam/OpenList/v4/internal/op"
|
||
"github.com/OpenListTeam/OpenList/v4/internal/stream"
|
||
"github.com/OpenListTeam/OpenList/v4/pkg/utils"
|
||
)
|
||
|
||
type Open123 struct {
|
||
model.Storage
|
||
Addition
|
||
}
|
||
|
||
func (d *Open123) Config() driver.Config {
|
||
return config
|
||
}
|
||
|
||
func (d *Open123) GetAddition() driver.Additional {
|
||
return &d.Addition
|
||
}
|
||
|
||
func (d *Open123) Init(ctx context.Context) error {
|
||
if d.UploadThread < 1 || d.UploadThread > 32 {
|
||
d.UploadThread = 3
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
func (d *Open123) Drop(ctx context.Context) error {
|
||
op.MustSaveDriverStorage(d)
|
||
return nil
|
||
}
|
||
|
||
func (d *Open123) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
||
fileLastId := int64(0)
|
||
parentFileId, err := strconv.ParseInt(dir.GetID(), 10, 64)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
res := make([]File, 0)
|
||
|
||
for fileLastId != -1 {
|
||
files, err := d.getFiles(parentFileId, 100, fileLastId)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
// 目前123panAPI请求，trashed失效，只能通过遍历过滤
|
||
for i := range files.Data.FileList {
|
||
if files.Data.FileList[i].Trashed == 0 {
|
||
res = append(res, files.Data.FileList[i])
|
||
}
|
||
}
|
||
fileLastId = files.Data.LastFileId
|
||
}
|
||
return utils.SliceConvert(res, func(src File) (model.Obj, error) {
|
||
return src, nil
|
||
})
|
||
}
|
||
|
||
func (d *Open123) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
||
fileId, _ := strconv.ParseInt(file.GetID(), 10, 64)
|
||
|
||
res, err := d.getDownloadInfo(fileId)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
link := model.Link{URL: res.Data.DownloadUrl}
|
||
return &link, nil
|
||
}
|
||
|
||
func (d *Open123) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
||
parentFileId, _ := strconv.ParseInt(parentDir.GetID(), 10, 64)
|
||
|
||
return d.mkdir(parentFileId, dirName)
|
||
}
|
||
|
||
func (d *Open123) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||
toParentFileID, _ := strconv.ParseInt(dstDir.GetID(), 10, 64)
|
||
|
||
return d.move(srcObj.(File).FileId, toParentFileID)
|
||
}
|
||
|
||
func (d *Open123) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
||
fileId, _ := strconv.ParseInt(srcObj.GetID(), 10, 64)
|
||
|
||
return d.rename(fileId, newName)
|
||
}
|
||
|
||
func (d *Open123) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
||
return errs.NotSupport
|
||
}
|
||
|
||
func (d *Open123) Remove(ctx context.Context, obj model.Obj) error {
|
||
fileId, _ := strconv.ParseInt(obj.GetID(), 10, 64)
|
||
|
||
return d.trash(fileId)
|
||
}
|
||
|
||
func (d *Open123) Put(ctx context.Context, dstDir model.Obj, file model.FileStreamer, up driver.UpdateProgress) error {
|
||
parentFileId, err := strconv.ParseInt(dstDir.GetID(), 10, 64)
|
||
etag := file.GetHash().GetHash(utils.MD5)
|
||
|
||
if len(etag) < utils.MD5.Width {
|
||
cacheFileProgress := model.UpdateProgressWithRange(up, 0, 50)
|
||
up = model.UpdateProgressWithRange(up, 50, 100)
|
||
_, etag, err = stream.CacheFullInTempFileAndHash(file, cacheFileProgress, utils.MD5)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
createResp, err := d.create(parentFileId, file.GetName(), etag, file.GetSize(), 2, false)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if createResp.Data.Reuse {
|
||
return nil
|
||
}
|
||
|
||
return d.Upload(ctx, file, createResp, up)
|
||
}
|
||
|
||
var _ driver.Driver = (*Open123)(nil)
|