import prisma from "indite-js/lib/prisma";
import { Prisma } from "indite-js/prisma";
import { SessionState } from "indite-js/schemas";

type Props = {
  id?: string;
  state: SessionState;
  isReplying?: boolean;
};

export const createSession = ({
  id,
  state,
  isReplying,
}: Props): Prisma.PrismaPromise<any> => {
  if (!id) {
    return prisma.chatSession.create({
      data: {
        id,
        state,
        isReplying,
      },
    });
  }
  return prisma.chatSession.upsert({
    where: { id },
    update: {
      state,
      isReplying,
    },
    create: {
      id,
      state,
      isReplying,
    },
  });
};