agent-activity

1import { atom, read, update } from 'claude-code'
2import type { Register } from 'claude-code'
3
4import type { Activity, ActivityStatus } from '../types'
5
6const PANE = 'agent-activity'
7const KEEP = 300
8const entries = atom({ plugin: 'agent-activity', key: 'entries' } as const, [] as Activity[])
9
10const STATUS_STYLE: Record<ActivityStatus, { mark: string; color: string }> = {
11  running: { mark: '…', color: 'cyan' },
12  ok: { mark: '✓', color: 'green' },
13  error: { mark: '✗', color: 'red' },
14  denied: { mark: '⛔', color: 'red' },
15  ask: { mark: '?', color: 'yellow' },
16}
17
18// The field that says most about a call, in the order the built-in tools use them.
19export function summarize(input: unknown): string {
20  if (!input || typeof input !== 'object') return ''
21  const fields = input as Record<string, unknown>
22  for (const key of ['command', 'skill', 'file_path', 'pattern', 'url', 'query', 'description', 'prompt']) {
23    const value = fields[key]
24    if (typeof value === 'string' && value.trim()) return value.replace(/\s+/g, ' ').trim().slice(0, 160)
25  }
26  return ''
27}
28
29export function kindOf(tool: string): Activity['kind'] {
30  if (tool === 'Skill') return 'skill'
31  if (tool === 'Agent' || tool === 'Task') return 'subagent'
32  return 'tool'
33}
34
35export function statusOf(result: { deny?: string; isError?: boolean } | undefined): ActivityStatus {
36  if (!result) return 'error'
37  if (typeof result.deny === 'string') return 'denied'
38  return result.isError ? 'error' : 'ok'
39}
40
41export function clock(at: number): string {
42  const time = new Date(at)
43  return [time.getHours(), time.getMinutes(), time.getSeconds()].map(n => String(n).padStart(2, '0')).join(':')
44}
45
46// /activity toggles: it closes the pane when it is already open and opens it otherwise.
47export function paneToggle(openPaneIds: readonly string[]): 'open' | 'close' {
48  return openPaneIds.includes(PANE) ? 'close' : 'open'
49}
50
51export const register: Register = on => {
52  let sequence = 0
53
54  on('session.start', async ($, e, next) => {
55    await $.command.register({ name: 'activity', description: 'Open or close the live pane of tool calls, skills, subagents and permission decisions' })
56    void $.ui.open({ id: PANE, title: 'Agent activity' })
57    return next(e)
58  })
59
60  on('command.run', { command: 'activity' }, async $ => {
61    const open = (await $.ui.panes()).map(pane => pane.id)
62    if (paneToggle(open) === 'close') {
63      await $.ui.close({ id: PANE })
64      return { text: 'Agent activity pane closed.' }
65    }
66    await $.ui.open({ id: PANE, title: 'Agent activity' })
67    return { text: 'Agent activity pane opened.' }
68  })
69
70  on('tool.call', async ($, e, next) => {
71    const id = e.tool_use_id || `call-${(sequence += 1)}`
72    const entry: Activity = {
73      id,
74      at: await $.clock.now(),
75      kind: kindOf(e.tool),
76      label: e.tool,
77      detail: summarize(e),
78      status: 'running',
79      isSubagent: Boolean(e.agentId),
80    }
81    await update($, entries, list => [...list, entry].slice(-KEEP))
82    const result = await next(e)
83    const status = statusOf(result)
84    await update($, entries, list => list.map(one => (one.id === id ? { ...one, status } : one)))
85    return result
86  })
87
88  on('tool.check', async ($, e, next) => {
89    const verdict = await next(e)
90    if (verdict.decision !== 'allow') {
91      const entry: Activity = {
92        id: `check-${(sequence += 1)}`,
93        at: await $.clock.now(),
94        kind: 'permission',
95        label: `${verdict.decision} ${e.tool}`,
96        detail: verdict.reason ?? summarize(e.input),
97        status: verdict.decision === 'deny' ? 'denied' : 'ask',
98        isSubagent: false,
99      }
100      await update($, entries, list => [...list, entry].slice(-KEEP))
101    }
102    return verdict
103  })
104
105  on('agent.spawn', async ($, e, next) => {
106    const decision = await next(e)
107    const entry: Activity = {
108      id: `spawn-${(sequence += 1)}`,
109      at: await $.clock.now(),
110      kind: 'subagent',
111      label: `spawn ${e.subagentType}`,
112      detail: `${e.description}${'model' in decision && decision.model ? ` (${decision.model})` : ''}`,
113      status: 'deny' in decision ? 'denied' : 'ok',
114      isSubagent: false,
115    }
116    await update($, entries, list => [...list, entry].slice(-KEEP))
117    return decision
118  })
119
120  on('ui.render', { component: 'Pane', requestId: PANE }, async ($, e) => {
121    const { Box, Text } = $.ui.resolve(e)
122    const list = await read($, entries)
123    const room = Math.max(1, (e.viewport?.rows ?? 24) - 3)
124    const tools = list.filter(one => one.kind !== 'permission').length
125    const asks = list.filter(one => one.kind === 'permission').length
126
127    return (
128      <Box flexDirection="column">
129        <Text dimColor>{`${tools} calls · ${asks} permission prompts or denials`}</Text>
130        {list.length === 0 && <Text dimColor>Nothing yet.</Text>}
131        {list.slice(-room).map(entry => {
132          const style = STATUS_STYLE[entry.status]
133          return (
134            <Box key={entry.id} flexDirection="row" gap={1}>
135              <Text dimColor>{clock(entry.at)}</Text>
136              <Text color={style.color}>{style.mark}</Text>
137              <Text bold color={entry.kind === 'tool' ? undefined : 'magenta'}>
138                {`${entry.isSubagent ? '↳ ' : ''}${entry.label}`}
139              </Text>
140              <Text dimColor wrap="truncate-end">{entry.detail}</Text>
141            </Box>
142          )
143        })}
144      </Box>
145    )
146  })
147}
148

1export type ActivityStatus = 'running' | 'ok' | 'error' | 'denied' | 'ask'
2
3export type Activity = {
4  id: string
5  at: number
6  kind: 'tool' | 'skill' | 'subagent' | 'permission'
7  label: string
8  detail: string
9  status: ActivityStatus
10  isSubagent: boolean
11}
12
13declare module 'claude-code' {
14  interface PluginState {
15    'agent-activity': { entries: Activity[] }
16  }
17}
18