next-steps

1import { atom, read, update } from 'claude-code'
2import type { Register } from 'claude-code'
3
4import type { NextSteps } from '../types'
5
6const RULES = `# Next-steps
7작업 단위를 끝낼 때 응답 마지막에 \`다음 단계\` 제목을 두고 체크박스(\`- [ ] \`) 2~3개로 제안한다. 단순 질의응답에는 붙이지 않는다.
8- 코드 작업: 아직 하지 않은 검증(빌드, 테스트, 실제 실행 확인), 영향받는 다른 모듈·호출부, 관련 이슈·PR. 이슈·PR 번호는 대화나 브랜치명에서 확인된 것만 쓰고 지어내지 않는다.
9- 문서 작업: 미확정 수치·출처, 검토자가 확인해야 할 항목, 다음에 쓸 절.
10- 각 항목은 바로 실행할 수 있는 한 줄로 쓴다. 가장 권장하는 항목을 맨 위에 둔다.`
11
12const items = atom({ plugin: 'next-steps', key: 'items' } as const, [])
13
14/** 응답에서 "다음 단계" 제목 아래의 체크박스 항목을 뽑는다. */
15export const parse = (answer: string): NextSteps => {
16  const at = answer.lastIndexOf('다음 단계')
17
18  if (at === -1) {
19    return []
20  }
21
22  return answer
23    .slice(at)
24    .split('\n')
25    .map(line => /^\s*[-*]\s*\[ \]\s*(.+)$/.exec(line)?.[1]?.trim())
26    .filter((one): one is string => one !== undefined && one !== '')
27    .slice(0, 5)
28}
29
30export const register: Register = on => {
31  on('session.start', async ($, e, next) => {
32    await $.command.register({
33      name: 'next',
34      description: '마지막으로 제안된 다음 단계를 보거나, 번호를 주면 입력창에 넣는다',
35      argumentHint: '[번호]',
36    })
37
38    return next(e)
39  })
40
41  on('prompt.compose', async ($, e, next) => {
42    const composed = await next(e)
43
44    return {
45      sections: [
46        ...composed.sections,
47        { id: 'next-steps:rules', text: RULES, scope: 'session' },
48      ],
49    }
50  })
51
52  on('turn.complete', async ($, e, next) => {
53    const done = await next(e)
54
55    if (e.agentId === undefined && e.reason === 'answer') {
56      const found = parse(e.answer)
57      await update($, items, () => found)
58      const first = found[0]
59
60      if (first !== undefined) {
61        // 턴이 끝난 뒤 빈 입력창에 흐리게 제안한다. Tab으로 받는다.
62        $.clock.after(500, () => {
63          void $.prompt.suggest({ text: first }).catch(() => undefined)
64        })
65      }
66    }
67
68    return done
69  })
70
71  on('command.run', { command: 'next' }, async ($, e) => {
72    const list = await read($, items)
73
74    if (list.length === 0) {
75      return { text: '제안된 다음 단계가 없습니다.' }
76    }
77
78    const picked = list[Number(e.args.trim()) - 1]
79
80    if (picked === undefined) {
81      const lines = list.map((one, i) => `${i + 1}. ${one}`).join('\n')
82
83      return { text: `다음 단계 (/next <번호>로 입력창에 넣기)\n${lines}` }
84    }
85
86    await $.prompt.fill({ text: picked, mode: 'replace' })
87
88    return { text: `입력창에 넣었습니다: ${picked}` }
89  })
90}
91

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