forked from hashicorp/terraform
-
Notifications
You must be signed in to change notification settings - Fork 7
/
resource_icinga2_host.go
121 lines (96 loc) · 2.5 KB
/
resource_icinga2_host.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
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
package icinga2
import (
"fmt"
"github.com/hashicorp/terraform/helper/schema"
"github.com/lrsmith/go-icinga2-api/iapi"
)
func resourceIcinga2Host() *schema.Resource {
return &schema.Resource{
Create: resourceIcinga2HostCreate,
Read: resourceIcinga2HostRead,
Delete: resourceIcinga2HostDelete,
Schema: map[string]*schema.Schema{
"hostname": &schema.Schema{
Type: schema.TypeString,
Required: true,
Description: "Hostname",
ForceNew: true,
},
"address": &schema.Schema{
Type: schema.TypeString,
Required: true,
ForceNew: true,
},
"check_command": &schema.Schema{
Type: schema.TypeString,
Required: true,
ForceNew: true,
},
"vars": &schema.Schema{
Type: schema.TypeMap,
Optional: true,
ForceNew: true,
},
},
}
}
func resourceIcinga2HostCreate(d *schema.ResourceData, meta interface{}) error {
client := meta.(*iapi.Server)
hostname := d.Get("hostname").(string)
address := d.Get("address").(string)
checkCommand := d.Get("check_command").(string)
vars := make(map[string]string)
// Normalize from map[string]interface{} to map[string]string
iterator := d.Get("vars").(map[string]interface{})
for key, value := range iterator {
vars[key] = value.(string)
}
// Call CreateHost with normalized data
hosts, err := client.CreateHost(hostname, address, checkCommand, vars)
if err != nil {
return err
}
found := false
for _, host := range hosts {
if host.Name == hostname {
d.SetId(hostname)
found = true
}
}
if !found {
return fmt.Errorf("Failed to Create Host %s : %s", hostname, err)
}
return nil
}
func resourceIcinga2HostRead(d *schema.ResourceData, meta interface{}) error {
client := meta.(*iapi.Server)
hostname := d.Get("hostname").(string)
hosts, err := client.GetHost(hostname)
if err != nil {
return err
}
found := false
for _, host := range hosts {
if host.Name == hostname {
d.SetId(hostname)
d.Set("hostname", host.Name)
d.Set("address", host.Attrs.Address)
d.Set("check_command", host.Attrs.CheckCommand)
d.Set("vars", host.Attrs.Vars)
found = true
}
}
if !found {
return fmt.Errorf("Failed to Read Host %s : %s", hostname, err)
}
return nil
}
func resourceIcinga2HostDelete(d *schema.ResourceData, meta interface{}) error {
client := meta.(*iapi.Server)
hostname := d.Get("hostname").(string)
err := client.DeleteHost(hostname)
if err != nil {
return fmt.Errorf("Failed to Delete Host %s : %s", hostname, err)
}
return nil
}