nonbinary.computer
forked from
oppi.li/claude-code
source dump of claude code
oppi.li
1import { useEffect, useRef } from 'react'
2import { getIsRemoteMode } from '../../bootstrap/state.js'
3import {
4 type Notification,
5 useNotifications,
6} from '../../context/notifications.js'
7import { useAppState } from '../../state/AppState.js'
8import { isInProcessTeammateTask } from '../../tasks/InProcessTeammateTask/types.js'
9
10function parseCount(notif: Notification): number {
11 if (!('text' in notif)) {
12 return 1
13 }
14 const match = notif.text.match(/^(\d+)/)
15 return match?.[1] ? parseInt(match[1], 10) : 1
16}
17
18function foldSpawn(acc: Notification, _incoming: Notification): Notification {
19 return makeSpawnNotif(parseCount(acc) + 1)
20}
21
22function makeSpawnNotif(count: number): Notification {
23 return {
24 key: 'teammate-spawn',
25 text: count === 1 ? '1 agent spawned' : `${count} agents spawned`,
26 priority: 'low',
27 timeoutMs: 5000,
28 fold: foldSpawn,
29 }
30}
31
32function foldShutdown(
33 acc: Notification,
34 _incoming: Notification,
35): Notification {
36 return makeShutdownNotif(parseCount(acc) + 1)
37}
38
39function makeShutdownNotif(count: number): Notification {
40 return {
41 key: 'teammate-shutdown',
42 text: count === 1 ? '1 agent shut down' : `${count} agents shut down`,
43 priority: 'low',
44 timeoutMs: 5000,
45 fold: foldShutdown,
46 }
47}
48
49/**
50 * Fires batched notifications when in-process teammates spawn or shut down.
51 * Uses fold() to combine repeated events into a single notification
52 * like "3 agents spawned" or "2 agents shut down".
53 */
54export function useTeammateLifecycleNotification(): void {
55 const tasks = useAppState(s => s.tasks)
56 const { addNotification } = useNotifications()
57 const seenRunningRef = useRef<Set<string>>(new Set())
58 const seenCompletedRef = useRef<Set<string>>(new Set())
59
60 useEffect(() => {
61 if (getIsRemoteMode()) return
62 for (const [id, task] of Object.entries(tasks)) {
63 if (!isInProcessTeammateTask(task)) {
64 continue
65 }
66
67 if (task.status === 'running' && !seenRunningRef.current.has(id)) {
68 seenRunningRef.current.add(id)
69 addNotification(makeSpawnNotif(1))
70 }
71
72 if (task.status === 'completed' && !seenCompletedRef.current.has(id)) {
73 seenCompletedRef.current.add(id)
74 addNotification(makeShutdownNotif(1))
75 }
76 }
77 }, [tasks, addNotification])
78}