const { Pool } = require('pg');
require('dotenv').config();

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

async function cleanDatabase() {
    const client = await pool.connect();
    try {
        console.log('Очистка базы данных...');
        await client.query('BEGIN');

        // Отключаем временно foreign key constraints
        await client.query('SET CONSTRAINTS ALL DEFERRED');

        // Очищаем таблицы в правильном порядке
        console.log('Очистка таблицы messages...');
        await client.query('DELETE FROM messages');
        
        console.log('Очистка таблицы profile_views...');
        await client.query('DELETE FROM profile_views');
        
        console.log('Очистка таблицы matches...');
        await client.query('DELETE FROM matches');
        
        console.log('Очистка таблицы swipes...');
        await client.query('DELETE FROM swipes');
        
        console.log('Очистка таблицы profiles...');
        await client.query('DELETE FROM profiles');
        
        console.log('Очистка таблицы users...');
        await client.query('DELETE FROM users');

        // Возвращаем foreign key constraints
        await client.query('SET CONSTRAINTS ALL IMMEDIATE');

        await client.query('COMMIT');
        console.log('✅ База данных успешно очищена');
    } catch (e) {
        await client.query('ROLLBACK');
        console.error('❌ Ошибка при очистке базы данных:', e);
    } finally {
        client.release();
        await pool.end();
    }
}

// Запускаем функцию очистки
cleanDatabase();