agent-progress

1import { atom, read, update } from 'claude-code'
2import type { Hook, Register } from 'claude-code'
3
4import type { RunView } from '../types'
5
6const PANE = 'agent-progress'
7const POLL_MS = 3000
8// Milestones that say where the loop is going, not what happened.
9const QUIET = /^#\d+: (next step|waiting for)/
10const run = atom({ plugin: 'agent-progress', key: 'run' } as const, null)
11
12type Event = { at: number; kind: string; parent: string; name: string; text: string }
13
14const parse = (tsv: string): Event[] =>
15  tsv
16    .split('\n')
17    .filter(Boolean)
18    .map(line => {
19      const [at, kind, parent, name, ...text] = line.split('\t')
20      return { at: Number(at), kind, parent, name, text: text.join('\t') }
21    })
22
23const summarise = (name: string, events: Event[], isActive: boolean): RunView => {
24  const steps: Event[] = []
25  const subSteps: Event[] = []
26  let tokens = 0
27  for (const e of events) {
28    if (e.kind === 'begin' && e.parent === '') steps.push(e)
29    if (e.kind === 'end' && e.parent === '') {
30      const at = steps.findLastIndex(s => s.name === e.name)
31      if (at >= 0) steps.splice(at, 1)
32    }
33    if (e.kind === 'begin' && e.parent !== '') subSteps.push(e)
34    if (e.kind === 'end' && e.parent !== '') {
35      const at = subSteps.findLastIndex(s => s.name === e.name && s.parent === e.parent)
36      if (at >= 0) subSteps.splice(at, 1)
37    }
38    if (e.kind === 'usage') tokens += Number(/tokens=(\d+)/.exec(e.text)?.[1] ?? 0)
39  }
40  const step = steps.at(-1)
41  const subStep = subSteps.findLast(s => s.parent === step?.name)
42  return {
43    name,
44    isActive,
45    step: step?.name ?? '',
46    subStep: subStep?.name ?? '',
47    stepSince: step?.at ?? 0,
48    tokens,
49    milestones: events
50      .filter(e => e.kind === 'milestone' && !QUIET.test(e.text))
51      .slice(-12)
52      .map(e => ({ at: e.at, text: e.text })),
53  }
54}
55
56const tokensText = (n: number) =>
57  n >= 1e6 ? `${(n / 1e6).toFixed(1)}M` : n >= 1e3 ? `${Math.round(n / 1e3)}k` : `${n}`
58const minutes = (seconds: number) => `${Math.max(0, Math.round(seconds / 60))}m`
59const clock = (at: number) => new Date(at * 1000).toTimeString().slice(0, 5)
60
61type Dollar = Parameters<Hook<'session.start'>>[0]
62
63// The state of every run is in the main repository's .git, shared by the worktrees.
64let agentDir: string | undefined
65const seen = new Map<string, { mtimeMs: number; milestones: number }>()
66let newest = { mtimeMs: 0, name: '' }
67
68async function poll($: Dollar) {
69  agentDir ??= await $.process
70    .run(['git', 'rev-parse', '--path-format=absolute', '--git-common-dir'])
71    .then(r => (r.exitCode === 0 ? `${r.stdout.trim()}/agent` : ''))
72  if (!agentDir) return
73  const dirs = await $.fs.list(agentDir).catch(() => [])
74  for (const dir of dirs) {
75    const tsv = `${agentDir}/${dir.name}/progress.tsv`
76    const stat = await $.fs.stat(tsv).catch(() => undefined)
77    if (!stat) continue
78    if (stat.mtimeMs > newest.mtimeMs) newest = { mtimeMs: stat.mtimeMs, name: dir.name }
79    if (seen.get(dir.name)?.mtimeMs === stat.mtimeMs) continue
80    const events = parse(await $.fs.read(tsv))
81    const milestones = events.filter(e => e.kind === 'milestone' && !QUIET.test(e.text))
82    const before = seen.get(dir.name)
83    seen.set(dir.name, { mtimeMs: stat.mtimeMs, milestones: milestones.length })
84    // The first look at a run is its baseline: its history is not announced.
85    for (const m of before ? milestones.slice(before.milestones) : []) {
86      $.ui.toast(`${dir.name.split('-')[0]}: ${m.text}`)
87    }
88  }
89  if (!newest.name) return
90  const pid = await $.fs.read(`${agentDir}/${newest.name}/run.pid`).catch(() => '')
91  const isActive = pid.trim() !== '' && (await $.process.run(['kill', '-0', pid.trim()])).exitCode === 0
92  const view = summarise(newest.name, parse(await $.fs.read(`${agentDir}/${newest.name}/progress.tsv`)), isActive)
93  await update($, run, () => view)
94  if (!isActive) return $.ui.status(undefined)
95  const now = await $.clock.now()
96  const where = view.subStep ? `${view.step} ${view.subStep}` : view.step
97  $.ui.status(`#${view.name.split('-')[0]} ${where} · ${minutes(now / 1000 - view.stepSince)} · ${tokensText(view.tokens)} tok`)
98}
99
100export const register: Register = on => {
101  on('session.start', async ($, e, next) => {
102    await $.command.register({
103      name: 'agent-progress',
104      description: 'Show the active agent run in a pane',
105    })
106    $.clock.every(POLL_MS, () => poll($).catch(() => undefined))
107    return next(e)
108  })
109
110  on('command.run', { command: 'agent-progress' }, async $ => {
111    await $.ui.open({ id: PANE, title: 'Agent run' })
112    return { text: 'Agent run pane opened.' }
113  })
114
115  on('ui.render', { component: 'Pane', requestId: PANE }, async ($, e) => {
116    const { Box, Text } = $.ui.resolve(e)
117    const view: RunView | null = await read($, run)
118    if (!view) return <Text dimColor>No agent run found.</Text>
119    const room = Math.max(1, (e.viewport?.rows ?? 24) - 6)
120    return (
121      <Box flexDirection="column">
122        <Text bold>{view.name}</Text>
123        <Text dimColor={!view.isActive}>
124          {view.isActive ? 'running' : 'not running'} · {view.step || 'no step'}
125          {view.subStep ? ` · ${view.subStep}` : ''} · {tokensText(view.tokens)} tokens
126        </Text>
127        {view.milestones.slice(-room).map(m => (
128          <Text>
129            {clock(m.at)} {m.text}
130          </Text>
131        ))}
132      </Box>
133    )
134  })
135}
136

1export type RunView = {
2  name: string
3  isActive: boolean
4  step: string
5  subStep: string
6  stepSince: number
7  tokens: number
8  milestones: { at: number; text: string }[]
9}
10
11declare module 'claude-code' {
12  interface PluginState {
13    'agent-progress': { run: RunView | null }
14  }
15}
16