tutti/packages/agent/gui/shared/workspaceAgentToolCallDisplay.ts

import type { WorkspaceAgentActivityTimelineItem } from "./workspaceAgentTimelineTypes";
import { translate } from "../i18n/index";
import {
  extractImageGenerationPreview,
  resolveImageGenerationCanonicalToolName
} from "./imageGenerationTool";
import {
  resolveCanonicalToolName,
  resolveLiveCanonicalToolName
} from "./workspaceAgentToolCallCanonical";
import { isAgentActivityTimelineItem } from "./workspaceAgentToolCallAgentActivity";
import { isOpaqueWorkspaceAgentToolCallIdentifier } from "./workspaceAgentToolCallIdentifiers";
import {
  AGENT_ACTIVITY_KINDS,
  TOOL_ACTIVITY_KIND_TRANSLATION_KEYS,
  legacyKindToToolName,
  type ToolActivityKind
} from "./workspaceAgentToolCallLabels";
import {
  isGenericToolLabel,
  itemId,
  looksLikeOpaqueToolCallID,
  normalizeToolNameToken,
  toolCallLabel
} from "./workspaceAgentToolCallPresentation";
import {
  collectToolPaths,
  firstPath,
  firstPresentString,
  recordValue,
  stringArrayFirstRecordValue,
  stringRecordValue,
  summarizeFileChangeCount,
  summarizeTodoProgress,
  summarizeToolInput,
  summarizeWebDomain
} from "./workspaceAgentToolCallSummary";
import { structuredToolText } from "./utils/structuredToolText";
export type { ToolActivityKind } from "./workspaceAgentToolCallLabels";
export type ToolCallStatusKind =
  | "working"
  | "completed"
  | "failed"
  | "canceled"
  | "waiting";

export interface WorkspaceAgentToolCallDisplay {
  id: string;
  name: string;
  status: string | null;
  statusKind: ToolCallStatusKind | null;
  detail?: string;
}

export function isWorkspaceAgentToolCallItem(
  item: WorkspaceAgentActivityTimelineItem
): boolean {
  const itemType = item.itemType.trim().toLowerCase();
  if (isAgentActivityTimelineItem(item)) {
    return false;
  }
  if (itemType === "call" || itemType.startsWith("call.")) {
    return true;
  }
  if (itemType.startsWith("approval.") || itemType.startsWith("interactive.")) {
    return true;
  }
  const payloadCallType = stringRecordValue(item.payload, "callType");
  if (payloadCallType === "approval" || payloadCallType === "interactive") {
    return true;
  }
  return isToolActivityTimelineItem(item);
}

export function buildWorkspaceAgentToolCallDisplay(
  item: WorkspaceAgentActivityTimelineItem
): WorkspaceAgentToolCallDisplay {
  const toolName = resolveWorkspaceAgentToolName(item);
  const detail = toolCallDetail(item, toolName);
  return {
    id: toolCallId(item),
    name: toolCallLabel(toolName),
    ...toolCallStatus(item),
    ...(detail ? { detail } : {})
  };
}

function toolCallId(item: WorkspaceAgentActivityTimelineItem): string {
  const metadata = recordValue(item.payload, "metadata");
  const callId = firstPresentString(
    stringRecordValue(item.payload, "callId"),
    stringRecordValue(item.payload, "callID"),
    stringRecordValue(item.payload, "call_id"),
    stringRecordValue(metadata, "callId"),
    stringRecordValue(metadata, "callID"),
    stringRecordValue(metadata, "call_id"),
    item.callId
  );
  return callId ? `call:${callId}` : itemId(item);
}

export function resolveWorkspaceAgentToolName(
  item: WorkspaceAgentActivityTimelineItem
): string | null {
  const metadata = recordValue(item.payload, "metadata");
  const input = recordValue(item.payload, "input");
  const output = recordValue(item.payload, "output");
  const rawPayloadToolName = firstPresentString(
    stringRecordValue(item.payload, "toolName"),
    stringRecordValue(metadata, "toolName")
  );
  const imageGenerationToolName = resolveImageGenerationCanonicalToolName({
    toolName: rawPayloadToolName,
    displayName: item.name,
    outputSavedPath: output?.savedPath,
    outputSavedPaths: output?.savedPaths,
    outputMimeType: output?.imageMimeType,
    outputText: output?.text,
    inputPrompt: input?.prompt
  });
  if (imageGenerationToolName) {
    return imageGenerationToolName;
  }
  const canonicalToolName = resolveCanonicalToolName(rawPayloadToolName);
  if (canonicalToolName) {
    return canonicalToolName;
  }
  const liveCanonicalToolName = resolveLiveCanonicalToolName(
    item,
    rawPayloadToolName,
    resolveLegacyACPToolName(item)
  );
  if (
    typeof liveCanonicalToolName === "string" &&
    liveCanonicalToolName.trim()
  ) {
    return liveCanonicalToolName;
  }
  if (liveCanonicalToolName === null && !rawPayloadToolName) {
    return null;
  }
  return resolveLegacyToolName(item) ?? resolveLegacyNamedToolName(item);
}

function resolveFetchToolName(
  payload: Record<string, unknown> | undefined
): string | null {
  const input = recordValue(payload, "input");
  const action = recordValue(input, "action");
  const actionType = normalizeToolToken(stringRecordValue(action, "type"));
  switch (actionType) {
    case "search":
    case "search_query":
    case "web_search":
      return "WebSearch";
    case "open_page":
    case "open":
    case "fetch":
    case "web_fetch":
      return "WebFetch";
    default:
      break;
  }
  if (
    firstPresentString(
      stringRecordValue(action, "url"),
      stringRecordValue(input, "url")
    )
  ) {
    return "WebFetch";
  }
  if (
    firstPresentString(
      stringRecordValue(action, "query"),
      stringRecordValue(input, "query"),
      stringRecordValue(input, "search_query"),
      stringRecordValue(input, "searchQuery")
    )
  ) {
    return "WebSearch";
  }
  return null;
}

function resolveLegacyToolName(
  item: WorkspaceAgentActivityTimelineItem
): string | null {
  const metadata = recordValue(item.payload, "metadata");
  const rawKind =
    stringRecordValue(item.payload, "activityKind") ??
    stringRecordValue(item.payload, "activity_kind") ??
    stringRecordValue(metadata, "activityKind") ??
    stringRecordValue(metadata, "activity_kind");
  const normalizedKind = normalizeToolToken(rawKind);
  if (AGENT_ACTIVITY_KINDS.has(normalizedKind as ToolActivityKind)) {
    return null;
  }
  if (isToolActivityKind(normalizedKind)) {
    return legacyKindToToolName(normalizedKind);
  }

  return resolveLegacyACPToolName(item);
}

function resolveLegacyACPToolName(
  item: WorkspaceAgentActivityTimelineItem
): string | null {
  const input = recordValue(item.payload, "input");
  const normalizedKind = normalizeToolToken(
    firstPresentString(
      stringRecordValue(item.payload, "kind"),
      stringRecordValue(input, "kind")
    )
  );
  const normalizedTitle = normalizeToolToken(
    firstPresentString(
      stringRecordValue(item.payload, "title"),
      stringRecordValue(input, "title"),
      item.name
    )
  );
  const fetchToolName = resolveFetchToolName(item.payload);

  if (fetchToolName) {
    return fetchToolName;
  }

  if (normalizedKind === "fetch") {
    return "WebFetch";
  }
  if (normalizedKind === "read") {
    return stringRecordValue(input, "pattern") ? "Glob" : "Read";
  }
  if (normalizedKind === "think") {
    return normalizedTitle === "update_todo" || Array.isArray(input?.todos)
      ? "TodoWrite"
      : "Think";
  }
  if (normalizedKind === "search") {
    switch (normalizedTitle) {
      case "glob":
      case "find":
      case "fd":
      case "file_search":
        return "Glob";
      case "grep":
      case "rg":
      case "ripgrep":
      case "codebase_search":
        return "Grep";
      default:
        return "Bash";
    }
  }
  if (normalizedKind === "other") {
    switch (normalizedTitle) {
      case "todowrite":
      case "updatetodo":
        return "TodoWrite";
      case "skill":
        return "Skill";
      case "task":
      case "subagent":
      case "runsubagent":
      case "delegatetask":
      case "delegateagent":
      case "agent":
        return "Agent";
      case "rg":
      case "ripgrep":
        return "Grep";
      case "find":
      case "fd":
        return "Glob";
      default:
        return null;
    }
  }
  switch (normalizedKind) {
    case "execute":
    case "exec":
    case "command":
    case "shell":
      return "Bash";
    case "edit":
    case "move":
      return "Edit";
    case "delete":
      return "Write";
    default:
      return null;
  }
}

function resolveLegacyNamedToolName(
  item: WorkspaceAgentActivityTimelineItem
): string | null {
  const metadata = recordValue(item.payload, "metadata");
  const rawTitle = firstPresentString(
    stringRecordValue(item.payload, "tool"),
    stringRecordValue(metadata, "tool"),
    item.name
  );
  if (looksLikeOpaqueToolCallID(rawTitle, item)) {
    return null;
  }
  const normalizedTitle = normalizeToolNameToken(rawTitle);
  switch (normalizedTitle) {
    case "approval":
      return "Approval";
    case "askuserquestion":
      return "AskUserQuestion";
    case "enterplanmode":
      return "EnterPlanMode";
    case "exitplanmode":
      return "ExitPlanMode";
    case "toolsearch":
      return "ToolSearch";
    case "skill":
      return "Skill";
    case "bash":
    case "shell":
    case "exec":
    case "execcommand":
    case "runshellcommand":
      return "Bash";
    case "read":
    case "readfile":
      return "Read";
    case "write":
    case "writefile":
      return "Write";
    case "edit":
    case "editfile":
    case "multiedit":
      return "Edit";
    case "grep":
      return "Grep";
    case "glob":
      return "Glob";
    case "websearch":
      return "WebSearch";
    case "webfetch":
      return "WebFetch";
    case "todowrite":
      return "TodoWrite";
    case "task":
    case "subagent":
    case "runsubagent":
    case "delegatetask":
    case "delegateagent":
    case "agent":
      return "Agent";
    default:
      break;
  }

  const lowerTitle = rawTitle.trim().toLowerCase();
  if (lowerTitle.startsWith("read ")) {
    return "Read";
  }
  if (lowerTitle.startsWith("write ")) {
    return "Write";
  }
  if (lowerTitle.startsWith("edit ")) {
    return "Edit";
  }

  if (
    !normalizedTitle ||
    isGenericToolLabel(normalizedTitle) ||
    isOpaqueWorkspaceAgentToolCallIdentifier(rawTitle, item)
  ) {
    return null;
  }
  return rawTitle.trim();
}

function toolCallStatus(item: WorkspaceAgentActivityTimelineItem): {
  status: string | null;
  statusKind: ToolCallStatusKind | null;
} {
  const payload = item.payload;
  const metadata = recordValue(payload, "metadata");
  const hasStructuredError = Boolean(
    structuredToolText(recordValue(payload, "error")) ||
    structuredToolText(recordValue(metadata, "error"))
  );
  const payloadStatus = firstPresentString(
    item.status,
    stringRecordValue(payload, "status"),
    stringRecordValue(payload, "activityStatus"),
    statusFromActivityEventType(activityEventType(item))
  );
  switch (normalizeToolToken(payloadStatus)) {
    case "active":
    case "running":
    case "working":
      return {
        status: translate("agentHost.agentTool.statusWorking"),
        statusKind: "working"
      };
    case "completed":
    case "done":
    case "success":
    case "succeeded":
      return {
        status: translate("agentHost.agentTool.statusCompleted"),
        statusKind: "completed"
      };
    case "failed":
    case "error":
    case "rejected":
      return {
        status: translate("agentHost.agentTool.statusFailed"),
        statusKind: "failed"
      };
    case "canceled":
    case "cancelled":
      return {
        status: translate("agentHost.agentTool.statusCanceled"),
        statusKind: "canceled"
      };
    case "pending":
    case "waiting":
    case "awaiting_approval":
    case "waiting_approval":
    case "waiting_input":
      return {
        status: translate("agentHost.agentTool.statusWaiting"),
        statusKind: "waiting"
      };
    default:
      return hasStructuredError
        ? {
            status: translate("agentHost.agentTool.statusFailed"),
            statusKind: "failed"
          }
        : { status: null, statusKind: null };
  }
}

function toolCallDetail(
  item: WorkspaceAgentActivityTimelineItem,
  toolName: string | null
): string | null {
  const metadata = recordValue(item.payload, "metadata");
  const metadataInput = recordValue(metadata, "input");
  const metadataError = recordValue(metadata, "error");
  const payloadInput = recordValue(item.payload, "input");
  const payloadOutput = recordValue(item.payload, "output");
  const payloadError = recordValue(item.payload, "error");
  const structuredError =
    structuredToolText(payloadError) ?? structuredToolText(metadataError);
  const paths = collectToolPaths(metadata, metadataInput, payloadInput);
  const rawToolName = firstPresentString(
    stringRecordValue(metadata, "tool"),
    stringRecordValue(item.payload, "tool"),
    toolNameFromActivityKey(toolActivityKey(item)),
    item.name
  );
  const visibleRawToolName = isOpaqueWorkspaceAgentToolCallIdentifier(
    rawToolName,
    item
  )
    ? ""
    : rawToolName;
  const fallback = firstPresentString(
    commandDetail(
      item.payload,
      metadata,
      metadataInput,
      metadataError,
      payloadInput,
      payloadError
    ),
    toolContentDetail(
      item.payload,
      metadata,
      metadataInput,
      metadataError,
      payloadInput,
      payloadError
    ),
    structuredError,
    item.content,
    stringRecordValue(item.payload, "summary"),
    stringRecordValue(item.payload, "text"),
    stringRecordValue(metadata, "summary")
  );
  const query = firstPresentString(
    stringRecordValue(item.payload, "query"),
    stringArrayFirstRecordValue(item.payload, "search_query"),
    stringArrayFirstRecordValue(item.payload, "searchQuery"),
    stringRecordValue(metadata, "query"),
    stringRecordValue(metadataInput, "query"),
    stringArrayFirstRecordValue(metadataInput, "search_query"),
    stringArrayFirstRecordValue(metadataInput, "searchQuery"),
    stringRecordValue(payloadInput, "query"),
    stringArrayFirstRecordValue(payloadInput, "search_query"),
    stringArrayFirstRecordValue(payloadInput, "searchQuery"),
    stringRecordValue(recordValue(payloadInput, "action"), "query"),
    stringRecordValue(recordValue(payloadInput, "action"), "url")
  );
  const webTarget = firstPresentString(
    query,
    stringRecordValue(metadataInput, "url"),
    stringRecordValue(payloadInput, "url")
  );
  const pattern = firstPresentString(
    stringRecordValue(item.payload, "pattern"),
    stringRecordValue(metadata, "pattern"),
    stringRecordValue(metadataInput, "pattern"),
    stringRecordValue(payloadInput, "pattern")
  );
  const command = firstPresentString(
    stringRecordValue(metadata, "command"),
    stringRecordValue(metadataInput, "cmd"),
    stringRecordValue(metadataInput, "command"),
    stringRecordValue(payloadInput, "cmd"),
    stringRecordValue(payloadInput, "command")
  );
  const path = firstPresentString(
    stringRecordValue(metadataInput, "path"),
    stringRecordValue(metadataInput, "file_path"),
    stringRecordValue(metadataInput, "fileName"),
    stringRecordValue(metadataInput, "filename"),
    stringRecordValue(payloadInput, "path"),
    stringRecordValue(payloadInput, "file_path"),
    stringRecordValue(payloadInput, "fileName"),
    stringRecordValue(payloadInput, "filename")
  );
  const inputSummary = firstPresentString(
    summarizeToolInput(metadataInput),
    summarizeToolInput(payloadInput)
  );
  const normalizedToolName = normalizeToolNameToken(toolName);
  const multiFileSummary = summarizeFileChangeCount(
    payloadOutput,
    payloadInput,
    item.payload,
    metadataInput
  );
  const todoSummary = summarizeTodoProgress(payloadInput);
  const webDomainSummary = summarizeWebDomain(webTarget);
  const imageGeneration = extractImageGenerationPreview({
    toolName,
    displayName: visibleRawToolName,
    outputSavedPath: payloadOutput?.savedPath,
    outputSavedPaths: payloadOutput?.savedPaths,
    outputMimeType: payloadOutput?.imageMimeType,
    outputText: payloadOutput?.text,
    inputPrompt: payloadInput?.prompt,
    payloadInputPrompt: metadataInput?.prompt
  });

  switch (normalizedToolName) {
    case "imagegeneration":
      return (
        imageGeneration.prompt ||
        imageGeneration.imageUri ||
        fallback ||
        inputSummary ||
        null
      );
    case "read":
    case "write":
    case "edit":
    case "multiedit":
      return (
        multiFileSummary ||
        path ||
        firstPath(paths) ||
        fallback ||
        inputSummary ||
        command ||
        null
      );
    case "glob":
      return (
        pattern ||
        query ||
        path ||
        firstPath(paths) ||
        fallback ||
        inputSummary ||
        null
      );
    case "bash":
      return (
        command || path || firstPath(paths) || fallback || inputSummary || null
      );
    case "grep":
      return (
        query ||
        pattern ||
        fallback ||
        inputSummary ||
        command ||
        firstPath(paths) ||
        null
      );
    case "websearch":
      return (
        query ||
        webDomainSummary ||
        pattern ||
        fallback ||
        inputSummary ||
        command ||
        firstPath(paths) ||
        null
      );
    case "webfetch":
      return (
        webDomainSummary ||
        query ||
        pattern ||
        fallback ||
        inputSummary ||
        command ||
        firstPath(paths) ||
        null
      );
    case "todowrite":
      return todoSummary || inputSummary || fallback || null;
    case "toolsearch":
    case "skill":
    case "task":
    case "agent":
      return (
        inputSummary ||
        fallback ||
        path ||
        firstPath(paths) ||
        query ||
        pattern ||
        command ||
        visibleRawToolName ||
        null
      );
    default:
      return (
        fallback ||
        command ||
        path ||
        firstPath(paths) ||
        query ||
        pattern ||
        inputSummary ||
        visibleRawToolName ||
        null
      );
  }
}

function commandDetail(
  payload: Record<string, unknown> | undefined,
  metadata: Record<string, unknown> | undefined,
  metadataInput: Record<string, unknown> | undefined,
  metadataError: Record<string, unknown> | undefined,
  payloadInput: Record<string, unknown> | undefined,
  payloadError: Record<string, unknown> | undefined
): string {
  return firstPresentString(
    stringRecordValue(payload, "command"),
    stringRecordValue(payload, "cmd"),
    stringRecordValue(metadata, "command"),
    stringRecordValue(metadata, "cmd"),
    stringRecordValue(metadataInput, "cmd"),
    stringRecordValue(metadataInput, "command"),
    stringRecordValue(metadataError, "cmd"),
    stringRecordValue(metadataError, "command"),
    stringRecordValue(payloadInput, "cmd"),
    stringRecordValue(payloadInput, "command"),
    stringRecordValue(payloadError, "cmd"),
    stringRecordValue(payloadError, "command")
  );
}

function toolContentDetail(
  payload: Record<string, unknown> | undefined,
  metadata: Record<string, unknown> | undefined,
  metadataInput: Record<string, unknown> | undefined,
  metadataError: Record<string, unknown> | undefined,
  payloadInput: Record<string, unknown> | undefined,
  payloadError: Record<string, unknown> | undefined
): string {
  return firstPresentString(
    stringRecordValue(payload, "query"),
    stringRecordValue(payload, "search_query"),
    stringRecordValue(payload, "searchQuery"),
    stringRecordValue(payload, "pattern"),
    stringRecordValue(payload, "glob"),
    stringRecordValue(payload, "regex"),
    stringRecordValue(payload, "url"),
    stringRecordValue(payload, "patch"),
    stringRecordValue(metadata, "query"),
    stringRecordValue(metadata, "pattern"),
    stringRecordValue(metadata, "glob"),
    stringRecordValue(metadata, "regex"),
    stringRecordValue(metadata, "url"),
    stringRecordValue(metadata, "patch"),
    stringRecordValue(metadataInput, "query"),
    stringRecordValue(metadataInput, "pattern"),
    stringRecordValue(metadataInput, "glob"),
    stringRecordValue(metadataInput, "regex"),
    stringRecordValue(metadataInput, "url"),
    stringRecordValue(metadataInput, "patch"),
    stringRecordValue(metadataError, "query"),
    stringRecordValue(metadataError, "pattern"),
    stringRecordValue(metadataError, "url"),
    stringRecordValue(payloadInput, "query"),
    stringRecordValue(payloadInput, "search_query"),
    stringRecordValue(payloadInput, "searchQuery"),
    stringRecordValue(payloadInput, "pattern"),
    stringRecordValue(payloadInput, "glob"),
    stringRecordValue(payloadInput, "regex"),
    stringRecordValue(payloadInput, "url"),
    stringRecordValue(payloadInput, "patch"),
    stringRecordValue(payloadError, "query"),
    stringRecordValue(payloadError, "pattern"),
    stringRecordValue(payloadError, "url")
  );
}

function isToolActivityTimelineItem(
  item: WorkspaceAgentActivityTimelineItem
): boolean {
  const eventType = activityEventType(item);
  if (!eventType.startsWith("activity.")) {
    return false;
  }
  const activityKey = toolActivityKey(item);
  if (activityKey.startsWith("tool.")) {
    return true;
  }
  return resolveLegacyToolName(item) !== null;
}

function activityEventType(item: WorkspaceAgentActivityTimelineItem): string {
  const itemType = item.itemType.trim().toLowerCase();
  if (itemType.startsWith("activity.")) {
    return itemType;
  }
  return firstPresentString(
    stringRecordValue(item.payload, "eventKey"),
    item.itemType
  )
    .trim()
    .toLowerCase();
}

function statusFromActivityEventType(eventType: string): string {
  switch (eventType) {
    case "activity.started":
    case "call.started":
      return "running";
    case "activity.completed":
    case "call.completed":
      return "completed";
    case "activity.failed":
    case "call.failed":
    case "call.errored":
      return "failed";
    default:
      return "";
  }
}

function toolActivityKey(item: WorkspaceAgentActivityTimelineItem): string {
  const metadata = recordValue(item.payload, "metadata");
  return firstPresentString(
    stringRecordValue(item.payload, "activityKey"),
    stringRecordValue(item.payload, "activity_key"),
    stringRecordValue(metadata, "activityKey"),
    stringRecordValue(metadata, "activity_key")
  )
    .trim()
    .toLowerCase();
}

function toolNameFromActivityKey(activityKey: string): string {
  return activityKey.startsWith("tool.")
    ? activityKey.slice("tool.".length)
    : "";
}

function isToolActivityKind(value: string): value is ToolActivityKind {
  return value in TOOL_ACTIVITY_KIND_TRANSLATION_KEYS;
}

function normalizeToolToken(value: string | undefined): string {
  return (
    value
      ?.trim()
      .toLowerCase()
      .replace(/[-\s]+/gu, "_") ?? ""
  );
}