package admin import ( "bytes" "encoding/json" "net/url" "time" model "github.com/cloudreve/Cloudreve/v3/models" "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)} }