watchtower/pkg/notifications/gotify.go

package notifications

import (
	"bytes"
	"crypto/tls"
	"encoding/json"
	"fmt"
	"github.com/spf13/viper"
	"net/http"
	"strings"

	t "github.com/containrrr/watchtower/pkg/types"
	log "github.com/sirupsen/logrus"
	"github.com/spf13/cobra"
)

const (
	gotifyType = "gotify"
)

type gotifyTypeNotifier struct {
	gotifyURL                string
	gotifyAppToken           string
	gotifyInsecureSkipVerify bool
	logLevels                []log.Level
}

func newGotifyNotifier(_ *cobra.Command, acceptedLogLevels []log.Level) t.Notifier {
	flags := viper.Sub(".")

	gotifyURL := flags.GetString("notification-gotify-url")
	if len(gotifyURL) < 1 {
		log.Fatal("Required argument --notification-gotify-url(cli) or WATCHTOWER_NOTIFICATION_GOTIFY_URL(env) is empty.")
	} else if !(strings.HasPrefix(gotifyURL, "http://") || strings.HasPrefix(gotifyURL, "https://")) {
		log.Fatal("Gotify URL must start with \"http://\" or \"https://\"")
	} else if strings.HasPrefix(gotifyURL, "http://") {
		log.Warn("Using an HTTP url for Gotify is insecure")
	}

	gotifyToken := flags.GetString("notification-gotify-token")
	if len(gotifyToken) < 1 {
		log.Fatal("Required argument --notification-gotify-token(cli) or WATCHTOWER_NOTIFICATION_GOTIFY_TOKEN(env) is empty.")
	}

	gotifyInsecureSkipVerify := flags.GetBool("notification-gotify-tls-skip-verify")

	n := &gotifyTypeNotifier{
		gotifyURL:                gotifyURL,
		gotifyAppToken:           gotifyToken,
		gotifyInsecureSkipVerify: gotifyInsecureSkipVerify,
		logLevels:                acceptedLogLevels,
	}

	log.AddHook(n)

	return n
}

func (n *gotifyTypeNotifier) StartNotification() {}

func (n *gotifyTypeNotifier) SendNotification() {}

func (n *gotifyTypeNotifier) Close() {}

func (n *gotifyTypeNotifier) Levels() []log.Level {
	return n.logLevels
}

func (n *gotifyTypeNotifier) getURL() string {
	url := n.gotifyURL
	if !strings.HasSuffix(url, "/") {
		url += "/"
	}
	return url + "message?token=" + n.gotifyAppToken
}

func (n *gotifyTypeNotifier) Fire(entry *log.Entry) error {

	go func() {
		jsonBody, err := json.Marshal(gotifyMessage{
			Message:  "(" + entry.Level.String() + "): " + entry.Message,
			Title:    "Watchtower",
			Priority: 0,
		})
		if err != nil {
			fmt.Println("Failed to create JSON body for Gotify notification: ", err)
			return
		}

		// Explicitly define the client so we can set InsecureSkipVerify to the desired value.
		client := &http.Client{
			Transport: &http.Transport{
				TLSClientConfig: &tls.Config{
					InsecureSkipVerify: n.gotifyInsecureSkipVerify,
				},
			},
		}
		jsonBodyBuffer := bytes.NewBuffer([]byte(jsonBody))
		resp, err := client.Post(n.getURL(), "application/json", jsonBodyBuffer)
		if err != nil {
			fmt.Println("Failed to send Gotify notification: ", err)
			return
		}
		defer resp.Body.Close()

		if resp.StatusCode < 200 || resp.StatusCode >= 300 {
			fmt.Printf("Gotify notification returned %d HTTP status code", resp.StatusCode)
		}

	}()
	return nil
}

type gotifyMessage struct {
	Message  string `json:"message"`
	Title    string `json:"title"`
	Priority int    `json:"priority"`
}