token-weather

1// Copyright 2026 Anthropic PBC
2// SPDX-License-Identifier: Apache-2.0
3//
4// Token Weather: a live forecast of the context window, above the prompt.
5// Based on mods/token-weather in anthropics/claude-code-playground (Apache-2.0, see LICENSE).
6//
7// japan-da-man による変更:
8// - 計測値を $.state に置く（プラグインを読み込み直しても履歴が消えない。/clear などでは初期化される）
9// - 前回と同じ計測値は追加しない（読み込み直した直後に同じ値が届いても二重に数えない）
10// - 帯に他の mod が描くものを残す（Replay Theater などと共存する）
11// - /weather で表示を切り替える
12// - 描画サイトの値は e.props から読む
13
14import { atom, read, update } from 'claude-code'
15
16const HISTORY = 12
17const BARS = '▁▂▃▄▅▆▇█'
18
19// 使用率ごとの天気。絵文字ではなく 1 文字幅の記号を使う（どのターミナルのフォントでも揃う）
20export const FORECAST = [
21  { upTo: 25, icon: '☀', word: 'Clear', color: 'yellow' },
22  { upTo: 50, icon: '☁', word: 'Cloudy', color: 'cyan' },
23  { upTo: 75, icon: '☂', word: 'Showers', color: 'blue' },
24  { upTo: 90, icon: '☇', word: 'Storm', color: 'magenta' },
25  { upTo: Infinity, icon: '↯', word: 'Compact soon', color: 'red' },
26]
27
28// 計測値 { tokens, window, percent } の並び（古い順）
29const readingsAtom = atom({ plugin: 'token-weather', key: 'readings' }, [])
30// /weather で表示を切り替える
31let hidden = false
32
33// 計測値を 1 つ足した次の並びを返す。足すものがなければ同じ配列を返す
34export function addReading(readings, context) {
35  if (!context || !context.window) return readings
36  const tokens = context.tokens ?? 0
37  const percent = Math.round(context.percent ?? (tokens / context.window) * 100)
38  // 最初の応答の前の 0 の計測は、本物の計測が来たら外す
39  const kept = readings.filter((r) => r.tokens > 0)
40  const last = kept[kept.length - 1]
41  if (last && last.tokens === tokens && last.window === context.window) return readings
42  if (tokens === 0 && readings.length) return readings
43  return [...kept, { tokens, window: context.window, percent }].slice(-HISTORY)
44}
45
46export function forecastFor(percent) {
47  return FORECAST.find((f) => percent < f.upTo) ?? FORECAST[FORECAST.length - 1]
48}
49
50// バーの高さは表示中で一番多いターンを基準にする。使用率が低くても増え方が見える
51export function chart(readings) {
52  const top = Math.max(...readings.map((r) => r.tokens), 1)
53  return readings.map((r) => BARS[Math.min(BARS.length - 1, Math.floor((r.tokens / top) * (BARS.length - 1)))]).join('')
54}
55
56export function trendWord(readings) {
57  if (readings.length < 2) return ''
58  const delta = readings[readings.length - 1].tokens - readings[readings.length - 2].tokens
59  if (delta > 0) return `▲ +${short(delta)} last turn`
60  if (delta < 0) return `▼ ${short(-delta)} last turn`
61  return 'steady'
62}
63
64export function short(n) {
65  if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(n % 1_000_000 === 0 ? 0 : 1)}M`
66  if (n >= 1_000) return `${(n / 1_000).toFixed(n % 1_000 === 0 ? 0 : 1)}k`
67  return String(n)
68}
69
70async function takeReading($) {
71  try {
72    const { context } = await $.session.usage()
73    // $.state に書くと、それを読んでいる帯は自動で描き直される
74    await update($, readingsAtom, (rs) => addReading(rs, context))
75  } catch {
76    // このターンは計測なし。帯は前回の値のまま
77  }
78}
79
80export function register(on) {
81  on('session.start', async ($, e, next) => {
82    const result = await next(e)
83    hidden = (await $.store.get('hidden')) === true
84    await takeReading($)
85    await $.command.register({ name: 'weather', description: 'Token Weather の表示を切り替える', immediate: true })
86    return result
87  })
88
89  // メインのターンが終わるたびに計測する。サブエージェントのターンは数えない
90  on('turn.complete', async ($, e, next) => {
91    const result = await next(e)
92    if (!e.agentId) await takeReading($)
93    return result
94  })
95
96  on('command.run', { command: 'weather' }, async ($) => {
97    hidden = !hidden
98    $.ui.invalidate('ui.render')
99    await $.store.set('hidden', hidden)
100    return { text: hidden ? 'Token Weather を非表示にしました' : 'Token Weather を表示しました' }
101  })
102
103  on('ui.render', { component: 'AbovePrompt' }, async ($, e, next) => {
104    const readings = await read($, readingsAtom)
105    if (hidden || e.props.hasSurvey || readings.length === 0) return next(e)
106    const { Box, Text } = $.ui.resolve(e)
107    const line = band(Box, Text, readings, e.props.bodyColumns ?? 80)
108    // 他の mod が帯に描くものも残す
109    const rest = await next(e)
110    return rest ? Box({ flexDirection: 'column', children: [line, rest] }) : line
111  })
112}
113
114function band(Box, Text, readings, columns) {
115  const now = readings[readings.length - 1]
116  const f = forecastFor(now.percent)
117  const trend = trendWord(readings)
118  const parts = [
119    Text({ color: f.color, bold: true, children: `${f.icon}  ${f.word}` }),
120    Text({ children: `  ${now.percent}% of context` }),
121    Text({ dimColor: true, children: `  ${short(now.tokens)} / ${short(now.window)}` }),
122  ]
123  // 狭い画面ではグラフと増減を省く
124  if (columns >= 60) {
125    parts.push(Text({ dimColor: true, children: '   last turns ' }))
126    parts.push(Text({ color: f.color, children: chart(readings) }))
127    if (trend) parts.push(Text({ dimColor: true, children: `  ${trend}` }))
128  }
129  return Box({ flexDirection: 'row', paddingX: 1, children: parts })
130}
131

1declare module 'claude-code' {
2  interface PluginState {
3    'token-weather': {
4      // 直近のターンの計測値（古い順）
5      readings: Array<{ tokens: number; window: number; percent: number }>
6    }
7  }
8}
9