@vezlo/assistant-server 1.3.0 → 2.0.0

package/scripts/validate-db.js

@@ -57,15 +57,24 @@ async function validateDatabase() {
       process.env.SUPABASE_SERVICE_KEY
     );
 
-    // Try to query a table
+    // Try to query any table (query will fail if connection is bad)
     const { error } = await supabase.from('vezlo_conversations').select('count').limit(0);
 
-    if (error && error.code !== 'PGRST116') {
-      throw error;
+    if (error) {
+      // Check for table not found errors (normal before migrations run)
+      if (error.code === 'PGRST116' || 
+          error.message.includes('does not exist') ||
+          error.message.includes('Could not find the table')) {
+        log('✅ Supabase connection successful', 'green');
+        log('⚠️  Note: Tables not found yet - this is normal before running migrations\n', 'yellow');
+      } else {
+        // Only throw if it's a real connection error
+        throw error;
+      }
+    } else {
+      log('✅ Supabase connection successful\n', 'green');
     }
 
-    log('✅ Supabase connection successful\n', 'green');
-
   } catch (error) {
     log(`❌ Supabase connection failed: ${error.message}\n`, 'red');
     process.exit(1);
@@ -74,10 +83,11 @@ async function validateDatabase() {
   // Test database connection and validate tables
   log('Validating database tables...', 'yellow');
 
+  let client;
   try {
     const { Client } = require('pg');
 
-    const client = new Client({
+    client = new Client({
       host: process.env.SUPABASE_DB_HOST,
       port: parseInt(process.env.SUPABASE_DB_PORT || '5432'),
       database: process.env.SUPABASE_DB_NAME || 'postgres',
@@ -86,14 +96,23 @@ async function validateDatabase() {
       ssl: { rejectUnauthorized: false }
     });
 
+    // Handle connection errors
+    client.on('error', (err) => {
+      console.error('Database connection error:', err.message);
+    });
+
     await client.connect();
 
     // Check required tables
     const requiredTables = [
-      'conversations',
-      'messages',
-      'message_feedback',
-      'knowledge_items'
+      'vezlo_companies',
+      'vezlo_users',
+      'vezlo_user_company_profiles',
+      'vezlo_api_keys',
+      'vezlo_conversations',
+      'vezlo_messages',
+      'vezlo_message_feedback',
+      'vezlo_knowledge_items'
     ];
 
     const result = await client.query(`
@@ -111,7 +130,9 @@ async function validateDatabase() {
       log(`\n❌ Missing required tables:`, 'red');
       missingTables.forEach(table => log(`   - ${table}`, 'red'));
       log('\nRun the setup wizard: ' + colors.bright + 'npx vezlo-setup' + colors.reset + '\n', 'yellow');
-      await client.end();
+      if (client) {
+        await client.end();
+      }
       process.exit(1);
     }
 
@@ -134,7 +155,10 @@ async function validateDatabase() {
       ORDER BY t.table_name
     `, [requiredTables]);
 
-    log('\n📊 Table Structure:', 'cyan');
+    log('\n📊 Database Tables Summary:', 'cyan');
+    log(`   Total tables found: ${schemaCheck.rows.length}`, 'green');
+    
+    log('\n📋 Table Details:', 'cyan');
     schemaCheck.rows.forEach(row => {
       log(`   ✓ ${row.table_name} (${row.column_count} columns)`, 'green');
     });
@@ -156,10 +180,19 @@ async function validateDatabase() {
     log('Your server is ready to start:', 'cyan');
     log('  ' + colors.bright + 'vezlo-server' + colors.reset + '\n');
 
-    await client.end();
+    if (client) {
+      await client.end();
+    }
 
   } catch (error) {
     log(`\n❌ Database validation failed: ${error.message}\n`, 'red');
+    if (client) {
+      try {
+        await client.end();
+      } catch (closeError) {
+        // Ignore close errors
+      }
+    }
     process.exit(1);
   }
 }