add untracked files

Author: LumePart

src/logging/notify.go

@@ -0,0 +1,124 @@
+package logging
+
+import (
+	"context"
+	"explo/src/config"
+	"fmt"
+	"log/slog"
+	"strings"
+
+	"github.com/nikoksr/notify"
+	"github.com/nikoksr/notify/service/discord"
+	"github.com/nikoksr/notify/service/matrix"
+	"github.com/nikoksr/notify/service/http"
+	"maunium.net/go/mautrix/id"
+)
+
+// TODO: reuse notifier instead of creating a new one every time. right now it's fine cause Explo sends 1 message per run
+
+
+type NotificationClient struct {
+	Cfg config.NotifyConfig
+}
+
+func InitNotify(cfg config.NotifyConfig) NotificationClient {
+	return NotificationClient{
+		Cfg: cfg,
+	}
+}
+
+func sendMatrix(cfg config.MatrixNotif, msg string) error {
+	// UserID and RoomID need to be cast as specific types
+	srvc, err := matrix.New(id.UserID(cfg.UserID), id.RoomID(cfg.RoomID), cfg.HomeServer, cfg.AccessToken)
+	if err != nil {
+    return fmt.Errorf("failed to create new Matrix notification service: %s", err.Error())
+  }
+
+  notifier := notify.New()
+  notifier.UseServices(srvc)
+
+  err = notifier.Send(context.Background(), "Explo", msg)
+  if err != nil {
+    return err
+  }
+
+  return nil
+}
+
+/* discordgo module (which notify uses) doesn't handle errors correctly
+ no errors are given even when authentication fails*/
+func sendDiscord(cfg config.DiscordNotif, msg string) error {
+	srvc := discord.New()
+	if err := srvc.AuthenticateWithBotToken(cfg.BotToken); err != nil {
+		return fmt.Errorf("failed to authenticate against Discord: %s", err.Error())
+	}
+
+	srvc.AddReceivers(cfg.ChannelIDs...)
+
+	notifier := notify.New()
+	notifier.UseServices(srvc)
+
+	if err := notifier.Send(context.Background(), "**Explo**", msg); err != nil {
+		return fmt.Errorf("failed to send Discord notification: %s", err.Error())
+	}
+
+	return nil	
+}
+
+func sendHttp(cfg config.HttpNotif, msg string) error {
+	httpNotify := http.New()
+
+	httpNotify.AddReceiversURLs(cfg.ReceiverURL)
+	httpNotify.AddReceivers(&http.Webhook{
+		URL: cfg.ReceiverURL,
+	})
+	notifier := notify.NewWithServices(httpNotify)
+
+	if err := notifier.Send(context.Background(), "Explo", msg); err != nil {
+		return fmt.Errorf("failed to send HTTP notification: %s", err.Error())
+	}
+	return nil
+}
+
+// format notification to JSON
+func formatNotification(r slog.Record) string {
+	var attrs []string
+
+	r.Attrs(func(a slog.Attr) bool {
+		attrs = append(attrs, fmt.Sprintf("%s=%v", a.Key, a.Value))
+		return true
+	})
+
+
+	return fmt.Sprintf(
+		"[%s] %s\n%s",
+		r.Level,
+		r.Message,
+		strings.Join(attrs, ", "),
+	)
+}
+
+func (c NotificationClient) SendNotification(r slog.Record) {
+	var err error
+	msg := formatNotification(r)
+	switch c.Cfg.Service {
+		case "matrix":
+			err = sendMatrix(c.Cfg.Matrix, msg)
+		
+		case "discord":
+			err = sendDiscord(c.Cfg.Discord, msg)
+
+		case "http":
+			err = sendHttp(c.Cfg.Http, msg)
+		
+		case "": // no system defined
+			return
+		default:
+			err = fmt.Errorf("wrong system defined for notifications: %s", c.Cfg.Service)
+	}
+	if err != nil {
+		slog.Error(err.Error())
+	} else {
+		slog.Info("notification sent", "service", c.Cfg.Service)
+	}
+}

src/logging/notify_handler.go

@@ -0,0 +1,53 @@
+package logging
+
+import (
+	"log/slog"
+	"context"
+)
+// slog handler that checks whether to send notifications
+
+type notifyHandler struct {
+	handler   slog.Handler
+	notify NotificationClient
+}
+
+func (h *notifyHandler) Enabled(ctx context.Context, level slog.Level) bool {
+	return h.handler.Enabled(ctx, level)
+}
+
+func (h *notifyHandler) Handle(ctx context.Context, r slog.Record) error {
+	if shouldNotify(r) {
+		// send notification in another goroutine
+		rec := r
+		go h.notify.SendNotification(rec) 
+	}
+	return h.handler.Handle(ctx, r)
+}
+
+func (h *notifyHandler) WithAttrs(attrs []slog.Attr) slog.Handler {
+	return &notifyHandler{
+		handler:   h.handler.WithAttrs(attrs),
+		notify: h.notify,
+	}
+}
+
+func (h *notifyHandler) WithGroup(name string) slog.Handler {
+	return &notifyHandler{
+		handler:   h.handler.WithGroup(name),
+		notify: h.notify,
+	}
+}
+
+func shouldNotify(r slog.Record) bool {
+	notify := false
+
+	r.Attrs(func(a slog.Attr) bool {
+		if a.Key == "notify" && a.Value.Kind() == slog.KindBool && a.Value.Bool() {
+			notify = true
+			return false // stop scanning
+		}
+		return true
+	})
+
+	return notify
+}