Описание изменений

backend/db/init.js

@@ -1,8 +1,13 @@
+const fs = require('fs');
+const path = require('path');
+const { pool } = require('./index');
+const logger = require('../utils/logger');
+
 // Инициализация таблицы roles
 async function initRoles() {
   try {
     // Проверяем, существует ли таблица roles
-    const tableExists = await db.query(`
+    const tableExists = await pool.query(`
       SELECT EXISTS (
         SELECT FROM information_schema.tables 
         WHERE table_name = 'roles'
@@ -11,7 +16,7 @@ async function initRoles() {
     
     if (!tableExists.rows[0].exists) {
       // Создаем таблицу roles
-      await db.query(`
+      await pool.query(`
         CREATE TABLE roles (
           id SERIAL PRIMARY KEY,
           name VARCHAR(50) NOT NULL UNIQUE,
@@ -21,7 +26,7 @@ async function initRoles() {
       `);
       
       // Добавляем роли
-      await db.query(`
+      await pool.query(`
         INSERT INTO roles (id, name, description) VALUES 
         (3, 'user', 'Обычный пользователь'),
         (4, 'admin', 'Администратор с полным доступом');
@@ -30,24 +35,24 @@ async function initRoles() {
       console.log('Таблица roles создана и заполнена');
     } else {
       // Проверяем наличие ролей
-      const rolesExist = await db.query(`
+      const rolesExist = await pool.query(`
         SELECT COUNT(*) FROM roles WHERE id IN (3, 4);
       `);
       
       if (rolesExist.rows[0].count < 2) {
         // Добавляем недостающие роли
-        const userRoleExists = await db.query(`SELECT EXISTS (SELECT FROM roles WHERE name = 'user');`);
-        const adminRoleExists = await db.query(`SELECT EXISTS (SELECT FROM roles WHERE name = 'admin');`);
+        const userRoleExists = await pool.query(`SELECT EXISTS (SELECT FROM roles WHERE name = 'user');`);
+        const adminRoleExists = await pool.query(`SELECT EXISTS (SELECT FROM roles WHERE name = 'admin');`);
         
         if (!userRoleExists.rows[0].exists) {
-          await db.query(`
+          await pool.query(`
             INSERT INTO roles (id, name, description) VALUES 
             (3, 'user', 'Обычный пользователь');
           `);
         }
         
         if (!adminRoleExists.rows[0].exists) {
-          await db.query(`
+          await pool.query(`
             INSERT INTO roles (id, name, description) VALUES 
             (4, 'admin', 'Администратор с полным доступом');
           `);
@@ -60,4 +65,55 @@ async function initRoles() {
     console.error('Ошибка при инициализации таблицы roles:', error);
     throw error;
   }
-} 
+}
+
+async function initializeDatabase() {
+  try {
+    // Создаем таблицу для отслеживания миграций, если её нет
+    await pool.query(`
+      CREATE TABLE IF NOT EXISTS migrations (
+        id SERIAL PRIMARY KEY,
+        name VARCHAR(255) NOT NULL UNIQUE,
+        executed_at TIMESTAMP NOT NULL DEFAULT NOW()
+      );
+    `);
+
+    // Путь к папке с миграциями
+    const migrationsPath = path.join(__dirname, 'migrations');
+    
+    // Получаем все файлы миграций
+    const migrationFiles = fs.readdirSync(migrationsPath)
+      .filter(file => file.endsWith('.sql'))
+      .sort();
+
+    // Получаем выполненные миграции
+    const { rows } = await pool.query('SELECT name FROM migrations');
+    const executedMigrations = new Set(rows.map(row => row.name));
+
+    // Выполняем только новые миграции
+    for (const file of migrationFiles) {
+      if (!executedMigrations.has(file)) {
+        const filePath = path.join(migrationsPath, file);
+        const sql = fs.readFileSync(filePath, 'utf8');
+        
+        logger.info(`Executing migration: ${file}`);
+        await pool.query(sql);
+        
+        // Записываем выполненную миграцию
+        await pool.query(
+          'INSERT INTO migrations (name) VALUES ($1)',
+          [file]
+        );
+        
+        logger.info(`Migration completed: ${file}`);
+      }
+    }
+
+    logger.info('All migrations completed successfully');
+  } catch (error) {
+    logger.error('Error during database initialization:', error);
+    throw error;
+  }
+}
+
+module.exports = { initializeDatabase };