/
inventory.go
92 lines (74 loc) · 2.02 KB
/
inventory.go
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
package runner
import (
"encoding/json"
"fmt"
"os"
"path"
"text/template"
log "github.com/sirupsen/logrus"
)
type InventoryContent struct {
Groups []*Group
Nodes []*Node // this seems unused
}
type Group struct {
Name string
Nodes []*Node
}
type Node struct {
Name string
AnsibleHost string
AnsibleUser string
Variables map[string]interface{}
}
const (
inventoryTemplate = `{{- range .Nodes }}
{{ .Name }} ansible_host={{ .AnsibleHost }} ansible_user={{ .AnsibleUser }} {{ range $key, $value := .Variables }}{{ $key }}={{ $value }} {{ end }}
{{- end }}
{{- range .Groups }}
[{{ .Name }}]
{{- range .Nodes }}
{{ .Name }} ansible_host={{ .AnsibleHost }} ansible_user={{ .AnsibleUser }} {{ range $key, $value := .Variables }}{{ $key }}={{ $value }} {{ end }}
{{- end }}
{{- end }}
`
clusterSelectedChecks string = "cluster_selected_checks"
provider string = "provider"
)
func CreateInventory(destination string, content *InventoryContent) error {
t := template.Must(template.New("").Parse(inventoryTemplate))
if err := os.MkdirAll(path.Dir(destination), 0755); err != nil {
return err
}
f, err := os.Create(destination)
if err != nil {
return err
}
if err := t.Execute(f, content); err != nil {
return nil
}
f.Close()
return nil
}
func NewClusterInventoryContent(e *ExecutionEvent) (*InventoryContent, error) {
content := &InventoryContent{}
nodes := []*Node{}
jsonChecks, err := json.Marshal(e.Checks)
if err != nil {
log.Errorf("error marshalling the cluster %s selected checks: %s", e.ClusterID.String(), err)
}
for _, host := range e.Hosts {
node := &Node{
Name: host.HostID.String(),
AnsibleHost: host.Address,
AnsibleUser: host.User,
Variables: make(map[string]interface{}),
}
node.Variables[clusterSelectedChecks] = fmt.Sprintf("'%s'", string(jsonChecks))
node.Variables[provider] = e.Provider
nodes = append(nodes, node)
}
group := &Group{Name: e.ClusterID.String(), Nodes: nodes}
content.Groups = append(content.Groups, group)
return content, nil
}