nonbinary.computer
forked from
oppi.li/claude-code
source dump of claude code
oppi.li
1/**
2 * Swarm Reconnection Module
3 *
4 * Handles initialization of swarm context for teammates.
5 * - Fresh spawns: Initialize from CLI args (set in main.tsx via dynamicTeamContext)
6 * - Resumed sessions: Initialize from teamName/agentName stored in the transcript
7 */
8
9import type { AppState } from '../../state/AppState.js'
10import { logForDebugging } from '../debug.js'
11import { logError } from '../log.js'
12import { getDynamicTeamContext } from '../teammate.js'
13import { getTeamFilePath, readTeamFile } from './teamHelpers.js'
14
15/**
16 * Computes the initial teamContext for AppState.
17 *
18 * This is called synchronously in main.tsx to compute the teamContext
19 * BEFORE the first render, eliminating the need for useEffect workarounds.
20 *
21 * @returns The teamContext object to include in initialState, or undefined if not a teammate
22 */
23export function computeInitialTeamContext():
24 | AppState['teamContext']
25 | undefined {
26 // dynamicTeamContext is set in main.tsx from CLI args
27 const context = getDynamicTeamContext()
28
29 if (!context?.teamName || !context?.agentName) {
30 logForDebugging(
31 '[Reconnection] computeInitialTeamContext: No teammate context set (not a teammate)',
32 )
33 return undefined
34 }
35
36 const { teamName, agentId, agentName } = context
37
38 // Read team file to get lead agent ID
39 const teamFile = readTeamFile(teamName)
40 if (!teamFile) {
41 logError(
42 new Error(
43 `[computeInitialTeamContext] Could not read team file for ${teamName}`,
44 ),
45 )
46 return undefined
47 }
48
49 const teamFilePath = getTeamFilePath(teamName)
50
51 const isLeader = !agentId
52
53 logForDebugging(
54 `[Reconnection] Computed initial team context for ${isLeader ? 'leader' : `teammate ${agentName}`} in team ${teamName}`,
55 )
56
57 return {
58 teamName,
59 teamFilePath,
60 leadAgentId: teamFile.leadAgentId,
61 selfAgentId: agentId,
62 selfAgentName: agentName,
63 isLeader,
64 teammates: {},
65 }
66}
67
68/**
69 * Initialize teammate context from a resumed session.
70 *
71 * This is called when resuming a session that has teamName/agentName stored
72 * in the transcript. It sets up teamContext in AppState so that heartbeat
73 * and other swarm features work correctly.
74 */
75export function initializeTeammateContextFromSession(
76 setAppState: (updater: (prev: AppState) => AppState) => void,
77 teamName: string,
78 agentName: string,
79): void {
80 // Read team file to get lead agent ID
81 const teamFile = readTeamFile(teamName)
82 if (!teamFile) {
83 logError(
84 new Error(
85 `[initializeTeammateContextFromSession] Could not read team file for ${teamName} (agent: ${agentName})`,
86 ),
87 )
88 return
89 }
90
91 // Find the member in the team file to get their agentId
92 const member = teamFile.members.find(m => m.name === agentName)
93 if (!member) {
94 logForDebugging(
95 `[Reconnection] Member ${agentName} not found in team ${teamName} - may have been removed`,
96 )
97 }
98 const agentId = member?.agentId
99
100 const teamFilePath = getTeamFilePath(teamName)
101
102 // Set teamContext in AppState
103 setAppState(prev => ({
104 ...prev,
105 teamContext: {
106 teamName,
107 teamFilePath,
108 leadAgentId: teamFile.leadAgentId,
109 selfAgentId: agentId,
110 selfAgentName: agentName,
111 isLeader: false,
112 teammates: {},
113 },
114 }))
115
116 logForDebugging(
117 `[Reconnection] Initialized agent context from session for ${agentName} in team ${teamName}`,
118 )
119}