@keyou007/agent-bot 0.1.0

package/src/codex/CodexProcessManager.ts

@@ -0,0 +1,63 @@
+import type { ChildProcessWithoutNullStreams } from "node:child_process";
+import os from "node:os";
+import path from "node:path";
+import readline from "node:readline";
+import type { Logger } from "pino";
+import { AppServerConnection } from "./AppServerConnection.js";
+import type { AppServerClient, AppServerClientProvider } from "./CodexRuntime.js";
+import { agentBotEnvironment } from "../runtime/agentEnvironment.js";
+import { spawnStdioCommand } from "../utils/spawnCommand.js";
+
+export class CodexProcessManager implements AppServerClientProvider {
+  private client?: AppServerConnection;
+  private child?: ChildProcessWithoutNullStreams;
+  private readonly disconnectListeners = new Set<(error: Error) => void>();
+
+  constructor(
+    private readonly command: string,
+    private readonly args: string[],
+    private readonly env: Record<string, string>,
+    private readonly logger: Logger,
+  ) {}
+
+  async getClient(): Promise<AppServerClient> {
+    if (this.client) return this.client;
+    const child = spawnStdioCommand(this.command, this.args, agentBotEnvironment(process.env, this.env));
+    this.child = child;
+    const client = new AppServerConnection(child, this.logger.child({ component: "codex-app-server" }));
+    this.client = client;
+    readline.createInterface({ input: child.stderr, crlfDelay: Infinity }).on("line", (line) => {
+      this.logger.debug({ line }, "Codex App Server stderr.");
+    });
+    child.once("error", (error) => this.logger.error({ error }, "Codex App Server process error."));
+    child.once("exit", (code, signal) => {
+      this.logger.warn({ code, signal }, "Codex App Server process exited.");
+      client.close();
+      if (this.client === client) this.client = undefined;
+      if (this.child === child) this.child = undefined;
+      const error = new Error(`Codex App Server exited (code=${code ?? "null"}, signal=${signal ?? "null"}).`);
+      for (const listener of this.disconnectListeners) listener(error);
+    });
+    await client.request("initialize", {
+      clientInfo: { name: "agent-bot", title: "Agent Bot", version: "0.1.0" },
+    });
+    client.notify("initialized", {});
+    return client;
+  }
+
+  getCodexHome(): string {
+    return this.env.CODEX_HOME ?? process.env.CODEX_HOME ?? path.join(os.homedir(), ".codex");
+  }
+
+  close(): void {
+    this.client?.close();
+    if (this.child && !this.child.killed) this.child.kill();
+    this.client = undefined;
+    this.child = undefined;
+  }
+
+  onDisconnect(listener: (error: Error) => void): () => void {
+    this.disconnectListeners.add(listener);
+    return () => this.disconnectListeners.delete(listener);
+  }
+}