Feat: aria2 download and transfer in slave node (#1040)

* Feat: retrieve nodes from data table

* Feat: master node ping slave node in REST API

* Feat: master send scheduled ping request

* Feat: inactive nodes recover loop

* Modify: remove database operations from aria2 RPC caller implementation

* Feat: init aria2 client in master node

* Feat: Round Robin load balancer

* Feat: create and monitor aria2 task in master node

* Feat: salve receive and handle heartbeat

* Fix: Node ID will be 0 in download record generated in older version

* Feat: sign request headers with all `X-` prefix

* Feat: API call to slave node will carry meta data in headers

* Feat: call slave aria2 rpc method from master

* Feat: get slave aria2 task status
Feat: encode slave response data using gob

* Feat: aria2 callback to master node / cancel or select task to slave node

* Fix: use dummy aria2 client when caller initialize failed in master node

* Feat: slave aria2 status event callback / salve RPC auth

* Feat: prototype for slave driven filesystem

* Feat: retry for init aria2 client in master node

* Feat: init request client with global options

* Feat: slave receive async task from master

* Fix: competition write in request header

* Refactor: dependency initialize order

* Feat: generic message queue implementation

* Feat: message queue implementation

* Feat: master waiting slave transfer result

* Feat: slave transfer file in stateless policy

* Feat: slave transfer file in slave policy

* Feat: slave transfer file in local policy

* Feat: slave transfer file in OneDrive policy

* Fix: failed to initialize update checker http client

* Feat: list slave nodes for dashboard

* Feat: test aria2 rpc connection in slave

* Feat: add and save node

* Feat: add and delete node in node pool

* Fix: temp file cannot be removed when aria2 task fails

* Fix: delete node in admin panel

* Feat: edit node and get node info

* Modify: delete unused settings

service/admin/aria2.go

@@ -1,43 +1,71 @@
 package admin
 
 import (
+	"bytes"
+	"encoding/json"
+	model "github.com/cloudreve/Cloudreve/v3/models"
 	"net/url"
+	"time"
 
 	"github.com/cloudreve/Cloudreve/v3/pkg/aria2"
+	"github.com/cloudreve/Cloudreve/v3/pkg/auth"
+	"github.com/cloudreve/Cloudreve/v3/pkg/request"
 	"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
 )
 
 // Aria2TestService aria2连接测试服务
 type Aria2TestService struct {
-	Server string `json:"server" binding:"required"`
-	Token  string `json:"token"`
+	Server string          `json:"server" binding:"required"`
+	RPC    string          `json:"rpc" binding:"required"`
+	Secret string          `json:"secret" binding:"required"`
+	Token  string          `json:"token"`
+	Type   model.ModelType `json:"type"`
 }
 
 // Test 测试aria2连接
-func (service *Aria2TestService) Test() serializer.Response {
-	testRPC := aria2.RPCService{}
-
-	// 解析RPC服务地址
-	server, err := url.Parse(service.Server)
+func (service *Aria2TestService) TestMaster() serializer.Response {
+	res, err := aria2.TestRPCConnection(service.RPC, service.Token, 5)
 	if err != nil {
-		return serializer.ParamErr("无法解析 aria2 RPC 服务地址, "+err.Error(), nil)
-	}
-	server.Path = "/jsonrpc"
-
-	if err := testRPC.Init(server.String(), service.Token, 5, map[string]interface{}{}); err != nil {
-		return serializer.ParamErr("无法初始化连接, "+err.Error(), nil)
+		return serializer.ParamErr(err.Error(), err)
 	}
 
-	defer testRPC.Caller.Close()
-
-	info, err := testRPC.Caller.GetVersion()
-	if err != nil {
-		return serializer.ParamErr("无法请求 RPC 服务, "+err.Error(), nil)
-	}
-
-	if info.Version == "" {
+	if res.Version == "" {
 		return serializer.ParamErr("RPC 服务返回非预期响应", nil)
 	}
 
-	return serializer.Response{Data: info.Version}
+	return serializer.Response{Data: res.Version}
+}
+
+func (service *Aria2TestService) TestSlave() serializer.Response {
+	slave, err := url.Parse(service.Server)
+	if err != nil {
+		return serializer.ParamErr("无法解析从机端地址，"+err.Error(), nil)
+	}
+
+	controller, _ := url.Parse("/api/v3/slave/ping/aria2")
+
+	// 请求正文
+	service.Type = model.MasterNodeType
+	bodyByte, _ := json.Marshal(service)
+
+	r := request.NewClient()
+	res, err := r.Request(
+		"POST",
+		slave.ResolveReference(controller).String(),
+		bytes.NewReader(bodyByte),
+		request.WithTimeout(time.Duration(10)*time.Second),
+		request.WithCredential(
+			auth.HMACAuth{SecretKey: []byte(service.Secret)},
+			int64(model.GetIntSetting("slave_api_timeout", 60)),
+		),
+	).DecodeResponse()
+	if err != nil {
+		return serializer.ParamErr("无连接到从机，"+err.Error(), nil)
+	}
+
+	if res.Code != 0 {
+		return serializer.ParamErr("成功接到从机，但是从机返回："+res.Msg, nil)
+	}
+
+	return serializer.Response{Data: res.Data.(string)}
 }