session-meter

1import { atom, read, update } from 'claude-code'
2import type { EngineInterface, Register } from 'claude-code'
3
4import { EMPTY, statusLine, turnToast, withFile } from './format'
5
6const REFRESH_MS = 30_000
7const EDITING_TOOLS = new Set(['Edit', 'Write', 'NotebookEdit'])
8const meter = atom({ plugin: 'session-meter', key: 'meter' } as const, EMPTY)
9
10async function refresh($: EngineInterface): Promise<void> {
11  const elapsed = (await $.clock.now()) - (await $.session.usage()).startedAt
12  $.ui.status(statusLine(await read($, meter), elapsed))
13}
14
15function editedPath(e: { tool: string; file_path?: unknown; notebook_path?: unknown }): string | undefined {
16  const path = e.file_path ?? e.notebook_path
17  return typeof path === 'string' ? path : undefined
18}
19
20export const register: Register = (on, options) => {
21  const toastAfterMs = Number(options.toastAfterSeconds ?? 60) * 1000
22
23  on('session.start', async ($, e, next) => {
24    await $.command.register({
25      name: 'meter',
26      description: 'Session Meter: elapsed time, tool calls and files edited this session',
27      immediate: true,
28    })
29    $.clock.every(REFRESH_MS, () => void refresh($).catch(() => undefined))
30    await refresh($)
31
32    return next(e)
33  })
34
35  on('turn.start', async ($, e, next) => {
36    await update($, meter, now => ({ ...now, turnTools: 0, turnFiles: [] }))
37
38    return next(e)
39  })
40
41  on('tool.call', async ($, e, next) => {
42    await update($, meter, now => ({ ...now, tools: now.tools + 1, turnTools: now.turnTools + 1 }))
43    const ran = await next(e)
44
45    const path = EDITING_TOOLS.has(e.tool) ? editedPath(e as { tool: string }) : undefined
46    if (path !== undefined && ran.deny === undefined && ran.isError !== true) {
47      await update($, meter, now => ({
48        ...now,
49        files: withFile(now.files, path),
50        turnFiles: withFile(now.turnFiles, path),
51      }))
52    }
53    await refresh($)
54
55    return ran
56  })
57
58  on('turn.complete', async ($, e, next) => {
59    const ran = await next(e)
60    const isLong = toastAfterMs > 0 && e.durationMs >= toastAfterMs
61    if (e.agentId === undefined && !e.isAborted && isLong) {
62      $.ui.toast(turnToast(await read($, meter), e.durationMs), { timeoutMs: 8000 })
63    }
64    await refresh($)
65
66    return ran
67  })
68
69  on('session.end', async ($, e, next) => {
70    if (e.reason === 'clear') await update($, meter, () => EMPTY)
71
72    return next(e)
73  })
74
75  on('command.run', { command: 'meter' }, async $ => {
76    const now = await read($, meter)
77    const elapsed = (await $.clock.now()) - (await $.session.usage()).startedAt
78    const files = now.files.length === 0 ? '' : `\nFiles: ${now.files.join(', ')}`
79
80    return { text: `Session Meter: ${statusLine(now, elapsed)}${files}` }
81  })
82}
83

1import type { Meter } from '../types'
2
3export const EMPTY: Meter = { tools: 0, files: [], turnTools: 0, turnFiles: [] }
4
5export function duration(ms: number): string {
6  const seconds = Math.max(0, Math.floor(ms / 1000))
7  if (seconds < 60) return `${seconds}s`
8  const minutes = Math.floor(seconds / 60)
9  if (minutes < 60) return seconds % 60 === 0 || minutes >= 10 ? `${minutes}m` : `${minutes}m ${seconds % 60}s`
10  return `${Math.floor(minutes / 60)}h ${String(minutes % 60).padStart(2, '0')}m`
11}
12
13function plural(count: number, word: string): string {
14  return `${count} ${word}${count === 1 ? '' : 's'}`
15}
16
17export function statusLine(meter: Meter, elapsedMs: number): string {
18  return `⏱ ${duration(elapsedMs)} · ${plural(meter.tools, 'tool call')} · ${plural(meter.files.length, 'file')} edited`
19}
20
21export function turnToast(meter: Meter, durationMs: number): string {
22  return `Done in ${duration(durationMs)} · ${plural(meter.turnTools, 'tool call')} · ${plural(meter.turnFiles.length, 'file')} edited`
23}
24
25export function withFile(files: readonly string[], file: string): string[] {
26  return files.includes(file) ? [...files] : [...files, file]
27}
28

1export type Meter = {
2  tools: number
3  files: string[]
4  turnTools: number
5  turnFiles: string[]
6}
7
8declare module 'claude-code' {
9  interface PluginState {
10    'session-meter': { meter: Meter }
11  }
12}
13