first commit

2025-11-06 11:08:59 +01:00
commit 3c5117c2c3
85 changed files with 13275 additions and 0 deletions
@@ -0,0 +1,212 @@
+const { initializeDatabase, resetDatabase, checkDatabaseStatus } = require('./databaseInit');
+const { sequelize } = require('../config/database');
+const logger = require('./logger');
+
+// Import seeders
+const createAdminUser = require('../seeders/001_create_admin_user');
+const createSampleTrainingData = require('../seeders/002_create_sample_training_data');
+
+const runMigrations = async () => {
+  try {
+    console.log('🔄 Running database migrations...');
+    
+    // Import all migration files
+    const migrations = [
+      require('../migrations/001_create_users'),
+      require('../migrations/002_create_conversations'),
+      require('../migrations/003_create_plans'),
+      require('../migrations/004_create_messages'),
+      require('../migrations/005_create_documents'),
+      require('../migrations/006_create_feedback'),
+      require('../migrations/007_create_model_versions'),
+      require('../migrations/008_create_training_data'),
+      require('../migrations/009_create_tool_executions')
+    ];
+
+    for (const migration of migrations) {
+      console.log(`   Running migration: ${migration.up.name || 'unnamed'}`);
+      await migration.up(sequelize.getQueryInterface(), sequelize.constructor);
+    }
+
+    console.log('✅ All migrations completed successfully');
+    return true;
+  } catch (error) {
+    console.error('❌ Migration failed:', error);
+    logger.error('Migration error:', error);
+    return false;
+  }
+};
+
+const runSeeders = async () => {
+  try {
+    console.log('🌱 Running database seeders...');
+    
+    // Run seeders
+    await createAdminUser.up(sequelize.getQueryInterface(), sequelize.constructor);
+    await createSampleTrainingData.up(sequelize.getQueryInterface(), sequelize.constructor);
+    
+    console.log('✅ All seeders completed successfully');
+    return true;
+  } catch (error) {
+    console.error('❌ Seeding failed:', error);
+    logger.error('Seeding error:', error);
+    return false;
+  }
+};
+
+const setupDatabase = async (options = {}) => {
+  try {
+    console.log('🚀 Setting up database...');
+    
+    const { 
+      force = false, 
+      seed = true, 
+      migrations = true 
+    } = options;
+
+    // Check database status
+    const status = await checkDatabaseStatus();
+    if (!status) {
+      throw new Error('Database connection failed');
+    }
+
+    // Run migrations if requested
+    if (migrations) {
+      const migrated = await runMigrations();
+      if (!migrated) {
+        throw new Error('Migrations failed');
+      }
+    }
+
+    // Run seeders if requested
+    if (seed) {
+      const seeded = await runSeeders();
+      if (!seeded) {
+        throw new Error('Seeding failed');
+      }
+    }
+
+    console.log('🎉 Database setup completed successfully!');
+    return true;
+  } catch (error) {
+    console.error('❌ Database setup failed:', error.message);
+    logger.error('Database setup error:', error);
+    return false;
+  }
+};
+
+const getDatabaseInfo = async () => {
+  try {
+    console.log('📊 Database Information:');
+    
+    // Get table information
+    const [results] = await sequelize.query(`
+      SELECT 
+        schemaname,
+        tablename,
+        tableowner
+      FROM pg_tables 
+      WHERE schemaname = 'public'
+      ORDER BY tablename;
+    `);
+
+    console.log(`   Tables: ${results.length}`);
+    results.forEach(table => {
+      console.log(`   - ${table.tablename}`);
+    });
+
+    // Get database size
+    const [sizeResult] = await sequelize.query(`
+      SELECT pg_size_pretty(pg_database_size(current_database())) as size;
+    `);
+    
+    console.log(`   Database size: ${sizeResult[0].size}`);
+
+    return {
+      tables: results,
+      size: sizeResult[0].size
+    };
+  } catch (error) {
+    console.error('❌ Error getting database info:', error);
+    return null;
+  }
+};
+
+const backupDatabase = async (backupPath) => {
+  try {
+    console.log('💾 Creating database backup...');
+    
+    // This would typically use pg_dump in a real implementation
+    console.log(`   Backup path: ${backupPath}`);
+    console.log('   Note: Implement actual backup logic with pg_dump');
+    
+    return true;
+  } catch (error) {
+    console.error('❌ Backup failed:', error);
+    return false;
+  }
+};
+
+// Command line interface
+if (require.main === module) {
+  const args = process.argv.slice(2);
+  const command = args[0] || 'help';
+  
+  const runCommand = async () => {
+    switch (command) {
+      case 'init':
+        await setupDatabase({ force: false, seed: true, migrations: true });
+        break;
+      case 'reset':
+        await resetDatabase();
+        break;
+      case 'migrate':
+        await runMigrations();
+        break;
+      case 'seed':
+        await runSeeders();
+        break;
+      case 'status':
+        await checkDatabaseStatus();
+        break;
+      case 'info':
+        await getDatabaseInfo();
+        break;
+      case 'backup':
+        const backupPath = args[1] || './backup.sql';
+        await backupDatabase(backupPath);
+        break;
+      case 'help':
+      default:
+        console.log(`
+Database Manager Commands:
+  init     - Initialize database with migrations and seeders
+  reset    - Reset database (drop and recreate)
+  migrate  - Run migrations only
+  seed     - Run seeders only
+  status   - Check database connection status
+  info     - Show database information
+  backup   - Create database backup
+  help     - Show this help message
+
+Usage: node databaseManager.js [command]
+        `);
+        break;
+    }
+  };
+
+  runCommand().then(() => {
+    process.exit(0);
+  }).catch(error => {
+    console.error('Command failed:', error);
+    process.exit(1);
+  });
+}
+
+module.exports = {
+  runMigrations,
+  runSeeders,
+  setupDatabase,
+  getDatabaseInfo,
+  backupDatabase
+};