delegation-dock

1import { atom, read, update } from 'claude-code'
2import type { Register, EngineInterface } from 'claude-code'
3
4const isEnabledRef = atom({ plugin: 'delegation-dock', key: 'isEnabled' } as const, true)
5
6type Job = { tool: string; startedAt: number; outputPath?: string; isRedirected: boolean }
7
8/** Which delegate a shell command starts, or undefined for any other command. */
9export function delegateOf(command: string): string | undefined {
10  if (/(^|[\s;&|(/\\"])agy(\.exe)?"?\s/.test(command) && !/agy(\.exe)?\s+(models|--help|-h\b|--version)/.test(command)) return 'agy'
11  if (/(^|[\s;&|(/\\"])codex(\.cmd)?"?\s+exec\b/.test(command)) return 'codex'
12  if (/(^|[\s;&|(/\\"])ollama(\.exe)?"?\s+run\b/.test(command) || /batch_llm\.py/.test(command)) return 'ollama'
13  return undefined
14}
15
16/** stdout sent to a file (`> out.txt`): an empty task output then says nothing about agy. */
17export const isRedirected = (command: string) => /(^|[^2&>])>>?\s*[^&\s]/.test(command)
18
19export const EMPTY_AGY =
20  'delegation-dock: agy terminou com saída VAZIA (exit 0). É a falha silenciosa conhecida do `agy -p` (CLAUDE.md, 2026-10-06): ' +
21  'não trate como resposta. Remonte como bundle via stdin, sem -p: `agy --mode plan --model gemini-3.1-pro-high --print-timeout 540s < bundle.txt`.'
22
23export function dockText(jobs: Iterable<Job>, now: number): string | undefined {
24  const parts = [...jobs].map(j => `${j.tool} ${Math.round((now - j.startedAt) / 1000)}s`)
25  return parts.length ? `⏳ ${parts.join(' · ')}` : undefined
26}
27
28// ponytail: module memory; a reload mid-delegation forgets the job (the model still gets its notification).
29const jobs = new Map<string, Job>()
30let ticker: { cancel: () => void } | undefined
31let fg = 0
32
33function refresh($: EngineInterface) {
34  $.ui.status(dockText(jobs.values(), Date.now()))
35  if (jobs.size && !ticker) ticker = $.clock.every(5000, () => refresh($))
36  if (!jobs.size && ticker) {
37    ticker.cancel()
38    ticker = undefined
39  }
40}
41
42export const register: Register = on => {
43  jobs.clear()
44
45  on('session.start', async ($, e, next) => {
46    const saved = await $.store.get('isEnabled').catch(() => undefined)
47    await update($, isEnabledRef, () => (typeof saved === 'boolean' ? saved : isEnabledRef.initial))
48    await $.command.register({
49      name: 'delegation-dock',
50      description: 'Liga/desliga a linha de status das delegações (agy/codex/ollama).',
51      argumentHint: 'on|off',
52    })
53    return next(e)
54  })
55
56  on('command.run', { command: 'delegation-dock' }, async ($, e) => {
57    const arg = e.args.trim().toLowerCase()
58    if (arg !== 'on' && arg !== 'off') {
59      const isEnabled = await read($, isEnabledRef)
60      return { text: `delegation-dock esta ${isEnabled ? 'ativado' : 'desativado'}. Uso: /delegation-dock on · /delegation-dock off` }
61    }
62    await update($, isEnabledRef, () => arg === 'on')
63    await $.store.set('isEnabled', arg === 'on') // $.state is per session; the store survives a restart
64    if (arg === 'off') $.ui.status(undefined)
65    return { text: `delegation-dock ${arg === 'on' ? 'ativado' : 'desativado'}.` }
66  })
67
68  on('tool.call', { tool: 'Bash' }, async ($, e, next) => {
69    if (!(await read($, isEnabledRef))) return next(e)
70    const tool = delegateOf(e.command)
71    if (!tool) return next(e)
72
73    const key = `fg${++fg}`
74    jobs.set(key, { tool, startedAt: Date.now(), isRedirected: isRedirected(e.command) })
75    refresh($)
76    const ran = await next(e).finally(() => {
77      jobs.delete(key)
78      refresh($)
79    })
80    if (ran.deny !== undefined || ran.isError) return ran
81
82    const out = ran.result as { stdout?: string; backgroundTaskId?: string }
83    if (out.backgroundTaskId) {
84      const outputPath = /Output is being written to: (.+?\.output)/.exec(out.stdout ?? '')?.[1]
85      jobs.set(out.backgroundTaskId, { tool, startedAt: Date.now(), isRedirected: isRedirected(e.command), ...(outputPath ? { outputPath } : {}) })
86      refresh($)
87      return ran
88    }
89    if (tool === 'agy' && !isRedirected(e.command) && !out.stdout?.trim()) {
90      $.ui.toast('agy devolveu vazio')
91      return { ...ran, context: [...(ran.context ?? []), EMPTY_AGY] }
92    }
93    return ran
94  }).catch(($, e, next) => next(e)) // a dock bug must never block the command
95
96  on('prompt.submit', async ($, e, next) => {
97    if (e.origin.kind !== 'task-notification') return next(e)
98    const id = /<task-id>([^<]+)<\/task-id>/.exec(e.text)?.[1]
99    const job = id ? jobs.get(id) : undefined
100    if (!id || !job) return next(e)
101
102    jobs.delete(id)
103    refresh($)
104    const completed = /<status>completed<\/status>/.test(e.text)
105    if (job.tool !== 'agy' || job.isRedirected || !completed || !job.outputPath) return next(e)
106    const text = await $.fs.read(job.outputPath).catch(() => undefined)
107    if (text === undefined || text.trim()) return next(e)
108    $.ui.toast('agy devolveu vazio')
109    return next({ ...e, context: [...(e.context ?? []), EMPTY_AGY] })
110  }).catch(($, e, next) => next(e))
111}
112

1export type DelegationDockState = {
2  isEnabled: boolean
3}
4
5declare module 'claude-code' {
6  interface PluginState {
7    'delegation-dock': {
8      isEnabled: boolean
9    }
10  }
11}
12