Initial import

Author: uubk

.gitignore

@@ -0,0 +1,3 @@
+vendor
+.idea
+config.yml

Gopkg.lock

@@ -0,0 +1,34 @@
+# Gopkg.toml example
+#
+# Refer to https://golang.github.io/dep/docs/Gopkg.toml.html
+# for detailed Gopkg.toml documentation.
+#
+# required = ["github.com/user/thing/cmd/thing"]
+# ignored = ["github.com/user/project/pkgX", "bitbucket.org/user/project/pkgA/pkgY"]
+#
+# [[constraint]]
+#   name = "github.com/user/project"
+#   version = "1.0.0"
+#
+# [[constraint]]
+#   name = "github.com/user/project2"
+#   branch = "dev"
+#   source = "github.com/myfork/project2"
+#
+# [[override]]
+#   name = "github.com/x/y"
+#   version = "2.4.0"
+#
+# [prune]
+#   non-go = false
+#   go-tests = true
+#   unused-packages = true
+
+
+[prune]
+  go-tests = true
+  unused-packages = true
+
+[[constraint]]
+  name = "gopkg.in/yaml.v2"
+  version = "2.2.1"

Gopkg.toml

@@ -0,0 +1,36 @@
+package main
+
+import "flag"
+
+type icingaHostCmd struct {
+	SimpleCFGLocation *string
+	icHostname *string
+	icHostV4 *string
+	icHostV6 *string
+	icHostState *string
+	icHostNotfnType *string
+	icHostCheckOutput *string
+	icHostNotfnUser *string
+	icHostNotfnComment *string
+	icURL *string
+}
+
+func (this *icingaHostCmd) Init(flagSet *flag.FlagSet) {
+	this.SimpleCFGLocation = flagSet.String("cfg", "/etc/sendmsg.yml", "Path to sendmsg config")
+	this.icHostname = flagSet.String("l", "", "Hostname")
+	this.icHostV4 = flagSet.String("4", "", "Host address (v4)")
+	this.icHostV6 = flagSet.String("6", "", "Host address (v6)")
+	this.icHostState = flagSet.String("s", "", "Host state")
+	this.icHostNotfnType = flagSet.String("t", "", "Notification type")
+	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")}
+
+func (this *icingaHostCmd) Parse() Message {
+	var msg Message
+
+	// TODO
+
+	return msg
+}

icinga_host.go

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"strings"
+	"bytes"
+)
+
+type fieldList []Field
+
+type FieldArgParseError struct {
+	err string
+}
+
+func (e *FieldArgParseError) Error() string {
+	return e.err
+}
+
+func (s *fieldList) String() string{
+	var buffer bytes.Buffer
+	for _, item := range *s {
+		buffer.WriteString("Element Title: ")
+		buffer.WriteString(item.Header)
+		buffer.WriteString(", Text: ")
+		buffer.WriteString(item.Text)
+		buffer.WriteString(", Short?: ")
+		if item.Short {
+			buffer.WriteString("yes\n")
+		} else {
+			buffer.WriteString("no\n")
+		}
+	}
+	return buffer.String()
+}
+
+func (s *fieldList) Set(val string) error {
+	arr := strings.Split(val, ",")
+	for _, elem := range arr {
+		values := strings.Split(elem, ":")
+		if len(values) != 3 {
+			return &FieldArgParseError{"Split produced unexpected number of parts (delimiter ':')"}
+		}
+		var field Field
+		field.Text = values[1]
+		field.Header = values[0]
+		field.Short = values[2] == "yes"
+		*s = append(*s, field)
+	}
+	return nil
+}

listarg.go

@@ -0,0 +1,71 @@
+package main
+
+import (
+	"flag"
+	"io/ioutil"
+	"fmt"
+	"os"
+	"gopkg.in/yaml.v2"
+)
+
+func main() {
+	// 'Simple' frontend - that is specifying each field manually
+	simpleCommand := flag.NewFlagSet("simple", flag.ExitOnError)
+	simpleFrontendCmd := simpleCmdData{}
+	simpleFrontendCmd.Init(simpleCommand)
+
+	// 'Icinga2' frontend - use this as a drop-in replacement for mails
+	icingaHostCommand := flag.NewFlagSet("icingaHost", flag.ExitOnError)
+	icingaHostCmd := icingaHostCmd{}
+	icingaHostCmd.Init(icingaHostCommand)
+
+	if len(os.Args) < 2 {
+		fmt.Print("Please use one of [simple, icingaHost]")
+		return
+	}
+
+	var cfglocation *string
+	switch os.Args[1] {
+	case "simple":
+		simpleCommand.Parse(os.Args[2:])
+		cfglocation = simpleFrontendCmd.SimpleCFGLocation
+		break
+	case "icingaHost":
+		simpleCommand.Parse(os.Args[2:])
+		cfglocation = icingaHostCmd.SimpleCFGLocation
+		break
+	default:
+		flag.PrintDefaults()
+		return
+	}
+
+	cfg := Config{}
+	if contents, err := ioutil.ReadFile(*cfglocation); err == nil {
+		err = yaml.UnmarshalStrict(contents, &cfg)
+		if err != nil {
+			fmt.Println("ERROR: Couldn't parse the configuration file!", err)
+			os.Exit(-1)
+		}
+	} else {
+		fmt.Println("ERROR: Couldn't read the configuration file!", err)
+		os.Exit(-1)
+	}
+
+	var msg Message
+	if simpleCommand.Parsed() {
+		msg = simpleFrontendCmd.Parse()
+	}
+	if icingaHostCommand.Parsed() {
+		msg = icingaHostCmd.Parse()
+	}
+
+	if msg.Body == "" || msg.Body_title == "" {
+		fmt.Println("ERROR: Either title or body are missing. Aborting now!")
+		return
+	}
+
+	switch cfg.Backend {
+	case "slack":
+		send_with_slack(msg, cfg, "simple", "")
+	}
+}

sendmsg.go

@@ -0,0 +1,2 @@
+backend: slack
+webhook: https://hooks.slack.com/services/something/secret/changeme

sendmsg.yml

@@ -0,0 +1,34 @@
+package main
+
+import "flag"
+
+type simpleCmdData struct {
+	SimpleCFGLocation *string
+	head *string
+	title *string
+	title_url *string
+	body *string
+	color *string
+	fields fieldList
+}
+
+func (this *simpleCmdData) Init(flagSet *flag.FlagSet) {
+	this.SimpleCFGLocation = flagSet.String("cfg", "/etc/sendmsg.yml", "Path to sendmsg config")
+	this.head = flagSet.String("head", "", "The header of the message to send (required)")
+	this.title = flagSet.String("title", "", "The title of the message to send (required)")
+	this.title_url = flagSet.String("title_url", "", "The url of the title of the message to send")
+	this.body = flagSet.String("body", "", "The body of the message to send")
+	this.color = flagSet.String("color", "", "The color of the message to send")
+	flagSet.Var(&this.fields, "fields", "A comma seperated list of fields (name:text) to be added")
+}
+
+func (this *simpleCmdData) Parse() Message {
+	var msg Message
+	msg.Body = *this.body
+	msg.Head = *this.head
+	msg.Color = *this.color
+	msg.Body_title = *this.title
+	msg.Body_link = *this.title_url
+	msg.Fields = this.fields
+	return msg
+}

simple.go

@@ -0,0 +1,84 @@
+package main
+
+import (
+	"net/http"
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+)
+
+type SlackField struct {
+	Text string `json:"value,omitempty"`
+	Title string `json:"title,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"`
+}
+
+type SlackMessage struct {
+	Text string `json:"text,omitempty"`
+	Attachments []SlackAttachments  `json:"attachments,omitempty"`
+}
+
+func send_with_slack(msg Message, cfg Config, frontend string, frontendIcon string) {
+	var body SlackMessage
+	var attachement SlackAttachments
+	attachement.Head = msg.Head
+	attachement.Text = msg.Body
+	if msg.Color != "" {
+		attachement.Color = msg.Color
+	}
+	if msg.Body_title != "" {
+		attachement.Title = msg.Body_title
+	}
+	if msg.Body_link != "" {
+		attachement.TitleLink = msg.Body_link
+	}
+	var fields []SlackField
+	for _, value := range msg.Fields {
+		var field SlackField
+		field.Title = value.Header
+		field.Text = value.Text
+		field.Short = value.Short
+		fields = append(fields, field)
+	}
+	attachement.Fields = fields
+	attachement.Footer = frontend + " (sendmsg)"
+	if frontendIcon != "" {
+		attachement.FooterIcon = frontendIcon
+	}
+	body.Attachments = []SlackAttachments{attachement}
+
+	jsonBody, err := json.Marshal(body)
+	if err != nil {
+		fmt.Errorf("Coudln't masrshal message for JSON transport: ", err)
+		return
+	}
+	resp, err := http.Post(cfg.Webhook, "application/reader", bytes.NewBuffer(jsonBody))
+	if err != nil {
+		fmt.Errorf("Coudln't POST message to webhook: ", err)
+		return
+	}
+	defer resp.Body.Close()
+
+	bodyBytes, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		fmt.Errorf("Couldn't read body", err)
+	}
+	bodyString := string(bodyBytes)
+	fmt.Print(bodyString)
+
+	if bodyString != "ok" {
+		fmt.Errorf("Slack didn't like what we sent, error: ", bodyString)
+	}
+}

slack.go

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