From 1b68302361851daeb1fc434764457606f4815df3 Mon Sep 17 00:00:00 2001
From: Tobias von der Krone <tobias.vonderkrone@cloud.ionos.com>
Date: Tue, 10 Sep 2019 20:56:59 +0200
Subject: [PATCH] Support setting "priority" in Icinga 2 integration

Add a parameter to set the priority of an alert in the Icinga 2
intergration. This is necessary since host down alerts mostly
are of priority P1.

refs #52

Signed-off-by: Tobias von der Krone <tobias.vonderkrone@cloud.ionos.com>
---
 icinga2/icinga2/icinga2opsgenie.go | 26 ++++++++++++++++++++++----
 icinga2/icinga2/opsgenie.conf      |  4 ++++
 2 files changed, 26 insertions(+), 4 deletions(-)

diff --git a/icinga2/icinga2/icinga2opsgenie.go b/icinga2/icinga2/icinga2opsgenie.go
index f29c743..fe5e18f 100644
--- a/icinga2/icinga2/icinga2opsgenie.go
+++ b/icinga2/icinga2/icinga2opsgenie.go
@@ -308,6 +308,7 @@ func parseFlags()map[string]string{
 
 	responders := flag.String("responders","","Responders")
 	tags := flag.String("tags","","Tags")
+	priority := flag.String("priority","","Priority")
 
 	flag.Parse()
 
@@ -388,6 +389,15 @@ func parseFlags()map[string]string{
 	parameters["service_perf_data"] = *servicePerfData
 	parameters["service_check_command"] = *serviceCheckCommand
 
+	if *priority == "" {
+		parameters["priority"] = ""
+	} else if IsValidPriority(*priority) {
+		parameters["priority"] = *priority
+	} else {
+		logger.Warning("Priority is not valid, needs to be one of P1, P2, P3, P4, P5.")
+		parameters["priority"] = ""
+	}
+
 	args := flag.Args()
 	for i := 0; i < len(args); i += 2 {
 		if(len(args)%2 != 0 && i==len(args)-1){
@@ -400,7 +410,15 @@ func parseFlags()map[string]string{
 	return parameters
 }
 
-
-
-
-
+func IsValidPriority(priority string) bool {
+	switch priority {
+	case
+		"P1",
+		"P2",
+		"P3",
+		"P4",
+		"P5":
+		return true
+	}
+	return false
+}
diff --git a/icinga2/icinga2/opsgenie.conf b/icinga2/icinga2/opsgenie.conf
index 3533409..c40675d 100644
--- a/icinga2/icinga2/opsgenie.conf
+++ b/icinga2/icinga2/opsgenie.conf
@@ -3,6 +3,7 @@ object NotificationCommand "opsgenie-service-notification" {
 
   vars.hgns = {{ host.groups.join(",") }}
   vars.sgns = {{ service.groups.join(",") }}
+  vars.priority = "P3"
   command = [ "/usr/bin/icinga2opsgenie" ]
   arguments = {
   "-entityType" = "service"
@@ -48,6 +49,7 @@ object NotificationCommand "opsgenie-service-notification" {
   "-lssc" = "$service.last_state_change$"
   "-so" = "$service.output$"
   "-spd" = "$service.perfdata$"
+  "-priority" = "$priority$"
   }
 }
 
@@ -55,6 +57,7 @@ object NotificationCommand "opsgenie-host-notification" {
   import "plugin-notification-command"
 
   vars.hgns = {{ host.groups.join(",") }}
+  vars.priority = "P3"
   command = [ "/usr/bin/icinga2opsgenie" ]
   arguments = {
   "-entityType" = "host"
@@ -81,6 +84,7 @@ object NotificationCommand "opsgenie-host-notification" {
   "-lhsc" = "$host.last_state_change$"
   "-ho" = "$host.output$"
   "-hpd" = "$host.perfdata$"
+  "-priority" = "$priority$"
   }
 }