/**
 * Prisma Seed Script - 初始化测试数据
 * 用于恢复账务系统的测试数据
 */
import { PrismaClient } from '@prisma/client';

const prisma = new PrismaClient();

async function main() {
  console.log('开始恢复测试数据...');

  // 清空现有数据（按依赖顺序）
  await prisma.record.deleteMany();
  await prisma.budget.deleteMany();
  await prisma.account.deleteMany();
  await prisma.user.deleteMany();

  // 1. 创建测试用户
  const user = await prisma.user.create({
    data: {
      name: '测试用户',
      email: 'test@example.com',
    },
  });
  console.log(`创建用户: ${user.name} (ID: ${user.id})`);

  // 2. 创建账户（支付宝、微信、银行卡）
  const accounts = await Promise.all([
    prisma.account.create({
      data: {
        userId: user.id,
        name: '支付宝',
        balance: 5000.00,
        type: 'Alipay',
        color: '#1677FF',
      },
    }),
    prisma.account.create({
      data: {
        userId: user.id,
        name: '微信支付',
        balance: 3000.00,
        type: 'WeChat',
        color: '#07C160',
      },
    }),
    prisma.account.create({
      data: {
        userId: user.id,
        name: '银行卡',
        balance: 10000.00,
        type: 'BankCard',
        color: '#722ED1',
      },
    }),
  ]);
  console.log(`创建 ${accounts.length} 个账户: ${accounts.map(a => a.name).join(', ')}`);

  // 3. 创建交易记录（至少6条）
  const records = await Promise.all([
    // 支付宝 - 支出
    prisma.record.create({
      data: {
        userId: user.id,
        accountId: accounts[0].id, // 支付宝
        type: 'expense',
        amount: 128.50,
        category: '餐饮',
        description: '午餐',
        date: new Date('2026-04-25'),
      },
    }),
    // 微信 - 支出
    prisma.record.create({
      data: {
        userId: user.id,
        accountId: accounts[1].id, // 微信
        type: 'expense',
        amount: 45.00,
        category: '交通',
        description: '地铁',
        date: new Date('2026-04-25'),
      },
    }),
    // 银行卡 - 收入
    prisma.record.create({
      data: {
        userId: user.id,
        accountId: accounts[2].id, // 银行卡
        type: 'income',
        amount: 15000.00,
        category: '工资',
        description: '月薪',
        date: new Date('2026-04-20'),
      },
    }),
    // 支付宝 - 支出
    prisma.record.create({
      data: {
        userId: user.id,
        accountId: accounts[0].id, // 支付宝
        type: 'expense',
        amount: 299.00,
        category: '购物',
        description: '网购商品',
        date: new Date('2026-04-23'),
      },
    }),
    // 微信 - 收入
    prisma.record.create({
      data: {
        userId: user.id,
        accountId: accounts[1].id, // 微信
        type: 'income',
        amount: 500.00,
        category: '转账',
        description: '朋友还款',
        date: new Date('2026-04-22'),
      },
    }),
    // 银行卡 - 支出
    prisma.record.create({
      data: {
        userId: user.id,
        accountId: accounts[2].id, // 银行卡
        type: 'expense',
        amount: 2000.00,
        category: '房租',
        description: '月租',
        date: new Date('2026-04-01'),
      },
    }),
    // 支付宝 - 支出（额外）
    prisma.record.create({
      data: {
        userId: user.id,
        accountId: accounts[0].id,
        type: 'expense',
        amount: 56.80,
        category: '娱乐',
        description: '电影票',
        date: new Date('2026-04-24'),
      },
    }),
  ]);
  console.log(`创建 ${records.length} 条交易记录`);

  // 4. 创建预算
  await prisma.budget.create({
    data: {
      userId: user.id,
      category: '餐饮',
      amount: 2000.00,
      month: '2026-04',
    },
  });
  console.log('创建预算数据');

  console.log('\n✅ 测试数据恢复完成!');
  console.log(`- 用户: 1 个`);
  console.log(`- 账户: ${accounts.length} 个`);
  console.log(`- 交易记录: ${records.length} 条`);
}

main()
  .catch((e) => {
    console.error('❌ 数据恢复失败:', e);
    process.exit(1);
  })
  .finally(async () => {
    await prisma.$disconnect();
  });