yoshiko-pg__difit/scripts/dev.js

#!/usr/bin/env node
import { spawn } from 'child_process';
import mri from 'mri';

const rawArgs = process.argv.slice(2);
const parsedArgs = mri(rawArgs, { '--': true });
const positionalArgs = parsedArgs._;
const commitish = positionalArgs[0] ?? 'HEAD';
const compareWith = positionalArgs[1];
const CLI_SERVER_READY_MESSAGE = 'difit server started';

// Check if we should read from stdin
const shouldReadStdin = !process.stdin.isTTY || commitish === '-';

console.log('🚀 Starting CLI server...');
if (shouldReadStdin) {
  console.log('📥 Reading diff from stdin...');
}

// Start CLI server
const cliArgs = ['run', 'dev:cli'];

// If not reading from stdin, add commitish and compareWith
if (!shouldReadStdin) {
  cliArgs.push(commitish);
  if (compareWith) {
    cliArgs.push(compareWith);
  }
} else {
  // For stdin mode, use '-' as the commitish
  cliArgs.push('-');
}
cliArgs.push('--no-open');

// Forward remaining args (preserves order and supports flag values)
const positionalToSkip = positionalArgs.slice(0, 2);
let skippedCount = 0;
for (let i = 0; i < rawArgs.length; i += 1) {
  const token = rawArgs[i];

  if (token === '--') {
    cliArgs.push(...rawArgs.slice(i));
    break;
  }

  if (skippedCount < positionalToSkip.length && token === positionalToSkip[skippedCount]) {
    skippedCount += 1;
    continue;
  }

  cliArgs.push(token);
}

const cliProcess = spawn('pnpm', cliArgs, {
  // For stdin mode, pipe stdin; otherwise inherit
  stdio: [shouldReadStdin ? 'pipe' : 'inherit', 'pipe', 'inherit'],
  shell: true,
});

// If reading from stdin, pipe it to the CLI process
if (shouldReadStdin) {
  process.stdin.pipe(cliProcess.stdin);
}

// Wait for CLI server to be ready, then start Vite
let cliReady = false;
let viteProcess = null;

cliProcess.stdout.on('data', (data) => {
  process.stdout.write(data);
  const output = data.toString();

  // Wait for CLI server before starting Vite to prevent proxy connection errors
  // Uses stdout parsing to keep dev orchestration separate from main CLI logic
  if (!cliReady && output.includes(CLI_SERVER_READY_MESSAGE)) {
    cliReady = true;
    console.log('🎨 Starting Vite dev server...');
    viteProcess = spawn('pnpm', ['exec', 'vite', '--open'], {
      stdio: 'inherit',
      shell: true,
    });
  }
});

// Handle graceful shutdown
process.on('SIGINT', () => {
  console.log('\n👋 Shutting down...');
  cliProcess.kill('SIGINT');
  viteProcess?.kill('SIGINT');
  process.exit(0);
});

cliProcess.on('exit', (code) => {
  if (code !== 0) {
    console.error(`CLI server exited with code ${code}`);
  }
  // Kill vite process when CLI exits
  viteProcess?.kill('SIGINT');
  process.exit(code || 0);
});