124 lines
3.3 KiB
Go
124 lines
3.3 KiB
Go
package hcloud
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"github.com/hetznercloud/hcloud-go/hcloud/schema"
|
|
)
|
|
|
|
// Datacenter represents a datacenter in the Hetzner Cloud.
|
|
type Datacenter struct {
|
|
ID int
|
|
Name string
|
|
Description string
|
|
Location *Location
|
|
ServerTypes DatacenterServerTypes
|
|
}
|
|
|
|
// DatacenterServerTypes represents the server types available and supported in a datacenter.
|
|
type DatacenterServerTypes struct {
|
|
Supported []*ServerType
|
|
Available []*ServerType
|
|
}
|
|
|
|
// DatacenterClient is a client for the datacenter API.
|
|
type DatacenterClient struct {
|
|
client *Client
|
|
}
|
|
|
|
// GetByID retrieves a datacenter by its ID. If the datacenter does not exist, nil is returned.
|
|
func (c *DatacenterClient) GetByID(ctx context.Context, id int) (*Datacenter, *Response, error) {
|
|
req, err := c.client.NewRequest(ctx, "GET", fmt.Sprintf("/datacenters/%d", id), nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
var body schema.DatacenterGetResponse
|
|
resp, err := c.client.Do(req, &body)
|
|
if err != nil {
|
|
if IsError(err, ErrorCodeNotFound) {
|
|
return nil, resp, nil
|
|
}
|
|
return nil, resp, err
|
|
}
|
|
return DatacenterFromSchema(body.Datacenter), resp, nil
|
|
}
|
|
|
|
// GetByName retrieves an datacenter by its name. If the datacenter does not exist, nil is returned.
|
|
func (c *DatacenterClient) GetByName(ctx context.Context, name string) (*Datacenter, *Response, error) {
|
|
datacenters, response, err := c.List(ctx, DatacenterListOpts{Name: name})
|
|
if len(datacenters) == 0 {
|
|
return nil, response, err
|
|
}
|
|
return datacenters[0], response, err
|
|
}
|
|
|
|
// Get retrieves a datacenter by its ID if the input can be parsed as an integer, otherwise it
|
|
// retrieves a datacenter by its name. If the datacenter does not exist, nil is returned.
|
|
func (c *DatacenterClient) Get(ctx context.Context, idOrName string) (*Datacenter, *Response, error) {
|
|
if id, err := strconv.Atoi(idOrName); err == nil {
|
|
return c.GetByID(ctx, int(id))
|
|
}
|
|
return c.GetByName(ctx, idOrName)
|
|
}
|
|
|
|
// DatacenterListOpts specifies options for listing datacenters.
|
|
type DatacenterListOpts struct {
|
|
ListOpts
|
|
Name string
|
|
}
|
|
|
|
func (l DatacenterListOpts) values() url.Values {
|
|
vals := l.ListOpts.values()
|
|
if l.Name != "" {
|
|
vals.Add("name", l.Name)
|
|
}
|
|
return vals
|
|
}
|
|
|
|
// List returns a list of datacenters for a specific page.
|
|
func (c *DatacenterClient) List(ctx context.Context, opts DatacenterListOpts) ([]*Datacenter, *Response, error) {
|
|
path := "/datacenters?" + opts.values().Encode()
|
|
req, err := c.client.NewRequest(ctx, "GET", path, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
var body schema.DatacenterListResponse
|
|
resp, err := c.client.Do(req, &body)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
datacenters := make([]*Datacenter, 0, len(body.Datacenters))
|
|
for _, i := range body.Datacenters {
|
|
datacenters = append(datacenters, DatacenterFromSchema(i))
|
|
}
|
|
return datacenters, resp, nil
|
|
}
|
|
|
|
// All returns all datacenters.
|
|
func (c *DatacenterClient) All(ctx context.Context) ([]*Datacenter, error) {
|
|
allDatacenters := []*Datacenter{}
|
|
|
|
opts := DatacenterListOpts{}
|
|
opts.PerPage = 50
|
|
|
|
_, err := c.client.all(func(page int) (*Response, error) {
|
|
opts.Page = page
|
|
datacenters, resp, err := c.List(ctx, opts)
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
allDatacenters = append(allDatacenters, datacenters...)
|
|
return resp, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return allDatacenters, nil
|
|
}
|