#!/usr/bin/env npx tsx
/**
 * Start Pod - Simulates a Kubernetes pod locally
 *
 * Starts 5 workers with a pod name from the predefined list.
 *
 * Usage:
 *   npx tsx src/tasks/start-pod.ts <pod-index>
 *   npx tsx src/tasks/start-pod.ts 0   # Starts pod "Aethelgard" with 5 workers
 *   npx tsx src/tasks/start-pod.ts 1   # Starts pod "Xylos" with 5 workers
 */

import { spawn } from 'child_process';
import path from 'path';

const POD_NAMES = [
  'Aethelgard',
  'Xylos',
  'Kryll',
  'Coriolis',
  'Dimidium',
  'Veridia',
  'Zetani',
  'Talos IV',
  'Onyx',
  'Celestia',
  'Gormand',
  'Betha',
  'Ragnar',
  'Syphon',
  'Axiom',
  'Nadir',
  'Terra Nova',
  'Acheron',
  'Nexus',
  'Vespera',
  'Helios Prime',
  'Oasis',
  'Mordina',
  'Cygnus',
  'Umbra',
];

const WORKERS_PER_POD = 5;

async function main() {
  const podIndex = parseInt(process.argv[2] ?? '0', 10);

  if (podIndex < 0 || podIndex >= POD_NAMES.length) {
    console.error(`Invalid pod index: ${podIndex}. Must be 0-${POD_NAMES.length - 1}`);
    process.exit(1);
  }

  const podName = POD_NAMES[podIndex];
  console.log(`[Pod] Starting pod "${podName}" with ${WORKERS_PER_POD} workers...`);

  const workerScript = path.join(__dirname, 'task-worker.ts');
  const workers: ReturnType<typeof spawn>[] = [];

  for (let i = 1; i <= WORKERS_PER_POD; i++) {
    const workerId = `${podName}-worker-${i}`;

    const worker = spawn('npx', ['tsx', workerScript], {
      env: {
        ...process.env,
        WORKER_ID: workerId,
        POD_NAME: podName,
      },
      stdio: 'inherit',
    });

    workers.push(worker);
    console.log(`[Pod] Started worker ${i}/${WORKERS_PER_POD}: ${workerId}`);
  }

  // Handle shutdown
  const shutdown = () => {
    console.log(`\n[Pod] Shutting down pod "${podName}"...`);
    workers.forEach(w => w.kill('SIGTERM'));
    setTimeout(() => process.exit(0), 2000);
  };

  process.on('SIGTERM', shutdown);
  process.on('SIGINT', shutdown);

  // Keep the process alive
  await new Promise(() => {});
}

main().catch(err => {
  console.error('[Pod] Fatal error:', err);
  process.exit(1);
});