-
Notifications
You must be signed in to change notification settings - Fork 24
/
template.go.disabled
132 lines (111 loc) · 3.52 KB
/
template.go.disabled
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
package civogo
import (
"bytes"
"encoding/json"
"errors"
"fmt"
"strings"
)
// Template represents a Template for launching instances from
type Template struct {
ID string `json:"id"`
Code string `json:"code"`
Name string `json:"name"`
Region string `json:"region"`
AccountID string `json:"account_id,omitempty"`
ImageID string `json:"image_id,omitempty"`
VolumeID string `json:"volume_id"`
ShortDescription string `json:"short_description"`
Description string `json:"description"`
DefaultUsername string `json:"default_username"`
CloudConfig string `json:"cloud_config"`
}
// ListTemplates return all template in system
func (c *Client) ListTemplates() ([]Template, error) {
resp, err := c.SendGetRequest("/v2/templates")
if err != nil {
return nil, decodeError(err)
}
templates := make([]Template, 0)
if err := json.NewDecoder(bytes.NewReader(resp)).Decode(&templates); err != nil {
return nil, err
}
return templates, nil
}
// NewTemplate will create a new template for the current user
func (c *Client) NewTemplate(conf *Template) (*SimpleResponse, error) {
if conf.ImageID == "" && conf.VolumeID == "" {
return nil, errors.New("if image id is not present, volume id must be")
}
resp, err := c.SendPostRequest("/v2/templates", conf)
if err != nil {
return nil, decodeError(err)
}
return c.DecodeSimpleResponse(resp)
}
// UpdateTemplate will update a template for the current user
func (c *Client) UpdateTemplate(id string, conf *Template) (*Template, error) {
if conf.ImageID == "" && conf.VolumeID == "" {
return nil, errors.New("if image id is not present, volume id must be")
}
resp, err := c.SendPutRequest(fmt.Sprintf("/v2/templates/%s", id), conf)
if err != nil {
return nil, decodeError(err)
}
template := &Template{}
if err := json.NewDecoder(bytes.NewReader(resp)).Decode(template); err != nil {
return nil, err
}
return template, nil
}
// GetTemplateByCode finds the Template for an account with the specified code
func (c *Client) GetTemplateByCode(code string) (*Template, error) {
resp, err := c.ListTemplates()
if err != nil {
return nil, decodeError(err)
}
for _, template := range resp {
if template.Code == code {
return &template, nil
}
}
return nil, errors.New("template not found")
}
// FindTemplate finds a template by either part of the ID or part of the code
func (c *Client) FindTemplate(search string) (*Template, error) {
templateList, err := c.ListTemplates()
if err != nil {
return nil, decodeError(err)
}
exactMatch := false
partialMatchesCount := 0
result := Template{}
for _, value := range templateList {
if value.Code == search || value.ID == search {
exactMatch = true
result = value
} else if strings.Contains(value.Code, search) || strings.Contains(value.ID, search) {
if !exactMatch {
result = value
partialMatchesCount++
}
}
}
if exactMatch || partialMatchesCount == 1 {
return &result, nil
} else if partialMatchesCount > 1 {
err := fmt.Errorf("unable to find %s because there were multiple matches", search)
return nil, MultipleMatchesError.wrap(err)
} else {
err := fmt.Errorf("unable to find %s, zero matches", search)
return nil, ZeroMatchesError.wrap(err)
}
}
// DeleteTemplate deletes requested template
func (c *Client) DeleteTemplate(id string) (*SimpleResponse, error) {
resp, err := c.SendDeleteRequest(fmt.Sprintf("/v2/templates/%s", id))
if err != nil {
return nil, decodeError(err)
}
return c.DecodeSimpleResponse(resp)
}