#!/usr/bin/env node
import { parseArgs } from 'node:util';
import { startHttpServer } from './http.js';
import { startStdioServer } from './mcp.js';
import { logError } from './observability.js';
const { values } = parseArgs({
options: {
stdio: { type: 'boolean', default: false },
},
});
const isStdioMode = values.stdio;
let isShuttingDown = false;
const shutdownHandlerRef: { current?: (signal: string) => Promise<void> } = {};
function shouldAttemptShutdown(): boolean {
return !isShuttingDown && !isStdioMode && Boolean(shutdownHandlerRef.current);
}
function attemptShutdown(signal: string): void {
if (!shutdownHandlerRef.current) return;
isShuttingDown = true;
process.stderr.write('Attempting graceful shutdown...\n');
void shutdownHandlerRef.current(signal);
}
function handleFatalError(label: string, error: Error, signal: string): void {
logError(label, error);
process.stderr.write(`${label}: ${error.message}\n`);
if (shouldAttemptShutdown()) {
attemptShutdown(signal);
return;
}
process.exit(1);
}
process.on('uncaughtException', (error) => {
handleFatalError('Uncaught exception', error, 'UNCAUGHT_EXCEPTION');
});
process.on('unhandledRejection', (reason) => {
const error = reason instanceof Error ? reason : new Error(String(reason));
handleFatalError('Unhandled rejection', error, 'UNHANDLED_REJECTION');
});
try {
if (isStdioMode) {
await startStdioServer();
} else {
const { shutdown } = await startHttpServer();
shutdownHandlerRef.current = shutdown;
}
} catch (error: unknown) {
logError(
'Failed to start server',
error instanceof Error ? error : undefined
);
const message = error instanceof Error ? error.message : String(error);
process.stderr.write(`Failed to start server: ${message}\n`);
process.exit(1);
}
