prompt-boost

1import { atom, read, update } from 'claude-code'
2import type { EngineInterface, Register } from 'claude-code'
3
4// Prompt-Verbesserer als Band UEBER dem Eingabefeld. Steht ein Entwurf im
5// Eingabefeld, zeigt das Band den Knopf "Prompt verbessern". Der Knopf fragt
6// Claude (ueber $.model.fork, kennt also Gespraech, Projekt und Skill-Liste),
7// den Entwurf besser zu schreiben: passende Skills/Agents nennen, Entscheidungen
8// vorher klaeren, Zielbild beschreiben. Das Ergebnis ERSETZT nur den Entwurf im
9// Eingabefeld; gesendet wird nichts. "rueckgaengig" holt den alten Entwurf
10// zurueck. Mit "/verbessern" an/aus.
11
12const FORK_PROMPT_HEAD = [
13  'Der Nutzer hat im Eingabefeld diesen Prompt-Entwurf getippt, aber NOCH NICHT',
14  'gesendet. Fuehre ihn NICHT aus. Schreibe ihn nur als besseren Prompt an dich',
15  'selbst um, in der Sprache des Entwurfs:',
16  '- Behalte Absicht und alle Fakten des Entwurfs. Erfinde nichts dazu.',
17  '- Nutze, was du aus dem bisherigen Gespraech und dem Projekt weisst.',
18  '- Nenne die Skills (als /name) und Agent-Typen aus deiner Liste, die wirklich',
19  '  passen. Keine, wenn keine passt.',
20  '- Liste die wichtigen Entscheidungen, die du vor dem Start fragen sollst.',
21  '- Beschreibe kurz, woran man erkennt, dass die Aufgabe fertig ist.',
22  '- Hoechstens etwa 15 kurze Zeilen.',
23  'Antworte NUR mit dem neuen Prompt-Text: ohne Einleitung, ohne Codeblock,',
24  'ohne Erklaerung. Nutze keine Werkzeuge.',
25  '',
26  'Entwurf:',
27].join('\n')
28
29const hasDraft = atom({ plugin: 'prompt-boost', key: 'hasDraft' } as const, false)
30const busy = atom({ plugin: 'prompt-boost', key: 'busy' } as const, false)
31// Der Entwurf vor dem Umschreiben; '' = nichts rueckgaengig zu machen.
32const original = atom({ plugin: 'prompt-boost', key: 'original' } as const, '')
33const enabled = atom({ plugin: 'prompt-boost', key: 'enabled' } as const, true)
34// Meldet das Eingabefeld Tippen (prompt.edit)? Manche Oberflaechen tun das
35// nicht; dann steht der Knopf immer da und prueft beim Druecken selbst.
36const editSeen = atom({ plugin: 'prompt-boost', key: 'editSeen' } as const, false)
37
38// Zaehler, damit eine alte Antwort nach "abbrechen" nichts mehr ersetzt.
39let generation = 0
40
41// Antwort des Forks saeubern: Codeblock-Zaeune und Rahmen-Leerzeilen weg.
42export function cleanRewrite(text: string): string {
43  let t = text.trim()
44  const fence = /^```[a-zA-Z]*\r?\n([\s\S]*?)\r?\n```$/.exec(t)
45  if (fence?.[1] !== undefined) t = fence[1].trim()
46  t = t.replace(/^(?:verbesserter prompt|neuer prompt|prompt)\s*:\s*\n/i, '')
47
48  return t.trim()
49}
50
51export function buildForkPrompt(draft: string): string {
52  return FORK_PROMPT_HEAD + '\n<<<\n' + draft + '\n>>>'
53}
54
55async function boost($: EngineInterface) {
56  const { text: draft } = await $.prompt.read()
57  if (!draft.trim()) {
58    $.ui.toast('Prompt verbessern: Erst einen Entwurf ins Eingabefeld tippen.')
59    return
60  }
61  const mine = ++generation
62  await update($, busy, () => true)
63
64  let rewrite = ''
65  try {
66    const r = await $.model.fork({ prompt: buildForkPrompt(draft) })
67    if (r.isAnswered) {
68      rewrite = cleanRewrite(r.text)
69    } else if (r.reason === 'api-error') {
70      $.ui.toast('Prompt verbessern: Fehler bei der Anfrage.')
71    }
72  } catch {
73    $.ui.toast('Prompt verbessern: Anfrage fehlgeschlagen.')
74  }
75
76  if (mine !== generation) return
77  await update($, busy, () => false)
78  if (!rewrite) return
79
80  // Hat der Nutzer inzwischen weitergetippt, nichts ueberschreiben.
81  const { text: now } = await $.prompt.read()
82  if (now !== draft) {
83    $.ui.toast('Prompt verbessern: Entwurf hat sich geaendert, nichts ersetzt.')
84    return
85  }
86  const filled = await $.prompt.fill({ text: rewrite, mode: 'replace' })
87  if (filled.isFilled) {
88    await update($, original, () => draft)
89  } else {
90    $.ui.toast('Prompt verbessern: Eingabefeld nimmt gerade keinen Text an.')
91  }
92}
93
94async function cancel($: EngineInterface) {
95  generation++
96  await update($, busy, () => false)
97}
98
99async function undo($: EngineInterface) {
100  const old = await read($, original)
101  if (!old) return
102  await update($, original, () => '')
103  await $.prompt.fill({ text: old, mode: 'replace' })
104}
105
106async function toggleEnabled($: EngineInterface) {
107  const next = !(await read($, enabled))
108  await update($, enabled, () => next)
109  await $.store.set('enabled', next)
110  if (!next) {
111    await cancel($)
112    await update($, original, () => '')
113  }
114}
115
116export const register: Register = on => {
117  on('session.start', async ($, e, next) => {
118    const saved = await $.store.get('enabled')
119    await update($, enabled, () => saved !== false)
120    await $.command.register({
121      name: 'verbessern',
122      description: 'Prompt-Verbesserer ueber dem Eingabefeld an- oder ausschalten',
123    })
124
125    return next(e)
126  })
127
128  on('command.run', { command: 'verbessern' }, async $ => {
129    await toggleEnabled($)
130    const isOn = await read($, enabled)
131
132    return { text: isOn ? 'Prompt-Verbesserer ist an.' : 'Prompt-Verbesserer ist aus.' }
133  })
134
135  // Merken, ob ein Entwurf im Eingabefeld steht (nur bei Wechsel schreiben).
136  on('prompt.edit', async ($, e, next) => {
137    const r = await next(e)
138    if (!(await read($, editSeen))) await update($, editSeen, () => true)
139    const has = r.text.trim().length > 0
140    if ((await read($, hasDraft)) !== has) await update($, hasDraft, () => has)
141
142    return r
143  })
144
145  // Gesendet: Rueckgaengig gilt nicht mehr, Entwurf ist weg.
146  on('prompt.submit', async ($, e, next) => {
147    await update($, original, () => '')
148    await update($, hasDraft, () => false)
149
150    return next(e)
151  })
152
153  // Nur EIN Band ueber dem Prompt: darunterliegendes Band durchreichen, unseres
154  // anhaengen.
155  on('ui.render', { component: 'AbovePrompt' }, async ($, e, next) => {
156    const below = await next(e)
157    if (e.props?.hasSurvey) return below
158    if (!(await read($, enabled))) return below
159
160    const isBusy = await read($, busy)
161    const draft = (await read($, hasDraft)) || !(await read($, editSeen))
162    const old = await read($, original)
163    if (!isBusy && !draft && !old) return below
164
165    const { Box, Text, Button } = $.ui.resolve(e)
166
167    return (
168      <Box flexDirection="column">
169        {below}
170        <Box gap={1}>
171          <Text color="magenta">{'✨'}</Text>
172          {isBusy ? <Text dimColor>verbessert den Prompt ...</Text> : null}
173          {isBusy ? (
174            <Button key="cancel" label="abbrechen" onPress={() => void cancel($)} />
175          ) : null}
176          {!isBusy && draft ? (
177            <Button key="boost" label="Prompt verbessern" onPress={() => void boost($)} />
178          ) : null}
179          {!isBusy && old ? (
180            <Button key="undo" label="rueckgaengig" onPress={() => void undo($)} />
181          ) : null}
182        </Box>
183      </Box>
184    )
185  })
186}
187

1declare module 'claude-code' {
2  interface PluginState {
3    'prompt-boost': {
4      hasDraft: boolean
5      busy: boolean
6      original: string
7      enabled: boolean
8      editSeen: boolean
9    }
10  }
11}
12