claude-code/tools/FileReadTool/UI.tsx

185 lines
22 KiB
TypeScript
Raw Normal View History

import type { ToolResultBlockParam } from '@anthropic-ai/sdk/resources/index.mjs';
import * as React from 'react';
import { extractTag } from 'src/utils/messages.js';
import { FallbackToolUseErrorMessage } from '../../components/FallbackToolUseErrorMessage.js';
import { FilePathLink } from '../../components/FilePathLink.js';
import { MessageResponse } from '../../components/MessageResponse.js';
import { Text } from '../../ink.js';
import { FILE_NOT_FOUND_CWD_NOTE, getDisplayPath } from '../../utils/file.js';
import { formatFileSize } from '../../utils/format.js';
import { getPlansDirectory } from '../../utils/plans.js';
import { getTaskOutputDir } from '../../utils/task/diskOutput.js';
import type { Input, Output } from './FileReadTool.js';
/**
* Check if a file path is an agent output file and extract the task ID.
* Agent output files follow the pattern: {projectTempDir}/tasks/{taskId}.output
*/
function getAgentOutputTaskId(filePath: string): string | null {
const prefix = `${getTaskOutputDir()}/`;
const suffix = '.output';
if (filePath.startsWith(prefix) && filePath.endsWith(suffix)) {
const taskId = filePath.slice(prefix.length, -suffix.length);
// Validate it looks like a task ID (alphanumeric, reasonable length)
if (taskId.length > 0 && taskId.length <= 20 && /^[a-zA-Z0-9_-]+$/.test(taskId)) {
return taskId;
}
}
return null;
}
export function renderToolUseMessage({
file_path,
offset,
limit,
pages
}: Partial<Input>, {
verbose
}: {
verbose: boolean;
}): React.ReactNode {
if (!file_path) {
return null;
}
// For agent output files, return empty string so no parentheses are shown
// The task ID is displayed separately by AssistantToolUseMessage
if (getAgentOutputTaskId(file_path)) {
return '';
}
const displayPath = verbose ? file_path : getDisplayPath(file_path);
if (pages) {
return <>
<FilePathLink filePath={file_path}>{displayPath}</FilePathLink>
{` · pages ${pages}`}
</>;
}
if (verbose && (offset || limit)) {
const startLine = offset ?? 1;
const lineRange = limit ? `lines ${startLine}-${startLine + limit - 1}` : `from line ${startLine}`;
return <>
<FilePathLink filePath={file_path}>{displayPath}</FilePathLink>
{` · ${lineRange}`}
</>;
}
return <FilePathLink filePath={file_path}>{displayPath}</FilePathLink>;
}
export function renderToolUseTag({
file_path
}: Partial<Input>): React.ReactNode {
const agentTaskId = file_path ? getAgentOutputTaskId(file_path) : null;
// Show agent task ID for Read tool when reading agent output
if (!agentTaskId) {
return null;
}
return <Text dimColor> {agentTaskId}</Text>;
}
export function renderToolResultMessage(output: Output): React.ReactNode {
// TODO: Render recursively
switch (output.type) {
case 'image':
{
const {
originalSize
} = output.file;
const formattedSize = formatFileSize(originalSize);
return <MessageResponse height={1}>
<Text>Read image ({formattedSize})</Text>
</MessageResponse>;
}
case 'notebook':
{
const {
cells
} = output.file;
if (!cells || cells.length < 1) {
return <Text color="error">No cells found in notebook</Text>;
}
return <MessageResponse height={1}>
<Text>
Read <Text bold>{cells.length}</Text> cells
</Text>
</MessageResponse>;
}
case 'pdf':
{
const {
originalSize
} = output.file;
const formattedSize = formatFileSize(originalSize);
return <MessageResponse height={1}>
<Text>Read PDF ({formattedSize})</Text>
</MessageResponse>;
}
case 'parts':
{
return <MessageResponse height={1}>
<Text>
Read <Text bold>{output.file.count}</Text>{' '}
{output.file.count === 1 ? 'page' : 'pages'} (
{formatFileSize(output.file.originalSize)})
</Text>
</MessageResponse>;
}
case 'text':
{
const {
numLines
} = output.file;
return <MessageResponse height={1}>
<Text>
Read <Text bold>{numLines}</Text>{' '}
{numLines === 1 ? 'line' : 'lines'}
</Text>
</MessageResponse>;
}
case 'file_unchanged':
{
return <MessageResponse height={1}>
<Text dimColor>Unchanged since last read</Text>
</MessageResponse>;
}
}
}
export function renderToolUseErrorMessage(result: ToolResultBlockParam['content'], {
verbose
}: {
verbose: boolean;
}): React.ReactNode {
if (!verbose && typeof result === 'string') {
// FileReadTool throws from call() so errors lack <tool_use_error> wrapping —
// check the raw string directly for the cwd note marker.
if (result.includes(FILE_NOT_FOUND_CWD_NOTE)) {
return <MessageResponse>
<Text color="error">File not found</Text>
</MessageResponse>;
}
if (extractTag(result, 'tool_use_error')) {
return <MessageResponse>
<Text color="error">Error reading file</Text>
</MessageResponse>;
}
}
return <FallbackToolUseErrorMessage result={result} verbose={verbose} />;
}
export function userFacingName(input: Partial<Input> | undefined): string {
if (input?.file_path?.startsWith(getPlansDirectory())) {
return 'Reading Plan';
}
if (input?.file_path && getAgentOutputTaskId(input.file_path)) {
return 'Read agent output';
}
return 'Read';
}
export function getToolUseSummary(input: Partial<Input> | undefined): string | null {
if (!input?.file_path) {
return null;
}
// For agent output files, just show the task ID
const agentTaskId = getAgentOutputTaskId(input.file_path);
if (agentTaskId) {
return agentTaskId;
}
return getDisplayPath(input.file_path);
}
//# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJuYW1lcyI6WyJUb29sUmVzdWx0QmxvY2tQYXJhbSIsIlJlYWN0IiwiZXh0cmFjdFRhZyIsIkZhbGxiYWNrVG9vbFVzZUVycm9yTWVzc2FnZSIsIkZpbGVQYXRoTGluayIsIk1lc3NhZ2VSZXNwb25zZSIsIlRleHQiLCJGSUxFX05PVF9GT1VORF9DV0RfTk9URSIsImdldERpc3BsYXlQYXRoIiwiZm9ybWF0RmlsZVNpemUiLCJnZXRQbGFuc0RpcmVjdG9yeSIsImdldFRhc2tPdXRwdXREaXIiLCJJbnB1dCIsIk91dHB1dCIsImdldEFnZW50T3V0cHV0VGFza0lkIiwiZmlsZVBhdGgiLCJwcmVmaXgiLCJzdWZmaXgiLCJzdGFydHNXaXRoIiwiZW5kc1dpdGgiLCJ0YXNrSWQiLCJzbGljZSIsImxlbmd0aCIsInRlc3QiLCJyZW5kZXJUb29sVXNlTWVzc2FnZSIsImZpbGVfcGF0aCIsIm9mZnNldCIsImxpbWl0IiwicGFnZXMiLCJQYXJ0aWFsIiwidmVyYm9zZSIsIlJlYWN0Tm9kZSIsImRpc3BsYXlQYXRoIiwic3RhcnRMaW5lIiwibGluZVJhbmdlIiwicmVuZGVyVG9vbFVzZVRhZyIsImFnZW50VGFza0lkIiwicmVuZGVyVG9vbFJlc3VsdE1lc3NhZ2UiLCJvdXRwdXQiLCJ0eXBlIiwib3JpZ2luYWxTaXplIiwiZmlsZSIsImZvcm1hdHRlZFNpemUiLCJjZWxscyIsImNvdW50IiwibnVtTGluZXMiLCJyZW5kZXJUb29sVXNlRXJyb3JNZXNzYWdlIiwicmVzdWx0IiwiaW5jbHVkZXMiLCJ1c2VyRmFjaW5nTmFtZSIsImlucHV0IiwiZ2V0VG9vbFVzZVN1bW1hcnkiXSwic291cmNlcyI6WyJVSS50c3giXSwic291cmNlc0NvbnRlbnQiOlsiaW1wb3J0IHR5cGUgeyBUb29sUmVzdWx0QmxvY2tQYXJhbSB9IGZyb20gJ0BhbnRocm9waWMtYWkvc2RrL3Jlc291cmNlcy9pbmRleC5tanMnXG5pbXBvcnQgKiBhcyBSZWFjdCBmcm9tICdyZWFjdCdcbmltcG9ydCB7IGV4dHJhY3RUYWcgfSBmcm9tICdzcmMvdXRpbHMvbWVzc2FnZXMuanMnXG5pbXBvcnQgeyBGYWxsYmFja1Rvb2xVc2VFcnJvck1lc3NhZ2UgfSBmcm9tICcuLi8uLi9jb21wb25lbnRzL0ZhbGxiYWNrVG9vbFVzZUVycm9yTWVzc2FnZS5qcydcbmltcG9ydCB7IEZpbGVQYXRoTGluayB9IGZyb20gJy4uLy4uL2NvbXBvbmVudHMvRmlsZVBhdGhMaW5rLmpzJ1xuaW1wb3J0IHsgTWVzc2FnZVJlc3BvbnNlIH0gZnJvbSAnLi4vLi4vY29tcG9uZW50cy9NZXNzYWdlUmVzcG9uc2UuanMnXG5pbXBvcnQgeyBUZXh0IH0gZnJvbSAnLi4vLi4vaW5rLmpzJ1xuaW1wb3J0IHsgRklMRV9OT1RfRk9VTkRfQ1dEX05PVEUsIGdldERpc3BsYXlQYXRoIH0gZnJvbSAnLi4vLi4vdXRpbHMvZmlsZS5qcydcbmltcG9ydCB7IGZvcm1hdEZpbGVTaXplIH0gZnJvbSAnLi4vLi4vdXRpbHMvZm9ybWF0LmpzJ1xuaW1wb3J0IHsgZ2V0UGxhbnNEaXJlY3RvcnkgfSBmcm9tICcuLi8uLi91dGlscy9wbGFucy5qcydcbmltcG9ydCB7IGdldFRhc2tPdXRwdXREaXIgfSBmcm9tICcuLi8uLi91dGlscy90YXNrL2Rpc2tPdXRwdXQuanMnXG5pbXBvcnQgdHlwZSB7IElucHV0LCBPdXRwdXQgfSBmcm9tICcuL0ZpbGVSZWFkVG9vbC5qcydcblxuLyoqXG4gKiBDaGVjayBpZiBhIGZpbGUgcGF0aCBpcyBhbiBhZ2VudCBvdXRwdXQgZmlsZSBhbmQgZXh0cmFjdCB0aGUgdGFzayBJRC5cbiAqIEFnZW50IG91dHB1dCBmaWxlcyBmb2xsb3cgdGhlIHBhdHRlcm46IHtwcm9qZWN0VGVtcERpcn0vdGFza3Mve3Rhc2tJZH0ub3V0cHV0XG4gKi9cbmZ1bmN0aW9uIGdldEFnZW50T3V0cHV0VGFza0lkKGZpbGVQYXRoOiBzdHJpbmcpOiBzdHJpbmcgfCBudWxsIHtcbiAgY29uc3QgcHJlZml4ID0gYCR7Z2V0VGFza091dHB1dERpcigpfS9gXG4gIGNvbnN0IHN1ZmZpeCA9ICcub3V0cHV0J1xuICBpZiAoZmlsZVBhdGguc3RhcnRzV2l0aChwcmVmaXgpICYmIGZpbGVQYXRoLmVuZHNXaXRoKHN1ZmZpeCkpIHtcbiAgICBjb25zdCB0YXNrSWQgPSBmaWxlUGF0aC5zbGljZShwcmVmaXgubGVuZ3RoLCAtc3VmZml4Lmxlbmd0aClcbiAgICAvLyBWYWxpZGF0ZSBpdCBsb29rcyBsaWtlIGEgdGFzayBJRCAoYWxwaGFudW1lcmljLCByZWFzb25hYmxlIGxlbmd0aClcbiAgICBpZiAoXG4gICAgICB0YXNrSWQubGVuZ3RoID4gMCAmJlxuICAgICAgdGFza0lkLmxlbmd0aCA8PSAyMCAmJlxuICAgICAgL15bYS16QS1aMC05Xy1dKyQvLnRlc3QodGFza0lkKVxuICAgICkge1xuICAgICAgcmV0dXJuIHRhc2tJZFxuICAgIH1cbiAgfVxuICByZXR1cm4gbnVsbFxufVxuXG5leHBvcnQgZnVuY3Rpb24gcmVuZGVyVG9vbFVzZU1lc3NhZ2UoXG4gIHsgZmlsZV9wYXRoLCBvZmZzZXQsIGxpbWl0LCBwYWdlcyB9OiBQYXJ0aWFsPElucHV0PixcbiAgeyB2ZXJib3NlIH06IHsgdmVyYm9zZTogYm9vbGVhbiB9LFxuKTogUmVhY3QuUmVhY3ROb2RlIHtcbiAgaWYgKCFmaWxlX3BhdGgpIHtcbiAgICByZXR1cm4gbnVsbFxuICB9XG5cbiAgLy8gRm9yIGFnZW50IG91dHB1dCBmaWxlcywgcmV0dXJuIGVtcHR5IHN0cmluZyBzbyBubyBwYXJlbnRoZXNlcyBhcmUgc2hvd25cbiAgLy8gVGhlIHRhc2sgSUQgaXMgZGlzcGxheWVkIHNlcGFyYXRlbHkgYnkgQXNzaXN0YW50VG9vbFVzZU1lc3NhZ2VcbiAgaWYgKGdldEFnZW50T3V0cHV0VGFza0lkKGZpbGVfcGF0aCkpIHtcbiAgICByZXR1cm4gJydcbiAgfVxuXG4gIGNvbnN0IGRpc3BsYXlQYXRoID0gdmVyYm9zZSA/IGZpbGVfcGF0aCA6IGdldERpc3BsYXlQYXRoKGZpbGVfcGF0aClcbiAgaWYgKHBhZ2VzKSB7XG4gICAgcmV0dXJuIChcbiAgICAgIDw+XG4gICAgICAgIDxGaWxlUGF0aExpbmsgZmlsZVBhdGg9e2ZpbGVfcGF0aH0+e2Rpc3BsYXlQYXRofTwvRmlsZVBhdGhMaW5rPlxuICAgICAgICB7YCDCtyBwYWdlcyAke3BhZ2VzfWB9XG4gICAgICA8Lz5cbiAgICApXG4gIH1cbiAgaWYgKHZlcmJvc2UgJiYgKG9mZnNldCB8fCBsaW1pdCkpIHtcbiAgICBjb25zdCBzdGFydExpbmUgPSBvZmZzZXQgPz8gMVxuICAgIGNvbnN0IGxpbmVSYW5nZSA9IGxpbWl0XG4gICAgICA/IGBsaW5lcyAke3N0YXJ0TGluZX0tJHtzdGFydExpbmUgKyBsaW1pdCAtIDF9YFxuICAgICAgOiBgZnJvbSBsaW5