// add-premium-columns.js
// Скрипт для добавления колонок premium и premium_expires_at в таблицу users

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

// Настройки подключения к базе данных из переменных окружения
require('dotenv').config();

const pool = new Pool({
  host: process.env.DB_HOST || 'localhost',
  port: process.env.DB_PORT || 5432,
  database: process.env.DB_NAME || 'telegram_tinder_bot',
  user: process.env.DB_USERNAME || 'postgres',
  password: process.env.DB_PASSWORD || ''
});

async function addPremiumColumns() {
  try {
    console.log('Подключение к базе данных...');
    const client = await pool.connect();
    
    console.log('Добавление колонок premium и premium_expires_at в таблицу users...');
    
    // SQL запрос для добавления колонок
    const sql = `
      ALTER TABLE users 
        ADD COLUMN IF NOT EXISTS premium BOOLEAN DEFAULT FALSE,
        ADD COLUMN IF NOT EXISTS premium_expires_at TIMESTAMP;
    `;
    
    await client.query(sql);
    console.log('✅ Колонки premium и premium_expires_at успешно добавлены в таблицу users');
    
    // Закрытие соединения
    client.release();
    await pool.end();
    console.log('Подключение к базе данных закрыто');
  } catch (error) {
    console.error('❌ Ошибка при добавлении колонок:', error);
    await pool.end();
    process.exit(1);
  }
}

// Запуск функции
addPremiumColumns();