cloudreve/service/admin/aria2.go

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"`
	RPC    string          `json:"rpc" binding:"required"`
	Secret string          `json:"secret"`
	Token  string          `json:"token"`
	Type   model.ModelType `json:"type"`
}

// Test 测试aria2连接
func (service *Aria2TestService) TestMaster() serializer.Response {
	res, err := aria2.TestRPCConnection(service.RPC, service.Token, 5)
	if err != nil {
		return serializer.ParamErr("Failed to connect to RPC server: "+err.Error(), err)
	}

	if res.Version == "" {
		return serializer.ParamErr("RPC server returns unexpected response", nil)
	}

	return serializer.Response{Data: res.Version}
}

func (service *Aria2TestService) TestSlave() serializer.Response {
	slave, err := url.Parse(service.Server)
	if err != nil {
		return serializer.ParamErr("Cannot parse slave server URL, "+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("Failed to connect to slave node, "+err.Error(), nil)
	}

	if res.Code != 0 {
		return serializer.ParamErr("Successfully connected to slave, but slave returns: "+res.Msg, nil)
	}

	return serializer.Response{Data: res.Data.(string)}
}