claude-code/tools/SkillTool/UI.tsx

128 lines
19 KiB
TypeScript
Raw Normal View History

import type { ToolResultBlockParam } from '@anthropic-ai/sdk/resources/index.mjs';
import * as React from 'react';
import { SubAgentProvider } from 'src/components/CtrlOToExpand.js';
import { FallbackToolUseErrorMessage } from 'src/components/FallbackToolUseErrorMessage.js';
import { FallbackToolUseRejectedMessage } from 'src/components/FallbackToolUseRejectedMessage.js';
import type { z } from 'zod/v4';
import type { Command } from '../../commands.js';
import { Byline } from '../../components/design-system/Byline.js';
import { Message as MessageComponent } from '../../components/Message.js';
import { MessageResponse } from '../../components/MessageResponse.js';
import { Box, Text } from '../../ink.js';
import type { Tools } from '../../Tool.js';
import type { ProgressMessage } from '../../types/message.js';
import { buildSubagentLookups, EMPTY_LOOKUPS } from '../../utils/messages.js';
import { plural } from '../../utils/stringUtils.js';
import type { inputSchema, Output, Progress } from './SkillTool.js';
type Input = z.infer<ReturnType<typeof inputSchema>>;
const MAX_PROGRESS_MESSAGES_TO_SHOW = 3;
const INITIALIZING_TEXT = 'Initializing…';
export function renderToolResultMessage(output: Output): React.ReactNode {
// Handle forked skill result
if ('status' in output && output.status === 'forked') {
return <MessageResponse height={1}>
<Text>
<Byline>{['Done']}</Byline>
</Text>
</MessageResponse>;
}
const parts: string[] = ['Successfully loaded skill'];
// Show tools count (only for inline skills)
if ('allowedTools' in output && output.allowedTools && output.allowedTools.length > 0) {
const count = output.allowedTools.length;
parts.push(`${count} ${plural(count, 'tool')} allowed`);
}
// Show model if non-default (only for inline skills)
if ('model' in output && output.model) {
parts.push(output.model);
}
return <MessageResponse height={1}>
<Text>
<Byline>{parts}</Byline>
</Text>
</MessageResponse>;
}
export function renderToolUseMessage({
skill
}: Partial<Input>, {
commands
}: {
commands?: Command[];
}): React.ReactNode {
if (!skill) {
return null;
}
// Look up the command to check if it came from the legacy /commands folder
const command = commands?.find(c => c.name === skill);
const displayName = command?.loadedFrom === 'commands_DEPRECATED' ? `/${skill}` : skill;
return displayName;
}
export function renderToolUseProgressMessage(progressMessages: ProgressMessage<Progress>[], {
tools,
verbose
}: {
tools: Tools;
verbose: boolean;
}): React.ReactNode {
if (!progressMessages.length) {
return <MessageResponse height={1}>
<Text dimColor>{INITIALIZING_TEXT}</Text>
</MessageResponse>;
}
// Take only the last few messages for display in non-verbose mode
const displayedMessages = verbose ? progressMessages : progressMessages.slice(-MAX_PROGRESS_MESSAGES_TO_SHOW);
const hiddenCount = progressMessages.length - displayedMessages.length;
const {
inProgressToolUseIDs
} = buildSubagentLookups(progressMessages.map(pm => pm.data));
return <MessageResponse>
<Box flexDirection="column">
<SubAgentProvider>
{displayedMessages.map(progressMessage => <Box key={progressMessage.uuid} height={1} overflow="hidden">
<MessageComponent message={progressMessage.data.message} lookups={EMPTY_LOOKUPS} addMargin={false} tools={tools} commands={[]} verbose={verbose} inProgressToolUseIDs={inProgressToolUseIDs} progressMessagesForMessage={[]} shouldAnimate={false} shouldShowDot={false} style="condensed" isTranscriptMode={false} isStatic={true} />
</Box>)}
</SubAgentProvider>
{hiddenCount > 0 && <Text dimColor>
+{hiddenCount} more tool {plural(hiddenCount, 'use')}
</Text>}
</Box>
</MessageResponse>;
}
export function renderToolUseRejectedMessage(_input: Input, {
progressMessagesForMessage,
tools,
verbose
}: {
progressMessagesForMessage: ProgressMessage<Progress>[];
tools: Tools;
verbose: boolean;
}): React.ReactNode {
return <>
{renderToolUseProgressMessage(progressMessagesForMessage, {
tools,
verbose
})}
<FallbackToolUseRejectedMessage />
</>;
}
export function renderToolUseErrorMessage(result: ToolResultBlockParam['content'], {
progressMessagesForMessage,
tools,
verbose
}: {
progressMessagesForMessage: ProgressMessage<Progress>[];
tools: Tools;
verbose: boolean;
}): React.ReactNode {
return <>
{renderToolUseProgressMessage(progressMessagesForMessage, {
tools,
verbose
})}
<FallbackToolUseErrorMessage result={result} verbose={verbose} />
</>;
}
//# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJuYW1lcyI6WyJUb29sUmVzdWx0QmxvY2tQYXJhbSIsIlJlYWN0IiwiU3ViQWdlbnRQcm92aWRlciIsIkZhbGxiYWNrVG9vbFVzZUVycm9yTWVzc2FnZSIsIkZhbGxiYWNrVG9vbFVzZVJlamVjdGVkTWVzc2FnZSIsInoiLCJDb21tYW5kIiwiQnlsaW5lIiwiTWVzc2FnZSIsIk1lc3NhZ2VDb21wb25lbnQiLCJNZXNzYWdlUmVzcG9uc2UiLCJCb3giLCJUZXh0IiwiVG9vbHMiLCJQcm9ncmVzc01lc3NhZ2UiLCJidWlsZFN1YmFnZW50TG9va3VwcyIsIkVNUFRZX0xPT0tVUFMiLCJwbHVyYWwiLCJpbnB1dFNjaGVtYSIsIk91dHB1dCIsIlByb2dyZXNzIiwiSW5wdXQiLCJpbmZlciIsIlJldHVyblR5cGUiLCJNQVhfUFJPR1JFU1NfTUVTU0FHRVNfVE9fU0hPVyIsIklOSVRJQUxJWklOR19URVhUIiwicmVuZGVyVG9vbFJlc3VsdE1lc3NhZ2UiLCJvdXRwdXQiLCJSZWFjdE5vZGUiLCJzdGF0dXMiLCJwYXJ0cyIsImFsbG93ZWRUb29scyIsImxlbmd0aCIsImNvdW50IiwicHVzaCIsIm1vZGVsIiwicmVuZGVyVG9vbFVzZU1lc3NhZ2UiLCJza2lsbCIsIlBhcnRpYWwiLCJjb21tYW5kcyIsImNvbW1hbmQiLCJmaW5kIiwiYyIsIm5hbWUiLCJkaXNwbGF5TmFtZSIsImxvYWRlZEZyb20iLCJyZW5kZXJUb29sVXNlUHJvZ3Jlc3NNZXNzYWdlIiwicHJvZ3Jlc3NNZXNzYWdlcyIsInRvb2xzIiwidmVyYm9zZSIsImRpc3BsYXllZE1lc3NhZ2VzIiwic2xpY2UiLCJoaWRkZW5Db3VudCIsImluUHJvZ3Jlc3NUb29sVXNlSURzIiwibWFwIiwicG0iLCJkYXRhIiwicHJvZ3Jlc3NNZXNzYWdlIiwidXVpZCIsIm1lc3NhZ2UiLCJyZW5kZXJUb29sVXNlUmVqZWN0ZWRNZXNzYWdlIiwiX2lucHV0IiwicHJvZ3Jlc3NNZXNzYWdlc0Zvck1lc3NhZ2UiLCJyZW5kZXJUb29sVXNlRXJyb3JNZXNzYWdlIiwicmVzdWx0Il0sInNvdXJjZXMiOlsiVUkudHN4Il0sInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB0eXBlIHsgVG9vbFJlc3VsdEJsb2NrUGFyYW0gfSBmcm9tICdAYW50aHJvcGljLWFpL3Nkay9yZXNvdXJjZXMvaW5kZXgubWpzJ1xuaW1wb3J0ICogYXMgUmVhY3QgZnJvbSAncmVhY3QnXG5pbXBvcnQgeyBTdWJBZ2VudFByb3ZpZGVyIH0gZnJvbSAnc3JjL2NvbXBvbmVudHMvQ3RybE9Ub0V4cGFuZC5qcydcbmltcG9ydCB7IEZhbGxiYWNrVG9vbFVzZUVycm9yTWVzc2FnZSB9IGZyb20gJ3NyYy9jb21wb25lbnRzL0ZhbGxiYWNrVG9vbFVzZUVycm9yTWVzc2FnZS5qcydcbmltcG9ydCB7IEZhbGxiYWNrVG9vbFVzZVJlamVjdGVkTWVzc2FnZSB9IGZyb20gJ3NyYy9jb21wb25lbnRzL0ZhbGxiYWNrVG9vbFVzZVJlamVjdGVkTWVzc2FnZS5qcydcbmltcG9ydCB0eXBlIHsgeiB9IGZyb20gJ3pvZC92NCdcbmltcG9ydCB0eXBlIHsgQ29tbWFuZCB9IGZyb20gJy4uLy4uL2NvbW1hbmRzLmpzJ1xuaW1wb3J0IHsgQnlsaW5lIH0gZnJvbSAnLi4vLi4vY29tcG9uZW50cy9kZXNpZ24tc3lzdGVtL0J5bGluZS5qcydcbmltcG9ydCB7IE1lc3NhZ2UgYXMgTWVzc2FnZUNvbXBvbmVudCB9IGZyb20gJy4uLy4uL2NvbXBvbmVudHMvTWVzc2FnZS5qcydcbmltcG9ydCB7IE1lc3NhZ2VSZXNwb25zZSB9IGZyb20gJy4uLy4uL2NvbXBvbmVudHMvTWVzc2FnZVJlc3BvbnNlLmpzJ1xuaW1wb3J0IHsgQm94LCBUZXh0IH0gZnJvbSAnLi4vLi4vaW5rLmpzJ1xuaW1wb3J0IHR5cGUgeyBUb29scyB9IGZyb20gJy4uLy4uL1Rvb2wuanMnXG5pbXBvcnQgdHlwZSB7IFByb2dyZXNzTWVzc2FnZSB9IGZyb20gJy4uLy4uL3R5cGVzL21lc3NhZ2UuanMnXG5pbXBvcnQgeyBidWlsZFN1YmFnZW50TG9va3VwcywgRU1QVFlfTE9PS1VQUyB9IGZyb20gJy4uLy4uL3V0aWxzL21lc3NhZ2VzLmpzJ1xuaW1wb3J0IHsgcGx1cmFsIH0gZnJvbSAnLi4vLi4vdXRpbHMvc3RyaW5nVXRpbHMuanMnXG5pbXBvcnQgdHlwZSB7IGlucHV0U2NoZW1hLCBPdXRwdXQsIFByb2dyZXNzIH0gZnJvbSAnLi9Ta2lsbFRvb2wuanMnXG5cbnR5cGUgSW5wdXQgPSB6LmluZmVyPFJldHVyblR5cGU8dHlwZW9mIGlucHV0U2NoZW1hPj5cblxuY29uc3QgTUFYX1BST0dSRVNTX01FU1NBR0VTX1RPX1NIT1cgPSAzXG5jb25zdCBJTklUSUFMSVpJTkdfVEVYVCA9ICdJbml0aWFsaXppbmfigKYnXG5cbmV4cG9ydCBmdW5jdGlvbiByZW5kZXJUb29sUmVzdWx0TWVzc2FnZShvdXRwdXQ6IE91dHB1dCk6IFJlYWN0LlJlYWN0Tm9kZSB7XG4gIC8vIEhhbmRsZSBmb3JrZWQgc2tpbGwgcmVzdWx0XG4gIGlmICgnc3RhdHVzJyBpbiBvdXRwdXQgJiYgb3V0cHV0LnN0YXR1cyA9PT0gJ2ZvcmtlZCcpIHtcbiAgICByZXR1cm4gKFxuICAgICAgPE1lc3NhZ2VSZXNwb25zZSBoZWlnaHQ9ezF9PlxuICAgICAgICA8VGV4dD5cbiAgICAgICAgICA8QnlsaW5lPntbJ0RvbmUnXX08L0J5bGluZT5cbiAgICAgICAgPC9UZXh0PlxuICAgICAgPC9NZXNzYWdlUmVzcG9uc2U+XG4gICAgKVxuICB9XG5cbiAgY29uc3QgcGFydHM6IHN0cmluZ1tdID0gWydTdWNjZXNzZnVsbHkgbG9hZGVkIHNraWxsJ11cblxuICAvLyBTaG93IHRvb2xzIGNvdW50IChvbmx5IGZvciBpbmxpbmUgc2tpbGxzKVxuICBpZiAoXG4gICAgJ2FsbG93ZWRUb29scycgaW4gb3V0cHV0ICYmXG4gICAgb3V0cHV0LmFsbG93ZWRUb29scyAmJlxuICAgIG91dHB1dC5hbGxvd2VkVG9vbHMubGVuZ3RoID4gMFxuICApIHtcbiAgICBjb25zdCBjb3VudCA9IG91dHB1dC5hbGxvd2VkVG9vbHMubGVuZ3RoXG4gICAgcGFydHMucHVzaChgJHtjb3VudH0gJHtwbHVyYWwoY291bnQsICd0b29sJyl9IGFsbG93ZWRgKVxuICB9XG5cbiAgLy8gU2hvdyBtb2RlbCBpZiBub24tZGVmYXVsdCAob25seSBmb3IgaW5saW5lIHNraWxscylcbiAgaWYgKCdtb2RlbCcgaW4gb3V0cHV0ICYmIG91dHB1dC5tb2RlbCkge1xuICAgIHBhcnRzLnB1c2gob3V0cHV0Lm1vZGVsKVxuICB9XG5cbiAgcmV0dXJuIChcbiAgICA8TWVzc2FnZVJlc3BvbnNlIGhlaWdodD17MX0+XG4gICAgICA8VGV4dD5cbiAgICAgICAgPEJ5bGluZT57cGFydHN9PC9CeWxpbmU+XG4gICA