Merge branch 'rewbycraft-master'

Author: benjojo

main.go

@@ -5,24 +5,26 @@ import (
 	"encoding/json"
 	"flag"
 	"fmt"
-        "os"
 	"io/ioutil"
 	"net/http"
+	"os"
 	"strings"
 )
 
+type alertManAlert struct {
+	Annotations struct {
+		Description string `json:"description"`
+		Summary     string `json:"summary"`
+	} `json:"annotations"`
+	EndsAt       string            `json:"endsAt"`
+	GeneratorURL string            `json:"generatorURL"`
+	Labels       map[string]string `json:"labels"`
+	StartsAt     string            `json:"startsAt"`
+	Status       string            `json:"status"`
+}
+
 type alertManOut struct {
-	Alerts []struct {
-		Annotations struct {
-			Description string `json:"description"`
-			Summary     string `json:"summary"`
-		} `json:"annotations"`
-		EndsAt       string            `json:"endsAt"`
-		GeneratorURL string            `json:"generatorURL"`
-		Labels       map[string]string `json:"labels"`
-		StartsAt     string            `json:"startsAt"`
-		Status       string            `json:"status"`
-	} `json:"alerts"`
+	Alerts            []alertManAlert `json:"alerts"`
 	CommonAnnotations struct {
 		Summary string `json:"summary"`
 	} `json:"commonAnnotations"`
@@ -48,7 +50,7 @@ func main() {
 	webhookUrl := os.Getenv("DISCORD_WEBHOOK")
 	if webhookUrl == "" {
 		fmt.Fprintf(os.Stderr, "error: environment variable DISCORD_WEBHOOK not found\n")
-        	os.Exit(1)
+		os.Exit(1)
 	}
 	whURL := flag.String("webhook.url", webhookUrl, "")
 	flag.Parse()
@@ -65,26 +67,34 @@ func main() {
 			panic(err)
 		}
 
-		DO := discordOut{
-			Name: amo.Status,
-		}
+		groupedAlerts := make(map[string][]alertManAlert)
 
-		Content := "```"
-		if amo.CommonAnnotations.Summary != "" {
-			Content = fmt.Sprintf(" === %s === \n```", amo.CommonAnnotations.Summary)
+		for _, alert := range amo.Alerts {
+			groupedAlerts[alert.Status] = append(groupedAlerts[alert.Status], alert)
 		}
 
-		for _, alert := range amo.Alerts {
-			realname := alert.Labels["instance"]
-			if strings.Contains(realname, "localhost") && alert.Labels["exported_instance"] != "" {
-				realname = alert.Labels["exported_instance"]
+		for status, alerts := range groupedAlerts {
+			DO := discordOut{
+				Name: status,
 			}
-			Content += fmt.Sprintf("[%s]: %s on %s\n%s\n\n", strings.ToUpper(amo.Status), alert.Labels["alertname"], realname, alert.Annotations.Description)
-		}
 
-		DO.Content = Content + "```"
+			Content := "```"
+			if amo.CommonAnnotations.Summary != "" {
+				Content = fmt.Sprintf(" === %s === \n```", amo.CommonAnnotations.Summary)
+			}
+
+			for _, alert := range alerts {
+				realname := alert.Labels["instance"]
+				if strings.Contains(realname, "localhost") && alert.Labels["exported_instance"] != "" {
+					realname = alert.Labels["exported_instance"]
+				}
+				Content += fmt.Sprintf("[%s]: %s on %s\n%s\n\n", strings.ToUpper(status), alert.Labels["alertname"], realname, alert.Annotations.Description)
+			}
 
-		DOD, _ := json.Marshal(DO)
-		http.Post(*whURL, "application/json", bytes.NewReader(DOD))
+			DO.Content = Content + "```"
+
+			DOD, _ := json.Marshal(DO)
+			http.Post(*whURL, "application/json", bytes.NewReader(DOD))
+		}
 	}))
 }