Feat: download torrent / multiple file / select file

service/aria2/add.go

@@ -11,11 +11,11 @@ import (
 // AddURLService 添加URL离线下载服务
 type AddURLService struct {
 	URL string `json:"url" binding:"required"`
-	Dst string `json:"dst" binding:"required,min=1,max=65535"`
+	Dst string `json:"dst" binding:"required,min=1"`
 }
 
 // Add 创建新的链接离线下载任务
-func (service *AddURLService) Add(c *gin.Context) serializer.Response {
+func (service *AddURLService) Add(c *gin.Context, taskType int) serializer.Response {
 	// 创建文件系统
 	fs, err := filesystem.NewFileSystemFromContext(c)
 	if err != nil {
@@ -36,7 +36,7 @@ func (service *AddURLService) Add(c *gin.Context) serializer.Response {
 	// 创建任务
 	task := &model.Download{
 		Status: aria2.Ready,
-		Type:   aria2.URLTask,
+		Type:   taskType,
 		Dst:    service.Dst,
 		UserID: fs.User.ID,
 		Source: service.URL,

service/aria2/manage.go

@@ -0,0 +1,37 @@
+package aria2
+
+import (
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/aria2"
+	"github.com/HFO4/cloudreve/pkg/serializer"
+	"github.com/gin-gonic/gin"
+)
+
+// SelectFileService 选择要下载的文件服务
+type SelectFileService struct {
+	Indexes []int `json:"indexes" binding:"required"`
+}
+
+// Select 选取要下载的文件
+func (service *SelectFileService) Select(c *gin.Context) serializer.Response {
+	userCtx, _ := c.Get("user")
+	user := userCtx.(*model.User)
+
+	// 查找下载记录
+	download, err := model.GetDownloadByGid(c.Param("gid"), user.ID)
+	if err != nil {
+		return serializer.Err(serializer.CodeNotFound, "下载记录不存在", err)
+	}
+
+	if download.StatusInfo.BitTorrent.Mode != "multi" || (download.Status != aria2.Downloading && download.Status != aria2.Paused) {
+		return serializer.Err(serializer.CodeNoPermissionErr, "此下载任务无法选取文件", err)
+	}
+
+	// 选取下载
+	if err := aria2.Instance.Select(download, service.Indexes); err != nil {
+		return serializer.Err(serializer.CodeNotSet, "操作失败", err)
+	}
+
+	return serializer.Response{}
+
+}