taigrr/wtf
1
0
Fork 0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #376 from Seanstoppable/pd-alerts

Browse Source 
Add incident showing to pagerduty
This commit is contained in:
Chris Cummer 2019-02-11 03:17:29 -08:00 committed by GitHub
commit c61e77469f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 70 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
pagerduty/client.go
View File

@ -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",

40
pagerduty/widget.go
View File

@ -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
}