126 lines
2.7 KiB
Go
126 lines
2.7 KiB
Go
// Copyright (C) 2020 Marius Schellenberger
|
|
|
|
package pve
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
)
|
|
|
|
var (
|
|
ErrPoolExists = errors.New("pool already exists")
|
|
ErrPoolNotExists = errors.New("pool does not exist")
|
|
ErrEmptyPoolName = errors.New("pool name is empty")
|
|
)
|
|
|
|
func MakePoolName(namespace, name string) string {
|
|
return fmt.Sprintf("%s-%s", namespace, name)
|
|
}
|
|
|
|
type PoolClient struct {
|
|
client *Client
|
|
}
|
|
|
|
func (c *PoolClient) exists(ctx context.Context, name string) (status int, err error) {
|
|
req, err := c.client.NewRequest(ctx, "GET", fmt.Sprintf("/pools/%s", name), nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
resp, err := c.client.Do(req, nil)
|
|
if err != nil {
|
|
if !errors.Is(err, HTTPErr) {
|
|
return
|
|
}
|
|
err = nil
|
|
}
|
|
status = resp.StatusCode
|
|
return
|
|
}
|
|
|
|
func (c *PoolClient) Create(ctx context.Context, name string) (err error) {
|
|
status, err := c.exists(ctx, name)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if status == http.StatusOK {
|
|
return ErrPoolExists
|
|
}
|
|
|
|
body := httpbody{"poolid": name}
|
|
req, err := c.client.NewRequest(ctx, "POST", "/pools", body.Reader())
|
|
if err != nil {
|
|
return
|
|
}
|
|
_, err = c.client.Do(req, nil)
|
|
return
|
|
}
|
|
|
|
func (c *PoolClient) Delete(ctx context.Context, name string) (err error) {
|
|
status, err := c.exists(ctx, name)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if status != http.StatusOK {
|
|
return ErrPoolNotExists
|
|
}
|
|
|
|
req, err := c.client.NewRequest(ctx, "DELETE", fmt.Sprintf("/pools/%s", name), nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
_, err = c.client.Do(req, nil)
|
|
return
|
|
}
|
|
|
|
type pool struct {
|
|
members ServerList `json:"members"`
|
|
}
|
|
|
|
func (c *PoolClient) List(ctx context.Context, name string) (sl ServerList, err error) {
|
|
if name == "" {
|
|
err = ErrEmptyPoolName
|
|
return
|
|
}
|
|
var p pool
|
|
req, err := c.client.NewRequest(ctx, "GET", fmt.Sprintf("/pools/%s", name), nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
_, err = c.client.Do(req, &p)
|
|
sl = p.members
|
|
return
|
|
}
|
|
|
|
func (c *PoolClient) FindServer(ctx context.Context, poolname, servername, id string) (s *Server, err error) {
|
|
sl, err := c.List(ctx, poolname)
|
|
if err != nil {
|
|
return
|
|
}
|
|
for _, srv := range sl {
|
|
if id != "" && srv.ID == id {
|
|
return srv, nil
|
|
}
|
|
if servername != "" && srv.Name == servername {
|
|
return srv, nil
|
|
}
|
|
}
|
|
return nil, ErrServerNotFound
|
|
}
|
|
|
|
func (c *PoolClient) FindServerByName(ctx context.Context, poolname, servername string) (s *Server, err error) {
|
|
return c.FindServer(ctx, poolname, servername, "")
|
|
}
|
|
|
|
func (c *PoolClient) FindServerByID(ctx context.Context, poolname, id string) (s *Server, err error) {
|
|
return c.FindServer(ctx, poolname, "", id)
|
|
}
|
|
|
|
func (c *PoolClient) FindServerByURL(ctx context.Context, url string) (s *Server, err error) {
|
|
pool, _, id, err := ParseURL(url)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return c.FindServerByID(ctx, pool, id)
|
|
}
|