DentalManagementMH06/apps/Backend/src/ai/aiHandoffStore.ts

import { prisma as db } from "@repo/db/client";

export type ConversationStage =
  | "initial"
  | "reminder_initial"
  | "greeted"
  | "done"
  | "new_patient_greeted"
  | "asked_new_or_existing"
  | "asked_new_patient_insurance"
  | "asked_existing_insurance"
  | "asked_appointment_time"
  | "awaiting_masshealth_info"
  | "asked_appointment_preference"
  | "asked_self_pay"
  | "asked_reschedule_confirm"
  | "asked_reschedule_preference"
  | "asked_reschedule_asap"
  | "asked_reschedule_next_week"
  | "asked_reschedule_time";

// ── Conversation stage + AI handoff per patient (DB-persisted) ────────────────

export async function getStage(userId: number, patientId: number): Promise<ConversationStage> {
  const row = await db.patientConversation.findUnique({ where: { patientId } });
  return (row?.stage as ConversationStage) ?? "initial";
}

export async function setStage(userId: number, patientId: number, stage: ConversationStage): Promise<void> {
  await db.patientConversation.upsert({
    where:  { patientId },
    update: { stage },
    create: { patientId, userId, stage, aiHandoff: true },
  });
}

export async function getHandoff(userId: number, patientId: number): Promise<boolean> {
  const row = await db.patientConversation.findUnique({ where: { patientId } });
  return row?.aiHandoff ?? true;
}

export async function setHandoff(userId: number, patientId: number, enabled: boolean): Promise<void> {
  await db.patientConversation.upsert({
    where:  { patientId },
    update: { aiHandoff: enabled },
    create: { patientId, userId, aiHandoff: enabled, stage: "initial" },
  });
}

// ── After-hours handoff per user (persisted in ai_settings) ──────────────────

export async function getAfterHoursHandoff(userId: number): Promise<boolean> {
  const row = await db.aiSettings.findUnique({ where: { userId } });
  return row?.afterHoursEnabled ?? true;
}

export async function setAfterHoursHandoff(userId: number, enabled: boolean): Promise<void> {
  await db.aiSettings.update({ where: { userId }, data: { afterHoursEnabled: enabled } });
}

// ── Conversation starters ─────────────────────────────────────────────────────

export async function resetConversation(userId: number, patientId: number): Promise<void> {
  await setStage(userId, patientId, "reminder_initial");
}

export async function startNewPatientConversation(userId: number, patientId: number): Promise<void> {
  await setStage(userId, patientId, "new_patient_greeted");
}

export async function startRescheduleConversation(userId: number, patientId: number): Promise<void> {
  await setStage(userId, patientId, "asked_reschedule_confirm");
}

// ── Pending reschedule (in-memory — seconds-lived within a single exchange) ───

interface PendingReschedule {
  newDate:  Date;
  dayLabel: string;
}

const pendingRescheduleStore = new Map<string, PendingReschedule>();

function convKey(userId: number, patientId: number): string {
  return `${userId}:${patientId}`;
}

export function setPendingReschedule(userId: number, patientId: number, data: PendingReschedule): void {
  pendingRescheduleStore.set(convKey(userId, patientId), data);
}

export function getPendingReschedule(userId: number, patientId: number): PendingReschedule | undefined {
  return pendingRescheduleStore.get(convKey(userId, patientId));
}

export function clearPendingReschedule(userId: number, patientId: number): void {
  pendingRescheduleStore.delete(convKey(userId, patientId));
}