Add Go command implementations

internal/icinga2/messages.go

@@ -0,0 +1,149 @@
+// Copyright Jan Dittberner
+// SPDX-License-Identifier: GPL-3.0-or-later
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+package icinga2
+
+import (
+	"fmt"
+	"strings"
+)
+
+const (
+	WarnIcon     = "⚠"
+	ErrorIcon    = "❌"
+	OKIcon       = "🆗"
+	QuestionIcon = "❓"
+)
+
+func getIcon(state string) string {
+	switch state {
+	case "UP", "OK":
+		return OKIcon
+	case "UNKNOWN":
+		return QuestionIcon
+	case "WARNING":
+		return WarnIcon
+	default:
+		return ErrorIcon
+	}
+}
+
+type CouldNotWriteToBufferErr struct {
+	Wrapped error
+}
+
+func (e CouldNotWriteToBufferErr) Error() string {
+	return fmt.Sprintf("could not write to message builder: %s", e.Wrapped)
+}
+
+func (e CouldNotWriteToBufferErr) Unwrap() error {
+	return e.Wrapped
+}
+
+func addOptionalData(message *strings.Builder, p *BaseParameters) error {
+	if p.HostAddress != "" {
+		if _, err := fmt.Fprintf(message, "<strong>IPv4:</strong> %s<br/>\n", p.HostAddress); err != nil {
+			return CouldNotWriteToBufferErr{err}
+		}
+	}
+
+	if p.HostAddress6 != "" {
+		if _, err := fmt.Fprintf(message, "<strong>IPv6:</strong> %s<br/>\n", p.HostAddress6); err != nil {
+			return CouldNotWriteToBufferErr{err}
+		}
+	}
+
+	if p.NotificationAuthorName != "" && p.NotificationComment != "" {
+		if _, err := fmt.Fprintf(
+			message,
+			"Comment by <strong>%s:</strong> %s<br/>\n",
+			p.NotificationAuthorName,
+			p.NotificationComment,
+		); err != nil {
+			return CouldNotWriteToBufferErr{err}
+		}
+	}
+
+	return nil
+}
+
+func BuildHostNotification(p *HostParameters) (string, error) {
+	message := &strings.Builder{}
+
+	if _, err := fmt.Fprintf(
+		message,
+		"%s <strong>HOST:</strong> %s is <strong>%s!</strong><br/>\n"+
+			"<strong>When:</strong> %s<br/>\n"+
+			"<strong>Info:</strong> %s<br/>\n",
+		getIcon(p.HostState),
+		p.HostDisplayName, p.HostState,
+		p.LongDateTime, p.HostOutput); err != nil {
+		return "", CouldNotWriteToBufferErr{err}
+	}
+
+	if err := addOptionalData(message, &p.BaseParameters); err != nil {
+		return "", err
+	}
+
+	if p.IcingaWeb2URL.URL != nil {
+		icinga2Url := p.IcingaWeb2URL.URL.JoinPath("monitoring/host/show")
+
+		q := icinga2Url.Query()
+		q.Set("host", p.Hostname)
+
+		icinga2Url.RawQuery = q.Encode()
+
+		if _, err := message.WriteString(icinga2Url.String()); err != nil {
+			return "", CouldNotWriteToBufferErr{err}
+		}
+	}
+
+	return message.String(), nil
+}
+
+func BuildServiceNotification(p *ServiceParameters) (string, error) {
+	message := &strings.Builder{}
+
+	if _, err := fmt.Fprintf(
+		message,
+		"%s <strong>Service:</strong> %s on %s is <strong>%s</strong>.<br/>\n"+
+			"<strong>When:</strong> %s<br/>\n"+
+			"<strong>Info:</strong> %s<br/>\n",
+		getIcon(p.ServiceState), p.ServiceDisplayName, p.HostDisplayName,
+		p.ServiceState, p.LongDateTime, p.ServiceOutput,
+	); err != nil {
+		return "", CouldNotWriteToBufferErr{err}
+	}
+
+	if err := addOptionalData(message, &p.BaseParameters); err != nil {
+		return "", err
+	}
+
+	if p.IcingaWeb2URL.URL != nil {
+		icinga2Url := p.IcingaWeb2URL.URL.JoinPath("monitoring/service/show")
+
+		q := icinga2Url.Query()
+		q.Set("host", p.Hostname)
+		q.Set("service", p.ServiceName)
+
+		icinga2Url.RawQuery = q.Encode()
+
+		if _, err := message.WriteString(icinga2Url.String()); err != nil {
+			return "", CouldNotWriteToBufferErr{err}
+		}
+	}
+
+	return message.String(), nil
+}

internal/icinga2/parameters.go

@@ -0,0 +1,104 @@
+// Copyright Jan Dittberner
+// SPDX-License-Identifier: GPL-3.0-or-later
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+package icinga2
+
+import (
+	"errors"
+	"fmt"
+	"net/url"
+)
+
+type URLValue struct {
+	URL *url.URL
+}
+
+func (u *URLValue) Set(value string) error {
+	var err error
+
+	u.URL, err = url.Parse(value)
+	if err != nil {
+		return fmt.Errorf("could not parse URL: %w", err)
+	}
+
+	return nil
+}
+
+func (u *URLValue) String() string {
+	if u.URL != nil {
+		return u.URL.String()
+	}
+
+	return ""
+}
+
+type MatrixParameters struct {
+	MatrixRoom   string
+	MatrixServer URLValue
+	MatrixToken  string
+}
+
+func (m *MatrixParameters) HasRequired() bool {
+	return m.MatrixRoom != "" && m.MatrixServer.URL != nil && m.MatrixToken != ""
+}
+
+type BaseParameters struct {
+	NotificationAuthorName string
+	NotificationComment    string
+	IcingaWeb2URL          URLValue
+	LongDateTime           string
+	Hostname               string
+	HostDisplayName        string
+	HostAddress            string
+	HostAddress6           string
+}
+
+func (p BaseParameters) HasRequired() bool {
+	return p.LongDateTime != "" && p.Hostname != "" && p.HostDisplayName != ""
+}
+
+type HostParameters struct {
+	HostOutput string
+	HostState  string
+	BaseParameters
+	MatrixParameters
+}
+
+func (p *HostParameters) ValidateRequired() error {
+	if !p.BaseParameters.HasRequired() || !p.MatrixParameters.HasRequired() ||
+		p.HostOutput == "" || p.HostState == "" {
+		return errors.New("missing required fields")
+	}
+
+	return nil
+}
+
+type ServiceParameters struct {
+	ServiceName        string
+	ServiceDisplayName string
+	ServiceOutput      string
+	ServiceState       string
+	BaseParameters
+	MatrixParameters
+}
+
+func (p *ServiceParameters) ValidateRequired() error {
+	if !p.BaseParameters.HasRequired() || !p.MatrixParameters.HasRequired() ||
+		p.ServiceName == "" || p.ServiceDisplayName == "" || p.ServiceState == "" || p.ServiceOutput == "" {
+		return errors.New("missing required fields")
+	}
+
+	return nil
+}