Merge pull request #376 from Seanstoppable/pd-alerts

Add incident showing to pagerduty

pagerduty/client.go

@@ -37,6 +37,34 @@ func GetOnCalls() ([]pagerduty.OnCall, error) {
 	return results, nil
 }
 
+// GetIncidents returns a list of people currently on call
+func GetIncidents() ([]pagerduty.Incident, error) {
+	client := pagerduty.NewClient(apiKey())
+
+	var results []pagerduty.Incident
+
+	var queryOpts pagerduty.ListIncidentsOptions
+	queryOpts.DateRange = "all"
+	queryOpts.Statuses = []string{"triggered", "acknowledged"}
+
+	items, err := client.ListIncidents(queryOpts)
+	if err != nil {
+		return nil, err
+	}
+	results = append(results, items.Incidents...)
+
+	for items.APIListObject.More == true {
+		queryOpts.APIListObject.Offset = items.APIListObject.Offset
+		items, err = client.ListIncidents(queryOpts)
+		if err != nil {
+			return nil, err
+		}
+		results = append(results, items.Incidents...)
+	}
+
+	return results, nil
+}
+
 func apiKey() string {
 	return wtf.Config.UString(
 		"wtf.mods.pagerduty.apiKey",

pagerduty/widget.go

@@ -24,18 +24,35 @@ func NewWidget(app *tview.Application) *Widget {
 /* -------------------- Exported Functions -------------------- */
 
 func (widget *Widget) Refresh() {
-	onCalls, err := GetOnCalls()
+	var onCalls []pagerduty.OnCall
+	var incidents []pagerduty.Incident
+
+	var err1 error
+	var err2 error
+
+	if wtf.Config.UBool("wtf.mods.pagerduty.showSchedules", true) {
+		onCalls, err1 = GetOnCalls()
+	}
+
+	if wtf.Config.UBool("wtf.mods.pagerduty.showIncidents") {
+		incidents, err2 = GetIncidents()
+	}
 
 	widget.View.SetTitle(widget.ContextualTitle(fmt.Sprintf("%s", widget.Name)))
 	widget.View.Clear()
 
 	var content string
-	if err != nil {
+	if err1 != nil || err2 != nil {
 		widget.View.SetWrap(true)
-		content = err.Error()
+		if err1 != nil {
+			content = content + err1.Error()
+		}
+		if err2 != nil {
+			content = content + err2.Error()
+		}
 	} else {
 		widget.View.SetWrap(false)
-		content = widget.contentFrom(onCalls)
+		content = widget.contentFrom(onCalls, incidents)
 	}
 
 	widget.View.SetText(content)
@@ -43,9 +60,19 @@ func (widget *Widget) Refresh() {
 
 /* -------------------- Unexported Functions -------------------- */
 
-func (widget *Widget) contentFrom(onCalls []pagerduty.OnCall) string {
+func (widget *Widget) contentFrom(onCalls []pagerduty.OnCall, incidents []pagerduty.Incident) string {
 	var str string
 
+	if len(incidents) > 0 {
+		str = str + "[yellow]Incidents[white]\n"
+		for _, incident := range incidents {
+			str = str + fmt.Sprintf("[red]%s[white]\n", incident.Summary)
+			str = str + fmt.Sprintf("Status: %s\n", incident.Status)
+			str = str + fmt.Sprintf("Service: %s\n", incident.Service.Summary)
+			str = str + fmt.Sprintf("Escalation: %s\n", incident.EscalationPolicy.Summary)
+		}
+	}
+
 	tree := make(map[string][]pagerduty.OnCall)
 
 	filtering := wtf.Config.UList("wtf.mods.pagerduty.escalationFilter")
@@ -69,6 +96,8 @@ func (widget *Widget) contentFrom(onCalls []pagerduty.OnCall) string {
 
 	sort.Strings(keys)
 
+	if len(keys) > 0 {
+		str = str + "[yellow]Schedules[white]\n"
 		// Print out policies, and escalation order of users
 		for _, key := range keys {
 			str = str + fmt.Sprintf("[red]%s\n", key)
@@ -78,6 +107,7 @@ func (widget *Widget) contentFrom(onCalls []pagerduty.OnCall) string {
 				str = str + fmt.Sprintf("[white]%d - %s\n", item.EscalationLevel, item.User.Summary)
 			}
 		}
+	}
 
 	return str
 }