const mysql = require('mysql2/promise');
const fs = require('fs');
const path = require('path');
require('dotenv').config();

async function runInitSQL() {
  const connection = await mysql.createConnection({
    host: process.env.DB_HOST,
    port: process.env.DB_PORT,
    user: process.env.DB_USER,
    password: process.env.DB_PASSWORD,
    database: process.env.DB_DATABASE,
    multipleStatements: true
  });

  try {
    console.log('✅ 数据库连接成功');
    
    const sqlFile = path.join(__dirname, 'init-high-page-data.sql');
    const sql = fs.readFileSync(sqlFile, 'utf8');
    
    console.log('📝 开始执行SQL脚本...');
    await connection.query(sql);
    
    console.log('✅ SQL脚本执行完成!');
    
    // 验证数据
    const [jobs] = await connection.query('SELECT COUNT(*) as count FROM high_salary_jobs');
    const [units] = await connection.query('SELECT COUNT(*) as count FROM training_units');
    const [stories] = await connection.query('SELECT COUNT(*) as count FROM success_stories');
    
    console.log('\n========== 数据验证 ==========');
    console.log(`✅ high_salary_jobs: ${jobs[0].count} 条`);
    console.log(`✅ training_units: ${units[0].count} 条`);
    console.log(`✅ success_stories: ${stories[0].count} 条`);
    console.log('========== 初始化完成 ==========\n');
    
  } catch (error) {
    console.error('❌ 执行失败:', error.message);
    process.exit(1);
  } finally {
    await connection.end();
  }
}

runInitSQL();