portabase/src/features/notifications/helpers.ts

import { DatabaseWith } from "@/db/schema/07_database";
import { EventKind, EventPayload } from "@/features/notifications/types";
import { dispatchNotification } from "@/features/notifications/dispatch";
import {db} from "@/db";
import {eq} from "drizzle-orm";
import * as drizzleDb from "@/db";

export async function sendNotificationsBackupRestore(database: DatabaseWith, event: EventKind) {

    const settings = await db.query.setting.findFirst({
        where: eq(drizzleDb.schemas.setting.name, "system"),
        with: { notificationChannel: true },
    });

    const defaultPolicy = settings?.notificationChannel
        ? [{
            id: null,
            notificationChannelId: settings.notificationChannel.id,
            enabled: settings.notificationChannel.enabled,
            eventKinds: ["error_backup" , "error_restore"]
        }]
        : [];

    const policiesToUse = (database.alertPolicies && database.alertPolicies.length > 0)
        ? database.alertPolicies.filter(policy => policy.enabled && policy.eventKinds.includes(event))
        : defaultPolicy;

    if (!policiesToUse || policiesToUse.length === 0) {
        return [];
    }

    const promises = policiesToUse.map(alertPolicy => {
        const date = new Date();
        let level: "info" | "critical" = "info";
        let message = "";
        let error: string | null = null;

        switch (event) {
            case "error_backup":
            case "error_restore":
                level = "critical";
                message = `An error occurred during ${event.includes("backup") ? "backup" : "restore"} on ${date.toISOString()}.`;
                error = "Check database connection or agent";
                break;
            case "success_backup":
            case "success_restore":
                level = "info";
                message = `${event.includes("backup") ? "Backup" : "Restore"} completed successfully at ${date.toISOString()}.`;
                break;
            case "weekly_report":
                level = "info";
                message = `Weekly report generated at ${date.toISOString()}.`;
                break;
        }

        const titleMap: Record<EventKind, string> = {
            error_backup: `Backup Notification`,
            error_restore: `Restore Notification`,
            success_backup: `Backup Notification`,
            success_restore: `Restore Notification`,
            weekly_report: `Weekly Report Notification`,
            error_health_agent: "Health Agent Notification",
        };

        const payload: EventPayload = {
            title: titleMap[event],
            message,
            level: level,
            event: event,
            data: {
                host: database.name,
                id: database.id,
                agentDatabaseId: database.agentDatabaseId,
                error,
            },
        };

        return dispatchNotification(payload, alertPolicy.id == null ? undefined : alertPolicy.id, alertPolicy.id ? undefined : alertPolicy.notificationChannelId, undefined);
    });


    return Promise.all(promises);
}