const fs = require('fs')
const path = require('path')
const mysql = require('mysql2/promise')
require('dotenv').config({ path: path.join(__dirname, '../.env') })

const categoriesData = require('../data/categories.json')
const goodsData = require('../data/goods.json')
const usersData = require('../data/users.json')

function requireEnv(name, fallback) {
  const value = process.env[name] || fallback
  if (!value && !fallback) {
    throw new Error(`Missing ${name} in .env`)
  }
  return value
}

const config = {
  host: requireEnv('DB_HOST'),
  port: parseInt(requireEnv('DB_PORT', '3306')),
  user: 'root',
  password: requireEnv('DB_ROOT_PASSWORD'),
  database: requireEnv('DB_NAME', 'miniprogram')
}

async function run() {
  let connection = null
  try {
    console.log('Connecting to MySQL...')
    connection = await mysql.createConnection({
      host: config.host,
      port: config.port,
      user: config.user,
      password: config.password
    })

    console.log('Creating database...')
    await connection.query(`CREATE DATABASE IF NOT EXISTS ${config.database} CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci`)
    await connection.query(`USE ${config.database}`)

    console.log('Granting permissions...')
    await connection.query('GRANT ALL PRIVILEGES ON miniprogram.* TO admin@"%";')
    await connection.query('FLUSH PRIVILEGES;')

    console.log('Creating tables...')
    const schema = fs.readFileSync(path.join(__dirname, '../config/schema.sql'), 'utf8')
    const statements = schema.split(';').filter(s => s.trim())
    
    for (const statement of statements) {
      await connection.query(statement)
    }

    console.log('Inserting categories...')
    for (const cat of categoriesData) {
      await connection.query(
        'INSERT INTO categories (id, name, icon, is_show, sort_order) VALUES (?, ?, ?, ?, ?) ON DUPLICATE KEY UPDATE name=VALUES(name), icon=VALUES(icon), is_show=VALUES(is_show), sort_order=VALUES(sort_order)',
        [cat.id, cat.name, cat.icon, cat.is_show, cat.sort_order]
      )
    }

    console.log('Inserting goods...')
    for (const good of goodsData) {
      await connection.query(
        'INSERT INTO goods (id, name, price, original_price, unit, category_id, images, stock, sales, is_hot, is_new, pricing_type, description) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ON DUPLICATE KEY UPDATE name=VALUES(name), price=VALUES(price), original_price=VALUES(original_price)',
        [good.id, good.name, good.price, good.original_price, good.unit, good.category_id, good.images, good.stock, good.sales, good.is_hot, good.is_new, good.pricing_type, good.description]
      )
    }

    console.log('Inserting users...')
    for (const user of usersData) {
      await connection.query(
        'INSERT INTO users (id, phone, password, name, avatar, points) VALUES (?, ?, ?, ?, ?, ?) ON DUPLICATE KEY UPDATE name=VALUES(name), points=VALUES(points)',
        [user.id, user.phone, user.password, user.name, user.avatar, user.points]
      )
    }

    console.log('Database initialization completed successfully!')
    process.exit(0)
  } catch (error) {
    console.error('Database initialization failed:', error)
    process.exit(1)
  } finally {
    if (connection) {
      await connection.end()
    }
  }
}

run()