Vendor main dependencies.

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

@@ -0,0 +1,188 @@
+package lib
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/url"
+	"strconv"
+)
+
+// 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"`
+}
+
+// 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
+	}
+	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
+}