status-band

1import { atom, read, update } from 'claude-code'
2import type { EngineInterface, Register, SessionContextUsage, SessionRateLimit } from 'claude-code'
3
4const usage = atom({ plugin: 'status-band', key: 'usage' } as const, null)
5
6// AGENTS.md「Claude Code Cloud Session への委譲」。期限を過ぎたら表示しない（この定数ごと消してよい）。
7const CLOUD_CREDIT_EXPIRES = Date.parse('2026-11-05T16:59:00+09:00')
8
9const JST_MS = 9 * 3600 * 1000
10const LABELS: Record<string, string> = { five_hour: '5時間', seven_day: '週間' }
11
12const jst = (iso: string, withDate: boolean) => {
13  const t = new Date(Date.parse(iso) + JST_MS).toISOString()
14  const hm = t.slice(11, 16)
15  return withDate ? `${t.slice(5, 7)}/${t.slice(8, 10)} ${hm}` : hm
16}
17
18const levelColor = (p: number) => (p >= 90 ? 'red' : p >= 70 ? 'yellow' : 'green')
19
20const save = async ($: EngineInterface, context: SessionContextUsage, limits: readonly SessionRateLimit[]) => {
21  const model = await $.session.model()
22  await update($, usage, () => ({
23    model,
24    contextPercent: context.percent,
25    limits: limits.map(l => ({ kind: l.kind, percentUsed: l.percentUsed, resetsAt: l.resetsAt })),
26  }))
27}
28
29export const register: Register = on => {
30  on('session.start', async ($, e, next) => {
31    const ran = await next(e)
32    const now = await $.session.usage()
33    await save($, now.context, now.rateLimits)
34
35    return ran
36  })
37
38  on('session.measure', async ($, e, next) => {
39    await save($, e.context, e.rateLimits)
40
41    return next(e)
42  })
43
44  on('ui.render', { component: 'AbovePrompt' }, async ($, e, next) => {
45    const u = await read($, usage)
46    if (e.props.hasSurvey || u === null) return next(e)
47
48    const { Box, Text } = $.ui.resolve(e)
49    const now = Date.now()
50    const ctx = u.contextPercent
51    const filled = ctx === undefined ? 0 : Math.round(ctx / 10)
52    const creditDays = Math.ceil((CLOUD_CREDIT_EXPIRES - now) / 86400000)
53
54    return (
55      <Box columnGap={1} flexWrap="wrap">
56        <Text dimColor>[{u.model}]</Text>
57        <Text>
58          文脈 {e.props.bodyColumns >= 65 && `[${'#'.repeat(filled)}${'-'.repeat(10 - filled)}] `}
59        </Text>
60        {ctx === undefined ? (
61          <Text dimColor>未取得</Text>
62        ) : (
63          <Text color={levelColor(ctx)}>{ctx}%</Text>
64        )}
65        {u.limits
66          .filter(l => LABELS[l.kind] !== undefined)
67          .map(l => {
68            const isPast = l.resetsAt !== undefined && Date.parse(l.resetsAt) <= now
69            const reset =
70              l.resetsAt === undefined ? '' : isPast ? '（更新待ち）' : `→${jst(l.resetsAt, l.kind === 'seven_day')}`
71            return (
72              <Text key={l.kind}>
73                | {LABELS[l.kind]} <Text color={levelColor(l.percentUsed)}>{Math.round(l.percentUsed)}%</Text>
74                {reset}
75              </Text>
76            )
77          })}
78        {creditDays > 0 && (
79          <Text color={creditDays <= 7 ? 'yellow' : undefined} dimColor={creditDays > 7}>
80            | Cloudクレジット失効まで{creditDays}日
81          </Text>
82        )}
83      </Box>
84    )
85  })
86}
87

1export type Usage = {
2  model: string
3  contextPercent?: number
4  limits: { kind: string; percentUsed: number; resetsAt?: string }[]
5}
6
7declare module 'claude-code' {
8  interface PluginState {
9    'status-band': {
10      usage: Usage | null
11    }
12  }
13}
14