import { PrismaClient, UserRole, CardTypeCategory } from '@prisma/client'

const prisma = new PrismaClient()

async function main() {
  console.log('🌱 Seeding database...')

  // 1. Studio Config
  const existingConfig = await prisma.studioConfig.findFirst()
  if (!existingConfig) {
    await prisma.studioConfig.create({
      data: {
        name: '普拉提工作室',
        address: '请在管理后台设置地址',
        phone: '请在管理后台设置电话',
        cancelHoursLimit: 2,
        photos: [],
      },
    })
    console.log('  ✅ Studio config created')
  } else {
    console.log('  ⏭️  Studio config already exists')
  }

  // 2. Card Types
  const existingCards = await prisma.cardType.count()
  if (existingCards === 0) {
    await prisma.cardType.createMany({
      data: [
        {
          name: '体验课',
          type: CardTypeCategory.TRIAL,
          totalTimes: 1,
          durationDays: 7,
          price: 9900,
          originalPrice: 29900,
          description: '首次体验，感受普拉提的魅力',
          isActive: true,
          sortOrder: 0,
        },
        {
          name: '私教10次卡',
          type: CardTypeCategory.TIMES,
          totalTimes: 10,
          durationDays: 90,
          price: 299900,
          originalPrice: 399900,
          description: '10节私教课，90天有效期',
          isActive: true,
          sortOrder: 1,
        },
        {
          name: '私教20次卡',
          type: CardTypeCategory.TIMES,
          totalTimes: 20,
          durationDays: 180,
          price: 499900,
          originalPrice: 699900,
          description: '20节私教课，180天有效期，更优惠',
          isActive: true,
          sortOrder: 2,
        },
        {
          name: '月卡',
          type: CardTypeCategory.DURATION,
          totalTimes: null,
          durationDays: 30,
          price: 99900,
          description: '30天不限次数',
          isActive: true,
          sortOrder: 3,
        },
        {
          name: '季卡',
          type: CardTypeCategory.DURATION,
          totalTimes: null,
          durationDays: 90,
          price: 249900,
          originalPrice: 299700,
          description: '90天不限次数，更实惠',
          isActive: true,
          sortOrder: 4,
        },
      ],
    })
    console.log('  ✅ Card types created')
  } else {
    console.log('  ⏭️  Card types already exist')
  }

  // 3. Admin User
  const existingAdmin = await prisma.user.findFirst({
    where: { role: UserRole.ADMIN },
  })
  if (!existingAdmin) {
    await prisma.user.create({
      data: {
        openid: 'admin_test_openid',
        nickname: '教练',
        role: UserRole.ADMIN,
        adminBookingSubscriptionCount: 0,
      },
    })
    console.log('  ✅ Admin user created')
  } else {
    console.log('  ⏭️  Admin user already exists')
  }

  // 4. Week Templates (weekdays 9-16, hourly slots)
  const existingTemplates = await prisma.weekTemplate.count()
  if (existingTemplates === 0) {
    const timeSlots = [
      { startTime: '09:00', endTime: '10:00' },
      { startTime: '10:00', endTime: '11:00' },
      { startTime: '14:00', endTime: '15:00' },
      { startTime: '15:00', endTime: '16:00' },
    ]

    const templates = []
    // Monday to Friday (1-5)
    for (let day = 1; day <= 5; day++) {
      for (const slot of timeSlots) {
        templates.push({
          dayOfWeek: day,
          startTime: slot.startTime,
          endTime: slot.endTime,
          capacity: 1,
          isActive: true,
        })
      }
    }

    // Saturday (6) — morning only
    templates.push(
      { dayOfWeek: 6, startTime: '09:00', endTime: '10:00', capacity: 1, isActive: true },
      { dayOfWeek: 6, startTime: '10:00', endTime: '11:00', capacity: 1, isActive: true },
    )

    await prisma.weekTemplate.createMany({ data: templates })
    console.log('  ✅ Week templates created')
  } else {
    console.log('  ⏭️  Week templates already exist')
  }

  console.log('🌱 Seeding complete!')
}

main()
  .catch((e) => {
    console.error('❌ Seed failed:', e)
    process.exit(1)
  })
  .finally(async () => {
    await prisma.$disconnect()
  })