// AddTeamAction adds a team to an alert at OpsGenie.
func AddTeamAction(c *gcli.Context) {
cli, err := NewAlertClient(c)
if err != nil {
os.Exit(1)
}
req := alerts.AddTeamAlertRequest{}
if val, success := getVal("id", c); success {
req.ID = val
}
if val, success := getVal("alias", c); success {
req.Alias = val
}
if val, success := getVal("team", c); success {
req.Team = val
}
req.User = grabUsername(c)
if val, success := getVal("source", c); success {
req.Source = val
}
if val, success := getVal("note", c); success {
req.Note = val
}
printVerboseMessage("Add team request prepared from flags, sending request to OpsGenie..")
_, err = cli.AddTeam(req)
if err != nil {
fmt.Printf("%s\n", err.Error())
os.Exit(1)
}
printVerboseMessage("Team added successfully.")
}
// AddTeam method adds a team to an alert at OpsGenie.
func (cli *OpsGenieAlertClient) AddTeam(req alerts.AddTeamAlertRequest) (*alerts.AddTeamAlertResponse, error) {
req.APIKey = cli.apiKey
resp, err := cli.sendRequest(cli.buildPostRequest(addTeamAlertURL, req))
if resp == nil {
return nil, err
}
defer resp.Body.Close()
var addTeamAlertResp alerts.AddTeamAlertResponse
if err = resp.Body.FromJsonTo(&addTeamAlertResp); err != nil {
message := "Server response can not be parsed, " + err.Error()
logging.Logger().Warn(message)
return nil, errors.New(message)
}
return &addTeamAlertResp, nil
}
func (cli *OpsGenieAlertClient) AddTeam(req alerts.AddTeamAlertRequest) (*alerts.AddTeamAlertResponse, error) {
req.ApiKey = cli.apiKey
// validate the mandatory parameters: apiKey, alertId/alias, team
if req.ApiKey == "" {
return nil, errors.New("ApiKey is a mandatory field and can not be empty.")
}
if req.Team == "" {
return nil, errors.New("Team is a mandatory field and can not be empty")
}
if req.AlertId == "" && req.Alias == "" {
return nil, errors.New("At least either Alert Id or Alias should be set in the request.")
}
if req.AlertId != "" && req.Alias != "" {
return nil, errors.New("Either Alert Id or Alias should be set in the request not both.")
}
var resp *goreq.Response
var err error
for i := 0; i < cli.retries; i++ {
resp, err = cli.buildRequest("POST", ADD_TEAM_ALERT_URL, req).Do()
if err == nil {
break
}
time.Sleep(TIME_SLEEP_BETWEEN_REQUESTS)
}
if err != nil {
return nil, errors.New("Team can not be added, unable to send the request")
}
// check the returning HTTP status code
httpStatusCode := resp.StatusCode
if httpStatusCode >= 400 && httpStatusCode < 500 {
return nil, errors.New(fmt.Sprintf("Client error %d returned", httpStatusCode))
}
if httpStatusCode >= 500 {
return nil, errors.New(fmt.Sprintf("Server error %d returned", httpStatusCode))
}
var addTeamAlertResp alerts.AddTeamAlertResponse
// check if the response can be unmarshalled
if err = resp.Body.FromJsonTo(&addTeamAlertResp); err != nil {
return nil, errors.New("Server response can not be parsed.")
}
return &addTeamAlertResp, nil
}