Feat: creat uri aria2 download task

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
+}