@juspay/shooter 1.0.0 → 1.2.0

package/bin/shooter.cjs

@@ -2,140 +2,509 @@
 
 // CLI entry point for the Shooter server.
 // Usage: shooter [command]
-//   shooter start    — Start the server (default)
-//   shooter setup    — Interactive setup wizard
-//   shooter version  — Show version
-//   shooter help     — Show usage
 
 'use strict';
 
+const os = require('os');
 const path = require('path');
-const { spawn } = require('child_process');
+const { spawn, execSync } = require('child_process');
 const fs = require('fs');
 
-// ── Resolve the package root ────────────────────────────────────────
-// Whether installed globally via npm or run from a local clone, the
-// package root is always one directory above this bin/ script.
+// ── Resolve paths ───────────────────────────────────────────────────
 const PKG_ROOT = path.resolve(__dirname, '..');
+const SHOOTER_HOME = process.env.SHOOTER_HOME || path.join(os.homedir(), '.shooter');
+const PID_FILE = path.join(SHOOTER_HOME, 'shooter.pid');
+const LOG_DIR = path.join(SHOOTER_HOME, 'logs');
+const LOG_FILE = path.join(LOG_DIR, 'shooter.log');
+const DEFAULT_PORT = 54007;
+const LAUNCHD_LABEL = 'com.juspay.shooter';
+const LAUNCHD_PLIST = path.join(os.homedir(), 'Library', 'LaunchAgents', `${LAUNCHD_LABEL}.plist`);
+const SYSTEMD_UNIT = path.join(os.homedir(), '.config', 'systemd', 'user', 'shooter.service');
 
 const pkg = require(path.join(PKG_ROOT, 'package.json'));
 
+// ── Signal Helpers ──────────────────────────────────────────────────
+function signalCode(sig) {
+  return os.constants.signals[sig] || 0;
+}
+
+// ── PID helpers ─────────────────────────────────────────────────────
+function readPid() {
+  try {
+    const pid = parseInt(fs.readFileSync(PID_FILE, 'utf8').trim(), 10);
+    if (isNaN(pid)) return null;
+    // Check if process is alive
+    process.kill(pid, 0);
+    // Verify it's actually a node/shooter process (prevent PID reuse attacks)
+    try {
+      const cmdline = execSync(`ps -p ${pid} -o command= 2>/dev/null`, { encoding: 'utf8' }).trim();
+      if (!cmdline.includes('shooter') && !cmdline.includes('server.ts') && !cmdline.includes('tsx')) {
+        // PID was reused by a different process — stale pidfile
+        fs.unlinkSync(PID_FILE);
+        return null;
+      }
+    } catch {
+      // ps failed — process may have exited between kill(0) and ps
+      return null;
+    }
+    return pid;
+  } catch {
+    return null;
+  }
+}
+
+function writePid(pid) {
+  fs.mkdirSync(SHOOTER_HOME, { recursive: true });
+  fs.writeFileSync(PID_FILE, String(pid));
+}
+
+function removePid() {
+  try { fs.unlinkSync(PID_FILE); } catch {}
+}
+
 // ── CLI argument parsing ────────────────────────────────────────────
 const args = process.argv.slice(2);
 const command = args[0] || 'start';
 
 switch (command) {
-	case 'start':
-		startServer();
-		break;
-	case 'setup':
-		runSetup();
-		break;
-	case 'version':
-	case '--version':
-	case '-v':
-		console.log(`shooter v${pkg.version}`);
-		break;
-	case 'help':
-	case '--help':
-	case '-h':
-		showHelp();
-		break;
-	default:
-		console.error(`Unknown command: ${command}\n`);
-		showHelp();
-		process.exit(1);
-}
-
-// ── Commands ────────────────────────────────────────────────────────
+  case 'start':
+    startServer();
+    break;
+  case 'stop':
+    stopServer();
+    break;
+  case 'status':
+    showStatus();
+    break;
+  case 'autostart':
+    manageAutostart(args[1]);
+    break;
+  case 'logs':
+    showLogs();
+    break;
+  case 'setup':
+    runSetup();
+    break;
+  case 'version':
+  case '--version':
+  case '-v':
+    console.log(`shooter v${pkg.version}`);
+    break;
+  case 'help':
+  case '--help':
+  case '-h':
+    showHelp();
+    break;
+  default:
+    console.error(`Unknown command: ${command}\n`);
+    showHelp();
+    process.exit(1);
+}
+
+// ── start ───────────────────────────────────────────────────────────
 
 function startServer() {
-	const serverEntry = path.join(PKG_ROOT, 'server.ts');
-
-	if (!fs.existsSync(serverEntry)) {
-		console.error('Error: server.ts not found at', serverEntry);
-		console.error('The Shooter package may not be installed correctly.');
-		process.exit(1);
-	}
-
-	// Spawn tsx with the server entry point, inheriting stdio so the
-	// user sees logs directly. Run from PKG_ROOT so relative imports
-	// (like ./build/handler.js) resolve correctly.
-	const child = spawn(
-		process.execPath,
-		['--import', 'tsx', serverEntry, ...args.slice(1)],
-		{
-			cwd: PKG_ROOT,
-			stdio: 'inherit',
-			env: {
-				...process.env,
-				// If the user has a .env in their cwd, dotenv in server.ts
-				// will pick it up. But also hint where the package lives so
-				// the server can find its own assets if needed.
-				SHOOTER_PKG_ROOT: PKG_ROOT,
-			},
-		}
-	);
-
-	child.on('error', (err) => {
-		console.error('Failed to start Shooter server:', err.message);
-		process.exit(1);
-	});
-
-	child.on('exit', (code) => {
-		process.exit(code ?? 0);
-	});
-
-	// Forward signals to the child so graceful shutdown works
-	for (const sig of ['SIGTERM', 'SIGINT', 'SIGHUP']) {
-		process.on(sig, () => child.kill(sig));
-	}
+  const serverEntry = path.join(PKG_ROOT, 'server.ts');
+
+  if (!fs.existsSync(serverEntry)) {
+    console.error('Error: server.ts not found at', serverEntry);
+    console.error('The Shooter package may not be installed correctly.');
+    process.exit(1);
+  }
+
+  // Check if already running
+  const existingPid = readPid();
+  if (existingPid) {
+    console.log(`Shooter is already running (PID ${existingPid}).`);
+    console.log('Run "shooter stop" first, or "shooter status" for details.');
+    process.exit(0);
+  }
+
+  const child = spawn(process.execPath, ['--import', 'tsx', serverEntry, ...args.slice(1)], {
+    cwd: PKG_ROOT,
+    stdio: 'inherit',
+    env: {
+      ...process.env,
+      SHOOTER_PKG_ROOT: PKG_ROOT,
+      SHOOTER_HOME,
+    },
+  });
+
+  // Write PID file for status/stop commands
+  writePid(child.pid);
+
+  child.on('error', (err) => {
+    removePid();
+    console.error('Failed to start Shooter server:', err.message);
+    process.exit(1);
+  });
+
+  child.on('exit', (code, signal) => {
+    removePid();
+    if (signal) {
+      process.exit(128 + (signalCode(signal) || 1));
+    }
+    process.exit(code ?? 1);
+  });
+
+  // Forward signals to the child so graceful shutdown works
+  for (const sig of ['SIGTERM', 'SIGINT', 'SIGHUP']) {
+    process.on(sig, () => child.kill(sig));
+  }
 }
 
+// ── stop ────────────────────────────────────────────────────────────
+
+function stopServer() {
+  const pid = readPid();
+  if (!pid) {
+    console.log('Shooter is not running.');
+    process.exit(0);
+  }
+
+  console.log(`Stopping Shooter (PID ${pid})...`);
+  try {
+    process.kill(pid, 'SIGTERM');
+    // Wait briefly for clean shutdown
+    let waited = 0;
+    const check = setInterval(() => {
+      try {
+        process.kill(pid, 0);
+        waited += 100;
+        if (waited > 5000) {
+          clearInterval(check);
+          console.log('Force-killing...');
+          try { process.kill(pid, 'SIGKILL'); } catch {}
+          removePid();
+          console.log('Shooter stopped.');
+        }
+      } catch {
+        clearInterval(check);
+        removePid();
+        console.log('Shooter stopped.');
+      }
+    }, 100);
+  } catch (err) {
+    removePid();
+    console.log('Shooter is not running (stale PID file removed).');
+  }
+}
+
+// ── status ──────────────────────────────────────────────────────────
+
+function resolvePort() {
+  // Read PORT from ~/.shooter/.env if not in environment
+  if (process.env.PORT) return process.env.PORT;
+  const envFile = path.join(SHOOTER_HOME, '.env');
+  try {
+    const contents = fs.readFileSync(envFile, 'utf8');
+    const match = contents.match(/^PORT=(\d+)/m);
+    if (match) return match[1];
+  } catch {}
+  return DEFAULT_PORT;
+}
+
+function showStatus() {
+  const pid = readPid();
+  const port = resolvePort();
+  const autostartEnabled = isAutostartInstalled();
+
+  if (pid) {
+    console.log(`Shooter is running`);
+    console.log(`  PID:        ${pid}`);
+    console.log(`  URL:        http://localhost:${port}`);
+    console.log(`  Autostart:  ${autostartEnabled ? 'enabled' : 'disabled'}`);
+    console.log(`  Logs:       ${LOG_FILE}`);
+    console.log(`  Home:       ${SHOOTER_HOME}`);
+  } else {
+    console.log('Shooter is not running.');
+    console.log(`  Autostart:  ${autostartEnabled ? 'enabled' : 'disabled'}`);
+    console.log(`  Home:       ${SHOOTER_HOME}`);
+    console.log('\nRun "shooter start" to start the server.');
+  }
+}
+
+// ── autostart ───────────────────────────────────────────────────────
+
+function isAutostartInstalled() {
+  if (os.platform() === 'darwin') {
+    return fs.existsSync(LAUNCHD_PLIST);
+  } else if (os.platform() === 'linux') {
+    return fs.existsSync(SYSTEMD_UNIT);
+  }
+  return false;
+}
+
+function manageAutostart(action) {
+  if (!action || (action !== 'on' && action !== 'off')) {
+    const status = isAutostartInstalled() ? 'enabled' : 'disabled';
+    console.log(`Autostart is currently ${status}.`);
+    console.log('\nUsage:');
+    console.log('  shooter autostart on    Enable autostart on login');
+    console.log('  shooter autostart off   Disable autostart');
+    return;
+  }
+
+  if (action === 'on') {
+    enableAutostart();
+  } else {
+    disableAutostart();
+  }
+}
+
+function enableAutostart() {
+  const platform = os.platform();
+
+  if (platform === 'darwin') {
+    enableLaunchAgent();
+  } else if (platform === 'linux') {
+    enableSystemdUnit();
+  } else {
+    console.error(`Autostart is not supported on ${platform}.`);
+    console.log('You can add "shooter start" to your system startup scripts manually.');
+    process.exit(1);
+  }
+}
+
+function disableAutostart() {
+  const platform = os.platform();
+
+  if (platform === 'darwin') {
+    disableLaunchAgent();
+  } else if (platform === 'linux') {
+    disableSystemdUnit();
+  } else {
+    console.error(`Autostart is not supported on ${platform}.`);
+    process.exit(1);
+  }
+}
+
+// ── macOS LaunchAgent ───────────────────────────────────────────────
+
+function enableLaunchAgent() {
+  const shooterBin = resolveShooterBin();
+  const nodeBin = process.execPath;
+
+  const plist = `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN"
+  "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+  <key>Label</key>
+  <string>${LAUNCHD_LABEL}</string>
+  <key>ProgramArguments</key>
+  <array>
+    <string>${nodeBin}</string>
+    <string>${shooterBin}</string>
+    <string>start</string>
+  </array>
+  <key>WorkingDirectory</key>
+  <string>${PKG_ROOT}</string>
+  <key>EnvironmentVariables</key>
+  <dict>
+    <key>PATH</key>
+    <string>${path.dirname(nodeBin)}:/usr/local/bin:/usr/bin:/bin</string>
+    <key>SHOOTER_HOME</key>
+    <string>${SHOOTER_HOME}</string>
+  </dict>
+  <key>KeepAlive</key>
+  <dict>
+    <key>SuccessfulExit</key>
+    <false/>
+  </dict>
+  <key>RunAtLoad</key>
+  <true/>
+  <key>StandardOutPath</key>
+  <string>${LOG_FILE}</string>
+  <key>StandardErrorPath</key>
+  <string>${LOG_FILE}</string>
+  <key>ThrottleInterval</key>
+  <integer>10</integer>
+</dict>
+</plist>`;
+
+  // Ensure directories exist
+  fs.mkdirSync(path.dirname(LAUNCHD_PLIST), { recursive: true });
+  fs.mkdirSync(LOG_DIR, { recursive: true });
+
+  // Unload existing if present
+  if (fs.existsSync(LAUNCHD_PLIST)) {
+    try { execSync(`launchctl unload "${LAUNCHD_PLIST}" 2>/dev/null`); } catch {}
+  }
+
+  fs.writeFileSync(LAUNCHD_PLIST, plist);
+  execSync(`launchctl load "${LAUNCHD_PLIST}"`);
+
+  console.log('Autostart enabled (macOS LaunchAgent).');
+  console.log(`  Plist:  ${LAUNCHD_PLIST}`);
+  console.log(`  Logs:   ${LOG_FILE}`);
+  console.log('\nShooter will start automatically on login.');
+  console.log('To start now: shooter start');
+}
+
+function disableLaunchAgent() {
+  if (!fs.existsSync(LAUNCHD_PLIST)) {
+    console.log('Autostart is not enabled.');
+    return;
+  }
+
+  try { execSync(`launchctl unload "${LAUNCHD_PLIST}" 2>/dev/null`); } catch {}
+  fs.unlinkSync(LAUNCHD_PLIST);
+  console.log('Autostart disabled. LaunchAgent removed.');
+}
+
+// ── Linux systemd user unit ─────────────────────────────────────────
+
+function enableSystemdUnit() {
+  const shooterBin = resolveShooterBin();
+
+  const unit = `[Unit]
+Description=Shooter — Mobile dev notifications & remote terminal
+After=network.target
+
+[Service]
+Type=simple
+ExecStart=${process.execPath} ${shooterBin} start
+WorkingDirectory=${PKG_ROOT}
+Environment=SHOOTER_HOME=${SHOOTER_HOME}
+Restart=on-failure
+RestartSec=10
+
+[Install]
+WantedBy=default.target
+`;
+
+  fs.mkdirSync(path.dirname(SYSTEMD_UNIT), { recursive: true });
+  fs.mkdirSync(LOG_DIR, { recursive: true });
+  fs.writeFileSync(SYSTEMD_UNIT, unit);
+
+  execSync('systemctl --user daemon-reload');
+  execSync('systemctl --user enable shooter.service');
+  execSync('systemctl --user start shooter.service');
+
+  console.log('Autostart enabled (systemd user unit).');
+  console.log(`  Unit:   ${SYSTEMD_UNIT}`);
+  console.log(`  Logs:   journalctl --user -u shooter.service`);
+  console.log('\nShooter is now running and will start automatically on login.');
+}
+
+function disableSystemdUnit() {
+  if (!fs.existsSync(SYSTEMD_UNIT)) {
+    console.log('Autostart is not enabled.');
+    return;
+  }
+
+  try { execSync('systemctl --user stop shooter.service 2>/dev/null'); } catch {}
+  try { execSync('systemctl --user disable shooter.service 2>/dev/null'); } catch {}
+  fs.unlinkSync(SYSTEMD_UNIT);
+  try { execSync('systemctl --user daemon-reload'); } catch {}
+  console.log('Autostart disabled. systemd unit removed.');
+}
+
+// ── logs ────────────────────────────────────────────────────────────
+
+function showLogs() {
+  const platform = os.platform();
+
+  // For systemd on Linux, use journalctl
+  if (platform === 'linux' && fs.existsSync(SYSTEMD_UNIT)) {
+    const child = spawn('journalctl', ['--user', '-u', 'shooter.service', '-f', '--no-pager', '-n', '50'], {
+      stdio: 'inherit',
+    });
+    child.on('exit', (code) => process.exit(code ?? 0));
+    for (const sig of ['SIGTERM', 'SIGINT']) {
+      process.on(sig, () => child.kill(sig));
+    }
+    return;
+  }
+
+  // For macOS LaunchAgent or manual runs, tail the log file
+  if (!fs.existsSync(LOG_FILE)) {
+    console.log('No log file found.');
+    console.log(`Expected at: ${LOG_FILE}`);
+    console.log('\nLogs are only written when running via autostart.');
+    console.log('For foreground runs, logs are printed directly to the terminal.');
+    return;
+  }
+
+  const child = spawn('tail', ['-f', '-n', '50', LOG_FILE], {
+    stdio: 'inherit',
+  });
+  child.on('exit', (code) => process.exit(code ?? 0));
+  for (const sig of ['SIGTERM', 'SIGINT']) {
+    process.on(sig, () => child.kill(sig));
+  }
+}
+
+// ── setup ───────────────────────────────────────────────────────────
+
 function runSetup() {
-	const setupScript = path.join(PKG_ROOT, 'scripts', 'setup.cjs');
+  const setupScript = path.join(PKG_ROOT, 'scripts', 'setup.cjs');
 
-	if (!fs.existsSync(setupScript)) {
-		console.error('Error: Setup wizard not found at', setupScript);
-		console.error('The setup script (scripts/setup.cjs) has not been created yet.');
-		process.exit(1);
-	}
+  if (!fs.existsSync(setupScript)) {
+    console.error('Error: Setup wizard not found at', setupScript);
+    console.error('The setup script (scripts/setup.cjs) has not been created yet.');
+    process.exit(1);
+  }
 
-	const child = spawn(process.execPath, [setupScript, ...args.slice(1)], {
-		cwd: process.cwd(),
-		stdio: 'inherit',
-		env: {
-			...process.env,
-			SHOOTER_PKG_ROOT: PKG_ROOT,
-		},
-	});
+  const child = spawn(process.execPath, [setupScript, ...args.slice(1)], {
+    cwd: process.cwd(),
+    stdio: 'inherit',
+    env: {
+      ...process.env,
+      SHOOTER_PKG_ROOT: PKG_ROOT,
+    },
+  });
 
-	child.on('error', (err) => {
-		console.error('Failed to run setup wizard:', err.message);
-		process.exit(1);
-	});
+  child.on('error', (err) => {
+    console.error('Failed to run setup wizard:', err.message);
+    process.exit(1);
+  });
 
-	child.on('exit', (code) => {
-		process.exit(code ?? 0);
-	});
+  child.on('exit', (code, signal) => {
+    if (signal) {
+      process.exit(128 + (signalCode(signal) || 1));
+    }
+    process.exit(code ?? 1);
+  });
 }
 
+// ── help ────────────────────────────────────────────────────────────
+
 function showHelp() {
-	console.log(`
+  console.log(
+    `
 Shooter v${pkg.version}
 
 Usage: shooter [command]
 
 Commands:
-  start     Start the Shooter server (default)
-  setup     Run the interactive setup wizard
-  version   Show version number
-  help      Show this help message
+  start            Start the server (default)
+  stop             Stop the running server
+  status           Show server status
+  autostart on     Start automatically on login (macOS/Linux)
+  autostart off    Disable autostart
+  logs             Tail server logs
+  setup            Run the interactive setup wizard
+  version          Show version number
+  help             Show this help message
 
 Examples:
-  shooter              Start the server
-  shooter start        Start the server (explicit)
-  shooter setup        Run setup wizard
-  shooter --version    Show version
-`.trim());
+  shooter                  Start the server
+  shooter status           Check if server is running
+  shooter autostart on     Enable autostart on login
+  shooter logs             Follow server logs
+  shooter setup            Configure credentials
+`.trim()
+  );
+}
+
+// ── Resolve shooter binary path ─────────────────────────────────────
+
+function resolveShooterBin() {
+  // Prefer the global symlink if it exists
+  const globalBin = path.join(os.homedir(), '.local', 'bin', 'shooter');
+  if (fs.existsSync(globalBin)) return globalBin;
+  // Fall back to this script
+  return path.join(PKG_ROOT, 'bin', 'shooter.cjs');
 }