@vezlo/assistant-server 2.1.0 → 2.2.2

package/scripts/setup.js

@@ -9,6 +9,29 @@ const fs = require('fs');
 const path = require('path');
 const readline = require('readline');
 const { createClient } = require('@supabase/supabase-js');
+const knexLib = require('knex');
+const pkgRoot = path.join(__dirname, '..');
+
+const migrationCandidates = [
+  path.join(pkgRoot, 'dist', 'src', 'migrations'),
+  path.join(pkgRoot, 'dist', 'migrations'),
+  path.join(pkgRoot, 'src', 'migrations'),
+  path.join(process.cwd(), 'dist', 'src', 'migrations'),
+  path.join(process.cwd(), 'src', 'migrations')
+];
+
+const fallbackMigrationDir = path.join(pkgRoot, 'dist', 'src', 'migrations');
+const migrationDirectory = migrationCandidates.find(dir => fs.existsSync(dir)) || fallbackMigrationDir;
+
+if (!fs.existsSync(migrationDirectory)) {
+  console.error(`❌ Migration directory not found. Expected one of:\n${migrationCandidates.concat(fallbackMigrationDir).join('\n')}`);
+  console.error('Run `npm run build` (or reinstall the package) to ensure compiled migrations exist.');
+  process.exit(1);
+}
+
+const usingDistMigrations = migrationDirectory.includes(`${path.sep}dist${path.sep}`);
+const migrationExtension = usingDistMigrations ? 'js' : 'ts';
+const migrationLoadExtensions = migrationExtension === 'ts' ? ['.ts'] : ['.js'];
 
 const rl = readline.createInterface({
   input: process.stdin,
@@ -300,9 +323,13 @@ async function saveEnvFile(envPath, config) {
     if (!config.MIGRATION_SECRET_KEY) {
       config.MIGRATION_SECRET_KEY = crypto.randomBytes(32).toString('hex');
     }
+    if (!config.JWT_SECRET) {
+      config.JWT_SECRET = crypto.randomBytes(32).toString('hex');
+    }
   } catch (_) {
-    // Fallback simple key if crypto unavailable (very unlikely)
+    // Fallback simple values if crypto unavailable (very unlikely)
     config.MIGRATION_SECRET_KEY = config.MIGRATION_SECRET_KEY || `msk_${Date.now()}`;
+    config.JWT_SECRET = config.JWT_SECRET || `jwt_${Date.now()}`;
   }
   const envContent = `# Vezlo Assistant Server Configuration
 # Generated by setup wizard on ${new Date().toISOString()}
@@ -348,8 +375,16 @@ ASSISTANT_NAME=Vezlo Assistant
 CHUNK_SIZE=1000
 CHUNK_OVERLAP=200
 
+# Chat History
+CHAT_HISTORY_LENGTH=${config.CHAT_HISTORY_LENGTH || '2'}
+
 # Migration Security
 MIGRATION_SECRET_KEY=${config.MIGRATION_SECRET_KEY}
+
+# Authentication / Defaults
+JWT_SECRET=${config.JWT_SECRET}
+DEFAULT_ADMIN_EMAIL=${config.DEFAULT_ADMIN_EMAIL || 'admin@vezlo.org'}
+DEFAULT_ADMIN_PASSWORD=${config.DEFAULT_ADMIN_PASSWORD || 'admin123'}
 `;
 
   fs.writeFileSync(envPath, envContent, 'utf8');
@@ -459,15 +494,7 @@ async function setupMigrations(config, validationStatus) {
       if (runMigrations.toLowerCase() === 'y') {
         log('🔄 Running migrations...', 'yellow');
         
-        // Set environment variables for migration
-        process.env.SUPABASE_DB_HOST = config.SUPABASE_DB_HOST;
-        process.env.SUPABASE_DB_PORT = config.SUPABASE_DB_PORT;
-        process.env.SUPABASE_DB_NAME = config.SUPABASE_DB_NAME;
-        process.env.SUPABASE_DB_USER = config.SUPABASE_DB_USER;
-        process.env.SUPABASE_DB_PASSWORD = config.SUPABASE_DB_PASSWORD;
-        
-        const { execSync } = require('child_process');
-        execSync('npm run migrate:latest', { stdio: 'inherit' });
+        await runMigrationsWithConfig(config);
         
         log('✅ Migrations completed successfully!', 'green');
         return { migrations: 'success' };
@@ -491,8 +518,7 @@ async function setupMigrations(config, validationStatus) {
       if (runPending.toLowerCase() === 'y') {
         log('🔄 Checking for pending migrations...', 'yellow');
         
-        const { execSync } = require('child_process');
-        execSync('npm run migrate:latest', { stdio: 'inherit' });
+        await runMigrationsWithConfig(config);
         
         log('✅ Migration check completed!', 'green');
         await client.end();
@@ -531,6 +557,34 @@ main().catch(error => {
   rl.close();
   process.exit(1);
 });
+
+async function runMigrationsWithConfig(config) {
+  const knexInstance = knexLib({
+    client: 'postgresql',
+    connection: {
+      host: config.SUPABASE_DB_HOST,
+      port: parseInt(config.SUPABASE_DB_PORT),
+      database: config.SUPABASE_DB_NAME,
+      user: config.SUPABASE_DB_USER,
+      password: config.SUPABASE_DB_PASSWORD,
+      ssl: { rejectUnauthorized: false }
+    },
+    migrations: {
+      directory: migrationDirectory,
+      tableName: 'knex_migrations',
+      extension: migrationExtension,
+      loadExtensions: migrationLoadExtensions
+    },
+    pool: { min: 0, max: 10 }
+  });
+
+  try {
+    await knexInstance.migrate.latest();
+  } finally {
+    await knexInstance.destroy();
+  }
+}
+
 async function setupDefaultData(config) {
   log('🔄 Setting up default company and admin user...', 'yellow');