84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
package iapi
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
)
|
|
|
|
// GetCheckcommand ...
|
|
func (server *Server) GetCheckcommand(name string) ([]CheckcommandStruct, error) {
|
|
|
|
var checkcommands []CheckcommandStruct
|
|
results, err := server.NewAPIRequest("GET", "/objects/checkcommands/"+name, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Contents of the results is an interface object. Need to convert it to json first.
|
|
jsonStr, marshalErr := json.Marshal(results.Results)
|
|
if marshalErr != nil {
|
|
return nil, marshalErr
|
|
}
|
|
|
|
// then the JSON can be pushed into the appropriate struct.
|
|
// Note : Results is a slice so much push into a slice.
|
|
|
|
if unmarshalErr := json.Unmarshal(jsonStr, &checkcommands); unmarshalErr != nil {
|
|
return nil, unmarshalErr
|
|
}
|
|
|
|
return checkcommands, err
|
|
|
|
}
|
|
|
|
// CreateCheckcommand ...
|
|
func (server *Server) CreateCheckcommand(name, command string, command_arguments map[string]string) ([]CheckcommandStruct, error) {
|
|
|
|
var newAttrs CheckcommandAttrs
|
|
newAttrs.Command = []string{command}
|
|
newAttrs.Arguments = command_arguments
|
|
|
|
var newCheckcommand CheckcommandStruct
|
|
newCheckcommand.Name = name
|
|
newCheckcommand.Type = "CheckCommand"
|
|
newCheckcommand.Attrs = newAttrs
|
|
|
|
// Create JSON from completed struct
|
|
payloadJSON, marshalErr := json.Marshal(newCheckcommand)
|
|
if marshalErr != nil {
|
|
return nil, marshalErr
|
|
}
|
|
|
|
//fmt.Printf("<payload> %s\n", payloadJSON)
|
|
|
|
// Make the API request to create the hosts.
|
|
results, err := server.NewAPIRequest("PUT", "/objects/checkcommands/"+name, []byte(payloadJSON))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if results.Code == 200 {
|
|
theCheckcommand, err := server.GetCheckcommand(name)
|
|
return theCheckcommand, err
|
|
}
|
|
|
|
return nil, fmt.Errorf("%s", results.ErrorString)
|
|
|
|
}
|
|
|
|
// DeleteCheckcommand ...
|
|
func (server *Server) DeleteCheckcommand(name string) error {
|
|
|
|
results, err := server.NewAPIRequest("DELETE", "/objects/checkcommands/"+name+"?cascade=1", nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if results.Code == 200 {
|
|
return nil
|
|
} else {
|
|
return fmt.Errorf("%s", results.ErrorString)
|
|
}
|
|
|
|
}
|