import { spawn } from 'node:child_process';
import { resolve } from 'node:path';

const viteEntry = resolve(process.cwd(), 'node_modules/vite/bin/vite.js');
const child = spawn(process.execPath, [viteEntry, ...process.argv.slice(2)], {
  stdio: 'inherit',
  env: {
    ...process.env,
    NIANCODE_IMAGE_WORKSPACE_MODE: 'local',
  },
});

for (const signal of ['SIGINT', 'SIGTERM']) {
  process.on(signal, () => {
    child.kill(signal);
  });
}

child.on('exit', (code, signal) => {
  if (signal) {
    process.kill(process.pid, signal);
    return;
  }
  process.exitCode = code ?? 1;
});