working-memory

1import { atom, read, update } from 'claude-code'
2import type { EngineInterface, Register, SessionContextUsage } from 'claude-code'
3
4import type { Brain, LogEntry, Region } from '../types'
5
6// ── 腦區與工具的對應（比喻，不是神經科學主張）────────────────────────
7const REGIONS: Region[] = ['frontal', 'motor', 'parietal', 'occipital', 'temporal', 'cerebellum']
8
9const INFO: Record<Region, { name: string; role: string; color: string }> = {
10  frontal: { name: '額葉', role: '規劃', color: 'magenta' },
11  motor: { name: '運動皮質', role: '動手改', color: 'yellow' },
12  parietal: { name: '頂葉', role: '操作環境', color: 'green' },
13  occipital: { name: '枕葉', role: '閱讀', color: 'cyan' },
14  temporal: { name: '顳葉', role: '聽外界', color: 'blue' },
15  cerebellum: { name: '小腦', role: '分身協作', color: 'white' },
16}
17
18const regionOfTool = (tool: string): Region => {
19  if (/^(Read|Grep|Glob|NotebookRead|LS)$/.test(tool)) return 'occipital'
20  if (/^(Edit|Write|MultiEdit|NotebookEdit)$/.test(tool)) return 'motor'
21  if (/^(WebFetch|WebSearch)$/.test(tool) || tool.startsWith('mcp__')) return 'temporal'
22  if (/^(Agent|Task|SendMessage)$/.test(tool)) return 'cerebellum'
23  if (/^(TaskCreate|TaskUpdate|TodoWrite|Skill|EnterPlanMode|ExitPlanMode)$/.test(tool)) return 'frontal'
24  return 'parietal'
25}
26
27// ── 腦的形狀：側面圖，左邊是前額 ─────────────────────────────────────
28const W = 46
29const H = 15
30const TICK_MS = 100
31const DECAY = 0.86
32const SLEEP_TICKS = 28
33
34type Cell = { region: Region | 'stem'; u: number; seed: number }
35
36const hash = (n: number) => {
37  let x = Math.imul(n ^ 0x9e3779b9, 0x85ebca6b)
38  x ^= x >>> 13
39  x = Math.imul(x, 0xc2b2ae35)
40  x ^= x >>> 16
41  return (x >>> 0) / 4294967296
42}
43
44const MAP: (Cell | null)[][] = Array.from({ length: H }, (_, y) =>
45  Array.from({ length: W }, (_, x) => {
46    const u = (x + 0.5) / W
47    const v = (y + 0.5) / H
48    const seed = y * W + x
49    // 大腦：外緣加一點起伏，像腦回
50    const dx = (u - 0.47) / 0.46
51    const dy = (v - 0.4) / 0.38
52    const wobble = 1 + 0.05 * Math.sin(Math.atan2(dy, dx) * 9)
53    const inCerebrum = dx * dx + dy * dy <= wobble * wobble
54    // 小腦與腦幹
55    const cx = (u - 0.77) / 0.15
56    const cy = (v - 0.8) / 0.17
57    const inCerebellum = cx * cx + cy * cy <= 1
58    const inStem = u > 0.57 && u < 0.65 && v > 0.7
59    if (inCerebrum) {
60      // 外側溝與中央溝留白，讓腦區分得出來
61      const sylvian = Math.abs(v - (0.62 - (u - 0.2) * 0.25)) < 0.035 && u > 0.2 && u < 0.62
62      const central = Math.abs(u - (0.42 + (v - 0.1) * 0.12)) < 0.012 && v < 0.55
63      if (sylvian || central) return null
64      let region: Region
65      if (v > 0.6 - (u - 0.2) * 0.25 && u > 0.2 && u < 0.74) region = 'temporal'
66      else if (u < 0.42 + (v - 0.1) * 0.12 - 0.09) region = 'frontal'
67      else if (u < 0.42 + (v - 0.1) * 0.12) region = 'motor'
68      else if (u > 0.76) region = 'occipital'
69      else region = 'parietal'
70      return { region, u, seed }
71    }
72    if (inCerebellum) return { region: 'cerebellum', u, seed }
73    if (inStem) return { region: 'stem', u, seed }
74    return null
75  }),
76)
77
78// ── 狀態 ─────────────────────────────────────────────────────────────
79const EMPTY: Brain = { tick: 0, act: REGIONS.map(() => 0), fill: 0, tokens: 0, window: 0, sleepAt: null, log: [] }
80const brain = atom({ plugin: 'working-memory', key: 'brain' } as const, EMPTY)
81
82const PANE = 'working-memory'
83
84async function fire($: EngineInterface, region: Region, entry: LogEntry | null) {
85  await update($, brain, b => ({
86    ...b,
87    act: b.act.map((a, i) => (REGIONS[i] === region ? 1 : a)),
88    log: entry ? [...b.log, entry].slice(-6) : b.log,
89  }))
90}
91
92async function measure($: EngineInterface, c: SessionContextUsage) {
93  if (c.percent === undefined || c.tokens === undefined) return
94  const { percent, tokens, window } = c
95  await update($, brain, b => ({ ...b, fill: percent / 100, tokens, window }))
96}
97
98let ticking = false
99
100// 每一格：活化衰減、推進睡眠動畫；沒有東西在動就不重畫
101async function step($: EngineInterface) {
102  if (ticking) return
103  ticking = true
104  try {
105    const b = await read($, brain)
106    const isMoving = b.act.some(a => a > 0.02) || b.sleepAt !== null || b.fill >= 0.85
107    if (!isMoving) return
108    await update($, brain, cur => {
109      const tick = cur.tick + 1
110      const isAwake = cur.sleepAt !== null && tick - cur.sleepAt > SLEEP_TICKS
111      return {
112        ...cur,
113        tick,
114        act: cur.act.map(a => (a * DECAY < 0.02 ? 0 : a * DECAY)),
115        sleepAt: isAwake ? null : cur.sleepAt,
116      }
117    })
118  } finally {
119    ticking = false
120  }
121}
122
123// ── 畫面 ─────────────────────────────────────────────────────────────
124type Run = { text: string; color?: string; bold?: boolean; dim?: boolean }
125
126const paintCell = (cell: Cell | null, b: Brain): Run => {
127  if (cell === null) return { text: ' ' }
128  if (cell.region === 'stem') return { text: '▒', dim: true }
129
130  // 睡眠：一道波從後腦掃到前額，掃過的地方清空
131  if (b.sleepAt !== null) {
132    const wave = 1 - (b.tick - b.sleepAt) / (SLEEP_TICKS * 0.8)
133    const d = cell.u - wave
134    if (Math.abs(d) < 0.04) return { text: '✦', color: 'white', bold: true }
135    if (d > 0) return { text: '·', color: 'blue', dim: true }
136  }
137
138  const color = INFO[cell.region].color
139  const act = b.act[REGIONS.indexOf(cell.region)] ?? 0
140  // 神經元放電：活化越強，同時閃的格子越多，每一格隨影格隨機
141  if (act > 0.02 && hash(cell.seed * 31 + b.tick) < act * 0.85) {
142    return { text: act > 0.55 ? '█' : '▓', color, bold: true }
143  }
144  // 記憶佔用：用量多少，就有多少比例的格子被填滿
145  if (hash(cell.seed) < b.fill) {
146    const isOverload = b.fill >= 0.85 && b.tick % 8 < 4
147    return { text: '░', color: isOverload ? 'red' : color }
148  }
149  return { text: '·', dim: true }
150}
151
152const runsOf = (cells: Run[]): Run[] =>
153  cells.reduce<Run[]>((acc, c) => {
154    const last = acc[acc.length - 1]
155    if (last && last.color === c.color && last.bold === c.bold && last.dim === c.dim) {
156      last.text += c.text
157    } else acc.push({ ...c })
158    return acc
159  }, [])
160
161const fmt = (n: number) =>
162  n >= 1_000_000 ? `${(n / 1_000_000).toFixed(1)}M` : n >= 1000 ? `${Math.round(n / 1000)}k` : `${n}`
163
164const mood = (b: Brain) =>
165  b.sleepAt !== null
166    ? '睡眠中：整理記憶…'
167    : b.fill >= 0.85
168      ? '過載！快要需要睡一覺（compact）'
169      : b.fill >= 0.6
170        ? '有點擁擠'
171        : '清醒、專注'
172
173const detailOf = (e: Record<string, unknown>): string => {
174  const raw = e.file_path ?? e.command ?? e.pattern ?? e.url ?? e.query ?? e.description ?? e.skill ?? ''
175  const s = String(raw).replace(/\s+/g, ' ')
176  const short = s.includes('/') && !s.includes(' ') ? (s.split('/').pop() ?? s) : s
177  return short.length > 36 ? short.slice(0, 35) + '…' : short
178}
179
180export const register: Register = on => {
181  on('session.start', async ($, e, next) => {
182    await $.command.register({ name: 'brain', description: '打開工作記憶腦圖' })
183    const { context } = await $.session.usage()
184    await measure($, context)
185    $.clock.every(TICK_MS, () => void step($))
186    return next(e)
187  })
188
189  on('command.run', { command: 'brain' }, async $ => {
190    await $.ui.open({ id: PANE, title: '工作記憶 · Working Memory' })
191    return { text: '工作記憶腦圖已打開。' }
192  })
193
194  on('prompt.submit', async ($, e, next) => {
195    await fire($, 'frontal', { region: 'frontal', tool: '思考', detail: '收到新任務' })
196    return next(e)
197  })
198
199  on('tool.call', async ($, e, next) => {
200    const region = regionOfTool(e.tool)
201    await fire($, region, { region, tool: e.tool.replace(/^mcp__[^_]+__/, ''), detail: detailOf(e as Record<string, unknown>) })
202    return next(e)
203  })
204
205  on('session.measure', async ($, e, next) => {
206    if (e.changed.includes('context')) await measure($, e.context)
207    return next(e)
208  })
209
210  on('session.compact', async ($, e, next) => {
211    if (e.trigger !== 'precompute') {
212      await update($, brain, b => ({ ...b, sleepAt: b.tick }))
213    }
214    return next(e)
215  })
216
217  on('ui.render', { component: 'Pane', requestId: PANE }, async ($, e) => {
218    const { Box, Text } = $.ui.resolve(e)
219    const b = await read($, brain)
220    const pct = Math.round(b.fill * 100)
221    const barCells = 20
222    const filled = Math.round(b.fill * barCells)
223    const barColor = b.fill >= 0.85 ? 'red' : b.fill >= 0.6 ? 'yellow' : 'green'
224
225    return (
226      <Box flexDirection="column">
227        <Box>
228          <Text bold>工作記憶 </Text>
229          <Text color={barColor}>{'█'.repeat(filled)}</Text>
230          <Text dimColor>{'░'.repeat(barCells - filled)}</Text>
231          <Text bold color={barColor}> {pct}%</Text>
232          <Text dimColor> {b.window > 0 ? `${fmt(b.tokens)}/${fmt(b.window)}` : ''}</Text>
233        </Box>
234        <Text dimColor>狀態：{mood(b)}</Text>
235        <Text> </Text>
236        {MAP.map((row, y) => (
237          <Box key={`row${y}`}>
238            {runsOf(row.map(cell => paintCell(cell, b))).map((r, i) => (
239              <Text key={`r${y}-${i}`} color={r.color} bold={r.bold} dimColor={r.dim}>
240                {r.text}
241              </Text>
242            ))}
243          </Box>
244        ))}
245        <Text> </Text>
246        <Box flexDirection="column">
247          {[REGIONS.slice(0, 3), REGIONS.slice(3)].map((group, gi) => (
248            <Box key={`legend${gi}`}>
249              {group.map(r => {
250                const isLit = (b.act[REGIONS.indexOf(r)] ?? 0) > 0.3
251                return (
252                  <Text key={`lg-${r}`} color={INFO[r].color} bold={isLit} dimColor={!isLit}>
253                    {isLit ? '● ' : '○ '}
254                    {INFO[r].name}·{INFO[r].role}
255                    {'   '}
256                  </Text>
257                )
258              })}
259            </Box>
260          ))}
261        </Box>
262        <Text> </Text>
263        <Text dimColor>最近的神經活動</Text>
264        {b.log.length === 0 && <Text dimColor>（還沒有活動）</Text>}
265        {b.log.map((l, i) => (
266          <Box key={`log${i}`}>
267            <Text color={INFO[l.region].color} bold={i === b.log.length - 1}>
268              {i === b.log.length - 1 ? '▸ ' : '  '}
269              {INFO[l.region].name.padEnd(4, '　')}
270            </Text>
271            <Text> {l.tool}</Text>
272            <Text dimColor> {l.detail}</Text>
273          </Box>
274        ))}
275      </Box>
276    )
277  })
278}
279

1export type Region = 'frontal' | 'motor' | 'parietal' | 'occipital' | 'temporal' | 'cerebellum'
2
3export type LogEntry = { region: Region; tool: string; detail: string }
4
5export type Brain = {
6  /** 動畫影格計數 */
7  tick: number
8  /** 各腦區活化程度 0–1，順序同 REGIONS */
9  act: number[]
10  /** context 用量 0–1 */
11  fill: number
12  tokens: number
13  window: number
14  /** compact 動畫開始的影格；null 表示沒在睡 */
15  sleepAt: number | null
16  log: LogEntry[]
17}
18
19declare module 'claude-code' {
20  interface PluginState {
21    'working-memory': { brain: Brain }
22  }
23}
24