72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
package lib
|
|
|
|
import (
|
|
"net/url"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
// Snapshot of a virtual machine on Vultr account
|
|
type Snapshot struct {
|
|
ID string `json:"SNAPSHOTID"`
|
|
Description string `json:"description"`
|
|
Size string `json:"size"`
|
|
Status string `json:"status"`
|
|
Created string `json:"date_created"`
|
|
}
|
|
|
|
type snapshots []Snapshot
|
|
|
|
func (s snapshots) Len() int { return len(s) }
|
|
func (s snapshots) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
|
|
func (s snapshots) Less(i, j int) bool {
|
|
// sort order: description, created
|
|
if strings.ToLower(s[i].Description) < strings.ToLower(s[j].Description) {
|
|
return true
|
|
} else if strings.ToLower(s[i].Description) > strings.ToLower(s[j].Description) {
|
|
return false
|
|
}
|
|
return s[i].Created < s[j].Created
|
|
}
|
|
|
|
// GetSnapshots retrieves a list of all snapshots on Vultr account
|
|
func (c *Client) GetSnapshots() (snapshotList []Snapshot, err error) {
|
|
var snapshotMap map[string]Snapshot
|
|
if err := c.get(`snapshot/list`, &snapshotMap); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, snapshot := range snapshotMap {
|
|
snapshotList = append(snapshotList, snapshot)
|
|
}
|
|
sort.Sort(snapshots(snapshotList))
|
|
return snapshotList, nil
|
|
}
|
|
|
|
// CreateSnapshot creates a new virtual machine snapshot
|
|
func (c *Client) CreateSnapshot(id, description string) (Snapshot, error) {
|
|
values := url.Values{
|
|
"SUBID": {id},
|
|
"description": {description},
|
|
}
|
|
|
|
var snapshot Snapshot
|
|
if err := c.post(`snapshot/create`, values, &snapshot); err != nil {
|
|
return Snapshot{}, err
|
|
}
|
|
snapshot.Description = description
|
|
|
|
return snapshot, nil
|
|
}
|
|
|
|
// DeleteSnapshot deletes an existing virtual machine snapshot
|
|
func (c *Client) DeleteSnapshot(id string) error {
|
|
values := url.Values{
|
|
"SNAPSHOTID": {id},
|
|
}
|
|
|
|
if err := c.post(`snapshot/destroy`, values, nil); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|