feat: rainyun server service — list/get/start/stop/reboot/vnc/upgrade
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
package rainyun
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"git.yoresee.cc/nightstar/ry/internal/provider"
|
||||
)
|
||||
|
||||
func init() {
|
||||
provider.Register("rainyun", New)
|
||||
}
|
||||
|
||||
// Rainyun provider — 雨云 cloud platform.
|
||||
type Rainyun struct {
|
||||
client *client
|
||||
server *ServerService
|
||||
}
|
||||
|
||||
// New creates a Rainyun provider from config.
|
||||
func New(cfg provider.ProviderConfig) (provider.Provider, error) {
|
||||
c := newClient(cfg.Endpoint, cfg.APIKey)
|
||||
r := &Rainyun{client: c}
|
||||
r.server = &ServerService{client: c}
|
||||
return r, nil
|
||||
}
|
||||
|
||||
func (r *Rainyun) Name() string { return "rainyun" }
|
||||
func (r *Rainyun) DisplayName() string { return "雨云" }
|
||||
|
||||
// ValidateAuth checks the API key by hitting the user info endpoint.
|
||||
func (r *Rainyun) ValidateAuth(ctx context.Context) error {
|
||||
var data any
|
||||
return r.client.get(ctx, "/user/", &data)
|
||||
}
|
||||
|
||||
func (r *Rainyun) Server() provider.ServerService {
|
||||
if r.server == nil {
|
||||
r.server = &ServerService{client: r.client}
|
||||
}
|
||||
return r.server
|
||||
}
|
||||
func (r *Rainyun) Domain() provider.DomainService { return nil }
|
||||
func (r *Rainyun) Storage() provider.StorageService { return nil }
|
||||
func (r *Rainyun) CDN() provider.CDNService { return nil }
|
||||
func (r *Rainyun) Billing() provider.BillingService { return nil }
|
||||
@@ -0,0 +1,120 @@
|
||||
package rainyun
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"git.yoresee.cc/nightstar/ry/internal/provider"
|
||||
)
|
||||
|
||||
// ServerService implements provider.ServerService for 雨云.
|
||||
type ServerService struct {
|
||||
client *client
|
||||
}
|
||||
|
||||
// rainyunServer mirrors the Rainyun RCS API JSON fields.
|
||||
type rainyunServer struct {
|
||||
ID int `json:"id"`
|
||||
Name string `json:"name"`
|
||||
Status string `json:"status"`
|
||||
IP string `json:"ip"`
|
||||
VCPU int `json:"vcpu"`
|
||||
Memory int `json:"memory"`
|
||||
Disk int `json:"disk"`
|
||||
OS string `json:"os"`
|
||||
Region string `json:"zone_name"`
|
||||
ExpireDate string `json:"expire_date"`
|
||||
}
|
||||
|
||||
// Response wrappers — the client unmarshals the full API envelope (code + data).
|
||||
type serverListResponse struct {
|
||||
Code int `json:"code"`
|
||||
Data []rainyunServer `json:"data"`
|
||||
}
|
||||
|
||||
type serverGetResponse struct {
|
||||
Code int `json:"code"`
|
||||
Data rainyunServer `json:"data"`
|
||||
}
|
||||
|
||||
type vncResponse struct {
|
||||
Code int `json:"code"`
|
||||
Data struct {
|
||||
VNCURL string `json:"vnc_url"`
|
||||
} `json:"data"`
|
||||
}
|
||||
|
||||
func (s *ServerService) List(ctx context.Context) ([]provider.Server, error) {
|
||||
var resp serverListResponse
|
||||
if err := s.client.get(ctx, "/product/rcs/", &resp); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
servers := make([]provider.Server, len(resp.Data))
|
||||
for i, r := range resp.Data {
|
||||
servers[i] = toServer(r)
|
||||
}
|
||||
return servers, nil
|
||||
}
|
||||
|
||||
func (s *ServerService) Get(ctx context.Context, id string) (*provider.Server, error) {
|
||||
var resp serverGetResponse
|
||||
path := fmt.Sprintf("/product/rcs/%s/", id)
|
||||
if err := s.client.get(ctx, path, &resp); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
sv := toServer(resp.Data)
|
||||
return &sv, nil
|
||||
}
|
||||
|
||||
func (s *ServerService) Start(ctx context.Context, id string) error {
|
||||
return s.client.post(ctx, fmt.Sprintf("/product/rcs/%s/start", id), nil, nil)
|
||||
}
|
||||
|
||||
func (s *ServerService) Stop(ctx context.Context, id string) error {
|
||||
return s.client.post(ctx, fmt.Sprintf("/product/rcs/%s/stop", id), nil, nil)
|
||||
}
|
||||
|
||||
func (s *ServerService) Reboot(ctx context.Context, id string) error {
|
||||
return s.client.post(ctx, fmt.Sprintf("/product/rcs/%s/reboot", id), nil, nil)
|
||||
}
|
||||
|
||||
func (s *ServerService) Reinstall(ctx context.Context, id, osTemplate string) error {
|
||||
body := map[string]string{"os": osTemplate}
|
||||
return s.client.post(ctx, fmt.Sprintf("/product/rcs/%s/changeos", id), body, nil)
|
||||
}
|
||||
|
||||
func (s *ServerService) ResetPassword(ctx context.Context, id string) error {
|
||||
return s.client.post(ctx, fmt.Sprintf("/product/rcs/%s/reset-password", id), nil, nil)
|
||||
}
|
||||
|
||||
func (s *ServerService) GetVNC(ctx context.Context, id string) (string, error) {
|
||||
var resp vncResponse
|
||||
if err := s.client.get(ctx, fmt.Sprintf("/product/rcs/%s/vnc", id), &resp); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return resp.Data.VNCURL, nil
|
||||
}
|
||||
|
||||
func (s *ServerService) Upgrade(ctx context.Context, id, plan string) error {
|
||||
body := map[string]string{"plan": plan}
|
||||
return s.client.post(ctx, fmt.Sprintf("/product/rcs/%s/upgrade", id), body, nil)
|
||||
}
|
||||
|
||||
// toServer maps a rainyunServer to the provider.Server model.
|
||||
func toServer(r rainyunServer) provider.Server {
|
||||
expire, _ := time.Parse("2006-01-02", r.ExpireDate)
|
||||
return provider.Server{
|
||||
ID: fmt.Sprintf("%d", r.ID),
|
||||
Name: r.Name,
|
||||
Status: r.Status,
|
||||
IP: r.IP,
|
||||
CPU: r.VCPU,
|
||||
Memory: r.Memory,
|
||||
Disk: r.Disk,
|
||||
OS: r.OS,
|
||||
Region: r.Region,
|
||||
ExpireAt: expire,
|
||||
Raw: r,
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,91 @@
|
||||
package rainyun
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestServerList(t *testing.T) {
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != "GET" || r.URL.Path != "/product/rcs/" {
|
||||
t.Errorf("unexpected request: %s %s", r.Method, r.URL.Path)
|
||||
}
|
||||
resp := map[string]any{
|
||||
"code": 0,
|
||||
"data": []map[string]any{
|
||||
{
|
||||
"id": 12001,
|
||||
"name": "web-server",
|
||||
"status": "running",
|
||||
"ip": "1.2.3.4",
|
||||
"vcpu": 2,
|
||||
"memory": 4096,
|
||||
"disk": 50,
|
||||
"os": "Ubuntu 22.04",
|
||||
"expire_date": "2026-08-15",
|
||||
},
|
||||
},
|
||||
}
|
||||
json.NewEncoder(w).Encode(resp)
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
rain := &Rainyun{client: newClient(srv.URL, "sk-test")}
|
||||
svc := rain.Server()
|
||||
servers, err := svc.List(context.Background())
|
||||
if err != nil {
|
||||
t.Fatalf("List() error: %v", err)
|
||||
}
|
||||
if len(servers) != 1 {
|
||||
t.Fatalf("len = %d, want 1", len(servers))
|
||||
}
|
||||
s := servers[0]
|
||||
if s.Name != "web-server" {
|
||||
t.Errorf("Name = %q", s.Name)
|
||||
}
|
||||
if s.Status != "running" {
|
||||
t.Errorf("Status = %q", s.Status)
|
||||
}
|
||||
if s.CPU != 2 {
|
||||
t.Errorf("CPU = %d", s.CPU)
|
||||
}
|
||||
}
|
||||
|
||||
func TestServerStart(t *testing.T) {
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != "POST" || r.URL.Path != "/product/rcs/12001/start" {
|
||||
t.Errorf("unexpected request: %s %s", r.Method, r.URL.Path)
|
||||
}
|
||||
json.NewEncoder(w).Encode(map[string]any{"code": 0})
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
rain := &Rainyun{client: newClient(srv.URL, "sk-test")}
|
||||
err := rain.Server().Start(context.Background(), "12001")
|
||||
if err != nil {
|
||||
t.Fatalf("Start() error: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestServerVNC(t *testing.T) {
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
resp := map[string]any{
|
||||
"code": 0,
|
||||
"data": map[string]any{"vnc_url": "https://vnc.example.com/abc"},
|
||||
}
|
||||
json.NewEncoder(w).Encode(resp)
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
rain := &Rainyun{client: newClient(srv.URL, "sk-test")}
|
||||
url, err := rain.Server().GetVNC(context.Background(), "12001")
|
||||
if err != nil {
|
||||
t.Fatalf("GetVNC() error: %v", err)
|
||||
}
|
||||
if url != "https://vnc.example.com/abc" {
|
||||
t.Errorf("VNC url = %q", url)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user