Feat: add local policy

service/admin/aria2.go

@@ -0,0 +1,42 @@
+package admin
+
+import (
+	"github.com/HFO4/cloudreve/pkg/aria2"
+	"github.com/HFO4/cloudreve/pkg/serializer"
+	"net/url"
+)
+
+// Aria2TestService aria2连接测试服务
+type Aria2TestService struct {
+	Server string `json:"server" binding:"required"`
+	Token  string `json:"token"`
+}
+
+// Test 测试aria2连接
+func (service *Aria2TestService) Test() serializer.Response {
+	testRPC := aria2.RPCService{}
+
+	// 解析RPC服务地址
+	server, err := url.Parse(service.Server)
+	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)
+	}
+
+	defer testRPC.Caller.Close()
+
+	info, err := testRPC.Caller.GetVersion()
+	if err != nil {
+		return serializer.ParamErr("无法请求 RPC 服务, "+err.Error(), nil)
+	}
+
+	if info.Version == "" {
+		return serializer.ParamErr("RPC 服务返回非预期响应", nil)
+	}
+
+	return serializer.Response{Data: info.Version}
+}