claude-mood

1import type { EngineInterface, Register } from 'claude-code'
2
3export type Tally = { reads: number; edits: number; runs: number; errors: number; agents: number }
4
5const READ_TOOLS: readonly string[] = ['Read', 'Grep', 'Glob', 'WebFetch', 'WebSearch', 'LS']
6const EDIT_TOOLS: readonly string[] = ['Edit', 'Write', 'NotebookEdit', 'MultiEdit']
7
8const empty = (): Tally => ({ reads: 0, edits: 0, runs: 0, errors: 0, agents: 0 })
9
10let tally: Tally = empty()
11let current = ''
12
13const isTest = (c: string) => /\b(test|tests|pytest|jest|vitest|go\s+test|cargo\s+test)\b/.test(c)
14
15/** The mood while a turn runs: errors weigh most, then what the turn mostly does. */
16export function workingMood(t: Tally, now: string): string {
17  if (t.errors >= 5) return '😵‍💫 头晕了'
18  if (t.errors >= 3) return '😤 有点烦躁'
19  if (now === 'test') return '🧪 紧张地跑测试'
20  if (now === 'agent' || t.agents >= 2) return '🧑‍🤝‍🧑 在摇人帮忙'
21  if (t.edits >= 10 && t.errors === 0) return '🔥 手感火热'
22  if (t.reads >= 15 && t.edits === 0) return '🧐 深度调研中'
23  if (t.edits > 0 && t.edits >= t.reads) return '✍️ 奋笔疾书'
24  if (t.reads > 0) return '🤔 在读代码'
25  return '💭 思考中'
26}
27
28export function doneMood(t: Tally, reason: string, aborted: boolean): string {
29  if (aborted) return '🫢 被打断了'
30  if (reason !== 'answer') return '😭 挂了'
31  if (t.errors >= 3) return '😮‍💨 磕磕绊绊，总算搞定'
32  if (t.edits >= 10) return '😎 大干一场'
33  if (t.edits === 0 && t.reads === 0 && t.runs === 0) return '😊 随口一答'
34  return '😌 搞定'
35}
36
37export function counts(t: Tally): string {
38  const parts = [
39    t.reads ? `📖${t.reads}` : '',
40    t.edits ? `✏️${t.edits}` : '',
41    t.runs ? `⚡${t.runs}` : '',
42    t.errors ? `💥${t.errors}` : '',
43  ].filter(Boolean)
44  return parts.length ? `  ${parts.join(' ')}` : ''
45}
46
47function show($: EngineInterface, mood: string) {
48  $.ui.status(`${mood}${counts(tally)}`)
49}
50
51export const register: Register = on => {
52  on('prompt.submit', ($, e, next) => {
53    tally = empty()
54    current = ''
55    show($, workingMood(tally, current))
56    return next(e)
57  })
58
59  on('tool.call', async ($, e, next) => {
60    const tool = String(e.tool)
61    current = tool === 'Agent' || tool === 'Task' ? 'agent' : e.tool === 'Bash' && isTest(e.command) ? 'test' : ''
62    if (current === 'agent') tally.agents += 1
63    show($, workingMood(tally, current))
64
65    const ran = await next(e)
66    if (ran.deny === undefined) {
67      if (READ_TOOLS.includes(tool)) tally.reads += 1
68      else if (EDIT_TOOLS.includes(tool)) tally.edits += 1
69      else if (tool === 'Bash') tally.runs += 1
70      if (ran.isError) tally.errors += 1
71    }
72    current = ''
73    show($, workingMood(tally, current))
74    return ran
75  })
76
77  on('turn.complete', async ($, e, next) => {
78    const done = await next(e)
79    if (e.agentId === undefined) show($, doneMood(tally, e.reason, e.isAborted))
80    return done
81  })
82}
83