Add support for selecting service template

icinga2/icinga.go

@@ -86,6 +86,7 @@ type MockClient struct {
 	Actions    map[string][]Action
 	mutex      sync.Mutex
 	URL        string
+	Templates  []string
 }
 
 type Vars map[string]interface{}
@@ -128,6 +129,7 @@ func NewMockClient() (c *MockClient) {
 	c.Services = make(map[string]Service)
 	c.Actions = make(map[string][]Action)
 	c.mutex = sync.Mutex{}
+	c.Templates = []string{}
 	return
 }
 

icinga2/service.go

@@ -6,22 +6,23 @@ import (
 )
 
 type Service struct {
-	Name               string  `json:"name,omitempty"`
-	DisplayName        string  `json:"display_name"`
-	HostName           string  `json:"host_name"`
-	CheckCommand       string  `json:"check_command"`
-	EnableActiveChecks bool    `json:"enable_active_checks"`
-	Notes              string  `json:"notes"`
-	NotesURL           string  `json:"notes_url"`
-	ActionURL          string  `json:"action_url"`
-	Vars               Vars    `json:"vars"`
-	Zone               string  `json:"zone,omitempty"`
-	CheckInterval      float64 `json:"check_interval"`
-	RetryInterval      float64 `json:"retry_interval"`
-	MaxCheckAttempts   float64 `json:"max_check_attempts"`
-	CheckPeriod        string  `json:"check_period,omitempty"`
-	State              float64 `json:"state,omitempty"`
-	LastStateChange    float64 `json:"last_state_change,omitempty"`
+	Name               string   `json:"name,omitempty"`
+	DisplayName        string   `json:"display_name"`
+	HostName           string   `json:"host_name"`
+	CheckCommand       string   `json:"check_command"`
+	EnableActiveChecks bool     `json:"enable_active_checks"`
+	Notes              string   `json:"notes"`
+	NotesURL           string   `json:"notes_url"`
+	ActionURL          string   `json:"action_url"`
+	Vars               Vars     `json:"vars,omitempty"`
+	Zone               string   `json:"zone,omitempty"`
+	CheckInterval      float64  `json:"check_interval"`
+	RetryInterval      float64  `json:"retry_interval"`
+	MaxCheckAttempts   float64  `json:"max_check_attempts"`
+	CheckPeriod        string   `json:"check_period,omitempty"`
+	State              float64  `json:"state,omitempty"`
+	LastStateChange    float64  `json:"last_state_change,omitempty"`
+	Templates          []string `json:"templates,omitempty"`
 }
 
 type ServiceResults struct {
@@ -68,10 +69,11 @@ func (s *WebClient) GetService(name string) (Service, error) {
 }
 
 func (s *WebClient) CreateService(service Service) error {
-	serviceCreate := ServiceCreate{Templates: []string{"generic-service"}, Attrs: service}
+	serviceCreate := ServiceCreate{Templates: service.Templates, Attrs: service}
 	// Strip "name" from create payload
 	serviceCreate.Attrs.Name = ""
 	err := s.CreateObject("/services/"+service.FullName(), serviceCreate)
+
 	return err
 }
 

icinga2/util.go

@@ -0,0 +1,55 @@
+package icinga2
+
+import (
+	"encoding/json"
+	"strconv"
+)
+
+func (s Service) MarshalJSON() ([]byte, error) {
+	// Prevent json.Marshal() recursion
+	type service Service
+
+	svc := service(s)
+	// Clear top-level Vars field, so it's not added to the marshalled JSON. We marshal all service variables into individual top-level `vars.<variable name>` fields below.
+
+	svc.Vars = Vars{}
+
+	serviceAsJson, err := json.Marshal(svc)
+	if err != nil {
+		return nil, err
+	}
+
+	var serviceAsMap map[string]interface{}
+	if err := json.Unmarshal(serviceAsJson, &serviceAsMap); err != nil {
+		return nil, err
+	}
+
+	// This loop flattens the json, so each var will be at the same level
+	for k, v := range Flatten(s.Vars) {
+		serviceAsMap["vars."+k] = v
+	}
+
+	return json.Marshal(serviceAsMap)
+}
+
+func Flatten(m map[string]interface{}) map[string]interface{} {
+	flat := map[string]interface{}{}
+
+	for k, v := range m {
+		switch child := v.(type) {
+		case map[string]interface{}:
+			flat_child := Flatten(child)
+			for ck, cv := range flat_child {
+				flat[k+"."+ck] = cv
+			}
+		case []interface{}:
+			for i := 0; i < len(child); i++ {
+				flat[k+"["+strconv.Itoa(i)+"]"] = child[i]
+			}
+		default:
+			flat[k] = v
+		}
+	}
+
+	return flat
+}