vector-admin/backend/models/notification.js

const prisma = require("../utils/prisma");

const Notification = {
  symbols: {
    info: "info",
    warning: "warning",
    error: "error",
    chroma: "chroma",
    pinecone: "pinecone",
    qdrant: "qdrant",
    weaviate: "weaviate",
  },
  create: async function (organizationId = 0, notificationData = {}) {
    try {
      const {
        textContent = "",
        symbol = null,
        link = null,
        target = null,
        seen = false,
      } = notificationData;
      const notification = await prisma.organization_notifications.create({
        data: {
          organization_id: Number(organizationId),
          textContent,
          symbol,
          link,
          target,
          seen,
        },
      });

      if (!notification) {
        await db.close();
        console.error("FAILED TO CREATE NOTIFICATION.");
        return { notification: null, message: "Could not create notification" };
      }

      return { notification, message: null };
    } catch (e) {
      console.error(e.message);
      return { notification: null, message: e.message };
    }
  },

  update: async function (notificationId = 0, updates = {}) {
    try {
      if (!notificationId)
        throw new Error("No notification id provided for update");
      const notification = await prisma.organization_notifications.update({
        where: { id: Number(notificationId) },
        data: updates,
      });
      return { success: !!notification, error: null };
    } catch (e) {
      console.error(e.message);
      return { success: false, error: e.message };
    }
  },

  get: async function (clause = {}) {
    try {
      const notification = await prisma.organization_notifications.findFirst({
        where: clause,
      });
      return notification ? { ...notification } : null;
    } catch (e) {
      console.error(e.message);
      return null;
    }
  },

  where: async function (
    clause = {},
    limit = null,
    offset = null,
    orderBy = null
  ) {
    try {
      const notifications = await prisma.organization_notifications.findMany({
        where: clause,
        ...(limit !== null ? { take: limit } : {}),
        ...(offset !== null ? { skip: offset } : {}),
        ...(orderBy !== null ? { orderBy } : {}),
      });
      return notifications;
    } catch (e) {
      console.error(e.message, e);
      return [];
    }
  },

  count: async function (clause = {}) {
    try {
      const count = await prisma.organization_notifications.count({
        where: clause,
      });
      return count;
    } catch (e) {
      console.error(e.message);
      return 0;
    }
  },

  delete: async function (clause = {}) {
    try {
      await prisma.organization_notifications.deleteMany({ where: clause });
      return true;
    } catch (e) {
      console.error(e.message);
      return false;
    }
  },

  markSeenForOrg: async function (organizationId = 0) {
    const unseenNotifications = await this.where({
      organization_id: Number(organizationId),
      seen: false,
    });
    const notificationIds = unseenNotifications.map((notif) => notif.id);
    if (notificationIds.length === 0) return;
    await prisma.organization_notifications.updateMany({
      where: {
        id: {
          in: notificationIds,
        },
      },
      data: {
        seen: true,
        lastUpdatedAt: new Date(),
      },
    });
  },
};

module.exports.Notification = Notification;