clientsflow/frontend/server/utils/prisma.ts

import { PrismaClient, type ClientTimelineContentType } from "../generated/prisma/client";
import { PrismaPg } from "@prisma/adapter-pg";

declare global {
  // eslint-disable-next-line no-var
  var __prisma: PrismaClient | undefined;
}

// ---------------------------------------------------------------------------
// Auto-sync ClientTimelineEntry for CalendarEvent and FeedCard
// ---------------------------------------------------------------------------
const TIMELINE_MODEL_MAP: Record<string, ClientTimelineContentType> = {
  calendarEvent: "CALENDAR_EVENT",
  feedCard: "RECOMMENDATION",
};

function timelineHook(model: string) {
  const contentType = TIMELINE_MODEL_MAP[model];
  if (!contentType) return undefined;

  const afterMutate = (result: any) => {
    if (!result || typeof result !== "object") return;
    const row = result as Record<string, any>;
    const teamId = row.teamId as string | undefined;
    const contactId = row.contactId as string | undefined;
    const id = row.id as string | undefined;
    if (!teamId || !contactId || !id) return;

    const datetime = row.startsAt ?? row.happenedAt ?? row.createdAt ?? new Date();

    basePrisma.clientTimelineEntry
      .upsert({
        where: {
          teamId_contentType_contentId: { teamId, contentType, contentId: id },
        },
        create: { teamId, contactId, contentType, contentId: id, datetime },
        update: { contactId, datetime },
        select: { id: true },
      })
      .catch(() => {});
  };

  return {
    async create({ args, query }: any) {
      const result = await query(args);
      afterMutate(result);
      return result;
    },
    async update({ args, query }: any) {
      const result = await query(args);
      afterMutate(result);
      return result;
    },
    async upsert({ args, query }: any) {
      const result = await query(args);
      afterMutate(result);
      return result;
    },
  };
}

const adapter = new PrismaPg({ connectionString: process.env.DATABASE_URL! });

const basePrisma =
  globalThis.__prisma ??
  new PrismaClient({
    adapter,
    log: ["error", "warn"],
  });

export const prisma = basePrisma.$extends({
  query: {
    calendarEvent: timelineHook("calendarEvent")!,
    feedCard: timelineHook("feedCard")!,
  },
}) as unknown as PrismaClient;

if (process.env.NODE_ENV !== "production") {
  globalThis.__prisma = basePrisma;
}