snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/system/automation/notification_handler.go

@@ -0,0 +1,104 @@
+package automation
+
+import (
+	"context"
+	"fmt"
+
+	composeTypes "github.com/cortezaproject/corteza/server/compose/types"
+	systemTypes "github.com/cortezaproject/corteza/server/system/types"
+	"go.uber.org/zap"
+)
+
+type (
+	notificationHandler struct {
+		reg    notificationHandlerRegistry
+		svc    notificationService
+		uSvc   notificationUserService
+		logger *zap.Logger
+	}
+
+	notificationService interface {
+		Create(context.Context, *systemTypes.Notification) (*systemTypes.Notification, error)
+	}
+
+	// Interface for user service that we need for recipient lookups
+	notificationUserService interface {
+		FindByID(ctx context.Context, userID uint64) (*systemTypes.User, error)
+		FindByHandle(ctx context.Context, handle string) (*systemTypes.User, error)
+		FindByEmail(ctx context.Context, email string) (*systemTypes.User, error)
+	}
+
+	notificationNamespaceService interface {
+		FindByHandle(ctx context.Context, handle string) (*composeTypes.Namespace, error)
+	}
+
+	notificationModuleService interface {
+		FindByHandle(ctx context.Context, handle string) (*composeTypes.Module, error)
+	}
+)
+
+// NotificationHandler initializes the notification handler
+func NotificationHandler(reg notificationHandlerRegistry, ntfSvc notificationService, uSvc notificationUserService, log *zap.Logger) *notificationHandler {
+	h := &notificationHandler{
+		reg:    reg,
+		svc:    ntfSvc,
+		uSvc:   uSvc,
+		logger: log.Named("notification"),
+	}
+
+	h.register()
+	return h
+}
+
+// send creates and sends a notification
+func (h notificationHandler) send(ctx context.Context, args *notificationSendArgs) error {
+	// Get the recipient user ID from the input parameter
+	var recipientID uint64
+
+	// Check if we have a direct user object
+	if args.recipientID > 0 {
+		// Direct ID passed
+		recipientID = args.recipientID
+	} else {
+		// Need to look up the user
+		var user *systemTypes.User
+		var err error
+
+		if args.recipientHandle != "" {
+			user, err = h.uSvc.FindByHandle(ctx, args.recipientHandle)
+		} else if args.recipientEmail != "" {
+			user, err = h.uSvc.FindByEmail(ctx, args.recipientEmail)
+		} else {
+			return fmt.Errorf("invalid recipient: unable to determine user ID")
+		}
+
+		if err != nil {
+			return err
+		}
+
+		if user == nil {
+			return fmt.Errorf("recipient not found")
+		}
+
+		recipientID = user.ID
+	}
+
+	// Create notification config directly with the expected fields for a simple notification
+	config := systemTypes.NotificationConfig{
+		Simple: systemTypes.SimpleNotificationConfig{
+			Title:       args.Title,
+			Description: args.Description,
+		},
+	}
+
+	// Create a simple notification
+	ntf := &systemTypes.Notification{
+		Kind:      systemTypes.NotificationKindSimple,
+		Config:    config,
+		Recipient: recipientID,
+	}
+
+	// Create the notification
+	_, err := h.svc.Create(ctx, ntf)
+	return err
+}