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