// Скрипт для добавления колонки premium в таблицу users и установки premium для всех пользователей
require('dotenv').config();
const { Pool } = require('pg');

// Создаем пул соединений
const pool = new Pool({
  user: process.env.DB_USERNAME,
  host: process.env.DB_HOST,
  database: process.env.DB_NAME,
  password: process.env.DB_PASSWORD,
  port: parseInt(process.env.DB_PORT || '5432')
});

async function setAllUsersToPremium() {
  try {
    console.log('Проверяем наличие столбца premium в таблице users...');
    
    const result = await pool.query(`
      SELECT EXISTS (
        SELECT FROM information_schema.columns 
        WHERE table_schema = 'public' 
        AND table_name = 'users' 
        AND column_name = 'premium'
      );
    `);
    
    if (!result.rows[0].exists) {
      console.log('🔄 Добавляем столбец premium...');
      await pool.query(`ALTER TABLE users ADD COLUMN premium BOOLEAN DEFAULT false;`);
      console.log('✅ Столбец premium успешно добавлен');
    } else {
      console.log('✅ Столбец premium уже существует');
    }

    console.log('Устанавливаем премиум-статус для всех пользователей...');
    
    const updateResult = await pool.query(`
      UPDATE users 
      SET premium = true 
      WHERE true
      RETURNING id, telegram_id, premium
    `);
    
    console.log(`✅ Успешно установлен премиум-статус для ${updateResult.rows.length} пользователей:`);
    updateResult.rows.forEach(row => {
      console.log(`ID: ${row.id.substr(0, 8)}... | Telegram ID: ${row.telegram_id} | Premium: ${row.premium}`);
    });
    
    console.log('🎉 Все пользователи теперь имеют премиум-статус!');
  } catch (error) {
    console.error('Ошибка при установке премиум-статуса:', error);
  } finally {
    await pool.end();
    console.log('Соединение с базой данных закрыто');
  }
}

setAllUsersToPremium();