Feat: creat uri aria2 download task

pkg/aria2/aria2.go

@@ -0,0 +1,75 @@
+package aria2
+
+import (
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/serializer"
+	"github.com/HFO4/cloudreve/pkg/util"
+	"net/url"
+)
+
+// Instance 默认使用的Aria2处理实例
+var Instance Aria2 = &DummyAria2{}
+
+// Aria2 离线下载处理接口
+type Aria2 interface {
+	// CreateTask 创建新的任务
+	CreateTask(task *model.Download) error
+}
+
+const (
+	// URLTask 从URL添加的任务
+	URLTask = iota
+	// TorrentTask 种子任务
+	TorrentTask
+)
+
+const (
+	// Ready 准备就绪
+	Ready = iota
+)
+
+var (
+	// ErrNotEnabled 功能未开启错误
+	ErrNotEnabled = serializer.NewError(serializer.CodeNoPermissionErr, "离线下载功能未开启", nil)
+)
+
+// DummyAria2 未开启Aria2功能时使用的默认处理器
+type DummyAria2 struct {
+}
+
+// CreateTask 创建新任务，此处直接返回未开启错误
+func (instance *DummyAria2) CreateTask(task *model.Download) error {
+	return ErrNotEnabled
+}
+
+// Init 初始化
+func Init() {
+	options := model.GetSettingByNames("aria2_rpcurl", "aria2_token", "aria2_options")
+	timeout := model.GetIntSetting("aria2_call_timeout", 5)
+	if options["aria2_rpcurl"] == "" {
+		// 未开启Aria2服务
+		return
+	}
+
+	util.Log().Info("初始化 aria2 RPC 服务[%s]", options["aria2_rpcurl"])
+	client := &RPCService{}
+	if previousClient, ok := Instance.(*RPCService); ok {
+		client = previousClient
+	}
+
+	// 解析RPC服务地址
+	server, err := url.Parse(options["aria2_rpcurl"])
+	if err != nil {
+		util.Log().Warning("无法解析 aria2 RPC 服务地址，%s", err)
+		return
+	}
+	server.Path = "/jsonrpc"
+
+	// todo 加载自定义下载配置
+	if err := client.Init(server.String(), options["aria2_token"], timeout, []interface{}{}); err != nil {
+		util.Log().Warning("初始化 aria2 RPC 服务失败，%s", err)
+		return
+	}
+
+	Instance = client
+}

pkg/aria2/caller.go

@@ -0,0 +1,58 @@
+package aria2
+
+import (
+	"context"
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/zyxar/argo/rpc"
+	"path/filepath"
+	"strconv"
+	"time"
+)
+
+// RPCService 通过RPC服务的Aria2任务管理器
+type RPCService struct {
+	options *clientOptions
+	caller  rpc.Client
+}
+
+type clientOptions struct {
+	Options []interface{} // 创建下载时额外添加的设置
+}
+
+// Init 初始化
+func (client *RPCService) Init(server, secret string, timeout int, options []interface{}) error {
+	// 客户端已存在，则关闭先前连接
+	if client.caller != nil {
+		client.caller.Close()
+	}
+
+	client.options = &clientOptions{
+		Options: options,
+	}
+	caller, err := rpc.New(context.Background(), server, secret, time.Duration(timeout)*time.Second,
+		rpc.DummyNotifier{})
+	client.caller = caller
+	return err
+}
+
+// CreateTask 创建新任务
+func (client *RPCService) CreateTask(task *model.Download) error {
+	// 生成存储路径
+	task.Path = filepath.Join(
+		model.GetSettingByName("aria2_temp_path"),
+		"aria2",
+		strconv.FormatInt(time.Now().UnixNano(), 10),
+	)
+
+	// 创建下载任务
+	gid, err := client.caller.AddURI(task.Source, map[string]string{"dir": task.Path})
+	if err != nil || gid == "" {
+		return err
+	}
+
+	// 保存到数据库
+	task.GID = gid
+	_, err = task.Create()
+
+	return err
+}

pkg/filesystem/driver/onedrive/api.go

@@ -290,7 +290,8 @@ func (client *Client) BatchDelete(ctx context.Context, dst []string) ([]string,
 	return finalRes, err
 }
 
-// Delete 并行删除文件，返回删除失败的文件，及第一个遇到的错误，最多删除20个
+// Delete 并行删除文件，返回删除失败的文件，及第一个遇到的错误，
+// 由于API限制，最多删除20个
 func (client *Client) Delete(ctx context.Context, dst []string) ([]string, error) {
 	body := client.makeBatchDeleteRequestsBody(dst)
 	res, err := client.requestWithStr(ctx, "POST", client.getRequestURL("$batch"), body, 200)