feat: set aria2 client and add url to aria2 api

server/controllers/aria2.go

@@ -0,0 +1,69 @@
+package controllers
+
+import (
+	"github.com/alist-org/alist/v3/internal/aria2"
+	"github.com/alist-org/alist/v3/internal/conf"
+	"github.com/alist-org/alist/v3/internal/db"
+	"github.com/alist-org/alist/v3/internal/model"
+	"github.com/alist-org/alist/v3/server/common"
+	"github.com/gin-gonic/gin"
+	stdpath "path"
+)
+
+type SetAria2Req struct {
+	Uri    string `json:"uri"`
+	Secret string `json:"secret"`
+}
+
+func SetAria2(c *gin.Context) {
+	var req SetAria2Req
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	items := []model.SettingItem{
+		{Key: conf.Aria2Uri, Value: req.Uri, Type: conf.TypeString, Group: model.ARIA2, Flag: model.PRIVATE},
+		{Key: conf.Aria2Secret, Value: req.Secret, Type: conf.TypeString, Group: model.ARIA2, Flag: model.PRIVATE},
+	}
+	if err := db.SaveSettingItems(items); err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	err := aria2.InitClient(2)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	common.SuccessResp(c)
+}
+
+type AddAria2Req struct {
+	Urls []string `json:"urls"`
+	Path string   `json:"path"`
+}
+
+func AddAria2(c *gin.Context) {
+	user := c.MustGet("user").(*model.User)
+	if !user.IsAdmin() && !user.Aira2 {
+		common.ErrorStrResp(c, "permission denied", 403)
+		return
+	}
+	if !aria2.IsAria2Ready() {
+		common.ErrorStrResp(c, "aria2 not ready", 500)
+		return
+	}
+	var req AddAria2Req
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	req.Path = stdpath.Join(user.BasePath, req.Path)
+	for _, url := range req.Urls {
+		err := aria2.AddURI(c, url, req.Path)
+		if err != nil {
+			common.ErrorResp(c, err, 500)
+			return
+		}
+	}
+	common.SuccessResp(c)
+}