package aria2 import ( "encoding/json" model "github.com/HFO4/cloudreve/models" "github.com/HFO4/cloudreve/pkg/aria2/rpc" "github.com/HFO4/cloudreve/pkg/serializer" "github.com/HFO4/cloudreve/pkg/util" "net/url" "sync" ) // Instance 默认使用的Aria2处理实例 var Instance Aria2 = &DummyAria2{} // Lock Instance的读写锁 var Lock sync.RWMutex // EventNotifier 任务状态更新通知处理器 var EventNotifier = &Notifier{} // Aria2 离线下载处理接口 type Aria2 interface { // CreateTask 创建新的任务 CreateTask(task *model.Download, options map[string]interface{}) error // 返回状态信息 Status(task *model.Download) (rpc.StatusInfo, error) // 取消任务 Cancel(task *model.Download) error // 选择要下载的文件 Select(task *model.Download, files []int) error } const ( // URLTask 从URL添加的任务 URLTask = iota // TorrentTask 种子任务 TorrentTask ) const ( // Ready 准备就绪 Ready = iota // Downloading 下载中 Downloading // Paused 暂停中 Paused // Error 出错 Error // Complete 完成 Complete // Canceled 取消/停止 Canceled // Unknown 未知状态 Unknown ) var ( // ErrNotEnabled 功能未开启错误 ErrNotEnabled = serializer.NewError(serializer.CodeNoPermissionErr, "离线下载功能未开启", nil) // ErrUserNotFound 未找到下载任务创建者 ErrUserNotFound = serializer.NewError(serializer.CodeNotFound, "无法找到任务创建者", nil) ) // DummyAria2 未开启Aria2功能时使用的默认处理器 type DummyAria2 struct { } // CreateTask 创建新任务,此处直接返回未开启错误 func (instance *DummyAria2) CreateTask(model *model.Download, options map[string]interface{}) error { return ErrNotEnabled } // Status 返回未开启错误 func (instance *DummyAria2) Status(task *model.Download) (rpc.StatusInfo, error) { return rpc.StatusInfo{}, ErrNotEnabled } // Cancel 返回未开启错误 func (instance *DummyAria2) Cancel(task *model.Download) error { return ErrNotEnabled } // Select 返回未开启错误 func (instance *DummyAria2) Select(task *model.Download, files []int) error { return ErrNotEnabled } // Init 初始化 func Init(isReload bool) { Lock.Lock() defer Lock.Unlock() // 关闭上个初始连接 if previousClient, ok := Instance.(*RPCService); ok { if previousClient.Caller != nil { util.Log().Debug("关闭上个 aria2 连接") previousClient.Caller.Close() } } options := model.GetSettingByNames("aria2_rpcurl", "aria2_token", "aria2_options") timeout := model.GetIntSetting("aria2_call_timeout", 5) if options["aria2_rpcurl"] == "" { Instance = &DummyAria2{} return } util.Log().Info("初始化 aria2 RPC 服务[%s]", options["aria2_rpcurl"]) client := &RPCService{} // 解析RPC服务地址 server, err := url.Parse(options["aria2_rpcurl"]) if err != nil { util.Log().Warning("无法解析 aria2 RPC 服务地址,%s", err) Instance = &DummyAria2{} return } server.Path = "/jsonrpc" // 加载自定义下载配置 var globalOptions map[string]interface{} err = json.Unmarshal([]byte(options["aria2_options"]), &globalOptions) if err != nil { util.Log().Warning("无法解析 aria2 全局配置,%s", err) Instance = &DummyAria2{} return } if err := client.Init(server.String(), options["aria2_token"], timeout, globalOptions); err != nil { util.Log().Warning("初始化 aria2 RPC 服务失败,%s", err) Instance = &DummyAria2{} return } Instance = client if !isReload { // 从数据库中读取未完成任务,创建监控 unfinished := model.GetDownloadsByStatus(Ready, Paused, Downloading) for i := 0; i < len(unfinished); i++ { // 创建任务监控 NewMonitor(&unfinished[i]) } } } // getStatus 将给定的状态字符串转换为状态标识数字 func getStatus(status string) int { switch status { case "complete": return Complete case "active": return Downloading case "waiting": return Ready case "paused": return Paused case "error": return Error case "removed": return Canceled default: return Unknown } }