import { PrismaClient } from '@prisma/client';
import * as bcrypt from 'bcryptjs';

const prisma = new PrismaClient();

async function main() {
  // 创建管理员
  const adminPassword = await bcrypt.hash('admin123456', 10);
  await prisma.adminUser.upsert({
    where: { username: 'admin' },
    update: {},
    create: {
      username: 'admin',
      passwordHash: adminPassword,
      nickname: '超级管理员',
      role: 'superadmin',
    },
  });

  // 创建分类
  const categories = [
    { name: 'AI 入门', slug: 'ai-basics', description: 'AI 基础知识与入门指南' },
    { name: '办公效率', slug: 'office-productivity', description: '用 AI 提升办公效率' },
    { name: '创意设计', slug: 'creative-design', description: 'AI 辅助创意与设计' },
    { name: '编程开发', slug: 'programming', description: 'AI 辅助编程开发' },
    { name: '教育学习', slug: 'education', description: 'AI 在教育中的应用' },
    { name: '提示词技巧', slug: 'prompt-engineering', description: '提示词工程与技巧' },
    { name: 'AI 工具', slug: 'ai-tools', description: 'AI 工具收录与评测' },
    { name: '模型百科', slug: 'model-encyclopedia', description: '大模型介绍与对比' },
  ];

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

  // 创建示例课程
  const basicsCategory = await prisma.category.findUnique({ where: { slug: 'ai-basics' } });
  const promptCategory = await prisma.category.findUnique({ where: { slug: 'prompt-engineering' } });

  if (basicsCategory) {
    const course = await prisma.course.upsert({
      where: { id: 1 },
      update: {},
      create: {
        title: 'AI 通识课：零基础入门人工智能',
        description: '面向零基础用户的 AI 入门课程，带你了解 AI 的基本概念、发展历程和实际应用。',
        categoryId: basicsCategory.id,
        isFree: true,
        status: 'PUBLISHED',
        sortOrder: 1,
        chapters: {
          create: [
            {
              title: '第一章：什么是人工智能',
              sortOrder: 1,
              lessons: {
                create: [
                  { title: 'AI 的定义与发展简史', content: '# AI 的定义\n\n人工智能...', sortOrder: 1, status: 'PUBLISHED' },
                  { title: '机器学习 vs 深度学习', content: '# 机器学习\n\n...', sortOrder: 2, status: 'PUBLISHED' },
                  { title: '大语言模型（LLM）是什么', content: '# 大语言模型\n\n...', sortOrder: 3, status: 'PUBLISHED' },
                ],
              },
            },
            {
              title: '第二章：主流 AI 工具介绍',
              sortOrder: 2,
              lessons: {
                create: [
                  { title: 'ChatGPT 与 GPT 系列', content: '# ChatGPT\n\n...', sortOrder: 1, status: 'PUBLISHED' },
                  { title: 'Claude 系列模型', content: '# Claude\n\n...', sortOrder: 2, status: 'PUBLISHED' },
                  { title: '国内大模型：通义千问、文心一言、GLM', content: '# 国内大模型\n\n...', sortOrder: 3, status: 'PUBLISHED' },
                ],
              },
            },
          ],
        },
      },
    });
  }

  // 创建示例提示词
  if (promptCategory) {
    const prompts = [
      { title: '写周报助手', content: '请帮我写一份本周工作总结，我的工作是：[描述你的工作内容]。\n\n要求：\n1. 按重点项目分类\n2. 包含数据成果\n3. 列出下周计划\n4. 语言简洁专业', description: '帮你快速生成高质量周报', categoryId: promptCategory.id, tags: '办公,周报', model: '通用', status: 'PUBLISHED' as const },
      { title: '会议纪要生成', content: '请根据以下会议内容生成会议纪要：\n[粘贴会议内容]\n\n格式要求：\n- 时间\n- 参会人\n- 会议议题\n- 讨论要点\n- 待办事项及负责人', description: '快速整理会议内容为结构化纪要', categoryId: promptCategory.id, tags: '办公,会议', model: '通用', status: 'PUBLISHED' as const },
    ];

    for (const prompt of prompts) {
      await prisma.prompt.create({ data: prompt });
    }
  }

  // 创建示例 AI 工具
  const toolsCategory = await prisma.category.findUnique({ where: { slug: 'ai-tools' } });
  if (toolsCategory) {
    const tools = [
      { name: 'ChatGPT', description: 'OpenAI 开发的对话式 AI 助手，支持文本生成、代码编写、分析等', url: 'https://chat.openai.com', categoryId: toolsCategory.id, tags: '对话,文本生成,代码', isFeatured: true, status: 'PUBLISHED' as const },
      { name: 'Claude', description: 'Anthropic 开发的 AI 助手，擅长深度分析、长文本处理', url: 'https://claude.ai', categoryId: toolsCategory.id, tags: '对话,分析,长文本', isFeatured: true, status: 'PUBLISHED' as const },
      { name: '通义千问', description: '阿里云开发的 AI 大模型，支持文本、图像、代码等多种任务', url: 'https://tongyi.aliyun.com', categoryId: toolsCategory.id, tags: '对话,文本,国内', isFeatured: true, status: 'PUBLISHED' as const },
      { name: '文心一言', description: '百度开发的 AI 对话产品，基于文心大模型', url: 'https://yiyan.baidu.com', categoryId: toolsCategory.id, tags: '对话,文本,国内', isFeatured: true, status: 'PUBLISHED' as const },
    ];

    for (const tool of tools) {
      await prisma.tool.create({ data: tool });
    }
  }

  console.log('Seed data created successfully!');
}

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