prompt-jump

1import { atom, read, update } from 'claude-code'
2import type { EngineInterface, Register } from 'claude-code'
3
4const PANE = 'prompt-jump'
5const HOTKEYS = '123456789abcdefghijklmnopqrstuvwxyz'
6const filter = atom({ plugin: 'prompt-jump', key: 'filter' } as const, '')
7
8// Transcript rows of the person's prompts, in the order the transcript drew them.
9// A render hook may not write $.state, so this lives in the module.
10const drawn: { requestId: string; text: string }[] = []
11// Every transcript row drawn so far, and the newest one first drawn since the last prompt: the bottom.
12const seenRows = new Set<string>()
13let isLive = false
14let bottomRow: string | undefined
15
16const isOwnPrompt = (text: string) => text.trim() !== '' && !text.trimStart().startsWith('<')
17
18function noteRow(requestId: string) {
19  if (seenRows.has(requestId)) return
20  seenRows.add(requestId)
21  if (isLive) bottomRow = requestId
22}
23
24async function scrollTo($: EngineInterface, requestId: string, block: 'start' | 'end') {
25  const moved = await $.ui
26    .scroll({ to: { requestId }, block })
27    .catch((err: unknown) => ({ deny: String(err) }))
28  if (moved.deny) $.ui.toast(`没法滚动过去：${moved.deny}（对话区需在全屏模式 /tui fullscreen）`)
29}
30
31async function scrollToBottom($: EngineInterface) {
32  if (!bottomRow) {
33    $.ui.toast('还不知道底部在哪：发一条消息后再试')
34    return
35  }
36  await scrollTo($, bottomRow, 'end')
37}
38
39const oneLine = (text: string, width: number) => {
40  const flat = text.replace(/\s+/g, ' ').trim()
41  return flat.length > width ? flat.slice(0, width - 1) + '…' : flat
42}
43
44export const register: Register = on => {
45  on('session.start', async ($, e, next) => {
46    await $.command.register({
47      name: 'prompts',
48      description: '列出本会话里你发过的消息，按编号跳到那条（可加关键词过滤）',
49      argumentHint: '[关键词]',
50    })
51    await $.command.register({ name: 'bottom', description: '对话区跳回最底部' })
52    return next(e)
53  })
54
55  on('command.run', { command: 'bottom' }, async $ => {
56    await scrollToBottom($)
57    return { text: '' }
58  })
59
60  on('prompt.submit', ($, e, next) => {
61    isLive = true
62    return next(e)
63  })
64
65  on('ui.render', { component: 'UserMessage' }, ($, e, next) => (noteRow(e.requestId), next(e)))
66  on('ui.render', { component: 'AssistantMessage' }, ($, e, next) => (noteRow(e.requestId), next(e)))
67  on('ui.render', { component: 'ToolUse' }, ($, e, next) => (noteRow(e.requestId), next(e)))
68  on('ui.render', { component: 'ToolResult' }, ($, e, next) => (noteRow(e.requestId), next(e)))
69
70  on('command.run', { command: 'prompts' }, async ($, e) => {
71    await update($, filter, () => (e.args ?? '').trim())
72    const opened = await $.ui.open({ id: PANE, title: '我的消息', focus: true, closeOnEscape: true })
73    return { text: opened.isPlaced ? '已打开消息列表。' : `消息列表暂时无法显示：${opened.reason}` }
74  })
75
76  on('ui.render', { component: 'UserMessage' }, async ($, e, next) => {
77    const kind = e.props.origin.kind
78    const isPerson = kind !== 'task-notification' && kind !== 'plugin' && !e.props.from
79    if (isPerson && isOwnPrompt(e.props.text) && !drawn.some(d => d.requestId === e.requestId)) {
80      drawn.push({ requestId: e.requestId, text: e.props.text })
81    }
82    return next(e)
83  })
84
85  on('ui.render', { component: 'Pane', requestId: PANE }, async ($, e) => {
86    const { Box, Text, Button } = $.ui.resolve(e)
87    const query = (await read($, filter)).toLowerCase()
88    const messages = await $.session.messages()
89
90    // The session's own prompts, matched in order to the transcript rows drawn so far.
91    const pool = [...drawn]
92    const prompts = messages
93      .filter(m => m.role === 'user' && isOwnPrompt(m.text))
94      .map((m, i) => {
95        const at = pool.findIndex(d => d.text.trim() === m.text.trim())
96        const requestId = at >= 0 ? pool.splice(at, 1)[0]?.requestId : undefined
97        return { n: i + 1, text: m.text, requestId }
98      })
99      .filter(p => !query || p.text.toLowerCase().includes(query))
100      .reverse()
101
102    const width = Math.max(20, (e.props.bodyColumns || e.viewport?.columns || 80) - 12)
103
104    const jump = async (requestId: string | undefined) => {
105      if (!requestId) {
106        $.ui.toast('这条还没在对话区加载过：往上滚动经过它一次后就能跳了')
107        return
108      }
109      await scrollTo($, requestId, 'start')
110    }
111
112    return (
113      <Box flexDirection="column">
114        <Text dimColor>
115          {query ? `过滤「${query}」· ` : ''}
116          {prompts.length} 条消息，最新的在上 · 按编号跳转，0 回到底部，Esc 关闭
117        </Text>
118        <Button key="bottom" plain hotkey="0" onPress={() => scrollToBottom($)}>
119          ↓ 回到底部
120        </Button>
121        {prompts.length === 0 && <Text dimColor>还没有消息。</Text>}
122        {prompts.map((p, i) => (
123          <Button key={`p${p.n}`} plain hotkey={HOTKEYS[i]} onPress={() => jump(p.requestId)}>
124            <Text dimColor>#{p.n} </Text>
125            {oneLine(p.text, width)}
126            {p.requestId ? '' : <Text dimColor> ·未加载</Text>}
127          </Button>
128        ))}
129      </Box>
130    )
131  })
132}
133

1export type PromptJumpState = Record<string, never>
2
3declare module 'claude-code' {
4  interface PluginState {
5    'prompt-jump': { filter: string }
6  }
7}
8