/**
 * Database Initialization Script
 * Run: npm run init-db
 * Creates the required MySQL database and tables.
 */
require('dotenv').config();
const mysql = require('mysql2/promise');

async function initDatabase() {
  const connection = await mysql.createConnection({
    host: process.env.MYSQL_HOST || 'localhost',
    port: parseInt(process.env.MYSQL_PORT) || 3306,
    user: process.env.MYSQL_USER || 'root',
    password: process.env.MYSQL_PASSWORD || ''
  });

  const dbName = process.env.MYSQL_DATABASE || 'display_wall';

  console.log('🔧 Initializing database...\n');

  // Create database
  await connection.query(`CREATE DATABASE IF NOT EXISTS \`${dbName}\` CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci`);
  console.log(`  ✅ Database "${dbName}" ready`);

  await connection.query(`USE \`${dbName}\``);

  // Create users table
  await connection.query(`
    CREATE TABLE IF NOT EXISTS users (
      id INT AUTO_INCREMENT PRIMARY KEY,
      username VARCHAR(255) NOT NULL UNIQUE,
      password VARCHAR(255) NOT NULL,
      salt VARCHAR(255) NOT NULL,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
  `);
  console.log('  ✅ Table "users" ready');

  // Create prometheus_sources table
  await connection.query(`
      CREATE TABLE IF NOT EXISTS prometheus_sources (
        id INT AUTO_INCREMENT PRIMARY KEY,
        name VARCHAR(255) NOT NULL,
        url VARCHAR(500) NOT NULL,
        description TEXT,
        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
        updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
      ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
  `);
  console.log('  ✅ Table "prometheus_sources" ready');

  // Create site_settings table
  await connection.query(`
    CREATE TABLE IF NOT EXISTS site_settings (
      id INT PRIMARY KEY DEFAULT 1,
      page_name VARCHAR(255) DEFAULT '数据可视化展示大屏',
      title VARCHAR(255) DEFAULT '数据可视化展示大屏',
      logo_url TEXT,
      default_theme VARCHAR(20) DEFAULT 'dark',
      show_page_name TINYINT(1) DEFAULT 1,
      logo_url_dark TEXT,
      favicon_url TEXT,
      show_95_bandwidth TINYINT(1) DEFAULT 0,
      p95_type VARCHAR(20) DEFAULT 'tx',
      require_login_for_server_details TINYINT(1) DEFAULT 1,
      blackbox_source_id INT,
      latency_source VARCHAR(100),
      latency_dest VARCHAR(100),
      latency_target VARCHAR(255),
      icp_filing VARCHAR(255),
      ps_filing VARCHAR(255),
      updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
  `);
  // Insert default settings if not exists
  await connection.query(`
    INSERT IGNORE INTO site_settings (id, page_name, title, default_theme) 
    VALUES (1, '数据可视化展示大屏', '数据可视化展示大屏', 'dark')
  `);
  console.log('  ✅ Table "site_settings" ready');

  // Create server_locations table
  await connection.query(`
    CREATE TABLE IF NOT EXISTS server_locations (
      id INT AUTO_INCREMENT PRIMARY KEY,
      ip VARCHAR(255) NOT NULL UNIQUE,
      country CHAR(2),
      country_name VARCHAR(100),
      region VARCHAR(100),
      city VARCHAR(100),
      latitude DOUBLE,
      longitude DOUBLE,
      last_updated TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
  `);
  console.log('  ✅ Table "server_locations" ready');

  console.log('\n🎉 Database initialization complete!\n');
  await connection.end();
}

initDatabase().catch(err => {
  console.error('❌ Database initialization failed:', err.message);
  process.exit(1);
});