#!/usr/bin/env node

/**
 * PM2 Management Script
 * Helper script để quản lý các PM2 processes
 */

const { execSync } = require('child_process');
const path = require('path');

const ECOSYSTEM_FILE = path.join(__dirname, '..', 'ecosystem.config.js');

// Colors for console output
const colors = {
  reset: '\x1b[0m',
  green: '\x1b[32m',
  yellow: '\x1b[33m',
  blue: '\x1b[34m',
  red: '\x1b[31m',
};

const log = (message, color = 'reset') => {
  console.log(`${colors[color]}${message}${colors.reset}`);
};

const exec = (command, options = {}) => {
  try {
    const result = execSync(command, {
      stdio: 'inherit',
      ...options,
    });
    return result;
  } catch (error) {
    log(`Error executing command: ${command}`, 'red');
    process.exit(1);
  }
};

// Commands
const commands = {
  start: () => {
    log('\n🚀 Starting all processes...', 'blue');
    exec(`pm2 start ${ECOSYSTEM_FILE}`);
    log('✅ All processes started!', 'green');
  },

  stop: () => {
    log('\n⏸️  Stopping all processes...', 'yellow');
    exec('pm2 stop ecosystem.config.js');
    log('✅ All processes stopped!', 'green');
  },

  restart: () => {
    log('\n🔄 Restarting all processes...', 'blue');
    exec(`pm2 restart ${ECOSYSTEM_FILE}`);
    log('✅ All processes restarted!', 'green');
  },

  delete: () => {
    log('\n🗑️  Deleting all processes...', 'yellow');
    exec('pm2 delete ecosystem.config.js');
    log('✅ All processes deleted!', 'green');
  },

  status: () => {
    log('\n📊 Process status:', 'blue');
    exec('pm2 list');
  },

  logs: (processName) => {
    if (processName) {
      log(`\n📋 Showing logs for ${processName}...`, 'blue');
      exec(`pm2 logs ${processName}`);
    } else {
      log('\n📋 Showing all logs...', 'blue');
      exec('pm2 logs');
    }
  },

  monit: () => {
    log('\n📈 Opening process monitor...', 'blue');
    exec('pm2 monit');
  },

  flush: () => {
    log('\n🧹 Flushing all logs...', 'yellow');
    exec('pm2 flush');
    log('✅ Logs flushed!', 'green');
  },

  save: () => {
    log('\n💾 Saving PM2 process list...', 'blue');
    exec('pm2 save');
    log('✅ Process list saved!', 'green');
  },

  startup: () => {
    log('\n⚙️  Configuring PM2 to start on system boot...', 'blue');
    exec('pm2 startup');
    log('✅ Run the command above, then run: npm run pm2:save', 'yellow');
  },

  'start:api': () => {
    log('\n🚀 Starting API server only...', 'blue');
    exec(`pm2 start ${ECOSYSTEM_FILE} --only sena-api`);
    log('✅ API server started!', 'green');
  },

  'start:workers': () => {
    log('\n🚀 Starting all workers...', 'blue');
    exec(`pm2 start ${ECOSYSTEM_FILE} --only worker-db-write,worker-lesson-fill,worker-process-data`);
    log('✅ All workers started!', 'green');
  },

  help: () => {
    console.log(`
╔════════════════════════════════════════════════════════════════╗
║              Sena PM2 Management Commands                      ║
╠════════════════════════════════════════════════════════════════╣
║                                                                ║
║  npm run pm2:start          - Start all processes              ║
║  npm run pm2:stop           - Stop all processes               ║
║  npm run pm2:restart        - Restart all processes            ║
║  npm run pm2:delete         - Delete all processes             ║
║  npm run pm2:status         - Show process status              ║
║  npm run pm2:logs [name]    - Show logs (optional: process)    ║
║  npm run pm2:monit          - Open process monitor             ║
║  npm run pm2:flush          - Flush all logs                   ║
║  npm run pm2:save           - Save process list                ║
║  npm run pm2:startup        - Configure startup script         ║
║                                                                ║
║  npm run pm2:start:api      - Start API server only            ║
║  npm run pm2:start:workers  - Start all workers only           ║
║                                                                ║
╠════════════════════════════════════════════════════════════════╣
║  Process Names:                                                ║
║    - sena-api                  (Main API Server)               ║
║    - worker-db-write           (Database Write Worker)         ║
║    - worker-lesson-fill        (Lesson Data Fill Worker)       ║
║    - worker-process-data       (Process Data Worker)           ║
╚════════════════════════════════════════════════════════════════╝
    `);
  },
};

// Parse command
const command = process.argv[2];
const arg = process.argv[3];

if (!command || command === 'help' || command === '--help' || command === '-h') {
  commands.help();
  process.exit(0);
}

if (commands[command]) {
  commands[command](arg);
} else {
  log(`Unknown command: ${command}`, 'red');
  commands.help();
  process.exit(1);
}