tool-lines

1import type { Register } from 'claude-code'
2
3const ENGINE_ROW_INDENT = 4
4const DOT_COLUMN_WIDTH = 2
5const MIN_TEXT_WIDTH = 20
6const MIN_LINE_CHARS = 120
7const MAX_LINE_ROWS = 3
8const MAX_SUBJECT_CHARS = 35
9const UNMEASURED_TERMINAL_COLUMNS = 120
10
11type ToolRow = {
12  tool: string
13  input: unknown
14  output?: unknown
15  isRunning: boolean
16  isErrored: boolean
17  isInterrupted: boolean
18}
19
20const SUBJECT_KEYS_BY_PRIORITY = [
21  'command',
22  'file_path',
23  'notebook_path',
24  'pattern',
25  'url',
26  'query',
27  'skill',
28  'description',
29  'path',
30  'action',
31  'prompt',
32]
33
34const isRecord = (value: unknown): value is Record<string, unknown> => typeof value === 'object' && value !== null
35
36const asOneLine = (text: string) => text.replace(/\s+/g, ' ').trim()
37
38function subjectOf(input: unknown): string {
39  if (!isRecord(input)) return ''
40  const subjectKey = SUBJECT_KEYS_BY_PRIORITY.find(key => typeof input[key] === 'string' && input[key] !== '')
41  if (subjectKey !== undefined) return asOneLine(String(input[subjectKey]))
42  return Object.keys(input).length === 0 ? '' : asOneLine(JSON.stringify(input))
43}
44
45const resultField = (row: ToolRow, name: string): unknown => (isRecord(row.output) ? row.output[name] : undefined)
46
47const printedOutput = (row: ToolRow) => `${resultField(row, 'stdout') ?? ''}\n${resultField(row, 'stderr') ?? ''}`
48
49type MetaProbe = (row: ToolRow) => string | undefined
50
51const META_PROBES: MetaProbe[] = [
52  row => (row.isInterrupted ? 'interrupted' : undefined),
53  row => (row.isErrored && !row.isInterrupted ? 'error' : undefined),
54  row => (resultField(row, 'backgroundTaskId') !== undefined ? 'background' : undefined),
55  row => (resultField(row, 'timedOutAfterMs') !== undefined ? 'timed out' : undefined),
56  row => (/Shell cwd was reset/.test(printedOutput(row)) ? 'shell cwd was reset' : undefined),
57  row => {
58    const note = resultField(row, 'returnCodeInterpretation')
59    return typeof note === 'string' && note !== '' ? note : undefined
60  },
61]
62
63type ToolLine = { tool: string; subject: string; meta: string }
64
65function clip(text: string, room: number): string {
66  if (text.length <= room) return text
67  return room <= 1 ? '…' : text.slice(0, room - 1) + '…'
68}
69
70function lineFor(row: ToolRow, columns: number): ToolLine {
71  const textWidth = Math.max(columns - ENGINE_ROW_INDENT - DOT_COLUMN_WIDTH, MIN_TEXT_WIDTH)
72  const lineBudget = Math.min(Math.max(textWidth, MIN_LINE_CHARS), textWidth * MAX_LINE_ROWS)
73  const meta = META_PROBES.map(probe => probe(row))
74    .filter(note => note !== undefined)
75    .join(', ')
76  const frameChars = `tool_call: ${row.tool}()`.length + (meta === '' ? 0 : ` - ${meta}`.length)
77  const subjectRoom = Math.min(MAX_SUBJECT_CHARS, lineBudget - frameChars)
78  return { tool: row.tool, subject: clip(subjectOf(row.input), subjectRoom), meta }
79}
80
81const hasFailed = (row: ToolRow) => row.isErrored || row.isInterrupted
82
83const hasFinished = (row: ToolRow) => !row.isRunning && row.output !== undefined
84
85function dotColor(row: ToolRow): 'error' | 'success' | 'warning' {
86  if (hasFailed(row)) return 'error'
87  return hasFinished(row) ? 'success' : 'warning'
88}
89
90let isTranscriptExpanded = false
91
92const isCompact = (surface: string) => surface === 'terminal' && !isTranscriptExpanded
93
94export const register: Register = on => {
95  on('ui.render', { component: 'UserMessage', props: { origin: { kind: 'composer' } } }, async ($, e, next) => {
96    if (e.surface === 'terminal' && e.props.isExpanded !== isTranscriptExpanded) {
97      isTranscriptExpanded = e.props.isExpanded
98      $.ui.invalidate('ui.render')
99    }
100    if (!isCompact(e.surface)) return next(e)
101    const { Box } = $.ui.resolve(e)
102    return Box({ flexDirection: 'column', marginBottom: 1, children: [await next(e)] })
103  })
104
105  on('ui.render', { component: 'ToolUse' }, async ($, e, next) => {
106    if (!isCompact(e.surface)) return next(e)
107    const { Box, Text } = $.ui.resolve(e)
108    const line = lineFor(e.props, e.viewport?.columns ?? UNMEASURED_TERMINAL_COLUMNS)
109    const metaStyle = e.props.isErrored ? { color: 'error' } : { dimColor: true }
110    const dotColumn = Box({
111      minWidth: DOT_COLUMN_WIDTH,
112      flexShrink: 0,
113      children: [Text({ color: dotColor(e.props), children: ['●'] })],
114    })
115    const textColumn = Box({
116      flexShrink: 1,
117      children: [
118        Text({
119          wrap: 'wrap',
120          children: [
121            Text({ dimColor: true, children: ['tool_call: '] }),
122            Text({ bold: true, children: [line.tool] }),
123            `(${line.subject})`,
124            ...(line.meta === '' ? [] : [Text({ ...metaStyle, children: [` - ${line.meta}`] })]),
125          ],
126        }),
127      ],
128    })
129    return Box({ flexDirection: 'row', marginLeft: 2, children: [dotColumn, textColumn] })
130  })
131
132  on('ui.render', { component: 'ToolGroup' }, async ($, e, next) => {
133    if (!isCompact(e.surface)) return next(e)
134    const unfoldedGroup = { ...e.props, isExpanded: true }
135    return next({ ...e, props: unfoldedGroup })
136  })
137
138  on('ui.render', { component: 'ToolResult' }, async ($, e, next) => {
139    if (!isCompact(e.surface)) return next(e)
140    const { Box } = $.ui.resolve(e)
141    return Box({})
142  })
143}
144