@ai-devkit/agent-manager 0.33.0 → 0.33.1

package/src/providers/gemini/GeminiSessionLocator.ts

@@ -0,0 +1,183 @@
+import * as crypto from "crypto";
+import * as fs from "fs";
+import * as path from "path";
+import type { ProcessInfo } from "../../adapters/AgentAdapter.js";
+import type { SessionFile } from "../../utils/session.js";
+import { isDirectory, safeReaddir } from "../../utils/session.js";
+
+export interface GeminiSessionDiscovery {
+  sessions: SessionFile[];
+  contentCache: Map<string, string>;
+}
+
+export interface GeminiSessionLocatorOptions {
+  geminiTmpDir?: string;
+}
+
+interface GeminiSessionMetadata {
+  sessionId?: string;
+  projectHash?: string;
+}
+
+const SESSION_FILE_PREFIX = "session-";
+const CHATS_DIR_NAME = "chats";
+
+export class GeminiSessionLocator {
+  private readonly geminiTmpDir: string;
+
+  constructor(options: GeminiSessionLocatorOptions = {}) {
+    const homeDir = process.env.HOME || process.env.USERPROFILE || "";
+    this.geminiTmpDir = options.geminiTmpDir ?? path.join(homeDir, ".gemini", "tmp");
+  }
+
+  /**
+   * Discover session files for the given processes.
+   *
+   * Gemini CLI writes sessions to ~/.gemini/tmp/<shortId>/chats/session-*.json
+   * where <shortId> is opaque (managed by a project registry). We scan every
+   * shortId directory and filter by matching session.projectHash against
+   * sha256(process.cwd) to bind each session to a candidate process CWD.
+   */
+  discoverSessions(processes: ProcessInfo[]): GeminiSessionDiscovery {
+    const empty = { sessions: [] as SessionFile[], contentCache: new Map<string, string>() };
+    if (!fs.existsSync(this.geminiTmpDir)) return empty;
+
+    const cwdHashMap = this.buildCwdHashMap(processes);
+    if (cwdHashMap.size === 0) return empty;
+
+    const contentCache = new Map<string, string>();
+    const sessions: SessionFile[] = [];
+
+    let shortIdEntries: string[];
+    try {
+      shortIdEntries = fs.readdirSync(this.geminiTmpDir);
+    } catch {
+      return empty;
+    }
+
+    for (const shortId of shortIdEntries) {
+      const chatsDir = path.join(this.geminiTmpDir, shortId, CHATS_DIR_NAME);
+      if (!isDirectory(chatsDir)) continue;
+
+      for (const fileName of safeReaddir(chatsDir)) {
+        if (!this.isSessionFile(fileName)) continue;
+
+        const session = this.readCandidateSession(chatsDir, fileName, cwdHashMap);
+        if (!session) continue;
+
+        contentCache.set(session.filePath, session.content);
+        sessions.push(session.sessionFile);
+      }
+    }
+
+    return { sessions, contentCache };
+  }
+
+  discoverHistoricalSessionFiles(): string[] {
+    if (!isDirectory(this.geminiTmpDir)) return [];
+
+    const files: string[] = [];
+    for (const shortId of safeReaddir(this.geminiTmpDir)) {
+      const chatsDir = path.join(this.geminiTmpDir, shortId, CHATS_DIR_NAME);
+      if (!isDirectory(chatsDir)) continue;
+
+      for (const fileName of safeReaddir(chatsDir)) {
+        if (this.isSessionFile(fileName)) {
+          files.push(path.join(chatsDir, fileName));
+        }
+      }
+    }
+
+    return files;
+  }
+
+  private readCandidateSession(
+    chatsDir: string,
+    fileName: string,
+    cwdHashMap: Map<string, string>,
+  ): { filePath: string; content: string; sessionFile: SessionFile } | null {
+    const filePath = path.join(chatsDir, fileName);
+
+    let content: string;
+    try {
+      content = fs.readFileSync(filePath, "utf-8");
+    } catch {
+      return null;
+    }
+
+    const parsed = this.parseSessionMetadata(content);
+    if (!parsed?.projectHash) return null;
+
+    const resolvedCwd = cwdHashMap.get(parsed.projectHash);
+    if (!resolvedCwd) return null;
+
+    let birthtimeMs = 0;
+    try {
+      birthtimeMs = fs.statSync(filePath).birthtimeMs;
+    } catch {
+      return null;
+    }
+
+    return {
+      filePath,
+      content,
+      sessionFile: {
+        sessionId: parsed.sessionId || fileName.replace(/\.json$/, ""),
+        filePath,
+        projectDir: chatsDir,
+        birthtimeMs,
+        resolvedCwd,
+      },
+    };
+  }
+
+  private buildCwdHashMap(processes: ProcessInfo[]): Map<string, string> {
+    const map = new Map<string, string>();
+    for (const proc of processes) {
+      if (!proc.cwd) continue;
+
+      // Gemini CLI resolves its project root by walking up from the
+      // startup directory looking for a `.git` boundary marker. A
+      // session's projectHash therefore tracks that ancestor rather
+      // than the process' actual CWD. Enumerate every ancestor as a
+      // candidate so subdirectory invocations still line up with the
+      // session the Gemini process wrote.
+      for (const candidate of this.candidateProjectRoots(proc.cwd)) {
+        const hash = this.hashProjectRoot(candidate);
+        if (!map.has(hash)) {
+          map.set(hash, proc.cwd);
+        }
+      }
+    }
+    return map;
+  }
+
+  private candidateProjectRoots(cwd: string): string[] {
+    const roots: string[] = [];
+    let current = path.resolve(cwd);
+    let parent = path.dirname(current);
+    while (parent !== current) {
+      roots.push(current);
+      current = parent;
+      parent = path.dirname(current);
+    }
+    roots.push(current);
+    return roots;
+  }
+
+  private hashProjectRoot(projectRoot: string): string {
+    return crypto.createHash("sha256").update(projectRoot).digest("hex");
+  }
+
+  private isSessionFile(fileName: string): boolean {
+    return fileName.startsWith(SESSION_FILE_PREFIX) && fileName.endsWith(".json");
+  }
+
+  private parseSessionMetadata(content: string): GeminiSessionMetadata | null {
+    try {
+      return JSON.parse(content) as GeminiSessionMetadata;
+    } catch {
+      return null;
+    }
+  }
+}

package/src/providers/gemini/GeminiSessionParser.ts

@@ -0,0 +1,271 @@
+import * as fs from "fs";
+import type { ConversationMessage, SessionSummary } from "../../adapters/AgentAdapter.js";
+import { AgentStatus } from "../../adapters/AgentAdapter.js";
+import { safeReadFile, safeStat } from "../../utils/session.js";
+
+/**
+ * A single Gemini CLI message content part. Mirrors the `{text?: string}`
+ * shape that Gemini writes for user message parts (derived from the
+ * Gemini API `Content.parts[]` schema). Non-text part variants (data,
+ * file, etc.) are preserved via the index signature but ignored by
+ * `resolveContent` since the adapter only surfaces human-readable text.
+ */
+interface GeminiContentPart {
+  text?: string;
+  [key: string]: unknown;
+}
+
+type GeminiMessageContent = string | GeminiContentPart[];
+
+export interface GeminiMessageEntry {
+  id?: string;
+  timestamp?: string;
+  type?: string;
+  /**
+   * Gemini CLI stores two different content shapes depending on the
+   * message origin:
+   * - `type: "user"` messages carry the raw Part[] from userContent.parts
+   *   (e.g. `[{ text: "hello" }]`).
+   * - `type: "gemini"` (assistant) messages carry a pre-joined string
+   *   built from `consolidatedParts.filter(p => p.text).join('').trim()`.
+   * Both forms must be normalized via resolveContent before any string
+   * operation is applied.
+   */
+  content?: GeminiMessageContent;
+  displayContent?: GeminiMessageContent;
+}
+
+export interface GeminiSessionFile {
+  sessionId?: string;
+  projectHash?: string;
+  startTime?: string;
+  lastUpdated?: string;
+  messages?: GeminiMessageEntry[];
+  directories?: string[];
+  kind?: string;
+}
+
+export interface GeminiSession {
+  sessionId: string;
+  projectPath: string;
+  summary: string;
+  sessionStart: Date;
+  lastActive: Date;
+  lastMessageType?: string;
+}
+
+const IDLE_THRESHOLD_MINUTES = 5;
+
+export class GeminiSessionParser {
+  /**
+   * Parse session file content into GeminiSession.
+   * Uses cached content if available, otherwise reads from disk.
+   */
+  parseSession(cachedContent: string | undefined, filePath: string): GeminiSession | null {
+    const content = cachedContent ?? this.readSessionFile(filePath);
+    if (content === null) return null;
+
+    const parsed = this.parseSessionJson(content);
+    if (!parsed?.sessionId) return null;
+
+    const messages = Array.isArray(parsed.messages) ? parsed.messages : [];
+    const lastEntry = messages.length > 0 ? messages[messages.length - 1] : undefined;
+    const fileStat = safeStat(filePath);
+
+    const lastActive =
+      this.parseTimestamp(parsed.lastUpdated) ||
+      this.parseTimestamp(lastEntry?.timestamp) ||
+      fileStat?.mtime ||
+      new Date();
+
+    const sessionStart = this.parseTimestamp(parsed.startTime) || lastActive;
+    const projectPath =
+      Array.isArray(parsed.directories) && parsed.directories.length > 0
+        ? parsed.directories[0]
+        : "";
+
+    return {
+      sessionId: parsed.sessionId,
+      projectPath,
+      summary: this.extractSummary(messages),
+      sessionStart,
+      lastActive,
+      lastMessageType: lastEntry?.type,
+    };
+  }
+
+  determineStatus(session: GeminiSession): AgentStatus {
+    const diffMs = Date.now() - session.lastActive.getTime();
+    const diffMinutes = diffMs / 60000;
+
+    if (diffMinutes > IDLE_THRESHOLD_MINUTES) {
+      return AgentStatus.IDLE;
+    }
+
+    if (session.lastMessageType === "gemini" || session.lastMessageType === "assistant") {
+      return AgentStatus.WAITING;
+    }
+
+    return AgentStatus.RUNNING;
+  }
+
+  /**
+   * Read the full conversation from a Gemini CLI session JSON file.
+   *
+   * Gemini sessions store messages in an array with `type` field — typically
+   * 'user' or 'gemini' for visible turns, with tool and system entries mixed in.
+   */
+  getConversation(sessionFilePath: string, options?: { verbose?: boolean }): ConversationMessage[] {
+    const verbose = options?.verbose ?? false;
+    const content = safeReadFile(sessionFilePath);
+    if (content === undefined) return [];
+
+    const parsed = this.parseSessionJson(content);
+    if (!Array.isArray(parsed?.messages)) return [];
+
+    const messages: ConversationMessage[] = [];
+    for (const entry of parsed.messages) {
+      const role = this.roleForEntry(entry?.type, verbose);
+      if (!role) continue;
+
+      const text = this.messageText(entry).trim();
+      if (!text) continue;
+
+      messages.push({
+        role,
+        content: text,
+        timestamp: entry.timestamp,
+      });
+    }
+
+    return messages;
+  }
+
+  /**
+   * Read a Gemini session JSON file and produce a {@link SessionSummary}.
+   * Returns null when the file is unreadable, the JSON doesn't parse,
+   * or the body lacks a sessionId.
+   */
+  fileToSessionSummary(filePath: string): SessionSummary | null {
+    const content = safeReadFile(filePath);
+    if (content === undefined) return null;
+
+    const parsed = this.parseSessionJson(content);
+    if (!parsed?.sessionId) return null;
+
+    const messages = Array.isArray(parsed.messages) ? parsed.messages : [];
+    const firstUserMessage = this.extractFirstUserMessage(messages);
+    const cwd =
+      Array.isArray(parsed.directories) && parsed.directories.length > 0
+        ? parsed.directories[0]
+        : "";
+
+    const stat = safeStat(filePath);
+    const lastEntryTimestamp = this.parseTimestamp(
+      messages.length > 0 ? messages[messages.length - 1]?.timestamp : undefined,
+    );
+    const lastActive =
+      this.parseTimestamp(parsed.lastUpdated) || lastEntryTimestamp || stat?.mtime || new Date();
+    const startedAt =
+      this.parseTimestamp(parsed.startTime) || stat?.birthtime || stat?.mtime || lastActive;
+
+    return {
+      type: "gemini_cli",
+      sessionId: parsed.sessionId,
+      cwd,
+      firstUserMessage,
+      lastActive,
+      startedAt,
+      sessionFilePath: filePath,
+    };
+  }
+
+  private parseSessionJson(content: string): GeminiSessionFile | null {
+    try {
+      return JSON.parse(content) as GeminiSessionFile;
+    } catch {
+      return null;
+    }
+  }
+
+  private readSessionFile(filePath: string): string | null {
+    try {
+      return fs.readFileSync(filePath, "utf-8");
+    } catch {
+      return null;
+    }
+  }
+
+  private roleForEntry(
+    entryType: string | undefined,
+    verbose: boolean,
+  ): ConversationMessage["role"] | null {
+    if (entryType === "user") return "user";
+    if (entryType === "gemini" || entryType === "assistant") return "assistant";
+    if (verbose && entryType) return "system";
+    return null;
+  }
+
+  private extractSummary(messages: GeminiMessageEntry[]): string {
+    for (let i = messages.length - 1; i >= 0; i--) {
+      const entry = messages[i];
+      if (entry?.type !== "user") continue;
+      const text = this.messageText(entry).trim();
+      if (text) return this.truncate(text, 120);
+    }
+
+    return "Gemini CLI session active";
+  }
+
+  /**
+   * Normalize an entry's content/displayContent into a plain string.
+   * Prefers displayContent when both are present (matches Gemini CLI's
+   * own rendering priority for the /chat UI).
+   */
+  private messageText(entry: GeminiMessageEntry): string {
+    const displayText = this.resolveContent(entry.displayContent);
+    if (displayText) return displayText;
+    return this.resolveContent(entry.content);
+  }
+
+  /**
+   * Collapse a Gemini message content field into plain text.
+   * Accepts either a pre-joined string (assistant turns) or a Part[]
+   * list (user turns carrying `[{text: "..."}]`). Non-text part
+   * variants (data, file) are dropped since this helper is only used
+   * for summary/conversation rendering.
+   */
+  private resolveContent(content: GeminiMessageContent | undefined): string {
+    if (!content) return "";
+    if (typeof content === "string") return content;
+    if (!Array.isArray(content)) return "";
+
+    const parts: string[] = [];
+    for (const part of content) {
+      if (part && typeof part.text === "string" && part.text) {
+        parts.push(part.text);
+      }
+    }
+    return parts.join("");
+  }
+
+  private extractFirstUserMessage(messages: GeminiMessageEntry[]): string {
+    for (const entry of messages) {
+      if (entry?.type !== "user") continue;
+      const text = this.messageText(entry).trim();
+      if (text) return text;
+    }
+    return "";
+  }
+
+  private parseTimestamp(value?: string): Date | null {
+    if (!value) return null;
+    const timestamp = new Date(value);
+    return Number.isNaN(timestamp.getTime()) ? null : timestamp;
+  }
+
+  private truncate(value: string, maxLength: number): string {
+    if (value.length <= maxLength) return value;
+    return `${value.slice(0, maxLength - 3)}...`;
+  }
+}

package/src/runtime/herdr/HerdrAgentDiscovery.ts

@@ -0,0 +1,178 @@
+import type { AgentInfo, AgentType } from "../../adapters/AgentAdapter.js";
+import { HerdrCliClient, type HerdrCommandRunner } from "./HerdrCliClient.js";
+import type { HerdrRuntimeRef } from "./HerdrRuntimeRef.js";
+
+export interface HerdrAgentPane {
+  agent: string;
+  agentSessionId?: string;
+  cwd?: string;
+  paneId: string;
+  workspaceId?: string;
+  tabId?: string;
+  foregroundPids?: number[];
+  shellPid?: number;
+}
+
+export async function fetchHerdrAgentPanes(runner?: HerdrCommandRunner): Promise<HerdrAgentPane[]> {
+  const client = new HerdrCliClient(runner);
+  const response = await client.runJson(["pane", "list"]);
+  return enrichHerdrAgentPanes(client, extractHerdrAgentPanes(response));
+}
+
+export function extractHerdrAgentPanes(value: unknown): HerdrAgentPane[] {
+  const snapshot = getObject(value, ["result", "snapshot"]);
+  const rawPanes = getArray(value, ["result", "panes"]) ?? snapshot?.agents;
+  if (!Array.isArray(rawPanes)) return [];
+
+  const panes: HerdrAgentPane[] = [];
+  for (const rawPane of rawPanes) {
+    if (!rawPane || typeof rawPane !== "object" || Array.isArray(rawPane)) continue;
+    const pane = rawPane as Record<string, unknown>;
+    const agent = nonEmptyString(pane.agent);
+    const paneId = nonEmptyString(pane.pane_id);
+    if (!agent || !paneId) continue;
+
+    const session = getObject(pane, ["agent_session"]);
+    const agentSessionId = nonEmptyString(session?.value);
+    const cwd = nonEmptyString(pane.cwd);
+    const workspaceId = nonEmptyString(pane.workspace_id);
+    const tabId = nonEmptyString(pane.tab_id);
+
+    panes.push({
+      agent,
+      paneId,
+      ...(agentSessionId ? { agentSessionId } : {}),
+      ...(cwd ? { cwd } : {}),
+      ...(workspaceId ? { workspaceId } : {}),
+      ...(tabId ? { tabId } : {}),
+    });
+  }
+
+  return panes;
+}
+
+export function findMatchingHerdrPane(
+  agent: Pick<AgentInfo, "type" | "projectPath" | "sessionId" | "pid">,
+  panes: readonly HerdrAgentPane[],
+): HerdrAgentPane | null {
+  if (agent.sessionId) {
+    const sessionMatches = panes.filter((pane) => pane.agentSessionId === agent.sessionId);
+    if (sessionMatches.length === 1) return sessionMatches[0]!;
+    if (sessionMatches.length > 1) return null;
+  }
+
+  const foregroundPidMatches = panes.filter((pane) => pane.foregroundPids?.includes(agent.pid));
+  if (foregroundPidMatches.length === 1) return foregroundPidMatches[0]!;
+  if (foregroundPidMatches.length > 1) return null;
+
+  const shellPidMatches = panes.filter((pane) => pane.shellPid === agent.pid);
+  if (shellPidMatches.length === 1) return shellPidMatches[0]!;
+  if (shellPidMatches.length > 1) return null;
+
+  const cwdMatches = panes.filter(
+    (pane) => pane.cwd === agent.projectPath && herdrAgentMatchesType(pane.agent, agent.type),
+  );
+  return cwdMatches.length === 1 ? cwdMatches[0]! : null;
+}
+
+export function herdrPaneToRuntimeRef(pane: HerdrAgentPane, agentName: string): HerdrRuntimeRef {
+  return {
+    session: "default",
+    paneId: pane.paneId,
+    agentName,
+    ...(pane.workspaceId ? { workspaceId: pane.workspaceId } : {}),
+    ...(pane.tabId ? { tabId: pane.tabId } : {}),
+  };
+}
+
+function herdrAgentMatchesType(herdrAgent: string, agentType: AgentType): boolean {
+  const normalized = herdrAgent.toLowerCase();
+  const candidates: Record<AgentType, readonly string[]> = {
+    claude: ["claude"],
+    codex: ["codex"],
+    copilot: ["copilot"],
+    gemini_cli: ["gemini", "gemini_cli"],
+    grok_cli: ["grok", "grok_cli"],
+    opencode: ["opencode"],
+    pi: ["pi"],
+    other: ["other"],
+  };
+  return candidates[agentType]?.includes(normalized) ?? false;
+}
+
+async function enrichHerdrAgentPanes(
+  client: HerdrCliClient,
+  panes: HerdrAgentPane[],
+): Promise<HerdrAgentPane[]> {
+  return Promise.all(
+    panes.map(async (pane) => {
+      try {
+        const response = await client.runJson(["pane", "process-info", "--pane", pane.paneId]);
+        return {
+          ...pane,
+          ...extractPaneProcessIds(response),
+        };
+      } catch {
+        return pane;
+      }
+    }),
+  );
+}
+
+function extractPaneProcessIds(value: unknown): Pick<HerdrAgentPane, "foregroundPids" | "shellPid"> {
+  const processInfo = getObject(value, ["result", "process_info"]);
+  if (!processInfo) return {};
+
+  const foregroundProcesses = processInfo.foreground_processes;
+  const foregroundPids: number[] = [];
+  if (Array.isArray(foregroundProcesses)) {
+    for (const foregroundProcess of foregroundProcesses) {
+      if (
+        !foregroundProcess ||
+        typeof foregroundProcess !== "object" ||
+        Array.isArray(foregroundProcess)
+      )
+        continue;
+      const pid = positiveIntegerOrNull((foregroundProcess as { pid?: unknown }).pid);
+      if (pid !== null) foregroundPids.push(pid);
+    }
+  }
+  const shellPid = positiveIntegerOrNull(processInfo.shell_pid);
+
+  return {
+    ...(foregroundPids.length > 0 ? { foregroundPids } : {}),
+    ...(shellPid !== null ? { shellPid } : {}),
+  };
+}
+
+function getArray(value: unknown, path: string[]): unknown[] | null {
+  let current = value;
+  for (const part of path) {
+    if (!current || typeof current !== "object" || Array.isArray(current) || !(part in current)) {
+      return null;
+    }
+    current = (current as Record<string, unknown>)[part];
+  }
+  return Array.isArray(current) ? current : null;
+}
+
+function getObject(value: unknown, path: string[]): Record<string, unknown> | null {
+  let current = value;
+  for (const part of path) {
+    if (!current || typeof current !== "object" || Array.isArray(current) || !(part in current)) {
+      return null;
+    }
+    current = (current as Record<string, unknown>)[part];
+  }
+  return current && typeof current === "object" && !Array.isArray(current)
+    ? (current as Record<string, unknown>)
+    : null;
+}
+
+function nonEmptyString(value: unknown): string | undefined {
+  return typeof value === "string" && value.trim() ? value : undefined;
+}
+
+function positiveIntegerOrNull(value: unknown): number | null {
+  return typeof value === "number" && Number.isInteger(value) && value > 0 ? value : null;
+}