Fix icinga2 host frontend

Author: uubk

Gopkg.lock

@@ -32,3 +32,7 @@
 [[constraint]]
   name = "gopkg.in/yaml.v2"
   version = "2.2.1"
+
+[[constraint]]
+  name = "github.com/sirupsen/logrus"
+  version = "1.0.5"

Gopkg.toml

@@ -0,0 +1,3 @@
+build:
+	go build -i -v -ldflags="-X main.gitversion=$(shell git describe --always --long --dirty)" 
+	strip sendmsg

Makefile

@@ -1,6 +1,23 @@
 package main
 
-import "flag"
+import (
+	"flag"
+	"bytes"
+	"net/url"
+	"fmt"
+	"time"
+	"github.com/sirupsen/logrus"
+	"strconv"
+)
+
+var icGoodState = map[string]bool {
+	"UP": true,
+}
+
+var icBadState = map[string]bool {
+	"UNREACHABLE": true,
+	"DOWN": true,
+}
 
 type icingaHostCmd struct {
 	SimpleCFGLocation *string
@@ -13,6 +30,12 @@ type icingaHostCmd struct {
 	icHostNotfnUser *string
 	icHostNotfnComment *string
 	icURL *string
+	// Fields below are unused currently but need to be parsed in order to be compliant
+	icDateTime *string
+	icHostDisplayname *string
+	icUserEmail *string
+	icMailFrom *string
+	icToSyslog *string
 }
 
 func (this *icingaHostCmd) Init(flagSet *flag.FlagSet) {
@@ -25,18 +48,80 @@ func (this *icingaHostCmd) Init(flagSet *flag.FlagSet) {
 	this.icHostCheckOutput = flagSet.String("o", "", "Check output")
 	this.icHostNotfnUser = flagSet.String("b", "", "Manual host notification user")
 	this.icHostNotfnComment = flagSet.String("c", "", "Manual host notification comment")
-	this.icHostNotfnComment = flagSet.String("i", "", "URL of Webinterface")}
+	this.icURL = flagSet.String("i", "", "URL of Webinterface")
+	this.icDateTime = flagSet.String("d", "", "Date/Time of event")
+	this.icHostDisplayname = flagSet.String("n", "", "Host display name")
+	this.icUserEmail = flagSet.String("r", "", "User email")
+	this.icMailFrom = flagSet.String("f", "", "Sender email")
+	this.icToSyslog = flagSet.String("v", "", "Send to syslog")
+}
 
 func (this *icingaHostCmd) Parse() Message {
 	var msg Message
 
-	msg.Fields = []Field {
-		{
-			Header: "IPv4",
-			Text: *this.icHostV4,
-		},
+	msg.Body_title = *this.icHostname + " is " + *this.icHostState
+
+	addFieldToMessage(&msg, true, "IPv4", this.icHostV4)
+	addFieldToMessage(&msg, true, "IPv6", this.icHostV6)
+	addFieldToMessage(&msg, true, "Notification type", this.icHostNotfnType)
+	addFieldToMessage(&msg, true, "Notification user", this.icHostNotfnUser)
+	timestamp, err := time.Parse("2006-01-02 15:04:05 -0700", *this.icDateTime)
+	if err != nil {
+		logrus.WithFields(logrus.Fields{
+			"error": err,
+		}).Warn("Couldn't parse supplied timestamp")
+		panic("Abort")
+	} else {
+		time := convertToSlackDate(timestamp)
+		addFieldToMessage(&msg, true, "Timestamp", &time)
 	}
-	// TODO
+
+	var buffer bytes.Buffer
+
+	buffer.WriteString("```\n")
+	buffer.WriteString(*this.icHostCheckOutput)
+	buffer.WriteString("```\n")
+	if *this.icHostNotfnComment != "" {
+		buffer.WriteString("Comment: ")
+		buffer.WriteString(*this.icHostNotfnComment)
+	}
+	msg.Body = buffer.String()
+
+	if *this.icURL != "" {
+		if err != nil {
+			fmt.Errorf("WARNING: Couldn't parse specified url, skipping link: ", err)
+		} else {
+			params := url.Values{}
+			params.Add("host", *this.icHostname)
+			msg.Body_link = *this.icURL + "/monitoring/host/show?"+ params.Encode()
+		}
+	}
+
+	if icBadState[*this.icHostState] {
+		msg.Color = "#ff5566"
+	} else if icGoodState[*this.icHostState] {
+		msg.Color = "#44bb77"
+	} else {
+		msg.Color = "#aa44ff"
+	}
+
+	msg.Frontend = "Icinga2 Host Notification"
+	msg.FrontendIconURL = "https://raw.githubusercontent.com/Icinga/icingaweb2/master/public/img/favicon.png"
 
 	return msg
+}
+
+func addFieldToMessage(msg *Message, short bool, header string, field *string) {
+	if *field != "" {
+		msg.Fields = append(msg.Fields, Field{
+			Header: header,
+			Text: *field,
+			Short: short,
+		})
+	}
+}
+
+func convertToSlackDate(timestamp time.Time) string {
+	return "<!date^" + strconv.FormatInt(timestamp.Unix(), 10) + "^{date_num} {time_secs}|" +
+		timestamp.Format("Mon Jan 2 15:04:05 -0700 MST 2006")+ ">"
 }

icinga_host.go

@@ -1,8 +1,8 @@
 package main
 
 import (
-	"strings"
 	"bytes"
+	"strings"
 )
 
 type fieldList []Field
@@ -15,7 +15,7 @@ func (e *FieldArgParseError) Error() string {
 	return e.err
 }
 
-func (s *fieldList) String() string{
+func (s *fieldList) String() string {
 	var buffer bytes.Buffer
 	for _, item := range *s {
 		buffer.WriteString("Element Title: ")
@@ -46,4 +46,4 @@ func (s *fieldList) Set(val string) error {
 		*s = append(*s, field)
 	}
 	return nil
-}
+}

listarg.go

@@ -2,13 +2,18 @@ package main
 
 import (
 	"flag"
-	"io/ioutil"
 	"fmt"
-	"os"
+	"github.com/sirupsen/logrus"
 	"gopkg.in/yaml.v2"
+	"io/ioutil"
+	"os"
 )
 
+var gitversion = "undefined"
+
 func main() {
+	logrus.SetOutput(os.Stdout)
+	logrus.SetLevel(logrus.WarnLevel)
 	// 'Simple' frontend - that is specifying each field manually
 	simpleCommand := flag.NewFlagSet("simple", flag.ExitOnError)
 	simpleFrontendCmd := simpleCmdData{}
@@ -66,6 +71,6 @@ func main() {
 
 	switch cfg.Backend {
 	case "slack":
-		send_with_slack(msg, cfg, "simple", "")
+		send_with_slack(msg, cfg)
 	}
 }

sendmsg.go

@@ -4,12 +4,12 @@ import "flag"
 
 type simpleCmdData struct {
 	SimpleCFGLocation *string
-	head *string
-	title *string
-	title_url *string
-	body *string
-	color *string
-	fields fieldList
+	head              *string
+	title             *string
+	title_url         *string
+	body              *string
+	color             *string
+	fields            fieldList
 }
 
 func (this *simpleCmdData) Init(flagSet *flag.FlagSet) {
@@ -30,5 +30,8 @@ func (this *simpleCmdData) Parse() Message {
 	msg.Body_title = *this.title
 	msg.Body_link = *this.title_url
 	msg.Fields = this.fields
+
+	msg.Frontend = "simple"
+
 	return msg
-}
+}

simple.go

@@ -1,36 +1,36 @@
 package main
 
 import (
-	"net/http"
 	"bytes"
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
+	"net/http"
 )
 
 type SlackField struct {
-	Text string `json:"value,omitempty"`
+	Text  string `json:"value,omitempty"`
 	Title string `json:"title,omitempty"`
-	Short bool `json:"short,omitempty"`
+	Short bool   `json:"short,omitempty"`
 }
 
 type SlackAttachments struct {
-	Color string `json:"color,omitempty"`
-	Text string `json:"text,omitempty"`
-	Title string  `json:"title,omitempty"`
-	TitleLink string  `json:"title_link,omitempty"`
-	Fields []SlackField `json:"fields,omitempty"`
-	Footer string `json:"footer,omitempty"`
-	FooterIcon string `json:"footer_icon,omitempty"`
-	Head string `json:"author_name,omitempty"`
+	Color      string       `json:"color,omitempty"`
+	Text       string       `json:"text,omitempty"`
+	Title      string       `json:"title,omitempty"`
+	TitleLink  string       `json:"title_link,omitempty"`
+	Fields     []SlackField `json:"fields,omitempty"`
+	Footer     string       `json:"footer,omitempty"`
+	FooterIcon string       `json:"footer_icon,omitempty"`
+	Head       string       `json:"author_name,omitempty"`
 }
 
 type SlackMessage struct {
-	Text string `json:"text,omitempty"`
-	Attachments []SlackAttachments  `json:"attachments,omitempty"`
+	Text        string             `json:"text,omitempty"`
+	Attachments []SlackAttachments `json:"attachments,omitempty"`
 }
 
-func send_with_slack(msg Message, cfg Config, frontend string, frontendIcon string) {
+func send_with_slack(msg Message, cfg Config) {
 	var body SlackMessage
 	var attachement SlackAttachments
 	attachement.Head = msg.Head
@@ -53,9 +53,9 @@ func send_with_slack(msg Message, cfg Config, frontend string, frontendIcon stri
 		fields = append(fields, field)
 	}
 	attachement.Fields = fields
-	attachement.Footer = frontend + " (sendmsg)"
-	if frontendIcon != "" {
-		attachement.FooterIcon = frontendIcon
+	attachement.Footer = msg.Frontend + " (sendmsg "+ gitversion+ ")"
+	if msg.FrontendIconURL != "" {
+		attachement.FooterIcon = msg.FrontendIconURL
 	}
 	body.Attachments = []SlackAttachments{attachement}
 
@@ -81,4 +81,4 @@ func send_with_slack(msg Message, cfg Config, frontend string, frontendIcon stri
 	if bodyString != "ok" {
 		fmt.Errorf("Slack didn't like what we sent, error: ", bodyString)
 	}
-}
+}

slack.go

@@ -1,23 +1,25 @@
 package main
 
 type Field struct {
-	Text string
+	Text   string
 	Header string
-	Short bool
+	Short  bool
 }
 
 type Message struct {
-	Head string
-	Body_title string
-	Body_link string
-	Body string
-	Color string
-	Fields []Field
+	Head            string
+	Body_title      string
+	Body_link       string
+	Body            string
+	Color           string
+	Fields          []Field
+	Frontend        string
+	FrontendIconURL string
 }
 
 type Config struct {
 	Backend string `yaml:"backend"`
 	Webhook string `yaml:"webhook"`
 }
 
-type CmdData struct {}
+type CmdData struct{}