@ai-devkit/agent-manager 0.31.0 → 0.33.0

package/src/__tests__/providers/codex/CodexSessionLocator.test.ts

@@ -0,0 +1,182 @@
+import type { MockedFunction } from "vitest";
+import * as fs from "fs";
+import * as os from "os";
+import * as path from "path";
+import type { ProcessInfo } from "../../../adapters/AgentAdapter.js";
+import { CodexSessionLocator } from "../../../providers/codex/CodexSessionLocator.js";
+import { batchGetSessionFileBirthtimes, type SessionFile } from "../../../utils/session.js";
+import { matchProcessesToSessions, type MatchResult } from "../../../utils/matching.js";
+
+vi.mock("../../../utils/session.js", async () => {
+  const actual = (await vi.importActual(
+    "../../../utils/session",
+  )) as typeof import("../../../utils/session");
+  return {
+    ...actual,
+    batchGetSessionFileBirthtimes: vi.fn(),
+  };
+});
+
+vi.mock("../../../utils/matching.js", async () => {
+  const actual = (await vi.importActual(
+    "../../../utils/matching",
+  )) as typeof import("../../../utils/matching");
+  return {
+    ...actual,
+    matchProcessesToSessions: vi.fn(),
+  };
+});
+
+const mockedBatchGetSessionFileBirthtimes = batchGetSessionFileBirthtimes as MockedFunction<
+  typeof batchGetSessionFileBirthtimes
+>;
+const mockedMatchProcessesToSessions = matchProcessesToSessions as MockedFunction<
+  typeof matchProcessesToSessions
+>;
+
+describe("CodexSessionLocator", () => {
+  let tmpDir: string;
+  let sessionsDir: string;
+  let locator: CodexSessionLocator;
+
+  beforeEach(() => {
+    tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "codex-locator-"));
+    sessionsDir = path.join(tmpDir, "sessions");
+    locator = new CodexSessionLocator({ sessionsDir });
+    mockedBatchGetSessionFileBirthtimes.mockReset();
+    mockedMatchProcessesToSessions.mockReset();
+  });
+
+  afterEach(() => {
+    fs.rmSync(tmpDir, { recursive: true, force: true });
+  });
+
+  it("discovers live sessions from the process start day window and applies session_meta cwd and timestamp", () => {
+    const dateDir = path.join(sessionsDir, "2026", "03", "18");
+    fs.mkdirSync(dateDir, { recursive: true });
+    const sessionFile = path.join(dateDir, "sess-meta-time.jsonl");
+    fs.writeFileSync(
+      sessionFile,
+      JSON.stringify({
+        type: "session_meta",
+        payload: { id: "sess-meta-time", timestamp: "2026-03-18T15:00:05.000Z", cwd: "/repo-a" },
+      }),
+    );
+    mockedBatchGetSessionFileBirthtimes.mockReturnValue([
+      {
+        sessionId: "sess-meta-time",
+        filePath: sessionFile,
+        projectDir: dateDir,
+        birthtimeMs: new Date("2026-03-18T15:05:30.000Z").getTime(),
+        resolvedCwd: "",
+      },
+    ]);
+
+    const { sessions, contentCache } = locator.discoverLiveSessions([
+      {
+        pid: 1,
+        command: "codex",
+        cwd: "/repo-a",
+        tty: "",
+        startTime: new Date("2026-03-18T15:00:00Z"),
+      },
+    ]);
+
+    expect(sessions).toHaveLength(1);
+    expect(sessions[0]).toMatchObject({
+      resolvedCwd: "/repo-a",
+      birthtimeMs: new Date("2026-03-18T15:00:05.000Z").getTime(),
+    });
+    expect(contentCache.has(sessionFile)).toBe(true);
+    expect(mockedBatchGetSessionFileBirthtimes.mock.calls[0][0]).toEqual([dateDir]);
+  });
+
+  it("matches resumed UUIDv7 sessions directly without legacy matching", () => {
+    const sessionId = "019eabed-4079-7071-9531-b853ddd9914e";
+    const sessionFile = writeSession(sessionId, "2026/06/09");
+    const processInfo: ProcessInfo = {
+      pid: 88018,
+      command: `codex resume ${sessionId}`,
+      cwd: "/repo-a",
+      tty: "ttys001",
+      startTime: new Date("2026-06-10T12:00:00.000Z"),
+    };
+
+    const result = locator.matchRunningProcesses([processInfo]);
+
+    expect(result.direct).toHaveLength(1);
+    expect(result.direct[0]).toMatchObject({
+      process: processInfo,
+      sessionFile: { sessionId, filePath: sessionFile, resolvedCwd: "/repo-a" },
+    });
+    expect(result.fallback).toEqual([]);
+    expect(mockedBatchGetSessionFileBirthtimes).not.toHaveBeenCalled();
+    expect(mockedMatchProcessesToSessions).not.toHaveBeenCalled();
+  });
+
+  it("falls back to walking all session files for non-time-sortable resume ids", () => {
+    const sessionId = "aaaaaaaa-bbbb-4ccc-dddd-eeeeeeeeeeee";
+    const sessionFile = writeSession(sessionId, "2026/01/02");
+
+    const result = locator.matchRunningProcesses([
+      { pid: 42, command: `codex resume ${sessionId}`, cwd: "/repo-a", tty: "ttys001" },
+    ]);
+
+    expect(result.direct[0].sessionFile.filePath).toBe(sessionFile);
+  });
+
+  it("passes unmatched processes and discovered sessions to legacy matching", () => {
+    const dateDir = path.join(sessionsDir, "2026", "03", "18");
+    fs.mkdirSync(dateDir, { recursive: true });
+    const candidate: SessionFile = {
+      sessionId: "sess",
+      filePath: path.join(dateDir, "sess.jsonl"),
+      projectDir: dateDir,
+      birthtimeMs: new Date("2026-03-18T15:00:05Z").getTime(),
+      resolvedCwd: "",
+    };
+    fs.writeFileSync(
+      candidate.filePath,
+      JSON.stringify({
+        type: "session_meta",
+        payload: { id: "sess", cwd: "/repo-a", timestamp: "2026-03-18T15:00:05Z" },
+      }),
+    );
+    const processInfo: ProcessInfo = {
+      pid: 100,
+      command: "codex",
+      cwd: "/repo-a",
+      tty: "ttys001",
+      startTime: new Date("2026-03-18T15:00:00Z"),
+    };
+    const legacyMatch: MatchResult = {
+      process: processInfo,
+      session: { ...candidate, resolvedCwd: "/repo-a" },
+      deltaMs: 5000,
+    };
+
+    mockedBatchGetSessionFileBirthtimes.mockReturnValue([candidate]);
+    mockedMatchProcessesToSessions.mockReturnValue([legacyMatch]);
+
+    const result = locator.matchRunningProcesses([processInfo]);
+
+    expect(result.direct).toEqual([]);
+    expect(result.fallback).toEqual([processInfo]);
+    expect(result.legacyMatches).toEqual([legacyMatch]);
+    expect(result.contentCache.has(candidate.filePath)).toBe(true);
+  });
+
+  function writeSession(sessionId: string, day: string): string {
+    const dateDir = path.join(sessionsDir, ...day.split("/"));
+    fs.mkdirSync(dateDir, { recursive: true });
+    const sessionFile = path.join(dateDir, `${sessionId}.jsonl`);
+    fs.writeFileSync(
+      sessionFile,
+      JSON.stringify({
+        type: "session_meta",
+        payload: { id: sessionId, timestamp: "2026-03-18T15:00:05.000Z", cwd: "/repo-a" },
+      }),
+    );
+    return sessionFile;
+  }
+});

package/src/__tests__/providers/codex/CodexSessionMapping.test.ts

@@ -0,0 +1,63 @@
+import * as fs from "fs";
+import * as os from "os";
+import * as path from "path";
+import type { ProcessInfo } from "../../../adapters/AgentAdapter.js";
+import { CodexSessionMapping } from "../../../providers/codex/CodexSessionMapping.js";
+
+describe("CodexSessionMapping", () => {
+  let tmpDir: string;
+  let sessionsDir: string;
+  let mappingPath: string;
+  let processInfo: ProcessInfo;
+
+  beforeEach(() => {
+    tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "codex-mapping-"));
+    sessionsDir = path.join(tmpDir, ".codex", "sessions");
+    mappingPath = path.join(tmpDir, ".codex", "ai-devkit", "sessions.json");
+    processInfo = { pid: 5151, command: "codex", cwd: "/repo", tty: "ttys001" };
+  });
+
+  afterEach(() => {
+    fs.rmSync(tmpDir, { recursive: true, force: true });
+  });
+
+  it("matches a process to an existing trusted session file", () => {
+    const sessionFile = path.join(sessionsDir, "2026", "06", "26", "session.jsonl");
+    fs.mkdirSync(path.dirname(sessionFile), { recursive: true });
+    fs.mkdirSync(path.dirname(mappingPath), { recursive: true });
+    fs.writeFileSync(sessionFile, "");
+    fs.writeFileSync(mappingPath, JSON.stringify({ 5151: sessionFile }));
+
+    expect(createMapping().match([processInfo])).toEqual({
+      matches: [{ process: processInfo, filePath: sessionFile }],
+      fallback: [],
+    });
+  });
+
+  it("falls back for malformed mappings, missing files, invalid pids, and untrusted paths", () => {
+    fs.mkdirSync(path.dirname(mappingPath), { recursive: true });
+    fs.writeFileSync(
+      mappingPath,
+      JSON.stringify({
+        invalid: path.join(sessionsDir, "ignored.jsonl"),
+        5151: path.join(tmpDir, "outside.jsonl"),
+        9999: path.join(sessionsDir, "missing.jsonl"),
+      }),
+    );
+
+    expect(createMapping().match([processInfo])).toEqual({
+      matches: [],
+      fallback: [processInfo],
+    });
+
+    fs.writeFileSync(mappingPath, "{not-json");
+    expect(createMapping().match([processInfo])).toEqual({
+      matches: [],
+      fallback: [processInfo],
+    });
+  });
+
+  function createMapping(): CodexSessionMapping {
+    return new CodexSessionMapping({ mappingPath, sessionsDir });
+  }
+});

package/src/__tests__/providers/codex/CodexSessionParser.test.ts

@@ -0,0 +1,192 @@
+import * as fs from "fs";
+import * as os from "os";
+import * as path from "path";
+import { AgentStatus } from "../../../adapters/AgentAdapter.js";
+import { CodexSessionParser } from "../../../providers/codex/CodexSessionParser.js";
+
+describe("CodexSessionParser", () => {
+  let tmpDir: string;
+  let parser: CodexSessionParser;
+
+  beforeEach(() => {
+    tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "codex-parser-"));
+    parser = new CodexSessionParser();
+  });
+
+  afterEach(() => {
+    fs.rmSync(tmpDir, { recursive: true, force: true });
+  });
+
+  it("reads session metadata, summary, timestamps, and status signal", () => {
+    const filePath = writeJsonl("session.jsonl", [
+      {
+        type: "session_meta",
+        payload: { id: "sess-1", timestamp: "2026-03-18T15:00:00Z", cwd: "/repo" },
+      },
+      {
+        type: "event",
+        timestamp: "2026-03-18T15:01:00Z",
+        payload: { type: "agent_reasoning", message: "Working on feature" },
+      },
+    ]);
+
+    const session = parser.readSession(filePath);
+
+    expect(session).toMatchObject({
+      sessionId: "sess-1",
+      projectPath: "/repo",
+      summary: "Working on feature",
+      lastPayloadType: "agent_reasoning",
+    });
+    expect(session?.sessionStart.toISOString()).toBe("2026-03-18T15:00:00.000Z");
+    expect(parser.determineStatus(session!)).toBe(AgentStatus.IDLE);
+  });
+
+  it("reads current response_item and event_msg messages without mirrored duplicates", () => {
+    const filePath = writeJsonl("conversation.jsonl", [
+      {
+        type: "session_meta",
+        payload: { id: "sess-1", cwd: "/repo", timestamp: "2026-03-27T10:00:00Z" },
+      },
+      {
+        type: "response_item",
+        timestamp: "2026-03-27T10:00:01Z",
+        payload: {
+          type: "message",
+          role: "user",
+          content: [{ type: "input_text", text: "Fix the bug" }],
+          internal_chat_message_metadata_passthrough: { turn_id: "turn-1" },
+        },
+      },
+      {
+        type: "event_msg",
+        timestamp: "2026-03-27T10:00:01.001Z",
+        payload: {
+          type: "item_completed",
+          turn_id: "turn-1",
+          item: { type: "UserMessage", content: [{ type: "text", text: "Fix the bug" }] },
+        },
+      },
+      {
+        type: "event_msg",
+        timestamp: "2026-03-27T10:00:05Z",
+        payload: {
+          type: "item_completed",
+          turn_id: "turn-1",
+          item: { type: "AgentMessage", content: [{ type: "Text", text: "I found the issue" }] },
+        },
+      },
+      {
+        type: "response_item",
+        timestamp: "2026-03-27T10:00:05.005Z",
+        payload: {
+          type: "message",
+          role: "assistant",
+          content: [{ type: "output_text", text: "I found the issue" }],
+          internal_chat_message_metadata_passthrough: { turn_id: "turn-1" },
+        },
+      },
+    ]);
+
+    expect(parser.getConversation(filePath)).toEqual([
+      { role: "user", content: "Fix the bug", timestamp: "2026-03-27T10:00:01Z" },
+      { role: "assistant", content: "I found the issue", timestamp: "2026-03-27T10:00:05.005Z" },
+    ]);
+  });
+
+  it("maps completed assistant messages as waiting while recent active events stay running", () => {
+    const waiting = parser.readSession(
+      "unused.jsonl",
+      [
+        JSON.stringify({
+          type: "session_meta",
+          payload: { id: "waiting", timestamp: new Date().toISOString(), cwd: "/repo" },
+        }),
+        JSON.stringify({
+          type: "event_msg",
+          timestamp: new Date().toISOString(),
+          payload: {
+            type: "item_completed",
+            item: {
+              type: "AgentMessage",
+              content: [{ type: "Text", text: "Waiting for the user now" }],
+            },
+          },
+        }),
+      ].join("\n"),
+    );
+
+    const running = parser.readSession(
+      "unused.jsonl",
+      [
+        JSON.stringify({
+          type: "session_meta",
+          payload: { id: "running", timestamp: new Date().toISOString(), cwd: "/repo" },
+        }),
+        JSON.stringify({
+          type: "event",
+          timestamp: new Date().toISOString(),
+          payload: { type: "token_count", message: "Working" },
+        }),
+      ].join("\n"),
+    );
+
+    expect(waiting?.summary).toBe("Waiting for the user now");
+    expect(parser.determineStatus(waiting!)).toBe(AgentStatus.WAITING);
+    expect(parser.determineStatus(running!)).toBe(AgentStatus.RUNNING);
+  });
+
+  it("builds historical summaries from the first non-synthetic user message", () => {
+    const filePath = writeJsonl("summary.jsonl", [
+      {
+        type: "session_meta",
+        payload: { id: "summary", cwd: "/repo", timestamp: "2025-01-01T00:00:00Z" },
+      },
+      {
+        type: "response_item",
+        timestamp: "2025-01-01T00:00:00.500Z",
+        payload: {
+          type: "message",
+          role: "user",
+          content: [
+            {
+              type: "input_text",
+              text: "<environment_context>\n  <cwd>/repo</cwd>\n</environment_context>",
+            },
+          ],
+        },
+      },
+      {
+        type: "event_msg",
+        timestamp: "2025-01-01T00:00:02Z",
+        payload: {
+          type: "item_completed",
+          item: { type: "UserMessage", content: [{ type: "text", text: "first event user" }] },
+        },
+      },
+    ]);
+
+    expect(parser.fileToSessionSummary(filePath)).toMatchObject({
+      type: "codex",
+      sessionId: "summary",
+      cwd: "/repo",
+      firstUserMessage: "first event user",
+      sessionFilePath: filePath,
+    });
+  });
+
+  it("returns null or empty results for unreadable and malformed inputs", () => {
+    expect(parser.readSession(path.join(tmpDir, "missing.jsonl"))).toBeNull();
+    expect(parser.readSession("unused.jsonl", "")).toBeNull();
+    expect(
+      parser.readSession("unused.jsonl", JSON.stringify({ type: "event", payload: {} })),
+    ).toBeNull();
+    expect(parser.getConversation(path.join(tmpDir, "missing.jsonl"))).toEqual([]);
+  });
+
+  function writeJsonl(name: string, entries: object[]): string {
+    const filePath = path.join(tmpDir, name);
+    fs.writeFileSync(filePath, entries.map((entry) => JSON.stringify(entry)).join("\n"));
+    return filePath;
+  }
+});