agent-bar

1import type { EngineInterface, Register } from 'claude-code'
2
3const PROGRESS = 'mcp__savvy-progress__progress'
4const FLOW = { plugin: 'savvy-progress', key: 'flow' } as const
5
6type Row = { title: string; tier: string; isDone: boolean }
7
8// The module's own memory of this session's rows; a reload starts it over.
9const s = {
10  lastPrompt: '',
11  title: '',
12  todos: [] as Row[],
13  tasks: new Map<string, Row>(),
14  agents: [] as Row[],
15  agentRow: new Map<string, Row>(),
16}
17
18// savvy-progress only knows its own tiers; map ours onto them.
19function tierOf(type: string, model: string | undefined): string {
20  const bare = type.replace(/^[^:]*:/, '')
21  if (bare.startsWith('savvy-')) return bare.slice('savvy-'.length)
22  if (model === 'fable') return 'fable'
23  if (model === 'sonnet' || model === 'haiku') return 'light'
24  if (bare === 'Explore') return 'medium'
25  return 'careful'
26}
27
28// "/plan Unframed ATS, HR, ..." -> "Unframed ATS, HR"
29function titleOf(prompt: string): string {
30  const words = prompt.replace(/^\/\S+\s*/, '').trim().split(/\s+/).slice(0, 5)
31  return words.join(' ').replace(/[,.;:!?]+$/, '') || 'Working'
32}
33
34function rows(): Row[] {
35  const seen = new Set<string>()
36  return [...s.todos, ...s.tasks.values(), ...s.agents].filter(row => !seen.has(row.title) && seen.add(row.title))
37}
38
39function reset(): void {
40  s.title = ''
41  s.todos = []
42  s.tasks = new Map()
43  s.agents = []
44  s.agentRow.clear()
45}
46
47// A skill that reports its own plan owns the bar: a running flow under
48// another title is theirs, so this mod stays out of it.
49async function isOwnedElsewhere($: EngineInterface): Promise<boolean> {
50  const { value } = await $.state.get(FLOW)
51  return Boolean(value && !value.isFinished && value.title !== s.title)
52}
53
54async function report($: EngineInterface): Promise<void> {
55  const list = rows()
56  if (list.length === 0) return
57  if (s.title === '') s.title = titleOf(s.lastPrompt)
58  if (await isOwnedElsewhere($)) return reset()
59
60  const done = list.filter(row => row.isDone).length
61  const isFinished = done === list.length
62  const isDelegating = s.agents.some(row => !row.isDone)
63
64  try {
65    await $.tool.call({
66      tool: PROGRESS,
67      title: s.title,
68      total: list.length,
69      done,
70      phase: isFinished ? 'close' : isDelegating ? 'delegate' : 'plan',
71      tasks: list.map(row => ({ title: row.title, tier: row.tier })),
72      ...(isFinished ? { finished: true } : {}),
73    } as never)
74  } catch {
75    return
76  }
77
78  if (isFinished) reset()
79}
80
81export const register: Register = on => {
82  on('prompt.submit', ($, e, next) => {
83    if (!e.text.trim().startsWith('/agents-info')) s.lastPrompt = e.text
84
85    return next(e)
86  })
87
88  on('tool.call', { tool: 'TodoWrite' }, async ($, e, next) => {
89    const ran = await next(e)
90
91    if (!e.agentId && ran.deny === undefined) {
92      s.todos = e.todos.map(todo => ({ title: todo.content, tier: 'careful', isDone: todo.status === 'completed' }))
93      await report($)
94    }
95
96    return ran
97  }).catch(($, e, next) => next(e))
98
99  on('tool.call', { tool: 'TaskCreate' }, async ($, e, next) => {
100    const ran = await next(e)
101    const id = /#?(\d+)/.exec(ran.text ?? '')?.[1]
102
103    if (!e.agentId && ran.deny === undefined && id) {
104      s.tasks.set(id, { title: e.subject, tier: 'careful', isDone: false })
105      await report($)
106    }
107
108    return ran
109  }).catch(($, e, next) => next(e))
110
111  on('tool.call', { tool: 'TaskUpdate' }, async ($, e, next) => {
112    const ran = await next(e)
113    const row = s.tasks.get(e.taskId)
114
115    if (!e.agentId && ran.deny === undefined && row && e.status) {
116      if (e.status === 'deleted') s.tasks.delete(e.taskId)
117      else s.tasks.set(e.taskId, { ...row, isDone: e.status === 'completed' })
118      await report($)
119    }
120
121    return ran
122  }).catch(($, e, next) => next(e))
123
124  on('tool.call', { tool: 'Agent' }, async ($, e, next) => {
125    if (!e.agentId && !s.agents.some(row => row.title === e.description)) {
126      const tier = tierOf(e.subagent_type ?? 'general-purpose', e.model)
127      s.agents = [...s.agents, { title: e.description, tier, isDone: false }]
128      await report($)
129    }
130
131    return next(e)
132  }).catch(($, e, next) => next(e))
133
134  on('agent.spawn', async ($, e, next) => {
135    const started = await next(e)
136    const row = s.agents.find(one => one.title === e.description)
137
138    if (started.deny === undefined && started.agentId && row) s.agentRow.set(started.agentId, row)
139
140    return started
141  })
142
143  on('turn.complete', async ($, e, next) => {
144    const result = await next(e)
145    const agentId = e.agentId
146    const row = agentId ? s.agentRow.get(agentId) : undefined
147
148    if (agentId && row) {
149      s.agents = s.agents.map(one => (one.title === row.title ? { ...one, isDone: true } : one))
150      s.agentRow.delete(agentId)
151      await report($)
152    }
153
154    return result
155  })
156}
157