Feat: cancel aria2 download task

pkg/aria2/Monitor.go

@@ -99,6 +99,8 @@ func (monitor *Monitor) Update() bool {
 	case "active", "waiting", "paused":
 		return false
 	case "removed":
+		monitor.Task.Status = Canceled
+		monitor.Task.Save()
 		return true
 	default:
 		util.Log().Warning("下载任务[%s]返回未知状态信息[%s]，", monitor.Task.GID, status.Status)
@@ -145,7 +147,7 @@ func (monitor *Monitor) UpdateTaskInfo(status rpc.StatusInfo) error {
 		// 文件大小更新后，对文件限制等进行校验
 		if err := monitor.ValidateFile(); err != nil {
 			// 验证失败时取消任务
-			monitor.Cancel()
+			Instance.Cancel(monitor.Task)
 			return err
 		}
 	}
@@ -153,20 +155,6 @@ func (monitor *Monitor) UpdateTaskInfo(status rpc.StatusInfo) error {
 	return nil
 }
 
-// Cancel 取消上传并尝试删除临时文件
-func (monitor *Monitor) Cancel() {
-	if err := Instance.Cancel(monitor.Task); err != nil {
-		util.Log().Warning("无法取消离线下载任务[%s], %s", monitor.Task.GID, err)
-	}
-	util.Log().Debug("离线下载任务[%s]已取消，1 分钟后删除临时文件", monitor.Task.GID)
-	go func(monitor *Monitor) {
-		select {
-		case <-time.After(time.Duration(60) * time.Second):
-			monitor.RemoveTempFolder()
-		}
-	}(monitor)
-}
-
 // ValidateFile 上传过程中校验文件大小、文件名
 func (monitor *Monitor) ValidateFile() error {
 	// 找到任务创建者

pkg/aria2/caller.go

@@ -5,6 +5,7 @@ import (
 	model "github.com/HFO4/cloudreve/models"
 	"github.com/HFO4/cloudreve/pkg/util"
 	"github.com/zyxar/argo/rpc"
+	"os"
 	"path/filepath"
 	"strconv"
 	"strings"
@@ -44,7 +45,24 @@ func (client *RPCService) Status(task *model.Download) (rpc.StatusInfo, error) {
 
 // Cancel 取消下载
 func (client *RPCService) Cancel(task *model.Download) error {
+	// 取消下载任务
 	_, err := client.caller.Remove(task.GID)
+	if err != nil {
+		util.Log().Warning("无法取消离线下载任务[%s], %s", task.GID, err)
+	}
+
+	// 删除临时文件
+	util.Log().Debug("离线下载任务[%s]已取消，1 分钟后删除临时文件", task.GID)
+	go func(task *model.Download) {
+		select {
+		case <-time.After(time.Duration(60) * time.Second):
+			err := os.RemoveAll(task.Parent)
+			if err != nil {
+				util.Log().Warning("无法删除离线下载临时目录[%s], %s", task.Parent, err)
+			}
+		}
+	}(task)
+
 	return err
 }