lego/vendor/github.com/JamesClonk/vultr/lib/block_storage.go

package lib

import (
	"encoding/json"
	"fmt"
	"net/url"
	"sort"
	"strconv"
	"strings"
)

// BlockStorage on Vultr account
type BlockStorage struct {
	ID         string `json:"SUBID,string"`
	Name       string `json:"label"`
	RegionID   int    `json:"DCID,string"`
	SizeGB     int    `json:"size_gb,string"`
	Created    string `json:"date_created"`
	Cost       string `json:"cost_per_month"`
	Status     string `json:"status"`
	AttachedTo string `json:"attached_to_SUBID"`
}

type blockstorages []BlockStorage

func (b blockstorages) Len() int      { return len(b) }
func (b blockstorages) Swap(i, j int) { b[i], b[j] = b[j], b[i] }
func (b blockstorages) Less(i, j int) bool {
	// sort order: name, size, status
	if strings.ToLower(b[i].Name) < strings.ToLower(b[j].Name) {
		return true
	} else if strings.ToLower(b[i].Name) > strings.ToLower(b[j].Name) {
		return false
	}
	if b[i].SizeGB < b[j].SizeGB {
		return true
	} else if b[i].SizeGB > b[j].SizeGB {
		return false
	}
	return b[i].Status < b[j].Status
}

// UnmarshalJSON implements json.Unmarshaller on BlockStorage.
// This is needed because the Vultr API is inconsistent in it's JSON responses.
// Some fields can change type, from JSON number to JSON string and vice-versa.
func (b *BlockStorage) UnmarshalJSON(data []byte) (err error) {
	if b == nil {
		*b = BlockStorage{}
	}

	var fields map[string]interface{}
	if err := json.Unmarshal(data, &fields); err != nil {
		return err
	}

	value := fmt.Sprintf("%v", fields["SUBID"])
	if len(value) == 0 || value == "<nil>" || value == "0" {
		b.ID = ""
	} else {
		id, err := strconv.ParseFloat(value, 64)
		if err != nil {
			return err
		}
		b.ID = strconv.FormatFloat(id, 'f', -1, 64)
	}

	value = fmt.Sprintf("%v", fields["DCID"])
	if len(value) == 0 || value == "<nil>" {
		value = "0"
	}
	region, err := strconv.ParseInt(value, 10, 64)
	if err != nil {
		return err
	}
	b.RegionID = int(region)

	value = fmt.Sprintf("%v", fields["size_gb"])
	if len(value) == 0 || value == "<nil>" {
		value = "0"
	}
	size, err := strconv.ParseInt(value, 10, 64)
	if err != nil {
		return err
	}
	b.SizeGB = int(size)

	value = fmt.Sprintf("%v", fields["attached_to_SUBID"])
	if len(value) == 0 || value == "<nil>" || value == "0" {
		b.AttachedTo = ""
	} else {
		attached, err := strconv.ParseFloat(value, 64)
		if err != nil {
			return err
		}
		b.AttachedTo = strconv.FormatFloat(attached, 'f', -1, 64)
	}

	b.Name = fmt.Sprintf("%v", fields["label"])
	b.Created = fmt.Sprintf("%v", fields["date_created"])
	b.Status = fmt.Sprintf("%v", fields["status"])
	b.Cost = fmt.Sprintf("%v", fields["cost_per_month"])

	return
}

// GetBlockStorages returns a list of all active block storages on Vultr account
func (c *Client) GetBlockStorages() (storages []BlockStorage, err error) {
	if err := c.get(`block/list`, &storages); err != nil {
		return nil, err
	}
	sort.Sort(blockstorages(storages))
	return storages, nil
}

// GetBlockStorage returns block storage with given ID
func (c *Client) GetBlockStorage(id string) (BlockStorage, error) {
	storages, err := c.GetBlockStorages()
	if err != nil {
		return BlockStorage{}, err
	}

	for _, s := range storages {
		if s.ID == id {
			return s, nil
		}
	}
	return BlockStorage{}, fmt.Errorf("BlockStorage with ID %v not found", id)
}

// CreateBlockStorage creates a new block storage on Vultr account
func (c *Client) CreateBlockStorage(name string, regionID, size int) (BlockStorage, error) {
	values := url.Values{
		"label":   {name},
		"DCID":    {fmt.Sprintf("%v", regionID)},
		"size_gb": {fmt.Sprintf("%v", size)},
	}

	var storage BlockStorage
	if err := c.post(`block/create`, values, &storage); err != nil {
		return BlockStorage{}, err
	}
	storage.RegionID = regionID
	storage.Name = name
	storage.SizeGB = size

	return storage, nil
}

// ResizeBlockStorage resizes an existing block storage
func (c *Client) ResizeBlockStorage(id string, size int) error {
	values := url.Values{
		"SUBID":   {id},
		"size_gb": {fmt.Sprintf("%v", size)},
	}

	if err := c.post(`block/resize`, values, nil); err != nil {
		return err
	}
	return nil
}

// LabelBlockStorage changes the label on an existing block storage
func (c *Client) LabelBlockStorage(id, name string) error {
	values := url.Values{
		"SUBID": {id},
		"label": {name},
	}

	if err := c.post(`block/label_set`, values, nil); err != nil {
		return err
	}
	return nil
}

// AttachBlockStorage attaches block storage to an existing virtual machine
func (c *Client) AttachBlockStorage(id, serverID string) error {
	values := url.Values{
		"SUBID":           {id},
		"attach_to_SUBID": {serverID},
	}

	if err := c.post(`block/attach`, values, nil); err != nil {
		return err
	}
	return nil
}

// DetachBlockStorage detaches block storage from virtual machine
func (c *Client) DetachBlockStorage(id string) error {
	values := url.Values{
		"SUBID": {id},
	}

	if err := c.post(`block/detach`, values, nil); err != nil {
		return err
	}
	return nil
}

// DeleteBlockStorage deletes an existing block storage
func (c *Client) DeleteBlockStorage(id string) error {
	values := url.Values{
		"SUBID": {id},
	}

	if err := c.post(`block/delete`, values, nil); err != nil {
		return err
	}
	return nil
}