import { terminalManager } from '../terminal-manager.js';
import { commandManager } from '../command-manager.js';
import { StartProcessArgsSchema, ReadProcessOutputArgsSchema, InteractWithProcessArgsSchema, ForceTerminateArgsSchema, ListSessionsArgsSchema } from './schemas.js';
import { capture } from "../utils/capture.js";
import { ServerResult } from '../types.js';
import { analyzeProcessState, cleanProcessOutput, formatProcessStateMessage, ProcessState } from '../utils/process-detection.js';
import * as os from 'os';
import { configManager } from '../config-manager.js';
import { spawn } from 'child_process';
import fs from 'fs/promises';
import path from 'path';
import { fileURLToPath } from 'url';
// Get the directory where the MCP is installed (for ES module imports)
const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);
const mcpRoot = path.resolve(__dirname, '..', '..');
// Track virtual Node sessions (PIDs that are actually Node fallback sessions)
const virtualNodeSessions = new Map<number, { timeout_ms: number }>();
let virtualPidCounter = -1000; // Use negative PIDs for virtual sessions
/**
* Execute Node.js code via temp file (fallback when Python unavailable)
* Creates temp .mjs file in MCP directory for ES module import access
*/
async function executeNodeCode(code: string, timeout_ms: number = 30000): Promise<ServerResult> {
const tempFile = path.join(mcpRoot, `.mcp-exec-${Date.now()}-${Math.random().toString(36).slice(2)}.mjs`);
try {
await fs.writeFile(tempFile, code, 'utf8');
const result = await new Promise<{ stdout: string; stderr: string; exitCode: number }>((resolve) => {
const proc = spawn(process.execPath, [tempFile], {
cwd: mcpRoot,
timeout: timeout_ms
});
let stdout = '';
let stderr = '';
proc.stdout.on('data', (data) => {
stdout += data.toString();
});
proc.stderr.on('data', (data) => {
stderr += data.toString();
});
proc.on('close', (exitCode) => {
resolve({ stdout, stderr, exitCode: exitCode ?? 1 });
});
proc.on('error', (err) => {
resolve({ stdout, stderr: stderr + '\n' + err.message, exitCode: 1 });
});
});
// Clean up temp file
await fs.unlink(tempFile).catch(() => {});
if (result.exitCode !== 0) {
return {
content: [{
type: "text",
text: `Execution failed (exit code ${result.exitCode}):\n${result.stderr}\n${result.stdout}`
}],
isError: true
};
}
return {
content: [{
type: "text",
text: result.stdout || '(no output)'
}]
};
} catch (error) {
// Clean up temp file on error
await fs.unlink(tempFile).catch(() => {});
return {
content: [{
type: "text",
text: `Failed to execute Node.js code: ${error instanceof Error ? error.message : String(error)}`
}],
isError: true
};
}
}
/**
* Start a new process (renamed from execute_command)
* Includes early detection of process waiting for input
*/
export async function startProcess(args: unknown): Promise<ServerResult> {
const parsed = StartProcessArgsSchema.safeParse(args);
if (!parsed.success) {
capture('server_start_process_failed');
return {
content: [{ type: "text", text: `Error: Invalid arguments for start_process: ${parsed.error}` }],
isError: true,
};
}
try {
const commands = commandManager.extractCommands(parsed.data.command).join(', ');
capture('server_start_process', {
command: commandManager.getBaseCommand(parsed.data.command),
commands: commands
});
} catch (error) {
capture('server_start_process', {
command: commandManager.getBaseCommand(parsed.data.command)
});
}
const isAllowed = await commandManager.validateCommand(parsed.data.command);
if (!isAllowed) {
return {
content: [{ type: "text", text: `Error: Command not allowed: ${parsed.data.command}` }],
isError: true,
};
}
const commandToRun = parsed.data.command;
// Handle node:local - runs Node.js code directly on MCP server
if (commandToRun.trim() === 'node:local') {
const virtualPid = virtualPidCounter--;
virtualNodeSessions.set(virtualPid, { timeout_ms: parsed.data.timeout_ms || 30000 });
return {
content: [{
type: "text",
text: `Node.js session started with PID ${virtualPid} (MCP server execution)
IMPORTANT: Each interact_with_process call runs as a FRESH script.
State is NOT preserved between calls. Include ALL code in ONE call:
- imports, file reading, processing, and output together.
Available libraries:
- ExcelJS for Excel files: import ExcelJS from 'exceljs'
- All Node.js built-ins: fs, path, http, crypto, etc.
π Ready for code - send complete self-contained script via interact_with_process.`
}],
};
}
let shellUsed: string | undefined = parsed.data.shell;
if (!shellUsed) {
const config = await configManager.getConfig();
if (config.defaultShell) {
shellUsed = config.defaultShell;
} else {
const isWindows = os.platform() === 'win32';
if (isWindows && process.env.COMSPEC) {
shellUsed = process.env.COMSPEC;
} else if (!isWindows && process.env.SHELL) {
shellUsed = process.env.SHELL;
} else {
shellUsed = isWindows ? 'cmd.exe' : '/bin/sh';
}
}
}
const result = await terminalManager.executeCommand(
commandToRun,
parsed.data.timeout_ms,
shellUsed,
parsed.data.verbose_timing || false
);
if (result.pid === -1) {
return {
content: [{ type: "text", text: result.output }],
isError: true,
};
}
// Analyze the process state to detect if it's waiting for input
const processState = analyzeProcessState(result.output, result.pid);
let statusMessage = '';
if (processState.isWaitingForInput) {
statusMessage = `\nπ ${formatProcessStateMessage(processState, result.pid)}`;
} else if (processState.isFinished) {
statusMessage = `\nβ
${formatProcessStateMessage(processState, result.pid)}`;
} else if (result.isBlocked) {
statusMessage = '\nβ³ Process is running. Use read_process_output to get more output.';
}
// Add timing information if requested
let timingMessage = '';
if (result.timingInfo) {
timingMessage = formatTimingInfo(result.timingInfo);
}
return {
content: [{
type: "text",
text: `Process started with PID ${result.pid} (shell: ${shellUsed})\nInitial output:\n${result.output}${statusMessage}${timingMessage}`
}],
};
}
function formatTimingInfo(timing: any): string {
let msg = '\n\nπ Timing Information:\n';
msg += ` Exit Reason: ${timing.exitReason}\n`;
msg += ` Total Duration: ${timing.totalDurationMs}ms\n`;
if (timing.timeToFirstOutputMs !== undefined) {
msg += ` Time to First Output: ${timing.timeToFirstOutputMs}ms\n`;
}
if (timing.firstOutputTime && timing.lastOutputTime) {
msg += ` Output Window: ${timing.lastOutputTime - timing.firstOutputTime}ms\n`;
}
if (timing.outputEvents && timing.outputEvents.length > 0) {
msg += `\n Output Events (${timing.outputEvents.length} total):\n`;
timing.outputEvents.forEach((event: any, idx: number) => {
msg += ` [${idx + 1}] +${event.deltaMs}ms | ${event.source} | ${event.length}b`;
if (event.matchedPattern) {
msg += ` | π― ${event.matchedPattern}`;
}
msg += `\n "${event.snippet}"\n`;
});
}
return msg;
}
/**
* Read output from a running process with file-like pagination
* Supports offset/length parameters for controlled reading
*/
export async function readProcessOutput(args: unknown): Promise<ServerResult> {
const parsed = ReadProcessOutputArgsSchema.safeParse(args);
if (!parsed.success) {
return {
content: [{ type: "text", text: `Error: Invalid arguments for read_process_output: ${parsed.error}` }],
isError: true,
};
}
// Get default line limit from config
const config = await configManager.getConfig();
const defaultLength = config.fileReadLineLimit ?? 1000;
const {
pid,
timeout_ms = 5000,
offset = 0, // 0 = from last read, positive = absolute, negative = tail
length = defaultLength, // Default from config, same as file reading
verbose_timing = false
} = parsed.data;
// Timing telemetry
const startTime = Date.now();
// For active sessions with no new output yet, optionally wait for output
const session = terminalManager.getSession(pid);
if (session && offset === 0) {
// Wait for new output to arrive (only for "new output" reads, not absolute/tail)
const waitForOutput = (): Promise<void> => {
return new Promise((resolve) => {
// Check if there's already new output
const currentLines = terminalManager.getOutputLineCount(pid) || 0;
if (currentLines > session.lastReadIndex) {
resolve();
return;
}
let resolved = false;
let interval: NodeJS.Timeout | null = null;
let timeout: NodeJS.Timeout | null = null;
const cleanup = () => {
if (interval) clearInterval(interval);
if (timeout) clearTimeout(timeout);
};
const resolveOnce = () => {
if (resolved) return;
resolved = true;
cleanup();
resolve();
};
// Poll for new output
interval = setInterval(() => {
const newLineCount = terminalManager.getOutputLineCount(pid) || 0;
if (newLineCount > session.lastReadIndex) {
resolveOnce();
}
}, 50);
// Timeout
timeout = setTimeout(() => {
resolveOnce();
}, timeout_ms);
});
};
await waitForOutput();
}
// Read output with pagination
const result = terminalManager.readOutputPaginated(pid, offset, length);
if (!result) {
return {
content: [{ type: "text", text: `No session found for PID ${pid}` }],
isError: true,
};
}
// Join lines back into string
const output = result.lines.join('\n');
// Generate status message similar to file reading
let statusMessage = '';
if (offset < 0) {
// Tail read - show actual line range
const endLine = result.readFrom + result.readCount - 1;
statusMessage = `[Reading lines ${result.readFrom}-${endLine} (${result.readCount} lines, starting ${Math.abs(offset)} from end, total: ${result.totalLines} lines)]`;
} else if (offset === 0) {
// "New output" read
if (result.remaining > 0) {
statusMessage = `[Reading ${result.readCount} new lines from line ${result.readFrom} (total: ${result.totalLines} lines, ${result.remaining} remaining)]`;
} else {
statusMessage = `[Reading ${result.readCount} new lines (total: ${result.totalLines} lines)]`;
}
} else {
// Absolute position read
statusMessage = `[Reading ${result.readCount} lines from line ${result.readFrom} (total: ${result.totalLines} lines, ${result.remaining} remaining)]`;
}
// Add process state info
let processStateMessage = '';
if (result.isComplete) {
const runtimeStr = result.runtimeMs !== undefined
? ` (runtime: ${(result.runtimeMs / 1000).toFixed(2)}s)`
: '';
processStateMessage = `\nβ
Process completed with exit code ${result.exitCode}${runtimeStr}`;
} else if (session) {
// Analyze state for running processes
const fullOutput = session.outputLines.join('\n');
const processState = analyzeProcessState(fullOutput, pid);
if (processState.isWaitingForInput) {
processStateMessage = `\nπ ${formatProcessStateMessage(processState, pid)}`;
}
}
// Add timing information if requested
let timingMessage = '';
if (verbose_timing) {
const endTime = Date.now();
timingMessage = `\n\nπ Timing: ${endTime - startTime}ms`;
}
const responseText = output || '(No output in requested range)';
return {
content: [{
type: "text",
text: `${statusMessage}\n\n${responseText}${processStateMessage}${timingMessage}`
}],
};
}
/**
* Interact with a running process (renamed from send_input)
* Automatically detects when process is ready and returns output
*/
export async function interactWithProcess(args: unknown): Promise<ServerResult> {
const parsed = InteractWithProcessArgsSchema.safeParse(args);
if (!parsed.success) {
capture('server_interact_with_process_failed', {
error: 'Invalid arguments'
});
return {
content: [{ type: "text", text: `Error: Invalid arguments for interact_with_process: ${parsed.error}` }],
isError: true,
};
}
const {
pid,
input,
timeout_ms = 8000,
wait_for_prompt = true,
verbose_timing = false
} = parsed.data;
// Get config for output line limit
const config = await configManager.getConfig();
const maxOutputLines = config.fileReadLineLimit ?? 1000;
// Check if this is a virtual Node session (node:local)
if (virtualNodeSessions.has(pid)) {
const session = virtualNodeSessions.get(pid)!;
capture('server_interact_with_process_node_fallback', {
pid: pid,
inputLength: input.length
});
// Execute code via temp file approach
// Respect per-call timeout if provided, otherwise use session default
const effectiveTimeout = timeout_ms ?? session.timeout_ms;
return executeNodeCode(input, effectiveTimeout);
}
// Timing telemetry
const startTime = Date.now();
let firstOutputTime: number | undefined;
let lastOutputTime: number | undefined;
const outputEvents: any[] = [];
let exitReason: 'early_exit_quick_pattern' | 'early_exit_periodic_check' | 'process_finished' | 'timeout' | 'no_wait' = 'timeout';
try {
capture('server_interact_with_process', {
pid: pid,
inputLength: input.length
});
// Capture output snapshot BEFORE sending input
// This handles REPLs where output is appended to the prompt line
const outputSnapshot = terminalManager.captureOutputSnapshot(pid);
const success = terminalManager.sendInputToProcess(pid, input);
if (!success) {
return {
content: [{ type: "text", text: `Error: Failed to send input to process ${pid}. The process may have exited or doesn't accept input.` }],
isError: true,
};
}
// If not waiting for response, return immediately
if (!wait_for_prompt) {
exitReason = 'no_wait';
let timingMessage = '';
if (verbose_timing) {
const endTime = Date.now();
const timingInfo = {
startTime,
endTime,
totalDurationMs: endTime - startTime,
exitReason,
firstOutputTime,
lastOutputTime,
timeToFirstOutputMs: undefined,
outputEvents: undefined
};
timingMessage = formatTimingInfo(timingInfo);
}
return {
content: [{
type: "text",
text: `β
Input sent to process ${pid}. Use read_process_output to get the response.${timingMessage}`
}],
};
}
// Smart waiting with immediate and periodic detection
let output = "";
let processState: ProcessState | undefined;
let earlyExit = false;
// Quick prompt patterns for immediate detection
const quickPromptPatterns = />>>\s*$|>\s*$|\$\s*$|#\s*$/;
const waitForResponse = (): Promise<void> => {
return new Promise((resolve) => {
let resolved = false;
let attempts = 0;
const pollIntervalMs = 50; // Poll every 50ms for faster response
const maxAttempts = Math.ceil(timeout_ms / pollIntervalMs);
let interval: NodeJS.Timeout | null = null;
let lastOutputLength = 0; // Track output length to detect new output
let resolveOnce = () => {
if (resolved) return;
resolved = true;
if (interval) clearInterval(interval);
resolve();
};
// Fast-polling check - check every 50ms for quick responses
interval = setInterval(() => {
if (resolved) return;
// Use snapshot-based reading to handle REPL prompt line appending
const newOutput = outputSnapshot
? terminalManager.getOutputSinceSnapshot(pid, outputSnapshot)
: terminalManager.getNewOutput(pid);
if (newOutput && newOutput.length > lastOutputLength) {
const now = Date.now();
if (!firstOutputTime) firstOutputTime = now;
lastOutputTime = now;
if (verbose_timing) {
outputEvents.push({
timestamp: now,
deltaMs: now - startTime,
source: 'periodic_poll',
length: newOutput.length - lastOutputLength,
snippet: newOutput.slice(lastOutputLength, lastOutputLength + 50).replace(/\n/g, '\\n')
});
}
output = newOutput; // Replace with full output since snapshot
lastOutputLength = newOutput.length;
// Analyze current state
processState = analyzeProcessState(output, pid);
// Exit early if we detect the process is waiting for input
if (processState.isWaitingForInput) {
earlyExit = true;
exitReason = 'early_exit_periodic_check';
if (verbose_timing && outputEvents.length > 0) {
outputEvents[outputEvents.length - 1].matchedPattern = 'periodic_check';
}
resolveOnce();
return;
}
// Also exit if process finished
if (processState.isFinished) {
exitReason = 'process_finished';
resolveOnce();
return;
}
}
attempts++;
if (attempts >= maxAttempts) {
exitReason = 'timeout';
resolveOnce();
}
}, pollIntervalMs);
});
};
await waitForResponse();
// Clean and format output
let cleanOutput = cleanProcessOutput(output, input);
const timeoutReached = !earlyExit && !processState?.isFinished && !processState?.isWaitingForInput;
// Apply output line limit to prevent context overflow
let truncationMessage = '';
const outputLines = cleanOutput.split('\n');
if (outputLines.length > maxOutputLines) {
const truncatedLines = outputLines.slice(0, maxOutputLines);
cleanOutput = truncatedLines.join('\n');
const remainingLines = outputLines.length - maxOutputLines;
truncationMessage = `\n\nβ οΈ Output truncated: showing ${maxOutputLines} of ${outputLines.length} lines (${remainingLines} hidden). Use read_process_output with offset/length for full output.`;
}
// Determine final state
if (!processState) {
processState = analyzeProcessState(output, pid);
}
let statusMessage = '';
if (processState.isWaitingForInput) {
statusMessage = `\nπ ${formatProcessStateMessage(processState, pid)}`;
} else if (processState.isFinished) {
statusMessage = `\nβ
${formatProcessStateMessage(processState, pid)}`;
} else if (timeoutReached) {
statusMessage = '\nβ±οΈ Response may be incomplete (timeout reached)';
}
// Add timing information if requested
let timingMessage = '';
if (verbose_timing) {
const endTime = Date.now();
const timingInfo = {
startTime,
endTime,
totalDurationMs: endTime - startTime,
exitReason,
firstOutputTime,
lastOutputTime,
timeToFirstOutputMs: firstOutputTime ? firstOutputTime - startTime : undefined,
outputEvents: outputEvents.length > 0 ? outputEvents : undefined
};
timingMessage = formatTimingInfo(timingInfo);
}
if (cleanOutput.trim().length === 0 && !timeoutReached) {
return {
content: [{
type: "text",
text: `β
Input executed in process ${pid}.\nπ (No output produced)${statusMessage}${timingMessage}`
}],
};
}
// Format response with better structure and consistent emojis
let responseText = `β
Input executed in process ${pid}`;
if (cleanOutput && cleanOutput.trim().length > 0) {
responseText += `:\n\nπ€ Output:\n${cleanOutput}`;
} else {
responseText += `.\nπ (No output produced)`;
}
if (statusMessage) {
responseText += `\n\n${statusMessage}`;
}
if (truncationMessage) {
responseText += truncationMessage;
}
if (timingMessage) {
responseText += timingMessage;
}
return {
content: [{
type: "text",
text: responseText
}],
};
} catch (error) {
const errorMessage = error instanceof Error ? error.message : String(error);
capture('server_interact_with_process_error', {
error: errorMessage
});
return {
content: [{ type: "text", text: `Error interacting with process: ${errorMessage}` }],
isError: true,
};
}
}
/**
* Force terminate a process
*/
export async function forceTerminate(args: unknown): Promise<ServerResult> {
const parsed = ForceTerminateArgsSchema.safeParse(args);
if (!parsed.success) {
return {
content: [{ type: "text", text: `Error: Invalid arguments for force_terminate: ${parsed.error}` }],
isError: true,
};
}
const pid = parsed.data.pid;
// Handle virtual Node.js sessions (node:local)
if (virtualNodeSessions.has(pid)) {
virtualNodeSessions.delete(pid);
return {
content: [{
type: "text",
text: `Cleared virtual Node.js session ${pid}`
}],
};
}
const success = terminalManager.forceTerminate(pid);
return {
content: [{
type: "text",
text: success
? `Successfully initiated termination of session ${pid}`
: `No active session found for PID ${pid}`
}],
};
}
/**
* List active sessions
*/
export async function listSessions(): Promise<ServerResult> {
const sessions = terminalManager.listActiveSessions();
// Include virtual Node.js sessions
const virtualSessions = Array.from(virtualNodeSessions.entries()).map(([pid, session]) => ({
pid,
type: 'node:local',
timeout_ms: session.timeout_ms
}));
const realSessionsText = sessions.map(s =>
`PID: ${s.pid}, Blocked: ${s.isBlocked}, Runtime: ${Math.round(s.runtime / 1000)}s`
);
const virtualSessionsText = virtualSessions.map(s =>
`PID: ${s.pid} (node:local), Timeout: ${s.timeout_ms}ms`
);
const allSessions = [...realSessionsText, ...virtualSessionsText];
return {
content: [{
type: "text",
text: allSessions.length === 0
? 'No active sessions'
: allSessions.join('\n')
}],
};
}