const { Pool } = require('pg');

const pool = new Pool({
  connectionString: process.env.DATABASE_URL || 'postgresql://kelly:kelly@localhost:5432/hub'
});

(async () => {
  try {
    console.log('🔄 Running location migration...');

    await pool.query(`
      ALTER TABLE proxies
      ADD COLUMN IF NOT EXISTS city VARCHAR(100),
      ADD COLUMN IF NOT EXISTS state VARCHAR(100),
      ADD COLUMN IF NOT EXISTS country VARCHAR(100),
      ADD COLUMN IF NOT EXISTS country_code VARCHAR(2),
      ADD COLUMN IF NOT EXISTS location_updated_at TIMESTAMP
    `);

    console.log('✅ Added columns to proxies table');

    await pool.query(`
      CREATE INDEX IF NOT EXISTS idx_proxies_location ON proxies(country_code, state, city)
    `);

    console.log('✅ Created location index');

    await pool.query(`
      ALTER TABLE failed_proxies
      ADD COLUMN IF NOT EXISTS city VARCHAR(100),
      ADD COLUMN IF NOT EXISTS state VARCHAR(100),
      ADD COLUMN IF NOT EXISTS country VARCHAR(100),
      ADD COLUMN IF NOT EXISTS country_code VARCHAR(2),
      ADD COLUMN IF NOT EXISTS location_updated_at TIMESTAMP
    `);

    console.log('✅ Added columns to failed_proxies table');
    console.log('✅ Migration complete!');

    process.exit(0);
  } catch (error) {
    console.error('❌ Migration failed:', error);
    process.exit(1);
  }
})();