From 9b97bffa892e513849251be98d4a4c28aabd3e6d Mon Sep 17 00:00:00 2001
From: Allan Frese <allan@frese.com>
Date: Tue, 5 May 2020 16:15:16 +0200
Subject: [PATCH] add warning icon for high urgency incidents

---
 .../Contents/Resources/warning.png                | Bin 0 -> 848 bytes
 src/app.go                                        |  12 ++++++++----
 src/conf.go                                       |   6 +++---
 src/pd.go                                         |   9 ++++++++-
 4 files changed, 19 insertions(+), 8 deletions(-)
 create mode 100644 dist/PagerdutyNotifier.app/Contents/Resources/warning.png

diff --git a/dist/PagerdutyNotifier.app/Contents/Resources/warning.png b/dist/PagerdutyNotifier.app/Contents/Resources/warning.png
new file mode 100644
index 0000000000000000000000000000000000000000..f50de29b64c5a1015c5f063a8ed0b7707fb48fee
GIT binary patch
literal 848
zcmV-W1F!svP)<h;3K|Lk000e1NJLTq002M$002M;1^@s6s%dfF00004b3#c}2nYxW
zd<bNS00009a7bBm000XU000XU0RWnu7ytkQ@<~KNRCt{2nm<TZQ5c7x&$2Qi!uV7w
z5Q4x#C<q#AXlM|!!KI-gAwpUlT3Q-fT#8%@T3i}hS_-Y9p`pN_p`jouhzcW!Kq4q4
z`}4kUL)W;O_w(F8=N_x`z@4u5e81=CIro0|C`2NWNF)-8L?X3e&`=l=xF;~JaLk|)
zI#+E7Tr{xsA#gr)u3AtSTr9tEC=7(o=M{tX0y9|!2)t0}2%Xa#24}P6QzeXr&glh(
zet~Ze1$;H=3!S%84B7;ymy=JGaOa<TDkvOxl~3IehePMC1cj!+`@97R%qnz-&R-gV
z%X#HfB}|0QUkZi&MdedVVQ*-ULKv(Ocv`jqfyV~zp?z|raJsB~Y8IRd?UM_G?E;^x
zDnMXPp*OUb69x@|TMkm!6`BfN0+S9uKNZ@|34_Cq)4I8EjY7BMY#0jZwk!%81>U;I
zEWXxpDZDY*5W>v~feWtddhxY6?g!%`+?K^)w}8}8z;}h6AsjO?SS9cv&z*gVG4ebW
z?i;j)a9t9G(L4fW3rKFFCqlUXfx;Gnk98Ly@Ij$Fg!69*T+JK4iUN`nu7q&@g~C9Q
z;8hinEQJFh+)?Ndcp16?f#(M6YA--wtZ3*e3rJ?cnOfxy`jXAd6;Xh|yuyyU<QoQe
z%LJ~vfTV=mR>wxF!f@HJ$+YcC$@f$=<`LiWodUB}gcX=f-U53BZdNqrJA+N$3J{p6
zBB;+p;F1@4g?*vr91Bo|J(bEEv?rUB!37;lcxbTN#m(uB%HU)h^p~r#KH6ZH!DuCe
zDf9}=dH5}*_XMXrjQ>etTPb<_xxsnE$22XtCz!W{Yh?=<^0iKVd=JR@oI{1=lb4;>
zzJdu%RsB66<8xjqtSwT&h_6s}9|}jaJTspH1_yJuwn8x&%sSWVG1zQz?r&pr)`?aH
zJn|4E6!~2gwkCgx3BHC)-UMd~Z=NRq-ZC!mOyEmsnH<vd0#5|aX^ZI}i9{liNF);Z
apZNjBM~#$0g0oHl0000<MNUMnLSTYp<!~zi

literal 0
HcmV?d00001

diff --git a/src/app.go b/src/app.go
index d7626de..a128527 100644
--- a/src/app.go
+++ b/src/app.go
@@ -25,7 +25,7 @@ var menuItems = []trayhost.MenuItem{
 	{
 		Title: "Pause",
 		Handler: func() {
-			appNotify("Pagerduty Notifier", "This is not implemeted yet.", "", 30*time.Second)
+			appNotify("Pagerduty Notifier", "This is not implemeted yet.", "", nil, 30*time.Second)
 		},
 	},
 	{
@@ -67,10 +67,10 @@ func appInit() {
 func toggleStartup() {
 	if existsLaunchConf() {
 		deleteLaunchConf()
-		appNotify("Pagerduty Notifier", "Removed from Launch configuration", "", 10*time.Second)
+		appNotify("Pagerduty Notifier", "Removed from Launch configuration", "", nil, 10*time.Second)
 	} else {
 		writeLaunchConf()
-		appNotify("Pagerduty Notifier", "Added to launch configuration", "", 10*time.Second)
+		appNotify("Pagerduty Notifier", "Added to launch configuration", "", nil, 10*time.Second)
 	}
 }
 
@@ -79,7 +79,7 @@ func appEnterLoop() {
 	trayhost.EnterLoop()
 }
 
-func appNotify(title string, message string, url string, timeout time.Duration) {
+func appNotify(title string, message string, url string, image *trayhost.Image, timeout time.Duration) {
 
 	notification := trayhost.Notification{
 		Title:   title,
@@ -91,6 +91,10 @@ func appNotify(title string, message string, url string, timeout time.Duration)
 		notification.Handler = func() { openBrowser(url) }
 	}
 
+	if image != nil {
+		notification.Image = *image
+	}
+
 	notification.Display()
 }
 
diff --git a/src/conf.go b/src/conf.go
index df779db..387f893 100644
--- a/src/conf.go
+++ b/src/conf.go
@@ -78,11 +78,11 @@ func cfgInit() *ini.File {
 
 		switch runtime.GOOS {
 		case "linux":
-			appNotify("HOME/.pagerduty.ini", "No ini file found, click here to see how.", "https://github.com/trustpilot/pagerduty-notifier", 0)
+			appNotify("HOME/.pagerduty.ini", "No ini file found, click here to see how.", "https://github.com/trustpilot/pagerduty-notifier", nil, 0)
 			os.Exit(1)
 
 		case "windows":
-			appNotify("HOME/.pagerduty.ini", "No ini file found, click here to see how.", "https://github.com/trustpilot/pagerduty-notifier", 0)
+			appNotify("HOME/.pagerduty.ini", "No ini file found, click here to see how.", "https://github.com/trustpilot/pagerduty-notifier", nil, 0)
 			os.Exit(1)
 
 		case "darwin":
@@ -99,7 +99,7 @@ func cfgInit() *ini.File {
 				os.Exit(1)
 			}
 
-			appNotify("HOME/.pagerduty.ini", "Created default config file, please edit and add token !", "https://github.com/trustpilot/pagerduty-notifier", 0)
+			appNotify("HOME/.pagerduty.ini", "Created default config file, please edit and add token !", "https://github.com/trustpilot/pagerduty-notifier", nil, 0)
 			os.Exit(0)
 
 		default:
diff --git a/src/pd.go b/src/pd.go
index 67a390d..a16ae41 100644
--- a/src/pd.go
+++ b/src/pd.go
@@ -9,6 +9,7 @@ import (
 
 	"github.com/PagerDuty/go-pagerduty"
 	"github.com/go-ini/ini"
+	"github.com/shurcooL/trayhost"
 )
 
 var pd *pagerduty.Client
@@ -208,6 +209,12 @@ func pdNotify(i pagerduty.Incident) {
 		title = fmt.Sprintf("Incident %s", i.Status)
 		message = removeCharacters(i.APIObject.Summary, "[]")
 	}
+	image := trayhost.Image{}
 
-	appNotify(title, message, i.APIObject.HTMLURL, 0)
+	if i.Urgency == "high" {
+		image.Kind = "png"
+		image.Bytes = getIcon("warning.png")
+	}
+
+	appNotify(title, message, i.APIObject.HTMLURL, &image, 0)
 }