toast-notify

1import type { EngineInterface, Register } from 'claude-code'
2
3// Windows PowerShell 5.1 の AppUserModelID。pwsh 7 は WinRT を読めないため powershell.exe を使う
4const APP_ID = '{1AC14E77-02E7-4E5D-B744-2EB1AE5198B7}\\WindowsPowerShell\\v1.0\\powershell.exe'
5// タイトル・本文は環境変数で渡し、スクリプト側で XML エスケープする（引数経由の注入を避ける）
6// rdpmanager で RDP 接続中なら仮想チャネル CCNOTIF でクライアント側へ送り、届かなければローカルにトーストを出す
7// （CCNOTIF の仕様は ntaksh42/rdp-manager の docs/remote-notifications.md）
8const SCRIPT = [
9  "$ErrorActionPreference = 'Stop'",
10  '$sent = $false',
11  'try { ' + [
12    'Add-Type -Namespace ToastNotify -Name Wts -MemberDefinition \'[DllImport("wtsapi32.dll", SetLastError = true, CharSet = CharSet.Ansi)] public static extern IntPtr WTSVirtualChannelOpen(IntPtr hServer, int sessionId, string name); [DllImport("wtsapi32.dll", SetLastError = true)] public static extern bool WTSVirtualChannelWrite(IntPtr h, byte[] buffer, int length, out int written); [DllImport("wtsapi32.dll")] public static extern bool WTSVirtualChannelClose(IntPtr h);\'',
13    '$m = if ($env:TOAST_NOTIFY_SESSION) { $env:TOAST_NOTIFY_BODY + [Environment]::NewLine + $env:TOAST_NOTIFY_SESSION } else { $env:TOAST_NOTIFY_BODY }',
14    '$j = @{ title = $env:TOAST_NOTIFY_TITLE; message = $m; level = $env:TOAST_NOTIFY_LEVEL } | ConvertTo-Json -Compress',
15    '$p = [Text.Encoding]::ASCII.GetBytes([Convert]::ToBase64String([Text.Encoding]::UTF8.GetBytes($j)))',
16    // 静的チャネルは 1 チャンク 1600 バイトを超えると分割され、受信側で捨てられる
17    '$h = if ($p.Length -le 1500) { [ToastNotify.Wts]::WTSVirtualChannelOpen([IntPtr]::Zero, -1, \'CCNOTIF\') } else { [IntPtr]::Zero }',
18    'if ($h -ne [IntPtr]::Zero) { try { $w = 0; $sent = [ToastNotify.Wts]::WTSVirtualChannelWrite($h, $p, $p.Length, [ref]$w) -and $w -eq $p.Length } finally { [void][ToastNotify.Wts]::WTSVirtualChannelClose($h) } }',
19  ].join('; ') + ' } catch { }',
20  'if ($sent) { exit 0 }',
21  '[void][Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType = WindowsRuntime]',
22  '[void][Windows.Data.Xml.Dom.XmlDocument, Windows.Data.Xml.Dom.XmlDocument, ContentType = WindowsRuntime]',
23  '$t = [Security.SecurityElement]::Escape($env:TOAST_NOTIFY_TITLE)',
24  '$b = [Security.SecurityElement]::Escape($env:TOAST_NOTIFY_BODY)',
25  '$s = [Security.SecurityElement]::Escape($env:TOAST_NOTIFY_SESSION)',
26  "$a = if ($s) { '<text placement=''attribution''>' + $s + '</text>' } else { '' }",
27  '$x = New-Object Windows.Data.Xml.Dom.XmlDocument',
28  "$x.LoadXml('<toast><visual><binding template=''ToastGeneric''><text>' + $t + '</text><text>' + $b + '</text>' + $a + '</binding></visual></toast>')",
29  '[Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier($env:TOAST_NOTIFY_APP).Show([Windows.UI.Notifications.ToastNotification]::new($x))',
30].join('; ')
31const MAX_BODY = 200
32// 停滞判定の間隔。活動の記録は同期的にフラグを立てるだけにし、時刻はここで読む
33const STALL_TICK_MS = 15_000
34
35// リロードで初期化されるため register で設定し直す
36const cfg = { minTurnMs: 30_000, minCommandMs: 60_000, stallMs: 300_000 }
37// メインのターン中に、最後に進捗（モデル出力・ツール呼び出し）があった時刻
38const stall = { active: false, last: 0, dirty: false, notified: false }
39const touch = () => { stall.dirty = true }
40// セッション名。/rename や auto-session-title の命名はプロンプト送信・セッション開始の入出力で拾う
41let sessionTitle = ''
42
43export const formatDuration = (ms: number) => {
44  const s = Math.round(ms / 1000)
45  return s < 60 ? `${s}秒` : `${Math.floor(s / 60)}分${s % 60 ? `${s % 60}秒` : ''}`
46}
47
48export const firstLine = (text: string) => {
49  const line = text.split('\n').map(l => l.trim()).find(l => l !== '') ?? ''
50  return line.length > MAX_BODY ? `${line.slice(0, MAX_BODY - 1)}…` : line
51}
52
53const project = async ($: EngineInterface) => {
54  const cwd = await $.session.cwd()
55  return cwd.split(/[\\/]/).filter(Boolean).pop() ?? cwd
56}
57
58type Level = 'info' | 'warn'
59
60async function toast($: EngineInterface, title: string, body: string, level: Level) {
61  const ran = await $.process.run(['powershell.exe', '-NoProfile', '-NonInteractive', '-Command', SCRIPT], {
62    env: { TOAST_NOTIFY_TITLE: `${title} - ${await project($)}`, TOAST_NOTIFY_BODY: body, TOAST_NOTIFY_SESSION: sessionTitle, TOAST_NOTIFY_LEVEL: level, TOAST_NOTIFY_APP: APP_ID },
63    timeoutMs: 15_000,
64  })
65  if (ran.exitCode !== 0) await $.ui.log(`toast-notify: ${ran.stderr.trim().split('\n')[0]}`, { to: 'debug' })
66}
67
68// 通知の失敗でフックを止めない。表示を待たずに戻る
69const notify = ($: EngineInterface, title: string, body: string, level: Level = 'info') => {
70  toast($, title, body, level).catch(error => $.ui.log(`toast-notify: ${String(error)}`, { to: 'debug' }))
71}
72
73export const register: Register = (on, options) => {
74  if (options.enabled === false) return
75  cfg.minTurnMs = Math.max(0, Number(options.minTurnSeconds ?? 30)) * 1000
76  cfg.minCommandMs = Math.max(0, Number(options.minCommandSeconds ?? 60)) * 1000
77  cfg.stallMs = Math.max(0, Number(options.stallSeconds ?? 300)) * 1000
78
79  if (cfg.stallMs > 0) {
80    on('session.start', async ($, e, next) => {
81      const started = await next(e)
82      $.clock.every(STALL_TICK_MS, () => {
83        $.clock.now().then(now => {
84          if (!stall.active) return
85          if (stall.dirty) Object.assign(stall, { last: now, dirty: false, notified: false })
86          else if (!stall.notified && now - stall.last >= cfg.stallMs) {
87            stall.notified = true
88            notify($, 'Claude: 停止の可能性', `${formatDuration(cfg.stallMs)}間進捗がありません`, 'warn')
89          }
90        }, error => $.ui.log(`toast-notify: ${String(error)}`, { to: 'debug' }))
91      })
92      return started
93    })
94
95    on('turn.start', async ($, e, next) => {
96      Object.assign(stall, { active: true, last: await $.clock.now(), dirty: false, notified: false })
97      return next(e)
98    })
99
100    // サブエージェントの出力も進捗とみなす
101    on('turn.step', async function* ($, e, next) {
102      const stream = next(e)
103      for await (const chunk of stream) {
104        touch()
105        yield chunk
106      }
107      return await stream.result
108    })
109  }
110
111  on('turn.complete', async ($, e, next) => {
112    const done = await next(e)
113    if (e.agentId === undefined) stall.active = false
114    if (e.agentId !== undefined || e.isAborted) return done
115    if (e.reason === 'error' || e.reason === 'refusal') {
116      const detail = e.reason === 'refusal' ? e.refusal.explanation ?? '' : firstLine(e.answer)
117      notify($, 'Claude: エラーで停止', detail || (e.reason === 'refusal' ? 'リクエストが拒否されました' : 'API エラー'), 'warn')
118    } else if (e.durationMs >= cfg.minTurnMs) {
119      notify($, `Claude: 完了 (${formatDuration(e.durationMs)})`, firstLine(e.answer) || '応答が完了しました')
120    }
121    return done
122  })
123
124  on('classic.SessionStart', async ($, e, next) => {
125    const ran = await next(e)
126    sessionTitle = ran.sessionTitle ?? e.session_title ?? sessionTitle
127    return ran
128  })
129  on('classic.UserPromptSubmit', async ($, e, next) => {
130    const ran = await next(e)
131    sessionTitle = ran.sessionTitle ?? e.session_title ?? sessionTitle
132    return ran
133  })
134
135  // 許可・質問ダイアログ。idle_prompt はターン完了通知と重なるので出さない
136  on('classic.Notification', async ($, e, next) => {
137    const done = await next(e)
138    if (e.agent_id === undefined && e.notification_type !== 'idle_prompt' && e.notification_type !== 'auth_success') {
139      notify($, 'Claude: 入力待ち', e.message, 'warn')
140    }
141    return done
142  })
143
144  // ツールの開始・終了も進捗とみなす（実行中のまま 5 分黙ったコマンドは停滞として通知する）
145  on('tool.call', async ($, e, next) => {
146    touch()
147    try {
148      return await next(e)
149    } finally {
150      touch()
151    }
152  })
153
154  for (const tool of ['Bash', 'PowerShell'] as const) {
155    on('tool.call', { tool }, async ($, e, next) => {
156      const started = await $.clock.now()
157      const ran = await next(e)
158      const elapsed = (await $.clock.now()) - started
159      if (ran.deny === undefined && elapsed >= cfg.minCommandMs) {
160        const failed = ran.isError === true
161        notify($, `${failed ? 'コマンド失敗' : 'コマンド完了'} (${formatDuration(elapsed)})`, firstLine(e.command), failed ? 'warn' : 'info')
162      }
163      return ran
164    })
165  }
166}
167