Feat: handle aria2 download complete

pkg/aria2/aria2.go

@@ -4,16 +4,22 @@ import (
 	model "github.com/HFO4/cloudreve/models"
 	"github.com/HFO4/cloudreve/pkg/serializer"
 	"github.com/HFO4/cloudreve/pkg/util"
+	"github.com/zyxar/argo/rpc"
 	"net/url"
 )
 
 // Instance 默认使用的Aria2处理实例
 var Instance Aria2 = &DummyAria2{}
 
+// EventNotifier 任务状态更新通知处理器
+var EventNotifier = &Notifier{}
+
 // Aria2 离线下载处理接口
 type Aria2 interface {
 	// CreateTask 创建新的任务
 	CreateTask(task *model.Download) error
+	// 返回状态信息
+	Status(task *model.Download) (rpc.StatusInfo, error)
 }
 
 const (
@@ -26,6 +32,18 @@ const (
 const (
 	// Ready 准备就绪
 	Ready = iota
+	// Downloading 下载中
+	Downloading
+	// Paused 暂停中
+	Paused
+	// Error 出错
+	Error
+	// Complete 完成
+	Complete
+	// Canceled 取消/停止
+	Canceled
+	// Unknown 未知状态
+	Unknown
 )
 
 var (
@@ -42,6 +60,11 @@ func (instance *DummyAria2) CreateTask(task *model.Download) error {
 	return ErrNotEnabled
 }
 
+// Status 返回未开启错误
+func (instance *DummyAria2) Status(task *model.Download) (rpc.StatusInfo, error) {
+	return rpc.StatusInfo{}, ErrNotEnabled
+}
+
 // Init 初始化
 func Init() {
 	options := model.GetSettingByNames("aria2_rpcurl", "aria2_token", "aria2_options")
@@ -72,4 +95,31 @@ func Init() {
 	}
 
 	Instance = client
+
+	// 从数据库中读取未完成任务，创建监控
+	unfinished := model.GetDownloadsByStatus(Ready, Paused, Downloading)
+	for _, task := range unfinished {
+		// 创建任务监控
+		NewMonitor(&task)
+	}
+}
+
+// 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
+	}
 }