cmds

1import { atom, read, update } from 'claude-code'
2import type { EngineInterface, Register } from 'claude-code'
3
4import type { CmdsEntry } from '../types'
5
6const OTHER = '其他'
7const PANE = 'cmds'
8const ROWS = 16
9
10const tip = atom({ plugin: 'cmds', key: 'tip' } as const, 0)
11const list = atom({ plugin: 'cmds', key: 'list' } as const, [] as CmdsEntry[])
12
13type Group = { title: string; names: string[] }
14
15// 「组名: 命令 命令; 组名: 命令」，组之间用分号或换行隔开，命令之间用空格、逗号或顿号
16function parseGroups(text: string): Group[] {
17  return text
18    .split(/[;；\n]/)
19    .map(part => part.match(/^\s*([^:：]+?)\s*[:：](.*)$/))
20    .filter(m => m !== null)
21    .map(m => ({
22      title: m[1] ?? '',
23      names: (m[2] ?? '').split(/[\s,，、]+/).map(n => n.replace(/^\//, '')).filter(Boolean),
24    }))
25    .filter(g => g.title && g.names.length > 0)
26}
27
28// description 在括号外的第一个冒号或分号之前那段
29function firstClause(text: string): string {
30  let depth = 0
31  for (let i = 0; i < text.length; i++) {
32    const ch = text[i] ?? ''
33    if ('（(【[「'.includes(ch)) depth++
34    else if ('）)】]」'.includes(ch)) depth = Math.max(0, depth - 1)
35    else if (depth === 0 && '：:；;'.includes(ch) && i > 0) return text.slice(0, i).trim()
36  }
37  return text.trim()
38}
39
40function field(front: string, key: string): string {
41  const m = front.match(new RegExp(`^${key}:[ \\t]*(.*)$`, 'm'))
42  return (m?.[1] ?? '').trim().replace(/^(['"])(.*)\1$/, '$2')
43}
44
45async function readDir($: EngineInterface, dir: string): Promise<CmdsEntry[]> {
46  const entries = await $.fs.list(dir).catch(() => [])
47  const found: CmdsEntry[] = []
48  for (const entry of entries) {
49    if (!entry.name.endsWith('.md') || (entry.kind !== 'file' && !entry.isLink)) continue
50    const text = await $.fs.read(`${dir}/${entry.name}`).catch(() => '')
51    const front = typeof text === 'string' ? (text.match(/^---\n([\s\S]*?)\n---/)?.[1] ?? '') : ''
52    found.push({
53      name: entry.name.slice(0, -3),
54      args: field(front, 'argument-hint'),
55      purpose: firstClause(field(front, 'description')),
56    })
57  }
58  return found
59}
60
61// 用户级 ~/.claude/commands（或 $CLAUDE_CONFIG_DIR/commands）和当前项目的 .claude/commands，同名以项目的为准
62async function load($: EngineInterface): Promise<void> {
63  const config = (await $.env.get('CLAUDE_CONFIG_DIR')) ?? `${(await $.env.get('HOME')) ?? ''}/.claude`
64  const byName = new Map<string, CmdsEntry>()
65  const project = `${await $.session.cwd()}/.claude/commands`
66  for (const entry of [...(await readDir($, `${config}/commands`)), ...(await readDir($, project))]) {
67    byName.set(entry.name, entry)
68  }
69  await update($, list, () => [...byName.values()].sort((a, b) => a.name.localeCompare(b.name)))
70}
71
72function grouped(entries: CmdsEntry[], groups: Group[]): { title: string; items: CmdsEntry[] }[] {
73  const byName = new Map(entries.map(e => [e.name, e]))
74  const placed = new Set<string>()
75  const result = groups.map(g => ({
76    title: g.title,
77    items: g.names.flatMap(n => {
78      const entry = byName.get(n)
79      if (!entry || placed.has(n)) return []
80      placed.add(n)
81      return [entry]
82    }),
83  }))
84  const rest = entries.filter(e => !placed.has(e.name))
85  result.push({ title: groups.length > 0 ? OTHER : '全部命令', items: rest })
86  return result.filter(g => g.items.length > 0)
87}
88
89export const register: Register = (on, options) => {
90  const groups = parseGroups(typeof options.groups === 'string' ? options.groups : '')
91
92  on('session.start', async ($, e, next) => {
93    // immediate：Claude 回答中途输入也立刻开关，不排到这一轮结束
94    await $.command.register({ name: 'cmds', description: '打开 / 关闭自定义短命令速查面板', immediate: true })
95    await load($)
96
97    return next(e)
98  })
99
100  // 同一个命令再输一次就关面板；面板拿着键盘：打开就能 ↑↓ 滚动，Esc 只关面板，不会中断 Claude
101  on('command.run', { command: 'cmds' }, async $ => {
102    if ((await $.ui.panes()).some(p => p.id === PANE)) {
103      await $.ui.close({ id: PANE })
104    } else {
105      await load($)
106      await $.ui.open({ id: PANE, title: '短命令', focus: true, closeOnEscape: true, rows: ROWS })
107    }
108
109    return {}
110  })
111
112  on('turn.complete', async ($, e, next) => {
113    await update($, tip, n => n + 1)
114
115    return next(e)
116  })
117
118  on('ui.render', { component: 'PromptHint' }, async ($, e, next) => {
119    if (e.props.isDraft || e.props.isWorking) {
120      return next(e)
121    }
122    const items = grouped(await read($, list), groups).flatMap(g => g.items)
123    const item = items[(await read($, tip)) % items.length]
124    if (!item) {
125      return next(e)
126    }
127
128    return next({ ...e, props: { ...e.props, tail: `/${item.name} ${item.purpose} · /cmds 看全部` } })
129  })
130
131  on('ui.render', { component: 'Pane', requestId: PANE }, async ($, e) => {
132    const { Box, Text } = $.ui.resolve(e)
133    const sections = grouped(await read($, list), groups)
134    // 右上角留给 Claude Code 自己画的关闭标记 ✕
135    const hint = (
136      <Box marginRight={3}>
137        <Text dimColor>↑↓ 滚动 · Esc 关闭</Text>
138      </Box>
139    )
140
141    if (sections.length === 0) {
142      return (
143        <Box flexDirection="row" justifyContent="space-between">
144          <Text dimColor>没找到自定义命令：~/.claude/commands/ 和项目的 .claude/commands/ 下都没有 .md 文件</Text>
145          {hint}
146        </Box>
147      )
148    }
149
150    return (
151      <Box flexDirection="column">
152        {sections.map((g, n) => (
153          <Box key={g.title} flexDirection="column" marginBottom={n === sections.length - 1 ? 0 : 1}>
154            <Box flexDirection="row" justifyContent="space-between">
155              <Text bold>{g.title}</Text>
156              {n === 0 ? hint : null}
157            </Box>
158            {g.items.map(item => (
159              <Box key={item.name} flexDirection="column">
160                <Text>
161                  <Text bold color="cyan">/{item.name}</Text>
162                  {item.args ? <Text dimColor> {item.args}</Text> : null}
163                </Text>
164                {item.purpose ? <Text>  {item.purpose}</Text> : null}
165              </Box>
166            ))}
167          </Box>
168        ))}
169      </Box>
170    )
171  })
172}
173

1export type CmdsEntry = { name: string; args: string; purpose: string }
2
3declare module 'claude-code' {
4  interface PluginState {
5    cmds: { tip: number; list: CmdsEntry[] }
6  }
7}
8