1 package icinga
2 
3 import (
4 	"encoding/json"
5 	"errors"
6 	"fmt"
7 	"io"
8 	"strings"
9 )
10 
11 type apiResponse struct {
12 	Results []struct {
13 		Name        string
14 		Type        string
15 		Errors      []string
16 		Permissions []string
17 		Attrs       json.RawMessage
18 	}
19 	Status string
20 }
21 
22 type response struct {
23 	Results []object
24 	Error   error
25 }
26 
27 func parseAPIResponse(r io.Reader) (apiResponse, error) {
28 	var apiresp apiResponse
29 	if err := json.NewDecoder(r).Decode(&apiresp); err != nil {
30 		return apiResponse{}, err
31 	}
32 	return apiresp, nil
33 }
34 
35 func parseResponse(r io.Reader) (*response, error) {
36 	apiresp, err := parseAPIResponse(r)
37 	if err != nil {
38 		return nil, err
39 	}
40 	// Confusingly the top-level status field in an API response contains
41 	// an error message. Successful statuses are actually held in the
42 	// status field in Results!
43 	if apiresp.Status != "" {
44 		return &response{Error: errors.New(apiresp.Status)}, nil
45 	}
46 	resp := &response{}
47 	for _, r := range apiresp.Results {
48 		if len(r.Errors) > 0 {
49 			resp.Error = errors.New(strings.Join(r.Errors, ", "))
50 			// got an error so nothing left in the API response
51 			break
52 		}
53 		if r.Type == "" {
54 			continue //
55 		}
56 		switch r.Type {
57 		case "Host":
58 			var h Host
59 			if err := json.Unmarshal(r.Attrs, &h); err != nil {
60 				return nil, err
61 			}
62 			resp.Results = append(resp.Results, h)
63 		case "Service":
64 			var s Service
65 			if err := json.Unmarshal(r.Attrs, &s); err != nil {
66 				return nil, err
67 			}
68 			resp.Results = append(resp.Results, s)
69 		case "User":
70 			var u User
71 			if err := json.Unmarshal(r.Attrs, &u); err != nil {
72 				return nil, err
73 			}
74 			resp.Results = append(resp.Results, u)
75 		case "HostGroup":
76 			var h HostGroup
77 			if err := json.Unmarshal(r.Attrs, &h); err != nil {
78 				return nil, err
79 			}
80 			resp.Results = append(resp.Results, h)
81 		default:
82 			return nil, fmt.Errorf("unsupported unmarshal of type %s", r.Type)
83 		}
84 	}
85 	return resp, nil
86 }
87 
88 func objectFromLookup(resp *response) (object, error) {
89 	if len(resp.Results) == 0 {
90 		return nil, errors.New("empty results")
91 	} else if len(resp.Results) > 1 {
92 		return nil, errors.New("too many results")
93 	}
94 	return resp.Results[0], nil
95 }