answer-buttons

1// answer-buttons: three small buttons under Claude's last answer.
2//   ✎ Plain English    asks for that answer again in plain words
3//   ✂ Shorter          asks for the same answer, cut down
4//   ✎✂ Plain & short   plain words, held to 2 or 3 sentences
5// The two Plain buttons send a built in prompt, or run the skill the plain_skill
6// setting names. From the keyboard: /answer-buttons:shorter and
7// /answer-buttons:plain-short (commands/), and that skill's own command.
8
9import type { Register } from 'claude-code'
10
11// Skills are commands, and a mod may not submit a '/' text, so the skill runs
12// through $.command.run
13export const PLAIN_ARGS = 'Re-explain your last answer.'
14export const PLAIN_PROMPT =
15  'Re-explain your last answer in plain English for someone who is not an expert in this area: ' +
16  'first what the thing is, in its own terms, then one analogy labelled as one and where it stops being true, ' +
17  'then where it actually is (a file, a command, a menu) and what breaks if it is wrong. Under 200 words. ' +
18  'Keep any yes/no question and any copyable draft exactly as they were.'
19export const PLAIN_SHORT_ARGS =
20  'Re-explain your last answer in 2 or 3 sentences only: the thing itself, then one labelled analogy. ' +
21  'Keep any yes/no question and any copyable draft exactly as they were.'
22export const PLAIN_SHORT_PROMPT =
23  'Re-explain your last answer in plain English, in 2 or 3 sentences only: the thing itself, then one ' +
24  'labelled analogy. Keep any yes/no question and any copyable draft exactly as they were.'
25export const SHORTER_PROMPT =
26  'Make your last answer shorter: lead with what it means for me in 2 or 3 sentences, ' +
27  'keep any yes/no question and any copyable draft exactly as they were, drop the rest. ' +
28  'Reply in the same language as before.'
29
30// The final text of the last main conversation answer, and whether a turn is
31// running. Module variables: a hot reload clears them, which only hides the
32// buttons until the next answer.
33let lastAnswer = ''
34let isWorking = false
35// The skill the Plain buttons run, from the plain_skill setting; empty sends the built in prompts
36let plainSkill = ''
37export function setPlainSkill(name: string | undefined) {
38  plainSkill = String(name ?? '').trim().replace(/^\//, '')
39}
40
41// True for the text block that ends the last answer
42export function isLastBlock(blockText: string, answer: string): boolean {
43  const block = blockText.trim()
44  return block.length > 0 && answer.trim().endsWith(block)
45}
46
47export const register: Register = (on, options) => {
48  setPlainSkill(options?.plain_skill as string | undefined)
49  on('turn.start', async ($, e, next) => {
50    isWorking = true
51    $.ui.invalidate('ui.render')
52    return next(e)
53  })
54
55  on('turn.complete', async ($, e, next) => {
56    // Subagent turns end inside the main one; only the main answer counts
57    if (!e.agentId) {
58      isWorking = false
59      lastAnswer = e.reason === 'answer' ? e.answer : ''
60      $.ui.invalidate('ui.render')
61    }
62    return next(e)
63  })
64
65  on('ui.render', { component: 'AssistantMessage' }, async ($, e, next) => {
66    const own = await next(e)
67    if (isWorking || e.surface === 'vscode' || e.surface === 'mobile') return own
68    if (!isLastBlock(e.props.text, lastAnswer)) return own
69
70    const { Box, Button } = $.ui.resolve(e)
71    return (
72      <Box flexDirection="column">
73        {own}
74        <Box gap={2}>
75          <Button key="plain" plain dimColor onPress={() => (plainSkill ? $.command.run({ command: plainSkill, args: PLAIN_ARGS }) : $.prompt.submit({ text: PLAIN_PROMPT, asUser: true }))}>
76            ✎ Plain English
77          </Button>
78          <Button key="shorter" plain dimColor onPress={() => $.prompt.submit({ text: SHORTER_PROMPT, asUser: true })}>
79            ✂ Shorter
80          </Button>
81          <Button key="plain-short" plain dimColor onPress={() => (plainSkill ? $.command.run({ command: plainSkill, args: PLAIN_SHORT_ARGS }) : $.prompt.submit({ text: PLAIN_SHORT_PROMPT, asUser: true }))}>
82            ✎✂ Plain & short
83          </Button>
84        </Box>
85      </Box>
86    )
87  })
88}
89