import { PrismaClient, UserRole, UserStatus, MaterialType } from '@prisma/client';

const prisma = new PrismaClient();

async function upsertUsers() {
  const usernames = ['SeedUser_Alpha', 'SeedUser_Beta', 'SeedUser_Gamma', 'SeedUser_Delta'];
  const authors = [] as Awaited<ReturnType<typeof prisma.user.upsert>>[];
  for (const username of usernames) {
    const user = await prisma.user.upsert({
      where: { username },
      update: {},
      create: {
        username,
        password: null,
        role: UserRole.CREATOR,
        status: UserStatus.ACTIVE,
        avatarUrl: `https://api.dicebear.com/7.x/avataaars/svg?seed=${encodeURIComponent(username)}`,
      },
    });
    authors.push(user);
  }
  return authors;
}

async function upsertTags() {
  const names = ['code', 'seed', 'demo'];
  const tags = [] as Awaited<ReturnType<typeof prisma.tag.upsert>>[];
  for (const name of names) {
    const tag = await prisma.tag.upsert({
      where: { name },
      update: {},
      create: { name },
    });
    tags.push(tag);
  }
  return tags;
}

async function seedMaterials(authors: Awaited<ReturnType<typeof upsertUsers>>, tags: Awaited<ReturnType<typeof upsertTags>>) {
  const total = 40;
  for (let i = 1; i <= total; i++) {
    const title = `Seed Code #${String(i).padStart(2, '0')}`;
    const existing = await prisma.material.findFirst({ where: { title, type: MaterialType.CODE } });
    if (existing) continue;

    const author = authors[i % authors.length];
    const snippet = `export const seed${i} = () => ${i};`;
    await prisma.material.create({
      data: {
        title,
        description: 'Seeded code snippet for pagination demo.',
        type: MaterialType.CODE,
        codeSnippet: snippet,
        language: 'ts',
        authorId: author.id,
        tags: {
          connect: tags.map(t => ({ id: t.id })),
        },
      },
    });
  }
}

async function main() {
  console.log('[SEED] Starting...');
  const authors = await upsertUsers();
  const tags = await upsertTags();
  await seedMaterials(authors, tags);
  const count = await prisma.material.count({ where: { type: MaterialType.CODE } });
  console.log(`[SEED] Done. CODE materials count: ${count}`);
}

main()
  .catch((e) => {
    console.error(e);
    process.exit(1);
  })
  .finally(async () => {
    await prisma.$disconnect();
  });