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

const adapter = new PrismaMariaDb(process.env.DATABASE_URL!);
const prisma = new PrismaClient({ adapter });

const categories = [
  { name: "文本生成", slug: "text-generation", icon: "", sortOrder: 1 },
  { name: "图像生成", slug: "image-generation", icon: "", sortOrder: 2 },
  { name: "视频生成", slug: "video-generation", icon: "", sortOrder: 3 },
  { name: "音频处理", slug: "audio-processing", icon: "", sortOrder: 4 },
  { name: "代码助手", slug: "code-assistant", icon: "", sortOrder: 5 },
  { name: "办公效率", slug: "office-efficiency", icon: "", sortOrder: 6 },
  { name: "AI 角色", slug: "ai-character", icon: "", sortOrder: 7 },
  { name: "设计工具", slug: "design-tools", icon: "", sortOrder: 8 },
  { name: "数据分析", slug: "data-analysis", icon: "", sortOrder: 9 },
  { name: "内容检测", slug: "content-detection", icon: "", sortOrder: 10 },
];

async function main() {
  console.log("Seeding categories...");

  for (const category of categories) {
    await prisma.category.upsert({
      where: { slug: category.slug },
      update: { name: category.name, sortOrder: category.sortOrder },
      create: category,
    });
  }

  console.log(`Seeded ${categories.length} categories.`);
}

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