@f5-sales-demo/xcsh 19.51.2

package/src/mcp/client.ts

@@ -0,0 +1,482 @@
+/**
+ * MCP Client.
+ *
+ * Handles connection initialization, tool listing, and tool calling.
+ */
+import * as path from "node:path";
+import * as url from "node:url";
+import { getProjectDir, logger, withTimeout } from "@f5xc-salesdemos/pi-utils";
+import { createHttpTransport } from "./transports/http";
+import { createStdioTransport } from "./transports/stdio";
+import type {
+	MCPGetPromptParams,
+	MCPGetPromptResult,
+	MCPHttpServerConfig,
+	MCPInitializeParams,
+	MCPInitializeResult,
+	MCPPrompt,
+	MCPPromptsListResult,
+	MCPRequestOptions,
+	MCPResource,
+	MCPResourceReadParams,
+	MCPResourceReadResult,
+	MCPResourceSubscribeParams,
+	MCPResourcesListResult,
+	MCPResourceTemplate,
+	MCPResourceTemplatesListResult,
+	MCPServerCapabilities,
+	MCPServerConfig,
+	MCPServerConnection,
+	MCPSseServerConfig,
+	MCPStdioServerConfig,
+	MCPToolCallParams,
+	MCPToolCallResult,
+	MCPToolDefinition,
+	MCPToolsListResult,
+	MCPTransport,
+} from "./types";
+
+/** MCP protocol version we support */
+const PROTOCOL_VERSION = "2025-03-26";
+
+/** Default connection timeout in ms */
+const CONNECTION_TIMEOUT_MS = 30_000;
+
+/** Client info sent during initialization */
+const CLIENT_INFO = {
+	name: "xcsh-coding-agent",
+	version: "1.0.0",
+};
+
+/**
+ * Default handler for standard MCP server-to-client requests.
+ * Handles `ping` and `roots/list`; rejects unknown methods with -32601.
+ * Reads getProjectDir() at call time so the root stays stable even if
+ * the process cwd changes during tool execution.
+ */
+async function defaultRequestHandler(method: string, _params: unknown): Promise<unknown> {
+	switch (method) {
+		case "ping":
+			return {};
+		case "roots/list": {
+			const cwd = getProjectDir();
+			return {
+				roots: [{ uri: url.pathToFileURL(cwd).href, name: path.basename(cwd) }],
+			};
+		}
+		default:
+			throw Object.assign(new Error(`Unsupported server request: ${method}`), { code: -32601 });
+	}
+}
+
+/**
+ * Create a transport for the given server config.
+ */
+async function createTransport(config: MCPServerConfig): Promise<MCPTransport> {
+	const serverType = config.type ?? "stdio";
+
+	switch (serverType) {
+		case "stdio":
+			return createStdioTransport(config as MCPStdioServerConfig);
+		case "http":
+		case "sse":
+			return createHttpTransport(config as MCPHttpServerConfig | MCPSseServerConfig);
+		default:
+			throw new Error(`Unknown server type: ${serverType}`);
+	}
+}
+
+/**
+ * Initialize connection with MCP server.
+ */
+async function initializeConnection(
+	transport: MCPTransport,
+	options?: {
+		signal?: AbortSignal;
+		/** Called after the initialize response (which sets the session ID) but before notifications/initialized. */
+		onInitialized?: () => void | Promise<void>;
+	},
+): Promise<MCPInitializeResult> {
+	const params: MCPInitializeParams = {
+		protocolVersion: PROTOCOL_VERSION,
+		capabilities: {
+			roots: { listChanged: false },
+		},
+		clientInfo: CLIENT_INFO,
+	};
+
+	const result = await transport.request<MCPInitializeResult>(
+		"initialize",
+		params as unknown as Record<string, unknown>,
+		{ signal: options?.signal },
+	);
+
+	if (options?.signal?.aborted) {
+		throw options.signal.reason instanceof Error ? options.signal.reason : new Error("Aborted");
+	}
+
+	// Hook point: the transport now has the session ID from the initialize response.
+	// For HTTP, this is the moment to open the SSE stream so server-to-client requests
+	// triggered by notifications/initialized (e.g. roots/list) can be delivered.
+	await options?.onInitialized?.();
+
+	// Send initialized notification
+	await transport.notify("notifications/initialized");
+
+	return result;
+}
+
+/**
+ * Connect to an MCP server.
+ * Has a 30 second timeout to prevent blocking startup.
+ */
+export async function connectToServer(
+	name: string,
+	config: MCPServerConfig,
+	options?: {
+		signal?: AbortSignal;
+		onNotification?: (method: string, params: unknown) => void;
+		onRequest?: (method: string, params: unknown) => Promise<unknown>;
+	},
+): Promise<MCPServerConnection> {
+	const timeoutMs = config.timeout ?? CONNECTION_TIMEOUT_MS;
+	let transport: MCPTransport | undefined;
+
+	const connect = async (): Promise<MCPServerConnection> => {
+		transport = await createTransport(config);
+		if (options?.onNotification) {
+			transport.onNotification = options.onNotification;
+		}
+
+		// Always handle standard MCP server-to-client requests (ping, roots/list).
+		// The initialize request declares roots capability, so we must respond to
+		// roots/list — even for short-lived test connections.
+		transport.onRequest = options?.onRequest ?? defaultRequestHandler;
+
+		try {
+			const initResult = await initializeConnection(transport, {
+				signal: options?.signal,
+				async onInitialized() {
+					// Open the SSE stream before sending initialized, so server-to-client
+					// requests triggered by on_initialized (e.g. roots/list) are delivered.
+					if ("startSSEListener" in transport! && typeof transport!.startSSEListener === "function") {
+						await (transport as { startSSEListener(): Promise<void> }).startSSEListener();
+					}
+				},
+			});
+
+			return {
+				name,
+				config,
+				transport,
+				serverInfo: initResult.serverInfo,
+				capabilities: initResult.capabilities,
+				instructions: initResult.instructions,
+			};
+		} catch (error) {
+			await transport.close();
+			throw error;
+		}
+	};
+
+	try {
+		return await withTimeout(
+			connect(),
+			timeoutMs,
+			`Connection to MCP server "${name}" timed out after ${timeoutMs}ms`,
+			options?.signal,
+		);
+	} catch (error) {
+		// If withTimeout rejected (timeout/abort) while connect() was still pending,
+		// the transport may be alive with an open SSE listener. Close it.
+		if (transport) {
+			void transport.close().catch(() => {});
+		}
+		throw error;
+	}
+}
+
+/**
+ * List tools from a connected server.
+ */
+export async function listTools(
+	connection: MCPServerConnection,
+	options?: { signal?: AbortSignal },
+): Promise<MCPToolDefinition[]> {
+	// Check if server supports tools
+	if (!connection.capabilities.tools) {
+		return [];
+	}
+
+	// Return cached tools if available
+	if (connection.tools) {
+		return connection.tools;
+	}
+
+	const allTools: MCPToolDefinition[] = [];
+	let cursor: string | undefined;
+
+	do {
+		const params: Record<string, unknown> = {};
+		if (cursor) {
+			params.cursor = cursor;
+		}
+
+		const result = await connection.transport.request<MCPToolsListResult>("tools/list", params, options);
+		allTools.push(...result.tools);
+		cursor = result.nextCursor;
+	} while (cursor);
+
+	// Cache tools
+	connection.tools = allTools;
+
+	return allTools;
+}
+
+/**
+ * Call a tool on a connected server.
+ */
+export async function callTool(
+	connection: MCPServerConnection,
+	toolName: string,
+	args: Record<string, unknown> = {},
+	options?: MCPRequestOptions,
+): Promise<MCPToolCallResult> {
+	const params: MCPToolCallParams = {
+		name: toolName,
+		arguments: args,
+	};
+
+	return connection.transport.request<MCPToolCallResult>(
+		"tools/call",
+		params as unknown as Record<string, unknown>,
+		options,
+	);
+}
+
+/**
+ * Disconnect from a server.
+ */
+export async function disconnectServer(connection: MCPServerConnection): Promise<void> {
+	await connection.transport.close();
+}
+
+/**
+ * Check if a server supports tools.
+ */
+export function serverSupportsTools(capabilities: MCPServerCapabilities): boolean {
+	return capabilities.tools !== undefined;
+}
+
+/**
+ * List resources from a connected server.
+ */
+export async function listResources(
+	connection: MCPServerConnection,
+	options?: { signal?: AbortSignal },
+): Promise<MCPResource[]> {
+	if (!connection.capabilities.resources) {
+		return [];
+	}
+
+	if (connection.resources) {
+		return connection.resources;
+	}
+
+	const allResources: MCPResource[] = [];
+	let cursor: string | undefined;
+
+	do {
+		const params: Record<string, unknown> = {};
+		if (cursor) {
+			params.cursor = cursor;
+		}
+
+		const result = await connection.transport.request<MCPResourcesListResult>("resources/list", params, options);
+		allResources.push(...result.resources);
+		cursor = result.nextCursor;
+	} while (cursor);
+
+	connection.resources = allResources;
+	return allResources;
+}
+
+/**
+ * List resource templates from a connected server.
+ */
+export async function listResourceTemplates(
+	connection: MCPServerConnection,
+	options?: { signal?: AbortSignal },
+): Promise<MCPResourceTemplate[]> {
+	if (!connection.capabilities.resources) {
+		return [];
+	}
+
+	if (connection.resourceTemplates) {
+		return connection.resourceTemplates;
+	}
+
+	const allTemplates: MCPResourceTemplate[] = [];
+	let cursor: string | undefined;
+
+	do {
+		const params: Record<string, unknown> = {};
+		if (cursor) {
+			params.cursor = cursor;
+		}
+
+		const result = await connection.transport.request<MCPResourceTemplatesListResult>(
+			"resources/templates/list",
+			params,
+			options,
+		);
+		allTemplates.push(...result.resourceTemplates);
+		cursor = result.nextCursor;
+	} while (cursor);
+
+	connection.resourceTemplates = allTemplates;
+	return allTemplates;
+}
+
+/**
+ * Read a resource from a connected server.
+ */
+export async function readResource(
+	connection: MCPServerConnection,
+	uri: string,
+	options?: MCPRequestOptions,
+): Promise<MCPResourceReadResult> {
+	const params: MCPResourceReadParams = { uri };
+	return connection.transport.request<MCPResourceReadResult>(
+		"resources/read",
+		params as unknown as Record<string, unknown>,
+		options,
+	);
+}
+
+/**
+ * Subscribe to resource update notifications.
+ */
+export async function subscribeToResources(
+	connection: MCPServerConnection,
+	uris: string[],
+	options?: MCPRequestOptions,
+): Promise<void> {
+	if (uris.length === 0 || !connection.capabilities.resources?.subscribe) return;
+	const results = await Promise.allSettled(
+		uris.map(uri => {
+			const params: MCPResourceSubscribeParams = { uri };
+			return connection.transport.request(
+				"resources/subscribe",
+				params as unknown as Record<string, unknown>,
+				options,
+			);
+		}),
+	);
+	for (const result of results) {
+		if (result.status === "rejected") {
+			logger.warn("Failed to subscribe to MCP resource", { error: result.reason });
+		}
+	}
+}
+
+/**
+ * Unsubscribe from resource update notifications.
+ */
+export async function unsubscribeFromResources(
+	connection: MCPServerConnection,
+	uris: string[],
+	options?: MCPRequestOptions,
+): Promise<void> {
+	if (uris.length === 0 || !connection.capabilities.resources?.subscribe) return;
+	const results = await Promise.allSettled(
+		uris.map(uri => {
+			const params: MCPResourceSubscribeParams = { uri };
+			return connection.transport.request(
+				"resources/unsubscribe",
+				params as unknown as Record<string, unknown>,
+				options,
+			);
+		}),
+	);
+	for (const result of results) {
+		if (result.status === "rejected") {
+			logger.warn("Failed to unsubscribe from MCP resource", { error: result.reason });
+		}
+	}
+}
+
+/**
+ * Check if a server supports resource subscriptions.
+ */
+export function serverSupportsResourceSubscriptions(capabilities: MCPServerCapabilities): boolean {
+	return capabilities.resources?.subscribe === true;
+}
+
+/**
+ * Check if a server supports resources.
+ */
+export function serverSupportsResources(capabilities: MCPServerCapabilities): boolean {
+	return capabilities.resources !== undefined;
+}
+
+/**
+ * List prompts from a connected server.
+ */
+export async function listPrompts(
+	connection: MCPServerConnection,
+	options?: { signal?: AbortSignal },
+): Promise<MCPPrompt[]> {
+	if (!connection.capabilities.prompts) {
+		return [];
+	}
+
+	if (connection.prompts) {
+		return connection.prompts;
+	}
+
+	const allPrompts: MCPPrompt[] = [];
+	let cursor: string | undefined;
+
+	do {
+		const params: Record<string, unknown> = {};
+		if (cursor) {
+			params.cursor = cursor;
+		}
+
+		const result = await connection.transport.request<MCPPromptsListResult>("prompts/list", params, options);
+		allPrompts.push(...result.prompts);
+		cursor = result.nextCursor;
+	} while (cursor);
+
+	connection.prompts = allPrompts;
+	return allPrompts;
+}
+
+/**
+ * Get a specific prompt from a connected server.
+ */
+export async function getPrompt(
+	connection: MCPServerConnection,
+	name: string,
+	args?: Record<string, string>,
+	options?: MCPRequestOptions,
+): Promise<MCPGetPromptResult> {
+	const params: MCPGetPromptParams = { name };
+	if (args && Object.keys(args).length > 0) {
+		params.arguments = args;
+	}
+
+	return connection.transport.request<MCPGetPromptResult>(
+		"prompts/get",
+		params as unknown as Record<string, unknown>,
+		options,
+	);
+}
+
+/**
+ * Check if a server supports prompts.
+ */
+export function serverSupportsPrompts(capabilities: MCPServerCapabilities): boolean {
+	return capabilities.prompts !== undefined;
+}

package/src/mcp/config-writer.ts

@@ -0,0 +1,225 @@
+/**
+ * MCP Configuration File Writer
+ *
+ * Utilities for reading/writing .xcsh/mcp.json files at user or project level.
+ */
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { isEnoent } from "@f5xc-salesdemos/pi-utils";
+import { invalidate as invalidateFsCache } from "../capability/fs";
+
+import { validateServerConfig } from "./config";
+import { MCP_CONFIG_SCHEMA_URL, type MCPConfigFile, type MCPServerConfig } from "./types";
+
+function withSchema(config: MCPConfigFile): MCPConfigFile {
+	return {
+		$schema: config.$schema ?? MCP_CONFIG_SCHEMA_URL,
+		...config,
+	};
+}
+
+/**
+ * Read an MCP config file.
+ * Returns empty config if file doesn't exist.
+ */
+export async function readMCPConfigFile(filePath: string): Promise<MCPConfigFile> {
+	try {
+		const content = await fs.promises.readFile(filePath, "utf-8");
+		const parsed = JSON.parse(content) as MCPConfigFile;
+		return parsed;
+	} catch (error) {
+		if (isEnoent(error)) {
+			// File doesn't exist, return empty config
+			return { mcpServers: {} };
+		}
+		throw error;
+	}
+}
+
+/**
+ * Write an MCP config file atomically.
+ * Creates parent directories if they don't exist.
+ */
+export async function writeMCPConfigFile(filePath: string, config: MCPConfigFile): Promise<void> {
+	// Ensure parent directory exists
+	const dir = path.dirname(filePath);
+	await fs.promises.mkdir(dir, { recursive: true, mode: 0o700 });
+
+	// Write to temp file first (atomic write)
+	const tmpPath = `${filePath}.tmp`;
+	const content = JSON.stringify(withSchema(config), null, 2);
+	await fs.promises.writeFile(tmpPath, content, { encoding: "utf-8", mode: 0o600 });
+
+	// Rename to final path (atomic on most systems)
+	await fs.promises.rename(tmpPath, filePath);
+	// Invalidate the capability fs cache so subsequent reads see the new content
+	invalidateFsCache(filePath);
+}
+
+/**
+ * Validate server name.
+ * @returns Error message if invalid, undefined if valid
+ */
+export function validateServerName(name: string): string | undefined {
+	if (!name) {
+		return "Server name cannot be empty";
+	}
+	if (name.length > 100) {
+		return "Server name is too long (max 100 characters)";
+	}
+	// Check for invalid characters (only allow alphanumeric, dash, underscore, dot)
+	if (!/^[a-zA-Z0-9_.-]+$/.test(name)) {
+		return "Server name can only contain letters, numbers, dash, underscore, and dot";
+	}
+	return undefined;
+}
+
+/**
+ * Add an MCP server to a config file.
+ * Validates the config before writing.
+ *
+ * @throws Error if server name already exists or validation fails
+ */
+export async function addMCPServer(filePath: string, name: string, config: MCPServerConfig): Promise<void> {
+	// Validate server name
+	const nameError = validateServerName(name);
+	if (nameError) {
+		throw new Error(nameError);
+	}
+
+	// Validate the config
+	const errors = validateServerConfig(name, config);
+	if (errors.length > 0) {
+		throw new Error(`Invalid server config: ${errors.join("; ")}`);
+	}
+
+	// Read existing config
+	const existing = await readMCPConfigFile(filePath);
+
+	// Check for duplicate name
+	if (existing.mcpServers?.[name]) {
+		throw new Error(`Server "${name}" already exists in ${filePath}`);
+	}
+
+	// Add server
+	const updated: MCPConfigFile = {
+		...existing,
+		mcpServers: {
+			...existing.mcpServers,
+			[name]: config,
+		},
+	};
+
+	// Write back
+	await writeMCPConfigFile(filePath, updated);
+}
+
+/**
+ * Update an existing MCP server in a config file.
+ * If the server doesn't exist, this will add it.
+ *
+ * @throws Error if validation fails
+ */
+export async function updateMCPServer(filePath: string, name: string, config: MCPServerConfig): Promise<void> {
+	// Validate server name
+	const nameError = validateServerName(name);
+	if (nameError) {
+		throw new Error(nameError);
+	}
+
+	// Validate the config
+	const errors = validateServerConfig(name, config);
+	if (errors.length > 0) {
+		throw new Error(`Invalid server config: ${errors.join("; ")}`);
+	}
+
+	// Read existing config
+	const existing = await readMCPConfigFile(filePath);
+
+	// Update server
+	const updated: MCPConfigFile = {
+		...existing,
+		mcpServers: {
+			...existing.mcpServers,
+			[name]: config,
+		},
+	};
+
+	// Write back
+	await writeMCPConfigFile(filePath, updated);
+}
+
+/**
+ * Remove an MCP server from a config file.
+ *
+ * @throws Error if server doesn't exist
+ */
+export async function removeMCPServer(filePath: string, name: string): Promise<void> {
+	// Read existing config
+	const existing = await readMCPConfigFile(filePath);
+
+	// Check if server exists
+	if (!existing.mcpServers?.[name]) {
+		throw new Error(`Server "${name}" not found in ${filePath}`);
+	}
+
+	// Remove server
+	const { [name]: _removed, ...remaining } = existing.mcpServers;
+	const updated: MCPConfigFile = {
+		...existing,
+		mcpServers: remaining,
+	};
+
+	// Write back
+	await writeMCPConfigFile(filePath, updated);
+}
+
+/**
+ * Get a specific server config from a file.
+ * Returns undefined if server doesn't exist.
+ */
+export async function getMCPServer(filePath: string, name: string): Promise<MCPServerConfig | undefined> {
+	const config = await readMCPConfigFile(filePath);
+	return config.mcpServers?.[name];
+}
+
+/**
+ * List all server names in a config file.
+ */
+export async function listMCPServers(filePath: string): Promise<string[]> {
+	const config = await readMCPConfigFile(filePath);
+	return Object.keys(config.mcpServers ?? {});
+}
+
+/**
+ * Read the disabled servers list from a config file.
+ */
+export async function readDisabledServers(filePath: string): Promise<string[]> {
+	const config = await readMCPConfigFile(filePath);
+	return Array.isArray(config.disabledServers) ? config.disabledServers : [];
+}
+
+/**
+ * Add or remove a server name from the disabled servers list.
+ */
+export async function setServerDisabled(filePath: string, name: string, disabled: boolean): Promise<void> {
+	const config = await readMCPConfigFile(filePath);
+	const current = new Set(config.disabledServers ?? []);
+
+	if (disabled) {
+		current.add(name);
+	} else {
+		current.delete(name);
+	}
+
+	const updated: MCPConfigFile = {
+		...config,
+		disabledServers: current.size > 0 ? Array.from(current).sort() : undefined,
+	};
+
+	if (!updated.disabledServers) {
+		delete updated.disabledServers;
+	}
+
+	await writeMCPConfigFile(filePath, updated);
+}