oh-my-opencode 4.13.0 → 4.14.0

package/packages/omo-codex/plugin/components/codegraph/dist/serve.js

@@ -1,14 +1,12 @@
 #!/usr/bin/env node
 
 // src/serve.ts
-import { spawn } from "node:child_process";
 import { existsSync as existsSync5, realpathSync } from "node:fs";
 import { homedir as homedir6 } from "node:os";
-import { basename as basename3, extname, join as join6, resolve as resolve2 } from "node:path";
+import { basename as basename3, join as join6, resolve as resolve2 } from "node:path";
 import {
   cwd as processCwd,
   env as processEnv,
-  execPath as processExecPath,
   stdin as processStdin,
   stderr as processStderr,
   stdout as processStdout
@@ -22,6 +20,41 @@ var CODEGRAPH_INSTALL_DIR_ENV = "CODEGRAPH_INSTALL_DIR";
 var CODEGRAPH_NO_DOWNLOAD_ENV = "CODEGRAPH_NO_DOWNLOAD";
 var CODEGRAPH_TELEMETRY_ENV = "CODEGRAPH_TELEMETRY";
 var DO_NOT_TRACK_ENV = "DO_NOT_TRACK";
+var SAFE_AMBIENT_ENV_KEYS = new Set([
+  "APPDATA",
+  "CI",
+  "CODEX_HOME",
+  "ComSpec",
+  "HOME",
+  "HOMEDRIVE",
+  "HOMEPATH",
+  "LANG",
+  "LC_ALL",
+  "LC_CTYPE",
+  "LOCALAPPDATA",
+  "PATH",
+  "PATHEXT",
+  "Path",
+  "SystemRoot",
+  "TEMP",
+  "TMP",
+  "TMPDIR",
+  "USERPROFILE",
+  "WINDIR",
+  "XDG_CACHE_HOME",
+  "XDG_CONFIG_HOME",
+  "XDG_DATA_HOME",
+  "XDG_STATE_HOME"
+]);
+var SAFE_CODEGRAPH_RUNTIME_ENV_KEYS = new Set([
+  "CODEGRAPH_ALLOW_UNSAFE_NODE",
+  "CODEGRAPH_BIN",
+  "CODEGRAPH_FAKE_LOG",
+  "CODEGRAPH_NODE_BIN",
+  "OMO_CODEGRAPH_BIN",
+  "OMO_CODEGRAPH_PROJECT_CWD",
+  "OMO_CODEGRAPH_SESSION_START_CWD"
+]);
 function buildCodegraphEnv(options = {}) {
   const homeDir = options.homeDir ?? homedir();
   return {
@@ -31,6 +64,26 @@ function buildCodegraphEnv(options = {}) {
     [DO_NOT_TRACK_ENV]: "1"
   };
 }
+function copyDefinedEnvKeys(output, input, allowedKeys) {
+  for (const key of allowedKeys) {
+    const value = input[key];
+    if (value !== undefined)
+      output[key] = value;
+  }
+}
+function copyDefinedEnv(output, input) {
+  for (const [key, value] of Object.entries(input)) {
+    if (value !== undefined)
+      output[key] = value;
+  }
+}
+function buildCodegraphChildEnv(options = {}) {
+  const env = {};
+  copyDefinedEnvKeys(env, options.ambientEnv ?? {}, SAFE_AMBIENT_ENV_KEYS);
+  copyDefinedEnvKeys(env, options.runtimeEnv ?? {}, SAFE_CODEGRAPH_RUNTIME_ENV_KEYS);
+  copyDefinedEnv(env, options.codegraphEnv ?? {});
+  return env;
+}
 
 // ../../../../utils/src/codegraph/node-support.ts
 var CODEGRAPH_MIN_NODE_MAJOR = 20;
@@ -360,6 +413,14 @@ var CODEGRAPH_PACKAGE = "@colbymchenry/codegraph";
 var CODEGRAPH_ENV_BIN = "OMO_CODEGRAPH_BIN";
 var CODEGRAPH_LEGACY_ENV_BIN = "CODEGRAPH_BIN";
 var CODEGRAPH_NODE_CANDIDATES = ["node24", "node22", "node20", "node"];
+var CODEGRAPH_NODE_PATH_CANDIDATES = [
+  "/opt/homebrew/opt/node@24/bin/node",
+  "/opt/homebrew/opt/node@22/bin/node",
+  "/opt/homebrew/opt/node@20/bin/node",
+  "/usr/local/opt/node@24/bin/node",
+  "/usr/local/opt/node@22/bin/node",
+  "/usr/local/opt/node@20/bin/node"
+];
 var requireFromHere = createRequire(import.meta.url);
 function defaultRequireResolve(specifier) {
   return requireFromHere.resolve(specifier);
@@ -410,7 +471,8 @@ function defaultNodeRuntime(env, fileExists, which, nodeVersion) {
   }
   const candidates = [
     ...isNodeExecutableName(process.execPath) ? [process.execPath] : [],
-    ...CODEGRAPH_NODE_CANDIDATES.map((commandName) => which(commandName)).filter((candidate) => candidate !== null)
+    ...CODEGRAPH_NODE_CANDIDATES.map((commandName) => which(commandName)).filter((candidate) => candidate !== null),
+    ...CODEGRAPH_NODE_PATH_CANDIDATES.filter((candidate) => fileExists(candidate))
   ];
   const seen = new Set;
   for (const candidate of candidates) {
@@ -1730,6 +1792,9 @@ function resolveHomeDir(options) {
   return options.homeDir ?? env["HOME"] ?? env["USERPROFILE"] ?? homedir5();
 }
 
+// src/mcp-bridge.ts
+import { spawn } from "node:child_process";
+
 // ../../../../mcp-stdio-core/src/record.ts
 function isPlainRecord(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
@@ -1930,6 +1995,114 @@ function createIdleTimer(idleTimeoutMs, log, onIdleTimeout) {
     closed: () => isClosed
   };
 }
+// src/serve-invocation.ts
+import { extname } from "node:path";
+import { execPath as processExecPath } from "node:process";
+var WINDOWS_CMD_EXTENSIONS = new Set([".bat", ".cmd"]);
+var WINDOWS_NODE_SCRIPT_EXTENSIONS = new Set([".cjs", ".js", ".mjs"]);
+function resolveServeProcessInvocation(command, args, platform = process.platform) {
+  if (platform !== "win32")
+    return { args: [...args], command };
+  const extension = extname(command).toLowerCase();
+  if (WINDOWS_NODE_SCRIPT_EXTENSIONS.has(extension)) {
+    return { args: [command, ...args], command: processExecPath };
+  }
+  if (WINDOWS_CMD_EXTENSIONS.has(extension)) {
+    return { args: ["/d", "/s", "/c", command, ...args], command: "cmd.exe" };
+  }
+  return { args: [...args], command };
+}
+
+// src/mcp-bridge.ts
+class CodegraphBridgeStdioError extends Error {
+  streamName;
+  name = "CodegraphBridgeStdioError";
+  constructor(streamName) {
+    super(`CodeGraph MCP bridge missing child ${streamName}`);
+    this.streamName = streamName;
+  }
+}
+async function runBridgedCodegraphProcess(command, args, options) {
+  const invocation = resolveServeProcessInvocation(command, args);
+  const child = spawn(invocation.command, invocation.args, {
+    cwd: options.cwd,
+    env: options.env,
+    stdio: ["pipe", "pipe", "inherit"]
+  });
+  const childInput = child.stdin;
+  const childOutput = child.stdout;
+  if (childInput === null)
+    throw new CodegraphBridgeStdioError("stdin");
+  if (childOutput === null)
+    throw new CodegraphBridgeStdioError("stdout");
+  const responseModes = new Map;
+  let defaultResponseMode = "framed";
+  const childExit = new Promise((resolveExit, reject) => {
+    child.once("error", reject);
+    child.once("exit", (code, signal) => {
+      if (code !== null) {
+        resolveExit(code);
+        return;
+      }
+      resolveExit(signal === null ? 0 : 1);
+    });
+  });
+  const bridgeDone = Promise.all([
+    forwardClientToCodegraph(options.input, childInput, responseModes, (mode) => {
+      defaultResponseMode = mode;
+    }),
+    forwardCodegraphToClient(childOutput, options.output, responseModes, () => defaultResponseMode)
+  ]);
+  const destroyChildPipes = () => {
+    childInput.destroy();
+    childOutput.destroy();
+  };
+  childExit.then(destroyChildPipes, destroyChildPipes);
+  return Promise.race([childExit, bridgeDone.then(() => childExit)]);
+}
+async function forwardClientToCodegraph(input, childInput, responseModes, setDefaultResponseMode) {
+  for await (const message of readStdioJsonRpcMessages(input)) {
+    if (message.kind === "parse_error") {
+      continue;
+    }
+    const responseMode = message.responseMode;
+    setDefaultResponseMode(responseMode);
+    const key = responseModeKey(message.payload);
+    if (key !== null)
+      responseModes.set(key, responseMode);
+    await writeLine(childInput, JSON.stringify(message.payload));
+  }
+  childInput.end();
+}
+async function forwardCodegraphToClient(childOutput, output, responseModes, defaultResponseMode) {
+  for await (const message of readStdioJsonRpcMessages(childOutput)) {
+    if (message.kind === "parse_error") {
+      writeStdioJsonRpcResponse(output, errorResponse(null, -32700, "Parse error", message.message), defaultResponseMode());
+      continue;
+    }
+    const key = responseModeKey(message.payload);
+    const responseMode = key === null ? defaultResponseMode() : responseModes.get(key) ?? defaultResponseMode();
+    if (key !== null)
+      responseModes.delete(key);
+    writeStdioJsonRpcResponse(output, message.payload, responseMode);
+  }
+}
+function responseModeKey(payload) {
+  if (!isPlainRecord(payload) || !("id" in payload))
+    return null;
+  const id = jsonRpcId(payload["id"]);
+  return `${typeof id}:${String(id)}`;
+}
+async function writeLine(output, line) {
+  if (output.write(`${line}
+`))
+    return;
+  await new Promise((resolveDrain, reject) => {
+    output.once("drain", resolveDrain);
+    output.once("error", reject);
+  });
+}
+
 // src/mcp-unavailable.ts
 async function runUnavailableCodegraphMcpServer(options) {
   await runJsonRpcStdioServer({
@@ -1976,18 +2149,24 @@ function requestedProtocolVersion(params) {
   return params["protocolVersion"];
 }
 
+// src/session-start-worker.ts
+var SESSION_START_CWD_ENV = "OMO_CODEGRAPH_SESSION_START_CWD";
+var WINDOWS_CMD_EXTENSIONS2 = new Set([".bat", ".cmd"]);
+var WINDOWS_NODE_SCRIPT_EXTENSIONS2 = new Set([".cjs", ".js", ".mjs"]);
+
 // src/serve.ts
 var CODEGRAPH_SKIP_HINT = `CodeGraph MCP skipped: codegraph binary not found. Install CodeGraph or set OMO_CODEGRAPH_BIN.
 `;
 var CODEGRAPH_DISABLED_HINT = `CodeGraph MCP skipped: disabled by OMO SOT config. Set [codex].codegraph.enabled=true to enable it.
 `;
 var CODEGRAPH_VERSION = "1.0.1";
-var WINDOWS_CMD_EXTENSIONS = new Set([".bat", ".cmd"]);
-var WINDOWS_NODE_SCRIPT_EXTENSIONS = new Set([".cjs", ".js", ".mjs"]);
+var PROJECT_CWD_ENV_KEYS = ["OMO_CODEGRAPH_PROJECT_CWD", SESSION_START_CWD_ENV, "PWD"];
 async function runCodegraphServe(options = {}) {
   const env = options.env ?? processEnv;
   const homeDir = options.homeDir ?? homedir6();
-  const config = options.config ?? getCodexOmoConfig({ cwd: options.cwd ?? processCwd(), env, homeDir });
+  const wrapperCwd = options.cwd ?? processCwd();
+  const projectCwd = resolveProjectCwd(env, wrapperCwd);
+  const config = options.config ?? getCodexOmoConfig({ cwd: projectCwd, env, homeDir });
   const codegraphConfig = config.codegraph ?? {};
   if (codegraphConfig.enabled === false) {
     return runUnavailableMcp(CODEGRAPH_DISABLED_HINT, options);
@@ -2019,15 +2198,16 @@ async function runCodegraphServe(options = {}) {
   if (codegraphCommandRequiresSupportedLocalNode(resolution) && !nodeSupport.supported) {
     return runUnavailableMcp(buildCodegraphNodeSkipHint(nodeSupport), options);
   }
-  const runProcess = options.runProcess ?? runChildProcess;
+  const runProcess = options.runProcess ?? runBridgedCodegraphProcess;
   const codegraphEnv = codegraphEnvForConfig(trustedInstallDir, homeDir, options.buildEnv);
-  const mergedEnv = {
-    ...env,
-    ...codegraphEnv
-  };
+  const mergedEnv = buildCodegraphChildEnv({ ambientEnv: env, codegraphEnv, runtimeEnv: env });
   return runProcess(resolution.command, [...resolution.argsPrefix, "serve", "--mcp"], {
+    cwd: projectCwd,
     env: mergedEnv,
-    stdio: "inherit"
+    input: options.stdin ?? processStdin,
+    output: options.stdout ?? processStdout,
+    stderr: options.stderr ?? processStderr,
+    stdio: "pipe"
   });
 }
 async function runUnavailableMcp(reason, options) {
@@ -2069,6 +2249,17 @@ function codegraphEnvForConfig(trustedInstallDir, homeDir, buildEnv) {
   const env = buildEnv?.({ homeDir }) ?? buildCodegraphEnv({ homeDir });
   return trustedInstallDir === undefined ? env : { ...env, CODEGRAPH_INSTALL_DIR: trustedInstallDir };
 }
+function resolveProjectCwd(env, fallback) {
+  for (const key of PROJECT_CWD_ENV_KEYS) {
+    const candidate = env[key]?.trim();
+    if (candidate === undefined || candidate.length === 0)
+      continue;
+    const resolved = resolve2(candidate);
+    if (existsSync5(resolved))
+      return resolved;
+  }
+  return resolve2(fallback);
+}
 function provisionedBinFromInstallDir(installDir) {
   if (installDir === undefined)
     return null;
@@ -2078,32 +2269,6 @@ function provisionedBinFromInstallDir(installDir) {
 async function runCodegraphServeCli() {
   process.exitCode = await runCodegraphServe();
 }
-async function runChildProcess(command, args, options) {
-  const invocation = resolveServeProcessInvocation(command, args);
-  const child = spawn(invocation.command, invocation.args, { env: options.env, stdio: options.stdio });
-  return new Promise((resolve3, reject) => {
-    child.once("error", reject);
-    child.once("exit", (code, signal) => {
-      if (code !== null) {
-        resolve3(code);
-        return;
-      }
-      resolve3(signal === null ? 0 : 1);
-    });
-  });
-}
-function resolveServeProcessInvocation(command, args, platform = process.platform) {
-  if (platform !== "win32")
-    return { args: [...args], command };
-  const extension = extname(command).toLowerCase();
-  if (WINDOWS_NODE_SCRIPT_EXTENSIONS.has(extension)) {
-    return { args: [command, ...args], command: processExecPath };
-  }
-  if (WINDOWS_CMD_EXTENSIONS.has(extension)) {
-    return { args: ["/d", "/s", "/c", command, ...args], command: "cmd.exe" };
-  }
-  return { args: [...args], command };
-}
 if (isDirectInvocation(process.argv[1])) {
   runCodegraphServeCli().catch((error) => {
     processStderr.write(`${error instanceof Error ? error.stack ?? error.message : String(error)}

package/packages/omo-codex/plugin/components/codegraph/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@sisyphuslabs/codex-codegraph",
-	"version": "4.13.0",
+	"version": "4.14.0",
 	"description": "Codex plugin MCP wrapper for CodeGraph.",
 	"type": "module",
 	"private": true,
@@ -15,7 +15,7 @@
 	],
 	"scripts": {
 		"build": "node -e \"require('node:fs').rmSync('dist',{recursive:true,force:true})\" && bun build src/serve.ts --target node --format esm --outfile dist/serve.js && bun build src/cli.ts --target node --format esm --outfile dist/cli.js",
-		"test": "bun test test/*.test.ts",
+		"test": "bun test ./test",
 		"typecheck": "tsc --noEmit"
 	},
 	"devDependencies": {

package/packages/omo-codex/plugin/components/codegraph/src/hook-types.ts

@@ -9,7 +9,7 @@ import type {
 import type { CodegraphWorkspacePreparation as SharedCodegraphWorkspacePreparation } from "../../../../../utils/src/codegraph/workspace.ts";
 import type { CodegraphConfig as SharedCodegraphConfig } from "../../../../../utils/src/omo-config.ts";
 
-export type SessionStartAction = "skipped-disabled" | "spawned";
+export type SessionStartAction = "skipped-disabled" | "skipped-initialized" | "spawned";
 export type PostToolUseAction = "emitted-guidance" | "skipped";
 export type WorkerAction = "failed" | "initialized" | "skipped-disabled" | "skipped-status" | "skipped-unavailable" | "skipped-unsupported-node" | "synced";
 
@@ -74,6 +74,12 @@ export interface SessionStartHookOptions {
 	readonly cwd?: string;
 	readonly env?: Record<string, string | undefined>;
 	readonly spawnWorker?: (invocation: WorkerSpawnInvocation) => void;
+	readonly statusProbe?: (options: {
+		readonly env: Record<string, string | undefined>;
+		readonly homeDir: string;
+		readonly projectRoot: string;
+		readonly trustedCodegraphInstallDir?: string;
+	}) => Promise<boolean>;
 	readonly stdin?: Readable & { readonly isTTY?: boolean };
 	readonly stdout?: HookStdout;
 	readonly workerCliPath?: string;

package/packages/omo-codex/plugin/components/codegraph/src/hook.ts

@@ -1,12 +1,15 @@
-import { spawn } from "node:child_process";
+import { execFile, spawn } from "node:child_process";
 import { homedir } from "node:os";
+import { join } from "node:path";
 import { cwd as processCwd, env as processEnv, stdin as processStdin, stdout as processStdout } from "node:process";
 import type { Readable } from "node:stream";
 import { fileURLToPath } from "node:url";
 
+import { buildCodegraphChildEnv, buildCodegraphEnv } from "../../../../../utils/src/codegraph/env.ts";
 import { buildCodegraphInitGuidanceForToolResult } from "../../../../../utils/src/codegraph/guidance.ts";
+import { resolveCodegraphCommand } from "../../../../../utils/src/codegraph/resolve.ts";
 import { getCodexOmoConfig } from "../../../shared/src/config-loader.ts";
-import { SESSION_START_CWD_ENV } from "./session-start-worker.js";
+import { resolveCodegraphCommandInvocation, SESSION_START_CWD_ENV } from "./session-start-worker.js";
 import type {
 	HookStdout,
 	PostToolUseHookOptions,
@@ -38,6 +41,7 @@ export type {
 export { resolveCodegraphCommandInvocation, runCodegraphSessionStartWorker } from "./session-start-worker.js";
 
 export const CODEGRAPH_SESSION_START_NOTICE = "LazyCodex CodeGraph bootstrap scheduled in background";
+const STATUS_PROBE_TIMEOUT_MS = 2_000;
 
 export async function runCodegraphSessionStartHook(options: SessionStartHookOptions = {}): Promise<number> {
 	return (await executeCodegraphSessionStartHook(options)).exitCode;
@@ -58,15 +62,102 @@ export async function executeCodegraphSessionStartHook(options: SessionStartHook
 		return { action: "skipped-disabled", exitCode: 0 };
 	}
 
+	const isInitialized = await (options.statusProbe ?? isCodegraphProjectInitialized)({
+		env,
+		homeDir,
+		projectRoot,
+		...(config.trustedCodegraphInstallDir === undefined ? {} : { trustedCodegraphInstallDir: config.trustedCodegraphInstallDir }),
+	});
+	if (isInitialized) {
+		return { action: "skipped-initialized", exitCode: 0 };
+	}
+
 	(options.spawnWorker ?? spawnDetachedWorker)({
 		args: [options.workerCliPath ?? defaultWorkerCliPath(), "hook", "session-start-worker"],
 		command: process.execPath,
-		env: { ...env, [SESSION_START_CWD_ENV]: projectRoot },
+		env: buildCodegraphChildEnv({
+			ambientEnv: env,
+			codegraphEnv: { [SESSION_START_CWD_ENV]: projectRoot },
+			runtimeEnv: env,
+		}),
 	});
 	writeHookJson(options.stdout ?? processStdout);
 	return { action: "spawned", exitCode: 0 };
 }
 
+async function isCodegraphProjectInitialized(options: {
+	readonly env: Record<string, string | undefined>;
+	readonly homeDir: string;
+	readonly projectRoot: string;
+	readonly trustedCodegraphInstallDir?: string;
+}): Promise<boolean> {
+	const resolved = resolveCodegraphCommand({
+		env: options.env,
+		homeDir: options.homeDir,
+		provisioned: () => provisionedBinFromInstallDir(options.trustedCodegraphInstallDir),
+	});
+	if (!resolved.exists) return false;
+
+	const invocation = resolveCodegraphCommandInvocation(resolved.command, [...resolved.argsPrefix, "status", "--json"]);
+	const codegraphEnv = {
+		...buildCodegraphEnv({ homeDir: options.homeDir }),
+		...(options.trustedCodegraphInstallDir === undefined ? {} : { CODEGRAPH_INSTALL_DIR: options.trustedCodegraphInstallDir }),
+	};
+	const status = await runStatusProbe(
+		options.projectRoot,
+		invocation.command,
+		invocation.args,
+		buildCodegraphChildEnv({ ambientEnv: options.env, codegraphEnv, runtimeEnv: options.env }),
+	);
+	if (status.exitCode !== 0 || status.timedOut) return false;
+	return codegraphStatusSaysInitialized(status.stdout);
+}
+
+function runStatusProbe(
+	projectRoot: string,
+	command: string,
+	args: readonly string[],
+	env: Record<string, string>,
+): Promise<{ readonly exitCode: number; readonly stdout: string; readonly timedOut: boolean }> {
+	return new Promise((resolveProbe) => {
+		execFile(
+			command,
+			[...args],
+			{
+				cwd: projectRoot,
+				encoding: "utf8",
+				env,
+				maxBuffer: 1024 * 1024,
+				timeout: STATUS_PROBE_TIMEOUT_MS,
+				windowsHide: true,
+			},
+			(error, stdout) => {
+				if (error === null) {
+					resolveProbe({ exitCode: 0, stdout: toOutputText(stdout), timedOut: false });
+					return;
+				}
+				resolveProbe({ exitCode: resolveExitCode(error), stdout: toOutputText(stdout), timedOut: error.killed === true });
+			},
+		);
+	});
+}
+
+function codegraphStatusSaysInitialized(stdout: string): boolean {
+	const parsed = parseJson(stdout);
+	if (!isRecord(parsed)) return false;
+	const initialized = parsed["initialized"] ?? parsed["isInitialized"] ?? parsed["ready"];
+	if (typeof initialized === "boolean") return initialized;
+	const status = parsed["status"];
+	if (typeof status !== "string") return false;
+	const normalized = status.toLowerCase();
+	return (normalized.includes("initialized") || normalized.includes("ready")) && !normalized.includes("not initialized") && !normalized.includes("uninitialized");
+}
+
+function provisionedBinFromInstallDir(installDir: string | undefined): string | null {
+	if (installDir === undefined) return null;
+	return join(installDir, "bin", process.platform === "win32" ? "codegraph.cmd" : "codegraph");
+}
+
 export async function executeCodegraphPostToolUseHook(options: PostToolUseHookOptions = {}): Promise<PostToolUseHookResult> {
 	const env = options.env ?? processEnv;
 	const input = await readHookInput(options.stdin ?? processStdin);
@@ -111,6 +202,15 @@ function resolveHomeDir(env: Record<string, string | undefined>): string {
 	return env["HOME"] ?? env["USERPROFILE"] ?? homedir();
 }
 
+function resolveExitCode(error: Error): number {
+	if ("code" in error && typeof error.code === "number") return error.code;
+	return 1;
+}
+
+function toOutputText(value: string | Buffer): string {
+	return Buffer.isBuffer(value) ? value.toString("utf8") : value;
+}
+
 function resolveProjectRoot(input: unknown, fallback: string): string {
 	if (!isRecord(input)) return fallback;
 	const cwd = input["cwd"];

package/packages/omo-codex/plugin/components/codegraph/src/mcp-bridge.ts

@@ -0,0 +1,114 @@
+import { spawn } from "node:child_process";
+import type { Readable, Writable } from "node:stream";
+
+import {
+	errorResponse,
+	isPlainRecord,
+	jsonRpcId,
+	readStdioJsonRpcMessages,
+	writeStdioJsonRpcResponse,
+	type StdioJsonRpcResponseMode,
+} from "../../../../../mcp-stdio-core/src/index.ts";
+import type { CodegraphServeProcessOptions } from "./serve.js";
+import { resolveServeProcessInvocation } from "./serve-invocation.js";
+
+class CodegraphBridgeStdioError extends Error {
+	override readonly name = "CodegraphBridgeStdioError";
+
+	constructor(readonly streamName: string) {
+		super(`CodeGraph MCP bridge missing child ${streamName}`);
+	}
+}
+
+export async function runBridgedCodegraphProcess(
+	command: string,
+	args: readonly string[],
+	options: CodegraphServeProcessOptions,
+): Promise<number> {
+	const invocation = resolveServeProcessInvocation(command, args);
+	const child = spawn(invocation.command, invocation.args, {
+		cwd: options.cwd,
+		env: options.env,
+		stdio: ["pipe", "pipe", "inherit"],
+	});
+	const childInput = child.stdin;
+	const childOutput = child.stdout;
+	if (childInput === null) throw new CodegraphBridgeStdioError("stdin");
+	if (childOutput === null) throw new CodegraphBridgeStdioError("stdout");
+
+	const responseModes = new Map<string, StdioJsonRpcResponseMode>();
+	let defaultResponseMode: StdioJsonRpcResponseMode = "framed";
+	const childExit = new Promise<number>((resolveExit, reject) => {
+		child.once("error", reject);
+		child.once("exit", (code, signal) => {
+			if (code !== null) {
+				resolveExit(code);
+				return;
+			}
+			resolveExit(signal === null ? 0 : 1);
+		});
+	});
+	const bridgeDone = Promise.all([
+		forwardClientToCodegraph(options.input, childInput, responseModes, (mode) => {
+			defaultResponseMode = mode;
+		}),
+		forwardCodegraphToClient(childOutput, options.output, responseModes, () => defaultResponseMode),
+	]);
+	const destroyChildPipes = (): void => {
+		childInput.destroy();
+		childOutput.destroy();
+	};
+	void childExit.then(destroyChildPipes, destroyChildPipes);
+	return Promise.race([childExit, bridgeDone.then(() => childExit)]);
+}
+
+async function forwardClientToCodegraph(
+	input: Readable,
+	childInput: Writable,
+	responseModes: Map<string, StdioJsonRpcResponseMode>,
+	setDefaultResponseMode: (mode: StdioJsonRpcResponseMode) => void,
+): Promise<void> {
+	for await (const message of readStdioJsonRpcMessages(input)) {
+		if (message.kind === "parse_error") {
+			continue;
+		}
+		const responseMode = message.responseMode;
+		setDefaultResponseMode(responseMode);
+		const key = responseModeKey(message.payload);
+		if (key !== null) responseModes.set(key, responseMode);
+		await writeLine(childInput, JSON.stringify(message.payload));
+	}
+	childInput.end();
+}
+
+async function forwardCodegraphToClient(
+	childOutput: Readable,
+	output: Writable,
+	responseModes: Map<string, StdioJsonRpcResponseMode>,
+	defaultResponseMode: () => StdioJsonRpcResponseMode,
+): Promise<void> {
+	for await (const message of readStdioJsonRpcMessages(childOutput)) {
+		if (message.kind === "parse_error") {
+			writeStdioJsonRpcResponse(output, errorResponse(null, -32700, "Parse error", message.message), defaultResponseMode());
+			continue;
+		}
+		const key = responseModeKey(message.payload);
+		const responseMode = key === null ? defaultResponseMode() : (responseModes.get(key) ?? defaultResponseMode());
+		if (key !== null) responseModes.delete(key);
+		writeStdioJsonRpcResponse(output, message.payload, responseMode);
+	}
+}
+
+function responseModeKey(payload: unknown): string | null {
+	if (!isPlainRecord(payload) || !("id" in payload)) return null;
+	const id = jsonRpcId(payload["id"]);
+	return `${typeof id}:${String(id)}`;
+}
+
+async function writeLine(output: Writable, line: string): Promise<void> {
+	if (output.write(`${line}\n`)) return;
+	await new Promise<void>((resolveDrain, reject) => {
+		output.once("drain", resolveDrain);
+		output.once("error", reject);
+	});
+}

package/packages/omo-codex/plugin/components/codegraph/src/serve-invocation.ts

@@ -0,0 +1,29 @@
+import { extname } from "node:path";
+import { execPath as processExecPath } from "node:process";
+
+export interface ServeProcessInvocation {
+	readonly args: readonly string[];
+	readonly command: string;
+}
+
+const WINDOWS_CMD_EXTENSIONS = new Set([".bat", ".cmd"]);
+const WINDOWS_NODE_SCRIPT_EXTENSIONS = new Set([".cjs", ".js", ".mjs"]);
+
+export function resolveServeProcessInvocation(
+	command: string,
+	args: readonly string[],
+	platform: NodeJS.Platform = process.platform,
+): ServeProcessInvocation {
+	if (platform !== "win32") return { args: [...args], command };
+
+	const extension = extname(command).toLowerCase();
+	if (WINDOWS_NODE_SCRIPT_EXTENSIONS.has(extension)) {
+		return { args: [command, ...args], command: processExecPath };
+	}
+
+	if (WINDOWS_CMD_EXTENSIONS.has(extension)) {
+		return { args: ["/d", "/s", "/c", command, ...args], command: "cmd.exe" };
+	}
+
+	return { args: [...args], command };
+}