next-steps

1import { atom, read, update } from 'claude-code'
2import type { Register } from 'claude-code'
3
4// Next Steps: после каждого ответа спрашиваем модель (через fork, из кэша)
5// о трёх следующих шагах и рисуем их кнопками над полем ввода.
6const steps = atom({ plugin: 'next-steps', key: 'steps' } as const, [])
7const isHidden = atom({ plugin: 'next-steps', key: 'isHidden' } as const, false)
8
9const ASK =
10  'Based on this conversation, suggest exactly 3 short next steps the user would most likely ask you to do next. ' +
11  'Answer in the language of the conversation. Each step is an imperative phrase under 60 characters. ' +
12  'Reply with a JSON array of 3 strings and nothing else.'
13
14function parse(text: string): string[] {
15  const match = text.match(/\[[\s\S]*\]/)
16  if (!match) return []
17  try {
18    const list = JSON.parse(match[0])
19    return Array.isArray(list) ? list.filter(s => typeof s === 'string').slice(0, 3) : []
20  } catch {
21    return []
22  }
23}
24
25export const register: Register = on => {
26  on('session.start', async ($, e, next) => {
27    await $.command.register({ name: 'next-steps', description: 'Показать или спрятать кнопки следующих шагов' })
28    return next(e)
29  })
30
31  on('command.run', { command: 'next-steps' }, async $ => {
32    const hidden = await update($, isHidden, v => !v)
33    return { text: hidden ? 'Next Steps выключен.' : 'Next Steps включён.' }
34  })
35
36  on('prompt.submit', async ($, e, next) => {
37    await update($, steps, () => [])
38    return next(e)
39  })
40
41  on('turn.complete', async ($, e, next) => {
42    const result = await next(e)
43    if (e.agentId || e.reason !== 'answer' || (await read($, isHidden))) return result
44
45    // fork живёт дольше хука, поэтому запускаем его таймером
46    $.clock.after(0, async () => {
47      try {
48        const answer = await $.model.fork({ prompt: ASK })
49        if (!answer.isAnswered) {
50          $.ui.log(`next-steps: модель не ответила (${answer.reason})`)
51          return
52        }
53        const list = parse(answer.text)
54        if (list.length === 0) $.ui.log(`next-steps: не разобрал ответ: ${answer.text.slice(0, 120)}`)
55        await update($, steps, () => list)
56      } catch (err) {
57        $.ui.log(`next-steps: ошибка ${String(err).slice(0, 160)}`)
58      }
59    })
60    return result
61  })
62
63  on('ui.render', { component: 'AbovePrompt' }, async ($, e, next) => {
64    const list = await read($, steps)
65    if (e.props.hasSurvey || e.props.isWorking || list.length === 0 || (await read($, isHidden))) {
66      return next(e)
67    }
68
69    const { Box, Button, Text } = $.ui.resolve(e)
70    const mine = (
71      <Box flexDirection="column" marginTop={1}>
72        <Text dimColor>дальше:</Text>
73        <Box flexDirection="column" paddingLeft={2}>
74          {list.map((step, i) => (
75            <Button
76              key={`step-${i}`}
77              plain
78              hotkey={String(i + 1)}
79              label={step}
80              onPress={async () => {
81                await update($, steps, () => [])
82                await $.prompt.submit({ text: step })
83              }}
84            />
85          ))}
86          <Button key="dismiss" plain dimColor hotkey="0" label="скрыть" onPress={() => update($, steps, () => [])} />
87        </Box>
88      </Box>
89    )
90
91    // место над полем ввода общее: дорисовываем под собой строки других модов
92    return (
93      <Box flexDirection="column">
94        {mine}
95        {await next(e)}
96      </Box>
97    )
98  })
99}
100

1export type NextStep = string
2
3declare module 'claude-code' {
4  interface PluginState {
5    'next-steps': { steps: NextStep[]; isHidden: boolean }
6  }
7}
8