initial gotify support

internal/notif/gotify.go

@@ -0,0 +1,111 @@
+package notif
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/url"
+
+	"github.com/gotify/server/v2/model"
+	"github.com/sirupsen/logrus"
+	E "github.com/yusing/go-proxy/internal/error"
+	U "github.com/yusing/go-proxy/internal/utils"
+)
+
+type (
+	GotifyClient struct {
+		GotifyConfig
+
+		url  *url.URL
+		http http.Client
+	}
+	GotifyConfig struct {
+		URL   string `json:"url" yaml:"url"`
+		Token string `json:"token" yaml:"token"`
+	}
+	GotifyMessage model.Message
+)
+
+const gotifyMsgEndpoint = "/message"
+
+func newGotifyClient(cfg map[string]any) (Provider, E.Error) {
+	client := new(GotifyClient)
+	err := U.Deserialize(cfg, &client.GotifyConfig)
+	if err != nil {
+		return nil, err
+	}
+
+	url, uErr := url.Parse(client.URL)
+	if uErr != nil {
+		return nil, E.FailWith("parse url", uErr)
+	}
+
+	client.url = url
+	return client, err
+}
+
+// Name implements NotifProvider.
+func (client *GotifyClient) Name() string {
+	return "gotify"
+}
+
+// Send implements NotifProvider.
+func (client *GotifyClient) Send(ctx context.Context, entry *logrus.Entry) error {
+	var priority int
+	var title string
+
+	switch entry.Level {
+	case logrus.WarnLevel:
+		priority = 2
+		title = "Warning"
+	case logrus.ErrorLevel:
+		priority = 5
+		title = "Error"
+	case logrus.FatalLevel, logrus.PanicLevel:
+		priority = 8
+		title = "Critical"
+	default:
+		return nil
+	}
+	if subjects := FieldsAsTitle(entry); subjects != "" {
+		title = subjects + " " + title
+	}
+
+	msg := &GotifyMessage{
+		Title:    title,
+		Message:  entry.Message,
+		Priority: priority,
+	}
+
+	data, err := json.Marshal(msg)
+	if err != nil {
+		return err
+	}
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodPost, client.url.String()+gotifyMsgEndpoint, bytes.NewReader(data))
+	if err != nil {
+		return fmt.Errorf("error creating request: %w", err)
+	}
+
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("Authorization", "Bearer "+client.Token)
+
+	resp, err := client.http.Do(req)
+	if err != nil {
+		return fmt.Errorf("failed to send gotify message: %w", err)
+	}
+
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		var errm model.Error
+		err = json.NewDecoder(resp.Body).Decode(&errm)
+		if err != nil {
+			return fmt.Errorf("gotify status %d, but failed to decode err response: %w", resp.StatusCode, err)
+		}
+		return fmt.Errorf("gotify status %d %s: %s", resp.StatusCode, errm.Error, errm.ErrorDescription)
+	}
+	return nil
+}