forked from alist-org/alist
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(115): add offline download tool (close alist-org#6888)
- Loading branch information
1 parent
d4285b7
commit cec0e50
Showing
7 changed files
with
166 additions
and
13 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,124 @@ | ||
package _115 | ||
|
||
import ( | ||
"context" | ||
"fmt" | ||
|
||
"github.com/alist-org/alist/v3/drivers/115" | ||
"github.com/alist-org/alist/v3/internal/errs" | ||
"github.com/alist-org/alist/v3/internal/model" | ||
"github.com/alist-org/alist/v3/internal/offline_download/tool" | ||
"github.com/alist-org/alist/v3/internal/op" | ||
) | ||
|
||
type Cloud115 struct { | ||
refreshTaskCache bool | ||
} | ||
|
||
func (p *Cloud115) Name() string { | ||
return "115 Cloud" | ||
} | ||
|
||
func (p *Cloud115) Items() []model.SettingItem { | ||
return nil | ||
} | ||
|
||
func (p *Cloud115) Run(task *tool.DownloadTask) error { | ||
return errs.NotSupport | ||
} | ||
|
||
func (p *Cloud115) Init() (string, error) { | ||
p.refreshTaskCache = false | ||
return "ok", nil | ||
} | ||
|
||
func (p *Cloud115) IsReady() bool { | ||
return true | ||
} | ||
|
||
func (p *Cloud115) AddURL(args *tool.AddUrlArgs) (string, error) { | ||
// 添加新任务刷新缓存 | ||
p.refreshTaskCache = true | ||
// args.TempDir 已经被修改为了 DstDirPath | ||
storage, actualPath, err := op.GetStorageAndActualPath(args.TempDir) | ||
if err != nil { | ||
return "", err | ||
} | ||
driver115, ok := storage.(*_115.Pan115) | ||
if !ok { | ||
return "", fmt.Errorf("unsupported storage driver for offline download, only 115 Cloud is supported") | ||
} | ||
|
||
ctx := context.Background() | ||
parentDir, err := op.GetUnwrap(ctx, storage, actualPath) | ||
if err != nil { | ||
return "", err | ||
} | ||
|
||
hashs, err := driver115.OfflineDownload(ctx, []string{args.Url}, parentDir) | ||
if err != nil || len(hashs) < 1 { | ||
return "", fmt.Errorf("failed to add offline download task: %w", err) | ||
} | ||
|
||
return hashs[0], nil | ||
} | ||
|
||
func (p *Cloud115) Remove(task *tool.DownloadTask) error { | ||
storage, _, err := op.GetStorageAndActualPath(task.DstDirPath) | ||
if err != nil { | ||
return err | ||
} | ||
driver115, ok := storage.(*_115.Pan115) | ||
if !ok { | ||
return fmt.Errorf("unsupported storage driver for offline download, only 115 Cloud is supported") | ||
} | ||
|
||
ctx := context.Background() | ||
if err := driver115.DeleteOfflineTasks(ctx, []string{task.GID}, false); err != nil { | ||
return err | ||
} | ||
return nil | ||
} | ||
|
||
func (p *Cloud115) Status(task *tool.DownloadTask) (*tool.Status, error) { | ||
storage, _, err := op.GetStorageAndActualPath(task.DstDirPath) | ||
if err != nil { | ||
return nil, err | ||
} | ||
driver115, ok := storage.(*_115.Pan115) | ||
if !ok { | ||
return nil, fmt.Errorf("unsupported storage driver for offline download, only 115 Cloud is supported") | ||
} | ||
|
||
tasks, err := driver115.OfflineList(context.Background()) | ||
if err != nil { | ||
return nil, err | ||
} | ||
|
||
s := &tool.Status{ | ||
Progress: 0, | ||
NewGID: "", | ||
Completed: false, | ||
Status: "the task has been deleted", | ||
Err: nil, | ||
} | ||
for _, t := range tasks { | ||
if t.InfoHash == task.GID { | ||
s.Progress = t.Percent | ||
s.Status = t.GetStatus() | ||
s.Completed = t.IsDone() | ||
if t.IsFailed() { | ||
s.Err = fmt.Errorf(t.GetStatus()) | ||
} | ||
return s, nil | ||
} | ||
} | ||
s.Err = fmt.Errorf("the task has been deleted") | ||
return nil, nil | ||
} | ||
|
||
var _ tool.Tool = (*Cloud115)(nil) | ||
|
||
func init() { | ||
tool.Tools.Add(&Cloud115{}) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters