prompt-optimizer

1import { atom, read, update } from 'claude-code'
2import type { Register } from 'claude-code'
3
4const COMMAND = 'prompt-optimizer'
5const BYPASS = '* '
6
7// Origins typed by the person; notifications, peers and schedules are left alone.
8const HUMAN_ORIGINS = new Set(['composer', 'bridge'])
9
10export const GUIDANCE = [
11  '[prompt-optimizer] Before acting, check whether this prompt states its goal, scope and what "done" looks like.',
12  '- If a decision only the user can make is missing and cannot be inferred from the code or a sensible default, ask at most 3 short clarifying questions first.',
13  '- Otherwise proceed directly, and open your reply with one line naming the assumptions you made.',
14].join('\n')
15
16const isPaused = atom({ plugin: 'prompt-optimizer', key: 'isPaused' } as const, false)
17
18export const register: Register = (on, options) => {
19  const mode = String(options.mode ?? 'short-only')
20  const minLength = Number(options.minLength ?? 80)
21
22  on('session.start', async ($, e, next) => {
23    await $.command.register({
24      name: COMMAND,
25      description: 'Pause or resume prompt-optimizer for this session',
26    })
27    // A /config change reloads the module and re-fires this; the pause survives it.
28    const paused = await read($, isPaused)
29    $.ui.status(mode === 'off' || paused ? undefined : 'PO: on')
30
31    return next(e)
32  })
33
34  on('command.run', { command: COMMAND }, async $ => {
35    const paused = await update($, isPaused, was => !was)
36    $.ui.status(mode === 'off' || paused ? undefined : 'PO: on')
37
38    return {
39      text: paused
40        ? 'prompt-optimizer paused for this session.'
41        : 'prompt-optimizer resumed for this session.',
42    }
43  })
44
45  on('prompt.submit', async ($, e, next) => {
46    if (!HUMAN_ORIGINS.has(e.origin.kind)) {
47      return next(e)
48    }
49
50    const text = e.text.trimStart()
51    const isWanted =
52      mode === 'always' || (mode === 'short-only' && text.length < minLength)
53
54    if (!isWanted || text.startsWith('/') || (await read($, isPaused))) {
55      return next(e)
56    }
57
58    // A leading "* " skips the guidance for one prompt; only stripped when
59    // guidance would otherwise apply, and never down to an empty prompt.
60    if (text.startsWith(BYPASS)) {
61      const rest = text.slice(BYPASS.length).trimStart()
62
63      return rest === '' ? next(e) : next({ ...e, text: rest })
64    }
65
66    return next({ ...e, context: [...(e.context ?? []), GUIDANCE] })
67  })
68}
69

1export type Paused = boolean
2
3declare module 'claude-code' {
4  interface PluginState {
5    'prompt-optimizer': { isPaused: Paused }
6  }
7}
8