brain-dump

1import type { Register } from 'claude-code'
2
3const MIN_CHARS = 500
4const MAX_QUESTIONS = 4
5const MODEL = 'haiku'
6
7const BULLET = /^\s*([-*•]|\d+[.)]|#{1,4})\s/
8const STRUCTURED = /^#{1,3}\s*(task|why|guardrails|done)\b/im
9
10// A long wall of text with few bullets, headings or numbered lines.
11const isBrainDump = (text: string) => {
12  if (text.length < MIN_CHARS || STRUCTURED.test(text)) return false
13  const lines = text.split('\n').filter(line => line.trim() !== '')
14  const bullets = lines.filter(line => BULLET.test(line)).length
15
16  return bullets < 3 && bullets / lines.length < 0.4
17}
18
19const QUESTIONS = (dump: string) => `A user typed this unstructured brain dump as a prompt for a coding assistant:
20
21<dump>
22${dump}
23</dump>
24
25Ask at most ${MAX_QUESTIONS} short clarifying questions that would most improve a prompt built from it. Cover whichever of these are unclear: the concrete task, why it matters, guardrails (what must not change), and what "done" looks like. Skip anything the dump already answers.
26
27Output one question per line in exactly this form, nothing else:
28question | suggested answer A | suggested answer B`
29
30const REWRITE = (dump: string, answers: string) => `Rewrite this brain dump into a clear, structured prompt for a coding assistant.
31
32<dump>
33${dump}
34</dump>
35
36<clarifications>
37${answers || '(none given)'}
38</clarifications>
39
40Output only the prompt, in markdown, with exactly these sections:
41## Task
42## Why
43## Guardrails
44## What done looks like
45
46Keep every concrete detail, name and constraint from the dump. Do not invent requirements; write "Not specified" where nothing was said.`
47
48export const register: Register = on => {
49  on('prompt.submit', async ($, e, next) => {
50    if (e.command || e.isMeta) return next(e)
51    if (e.text.startsWith('!raw ')) return next({ ...e, text: e.text.slice(5) })
52    if (!isBrainDump(e.text)) return next(e)
53
54    $.ui.toast('Brain Dump: a cluttered mind, this is. Sorting it, I am…')
55
56    const asked = await $.model.complete({
57      model: MODEL,
58      prompt: QUESTIONS(e.text),
59      maxTokens: 600,
60      effort: 'low',
61    })
62    if (!asked.isAnswered) return next(e)
63
64    const questions = asked.text
65      .split('\n')
66      .map(line => line.split('|').map(part => part.trim()))
67      .filter(parts => parts[0] && parts[0].endsWith('?'))
68      .slice(0, MAX_QUESTIONS)
69
70    const answers: string[] = []
71    for (const [question = '', ...options] of questions) {
72      const answer = await $.ui.ask(question, {
73        options: options.filter(Boolean).slice(0, 3),
74        header: 'Brain Dump',
75      })
76      answers.push(`Q: ${question}\nA: ${answer}`)
77    }
78
79    const rewritten = await $.model.complete({
80      model: MODEL,
81      prompt: REWRITE(e.text, answers.join('\n\n')),
82      maxTokens: 2000,
83    })
84    if (!rewritten.isAnswered) return next(e)
85
86    await $.prompt.fill({ text: rewritten.text.trim(), mode: 'replace' })
87
88    return {
89      drop: 'Brain Dump: your thoughts are sorted. The structured prompt is in your prompt box; edit it, then press Enter. (Prefix a prompt with "!raw " to skip Brain Dump.)',
90    }
91  })
92}
93