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