linkshell-cli 0.3.11 → 0.3.12

package/src/runtime/acp/agent-workspace.ts

@@ -207,6 +207,8 @@ interface PendingStructuredInputWaiter {
 
 const PERMISSION_TIMEOUT_MS = 5 * 60_000;
 const MAX_TIMELINE_ITEMS = 200;
+const MAX_SNAPSHOT_ITEMS = 80;
+const MAX_SNAPSHOT_TEXT_BYTES = 128 * 1024;
 
 function id(prefix: string): string {
   return `${prefix}-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
@@ -221,6 +223,71 @@ function stringify(value: unknown): string {
   }
 }
 
+function truncateUtf8(value: string | undefined, maxBytes = MAX_SNAPSHOT_TEXT_BYTES): string | undefined {
+  if (!value) return value;
+  if (Buffer.byteLength(value, "utf8") <= maxBytes) return value;
+  let end = Math.min(value.length, maxBytes);
+  while (end > 0 && Buffer.byteLength(value.slice(0, end), "utf8") > maxBytes) {
+    end = Math.floor(end * 0.9);
+  }
+  return `${value.slice(0, end)}\n\n[truncated by LinkShell: original ${Buffer.byteLength(value, "utf8")} bytes]`;
+}
+
+function snapshotContentBlocks(
+  blocks: AgentContentBlock[] | undefined,
+  options: { stripImages?: boolean } = {},
+): AgentContentBlock[] | undefined {
+  if (!blocks) return undefined;
+  return blocks.map((block) =>
+    block.type === "image" && options.stripImages !== false
+      ? { ...block, data: undefined, text: block.text || "图片附件" }
+      : { ...block, text: truncateUtf8(block.text) },
+  );
+}
+
+function snapshotTimelineItem(
+  item: AgentTimelineItem,
+  options: { stripImages?: boolean } = {},
+): AgentTimelineItem {
+  return {
+    ...item,
+    content: snapshotContentBlocks(item.content, options),
+    text: truncateUtf8(item.text),
+    toolCall: item.toolCall
+      ? {
+          ...item.toolCall,
+          input: truncateUtf8(item.toolCall.input),
+          output: truncateUtf8(item.toolCall.output),
+        }
+      : undefined,
+    commandExecution: item.commandExecution
+      ? {
+          ...item.commandExecution,
+          command: truncateUtf8(item.commandExecution.command, 16 * 1024),
+          output: truncateUtf8(item.commandExecution.output),
+        }
+      : undefined,
+    fileChange: item.fileChange
+      ? {
+          ...item.fileChange,
+          diff: truncateUtf8(item.fileChange.diff),
+          summary: truncateUtf8(item.fileChange.summary),
+        }
+      : undefined,
+    permission: item.permission
+      ? {
+          ...item.permission,
+          toolInput: truncateUtf8(item.permission.toolInput),
+          context: truncateUtf8(item.permission.context),
+        }
+      : undefined,
+  };
+}
+
+function snapshotTimelineItems(items: AgentTimelineItem[]): AgentTimelineItem[] {
+  return items.slice(-MAX_SNAPSHOT_ITEMS).map((item) => snapshotTimelineItem(item));
+}
+
 function asRecord(value: unknown): Record<string, unknown> | undefined {
   return typeof value === "object" && value ? value as Record<string, unknown> : undefined;
 }
@@ -1583,7 +1650,7 @@ export class AgentWorkspaceProxy {
         hostDeviceId: this.input.hostDeviceId,
         payload: {
           conversation: existingConversation,
-          snapshot: this.timelines.get(existingConversation.id) ?? [],
+          snapshot: snapshotTimelineItems(this.timelines.get(existingConversation.id) ?? []),
         },
       }));
       return existingConversation;
@@ -1639,7 +1706,7 @@ export class AgentWorkspaceProxy {
       this.input.send(createEnvelope({
         type: "agent.v2.conversation.opened",
         hostDeviceId: this.input.hostDeviceId,
-        payload: { conversation, snapshot: this.timelines.get(conversation.id) ?? [] },
+        payload: { conversation, snapshot: snapshotTimelineItems(this.timelines.get(conversation.id) ?? []) },
       }));
       return conversation;
     } catch (error) {
@@ -1691,7 +1758,7 @@ export class AgentWorkspaceProxy {
     this.input.send(createEnvelope({
       type: "agent.v2.conversation.opened",
       hostDeviceId: this.input.hostDeviceId,
-      payload: { conversation, snapshot: this.timelines.get(conversation.id) ?? [] },
+      payload: { conversation, snapshot: snapshotTimelineItems(this.timelines.get(conversation.id) ?? []) },
     }));
     return conversation;
   }
@@ -2935,7 +3002,7 @@ export class AgentWorkspaceProxy {
     this.input.send(createEnvelope({
       type: "agent.v2.event",
       hostDeviceId: this.input.hostDeviceId,
-      payload: { conversationId, conversation, item },
+      payload: { conversationId, conversation, item: snapshotTimelineItem(item, { stripImages: false }) },
     }));
   }
 
@@ -2994,8 +3061,8 @@ export class AgentWorkspaceProxy {
     }
     const conversations = [...this.conversations.values()];
     const items = conversationId
-      ? this.timelines.get(conversationId) ?? []
-      : [...this.timelines.values()].flat();
+      ? snapshotTimelineItems(this.timelines.get(conversationId) ?? [])
+      : [];
     this.input.send(createEnvelope({
       type: "agent.v2.snapshot",
       hostDeviceId: this.input.hostDeviceId,

package/src/runtime/acp/codex-sessions.ts

@@ -62,6 +62,8 @@ interface CodexIndexEntry {
   updatedAt?: number;
 }
 
+type StoredToolStatus = NonNullable<StoredAgentTimelineItem["toolCall"]>["status"];
+
 function asRecord(value: unknown): Record<string, unknown> | undefined {
   return value && typeof value === "object" && !Array.isArray(value)
     ? value as Record<string, unknown>
@@ -366,7 +368,8 @@ function historyToolName(name: string | undefined): string {
   return name;
 }
 
-function historyToolKind(name: string | undefined): "tool_activity" | "command_execution" {
+function historyToolKind(name: string | undefined): "tool_activity" | "command_execution" | "file_change" {
+  if (name === "apply_patch") return "file_change";
   return name?.endsWith("exec_command") || name?.endsWith("write_stdin") ? "command_execution" : "tool_activity";
 }
 
@@ -390,6 +393,75 @@ function commandFromCodexTool(name: string | undefined, rawInput: unknown, outpu
   return { command, cwd, output, status: output === undefined ? "running" : "completed" };
 }
 
+function patchTextFromCodexTool(name: string | undefined, rawInput: unknown, input: string | undefined): string | undefined {
+  if (name !== "apply_patch") return undefined;
+  if (typeof rawInput === "string") return rawInput;
+  const record = asRecord(rawInput);
+  for (const key of ["patch", "input", "text", "content"]) {
+    const value = record?.[key];
+    if (typeof value === "string" && value.trim()) return value;
+  }
+  return input;
+}
+
+function fileChangeFromApplyPatch(
+  patchText: string | undefined,
+  status: StoredToolStatus,
+): StoredAgentTimelineItem["fileChange"] | undefined {
+  if (!patchText?.trim()) return undefined;
+  const entries: NonNullable<StoredAgentTimelineItem["fileChange"]>["entries"] = [];
+  let current: NonNullable<StoredAgentTimelineItem["fileChange"]>["entries"][number] | undefined;
+
+  const flush = () => {
+    if (!current?.path) return;
+    const existing = entries.find((entry) => entry.path === current!.path);
+    if (existing) {
+      existing.added = (existing.added ?? 0) + (current.added ?? 0);
+      existing.removed = (existing.removed ?? 0) + (current.removed ?? 0);
+      existing.kind ??= current.kind;
+    } else {
+      entries.push(current);
+    }
+  };
+
+  for (const rawLine of patchText.split(/\r?\n/)) {
+    const add = rawLine.match(/^\*\*\* Add File:\s+(.+)$/);
+    const update = rawLine.match(/^\*\*\* Update File:\s+(.+)$/);
+    const del = rawLine.match(/^\*\*\* Delete File:\s+(.+)$/);
+    const move = rawLine.match(/^\*\*\* Move to:\s+(.+)$/);
+    if (add || update || del) {
+      flush();
+      current = {
+        path: (add?.[1] ?? update?.[1] ?? del?.[1] ?? "").trim(),
+        kind: add ? "create" : del ? "delete" : "update",
+        added: 0,
+        removed: 0,
+      };
+      continue;
+    }
+    if (move?.[1] && current) {
+      current.path = move[1].trim();
+      current.kind = "move";
+      continue;
+    }
+    if (!current) continue;
+    if (rawLine.startsWith("+") && !rawLine.startsWith("+++")) {
+      current.added = (current.added ?? 0) + 1;
+    } else if (rawLine.startsWith("-") && !rawLine.startsWith("---")) {
+      current.removed = (current.removed ?? 0) + 1;
+    }
+  }
+  flush();
+
+  if (entries.length === 0) return undefined;
+  return {
+    entries,
+    diff: patchText,
+    summary: entries.map((entry) => [entry.kind, entry.path].filter(Boolean).join(" ")).join("\n"),
+    status,
+  };
+}
+
 function upsertHistoryTool(
   itemsById: Map<string, StoredAgentTimelineItem>,
   conversationId: string,
@@ -402,21 +474,27 @@ function upsertHistoryTool(
   const id = `history-tool:${callId}`;
   const existing = itemsById.get(id);
   const commandExecution = commandFromCodexTool(name, rawInput, existing?.commandExecution?.output);
+  const status: StoredToolStatus = existing?.toolCall?.status ?? "running";
+  const fileChange = fileChangeFromApplyPatch(patchTextFromCodexTool(name, rawInput, input), status);
   itemsById.set(id, {
     id,
     conversationId,
     type: "tool_call",
-    kind: historyToolKind(name),
+    kind: fileChange ? "file_change" : historyToolKind(name),
     itemId: callId,
     toolCall: {
       id: callId,
-      name: historyToolName(name),
-      input: input ?? existing?.toolCall?.input,
+      name: fileChange ? "文件修改" : historyToolName(name),
+      input: fileChange?.summary ?? input ?? existing?.toolCall?.input,
       output: existing?.toolCall?.output,
       createdAt: existing?.toolCall?.createdAt ?? createdAt,
-      status: existing?.toolCall?.status ?? "running",
+      status,
     },
     commandExecution: commandExecution ?? existing?.commandExecution,
+    fileChange: fileChange ?? existing?.fileChange,
+    text: fileChange
+      ? `已编辑 ${fileChange.entries.length} 个文件`
+      : existing?.text,
     createdAt: existing?.createdAt ?? createdAt,
     updatedAt: createdAt,
     metadata: { source: "device-history", provider: "codex" },
@@ -435,6 +513,9 @@ function completeHistoryTool(
   const commandExecution = existing?.commandExecution
     ? { ...existing.commandExecution, output, status: "completed" as const }
     : undefined;
+  const fileChange = existing?.fileChange
+    ? { ...existing.fileChange, summary: existing.fileChange.summary ?? output, status: "completed" as const }
+    : undefined;
   itemsById.set(id, {
     id,
     conversationId,
@@ -450,6 +531,8 @@ function completeHistoryTool(
       status: "completed",
     },
     commandExecution,
+    fileChange,
+    text: fileChange ? existing?.text ?? `已编辑 ${fileChange.entries.length} 个文件` : existing?.text,
     createdAt: existing?.createdAt ?? createdAt,
     updatedAt: createdAt,
     metadata: { source: "device-history", provider: "codex" },