import { prisma } from '@repo/db';
import { StorageAdapter } from 'grammy';

export class PrismaStorageAdapter<T extends object>
  implements StorageAdapter<T>
{
  async read(key: string): Promise<T | undefined> {
    return prisma.telegramSession
      .findUnique({
        where: {
          id: key,
        },
      })
      .then((session) => (session?.data as T) || undefined);
  }
  async write(key: string, data: T): Promise<void> {
    await prisma.telegramSession.upsert({
      where: {
        id: key,
      },
      create: {
        id: key,
        data,
      },
      update: {
        data,
      },
    });
  }
  async delete(key: string): Promise<void> {
    await prisma.telegramSession.delete({ where: { id: key } });
  }
  async has(key: string): Promise<boolean> {
    return prisma.telegramSession.exists({ id: key });
  }
}