import { PrismaClient } from "@prisma/client";
import { PrismaMariaDb } from "@prisma/adapter-mariadb";
import "dotenv/config";

const base = process.env.DATABASE_URL.replace("mysql://", "mariadb://");
const connectionString = `${base}&connection_limit=2`;
const adapter = new PrismaMariaDb(connectionString);
const prisma = new PrismaClient({ adapter });

const count = await prisma.botMemory.count();
const groups = await prisma.botMemory.groupBy({
  by: ["memoryType"],
  _count: true,
});
console.log(`Total memories: ${count}`);
console.log(groups.map((r) => `${r.memoryType}: ${r._count}`).join(", "));

const recent = await prisma.botMemory.findMany({
  orderBy: { createdAt: "desc" },
  take: 5,
  include: { bot: { select: { displayName: true } } },
});
for (const r of recent) {
  const c = r.content;
  console.log(
    `  ${r.bot.displayName} | ${r.memoryType} | ${c.title || c.topicTitle || ""}`
  );
}

await prisma.$disconnect();