mirror of
http://10.0.2.1:3031/sauer/claude-code.git
synced 2026-06-30 16:06:57 +10:00
185 lines
22 KiB
TypeScript
185 lines
22 KiB
TypeScript
|
|
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
|