@agentuity/coder 1.0.37

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (92) hide show
  1. package/README.md +57 -0
  2. package/dist/chain-preview.d.ts +55 -0
  3. package/dist/chain-preview.d.ts.map +1 -0
  4. package/dist/chain-preview.js +472 -0
  5. package/dist/chain-preview.js.map +1 -0
  6. package/dist/client.d.ts +43 -0
  7. package/dist/client.d.ts.map +1 -0
  8. package/dist/client.js +402 -0
  9. package/dist/client.js.map +1 -0
  10. package/dist/commands.d.ts +22 -0
  11. package/dist/commands.d.ts.map +1 -0
  12. package/dist/commands.js +99 -0
  13. package/dist/commands.js.map +1 -0
  14. package/dist/footer.d.ts +34 -0
  15. package/dist/footer.d.ts.map +1 -0
  16. package/dist/footer.js +249 -0
  17. package/dist/footer.js.map +1 -0
  18. package/dist/handlers.d.ts +24 -0
  19. package/dist/handlers.d.ts.map +1 -0
  20. package/dist/handlers.js +83 -0
  21. package/dist/handlers.js.map +1 -0
  22. package/dist/hub-overlay.d.ts +107 -0
  23. package/dist/hub-overlay.d.ts.map +1 -0
  24. package/dist/hub-overlay.js +1794 -0
  25. package/dist/hub-overlay.js.map +1 -0
  26. package/dist/index.d.ts +4 -0
  27. package/dist/index.d.ts.map +1 -0
  28. package/dist/index.js +1585 -0
  29. package/dist/index.js.map +1 -0
  30. package/dist/output-viewer.d.ts +49 -0
  31. package/dist/output-viewer.d.ts.map +1 -0
  32. package/dist/output-viewer.js +389 -0
  33. package/dist/output-viewer.js.map +1 -0
  34. package/dist/overlay.d.ts +40 -0
  35. package/dist/overlay.d.ts.map +1 -0
  36. package/dist/overlay.js +225 -0
  37. package/dist/overlay.js.map +1 -0
  38. package/dist/protocol.d.ts +118 -0
  39. package/dist/protocol.d.ts.map +1 -0
  40. package/dist/protocol.js +3 -0
  41. package/dist/protocol.js.map +1 -0
  42. package/dist/remote-session.d.ts +113 -0
  43. package/dist/remote-session.d.ts.map +1 -0
  44. package/dist/remote-session.js +645 -0
  45. package/dist/remote-session.js.map +1 -0
  46. package/dist/remote-tui.d.ts +40 -0
  47. package/dist/remote-tui.d.ts.map +1 -0
  48. package/dist/remote-tui.js +606 -0
  49. package/dist/remote-tui.js.map +1 -0
  50. package/dist/renderers.d.ts +34 -0
  51. package/dist/renderers.d.ts.map +1 -0
  52. package/dist/renderers.js +669 -0
  53. package/dist/renderers.js.map +1 -0
  54. package/dist/review.d.ts +15 -0
  55. package/dist/review.d.ts.map +1 -0
  56. package/dist/review.js +154 -0
  57. package/dist/review.js.map +1 -0
  58. package/dist/titlebar.d.ts +3 -0
  59. package/dist/titlebar.d.ts.map +1 -0
  60. package/dist/titlebar.js +59 -0
  61. package/dist/titlebar.js.map +1 -0
  62. package/dist/todo/index.d.ts +3 -0
  63. package/dist/todo/index.d.ts.map +1 -0
  64. package/dist/todo/index.js +3 -0
  65. package/dist/todo/index.js.map +1 -0
  66. package/dist/todo/store.d.ts +6 -0
  67. package/dist/todo/store.d.ts.map +1 -0
  68. package/dist/todo/store.js +43 -0
  69. package/dist/todo/store.js.map +1 -0
  70. package/dist/todo/types.d.ts +13 -0
  71. package/dist/todo/types.d.ts.map +1 -0
  72. package/dist/todo/types.js +2 -0
  73. package/dist/todo/types.js.map +1 -0
  74. package/package.json +44 -0
  75. package/src/chain-preview.ts +621 -0
  76. package/src/client.ts +515 -0
  77. package/src/commands.ts +132 -0
  78. package/src/footer.ts +305 -0
  79. package/src/handlers.ts +113 -0
  80. package/src/hub-overlay.ts +2324 -0
  81. package/src/index.ts +1907 -0
  82. package/src/output-viewer.ts +480 -0
  83. package/src/overlay.ts +294 -0
  84. package/src/protocol.ts +157 -0
  85. package/src/remote-session.ts +800 -0
  86. package/src/remote-tui.ts +707 -0
  87. package/src/renderers.ts +740 -0
  88. package/src/review.ts +201 -0
  89. package/src/titlebar.ts +63 -0
  90. package/src/todo/index.ts +2 -0
  91. package/src/todo/store.ts +49 -0
  92. package/src/todo/types.ts +14 -0
@@ -0,0 +1,707 @@
1
+ /**
2
+ * Remote TUI — Native Pi Coding Agent Renderer for Remote Sessions
3
+ *
4
+ * Creates a real AgentSession + InteractiveMode backed by a remote sandbox
5
+ * via Hub WebSocket, with the coder extension loaded for Hub UI (footer,
6
+ * /hub overlay, commands, titlebar).
7
+ *
8
+ * Architecture:
9
+ * Remote TUI → Hub WebSocket (controller) → Sandbox (Pi RPC mode)
10
+ * - User input → agent.prompt() (monkey-patched) → RPC `prompt` → Hub → sandbox
11
+ * - Sandbox Pi → AgentEvent stream → Hub broadcast → Agent.emit() → InteractiveMode renders natively
12
+ * - Hub UI → coder extension (loaded via DefaultResourceLoader) provides footer, /hub, commands
13
+ *
14
+ * The local Agent never calls an LLM. Its prompt/steer/abort are monkey-patched
15
+ * to send RPC commands. Its internal state is kept in sync with the remote agent
16
+ * by mirroring state updates from each received event.
17
+ *
18
+ * The coder extension sees AGENTUITY_CODER_REMOTE_SESSION (Hub UI mode) +
19
+ * AGENTUITY_CODER_NATIVE_REMOTE (skip legacy event rendering). It provides
20
+ * Hub connection, footer, /hub overlay, commands, titlebar — but does NOT
21
+ * intercept input or render events (this module handles both).
22
+ *
23
+ * IMPORTANT: Initialization order matters!
24
+ * 1. Create RemoteSession (no connection yet)
25
+ * 2. Create AgentSession, patch Agent/Session methods
26
+ * 3. Register ALL event handlers on RemoteSession
27
+ * 4. THEN connect — so hydration + replay events are captured
28
+ */
29
+
30
+ import {
31
+ createAgentSession,
32
+ DefaultResourceLoader,
33
+ InteractiveMode,
34
+ SessionManager,
35
+ } from '@mariozechner/pi-coding-agent';
36
+ import { RemoteSession } from './remote-session.ts';
37
+ import type { RpcEvent } from './remote-session.ts';
38
+ import { agentuityCoderHub } from './index.ts';
39
+
40
+ const DEBUG = !!process.env['AGENTUITY_DEBUG'];
41
+
42
+ function log(msg: string): void {
43
+ if (DEBUG) console.error(`[remote-tui] ${msg}`);
44
+ }
45
+
46
+ /**
47
+ * Run the native Pi TUI connected to a remote sandbox session.
48
+ *
49
+ * This is the entry point for `agentuity coder start --remote <sessionId>`.
50
+ * Creates an AgentSession with the coder extension loaded (Hub UI), then
51
+ * monkey-patches the Agent for remote-backed execution.
52
+ */
53
+ export async function runRemoteTui(options: {
54
+ hubWsUrl: string;
55
+ sessionId: string;
56
+ }): Promise<void> {
57
+ const { hubWsUrl, sessionId } = options;
58
+
59
+ log(`Starting remote TUI for session ${sessionId}`);
60
+ log(`Hub URL: ${hubWsUrl}`);
61
+
62
+ // Set env vars BEFORE loading the extension so it enters native remote mode
63
+ process.env.AGENTUITY_CODER_HUB_URL = hubWsUrl;
64
+ process.env.AGENTUITY_CODER_REMOTE_SESSION = sessionId;
65
+ process.env.AGENTUITY_CODER_NATIVE_REMOTE = '1';
66
+
67
+ // ── 1. Create RemoteSession (NOT connected yet) ──
68
+ // We register all handlers BEFORE connecting so that the hydration
69
+ // message from the Hub (sent immediately after init) is captured.
70
+ const remote = new RemoteSession(sessionId);
71
+ // TODO: Remove/Change when we get Agentuity service level auth enabled, this is just temporary
72
+ remote.apiKey = process.env.AGENTUITY_CODER_API_KEY || null;
73
+ let hydrationStreamingDetected = false;
74
+
75
+ // ── 2. Create AgentSession with coder extension loaded ──
76
+ // The extension provides Hub UI (footer, /hub overlay, commands, titlebar).
77
+ // AGENTUITY_CODER_NATIVE_REMOTE=1 tells it to skip legacy event rendering.
78
+ const cwd = process.cwd();
79
+ const resourceLoader = new DefaultResourceLoader({
80
+ cwd,
81
+ noExtensions: true, // Skip file-system extension discovery
82
+ extensionFactories: [agentuityCoderHub], // Load coder extension directly
83
+ });
84
+ await resourceLoader.reload();
85
+
86
+ const { session } = await createAgentSession({
87
+ sessionManager: SessionManager.inMemory(),
88
+ tools: [], // No local tools — sandbox has all the tools
89
+ resourceLoader,
90
+ });
91
+ log('AgentSession created');
92
+
93
+ // NOTE: Do NOT call session.bindExtensions() here.
94
+ // InteractiveMode.initExtensions() calls it with the proper uiContext.
95
+ // Calling it early fires session_start twice, duplicating extension init.
96
+
97
+ // Access the Agent instance (typed as `any` for monkey-patching)
98
+ const agent: any = session.agent;
99
+
100
+ // ── 3. Patch Agent to be remote-backed ──
101
+ // Track the running prompt promise so InteractiveMode waits correctly
102
+ let runningPromptResolve: (() => void) | null = null;
103
+ let syntheticAgentStartEmitted = false;
104
+
105
+ // Override Agent.prompt — send RPC prompt command, block until agent_end
106
+ agent.prompt = async (input: any): Promise<void> => {
107
+ const text = extractPromptText(input);
108
+ log(`agent.prompt called, extracted text: ${text ? text.slice(0, 80) : '(empty)'}`);
109
+ if (!text) return;
110
+
111
+ // Set streaming state — InteractiveMode checks this
112
+ agent._state.isStreaming = true;
113
+ agent._state.streamMessage = null;
114
+ agent._state.error = undefined;
115
+
116
+ // Create runningPrompt so waitForIdle() works
117
+ const runPromise = new Promise<void>((resolve) => {
118
+ runningPromptResolve = resolve;
119
+ });
120
+ agent.runningPrompt = runPromise;
121
+
122
+ // Emit synthetic agent_start so InteractiveMode shows "working" immediately
123
+ syntheticAgentStartEmitted = true;
124
+ agent.emit({ type: 'agent_start', agentName: 'lead', timestamp: Date.now() });
125
+
126
+ // Send RPC command to sandbox
127
+ remote.prompt(text);
128
+ log(`Sent prompt: ${text.slice(0, 100)}`);
129
+
130
+ // Block until agent_end received from remote
131
+ await runPromise;
132
+ };
133
+
134
+ // Override Agent.steer — send RPC steer command
135
+ agent.steer = (m: any) => {
136
+ const text = extractMessageText(m);
137
+ if (text) {
138
+ remote.steer(text);
139
+ log(`Sent steer: ${text.slice(0, 100)}`);
140
+ }
141
+ };
142
+
143
+ // Override Agent.abort — send RPC abort command
144
+ agent.abort = () => {
145
+ remote.abort();
146
+ log('Sent abort');
147
+ resolveRunningPrompt();
148
+ };
149
+
150
+ // Override Agent.waitForIdle
151
+ agent.waitForIdle = () => {
152
+ return agent.runningPrompt ?? Promise.resolve();
153
+ };
154
+
155
+ // ── 4. Patch AgentSession methods ──
156
+ // session.prompt() does model/API key validation before calling agent.prompt().
157
+ // In remote mode, skip validation — the sandbox has the model/key.
158
+ // InteractiveMode calls session.prompt(text, { streamingBehavior: "steer" })
159
+ // when user types during streaming, and session.prompt(text, { streamingBehavior: "followUp" })
160
+ // for Alt+Enter follow-ups. Handle these by routing to steer/followUp commands.
161
+ (session as any).prompt = async (text: string, options?: any) => {
162
+ const behavior = options?.streamingBehavior;
163
+ log(`session.prompt called (behavior=${behavior ?? 'normal'}): ${text.slice(0, 80)}`);
164
+
165
+ // Extension commands (start with /) — let AgentSession handle them
166
+ // so extension-registered slash commands still work in remote mode
167
+ if (text.startsWith('/') && (session as any)._tryExecuteExtensionCommand) {
168
+ try {
169
+ const handled = await (session as any)._tryExecuteExtensionCommand(text);
170
+ if (handled) {
171
+ log(`Extension command handled: ${text}`);
172
+ return;
173
+ }
174
+ } catch (err) {
175
+ log(`Extension command error: ${err}`);
176
+ }
177
+ }
178
+
179
+ if (behavior === 'steer') {
180
+ remote.steer(text);
181
+ log(`Sent steer: ${text.slice(0, 80)}`);
182
+ return;
183
+ }
184
+ if (behavior === 'followUp') {
185
+ remote.sendCommand({ type: 'follow_up', message: text });
186
+ log(`Sent follow-up: ${text.slice(0, 80)}`);
187
+ return;
188
+ }
189
+
190
+ // Normal prompt — send to sandbox
191
+ await agent.prompt(text);
192
+ };
193
+
194
+ (session as any).steer = async (text: string) => {
195
+ remote.steer(text);
196
+ };
197
+
198
+ (session as any).followUp = async (text: string) => {
199
+ remote.sendCommand({ type: 'follow_up', message: text });
200
+ };
201
+
202
+ (session as any).abort = async () => {
203
+ remote.abort();
204
+ resolveRunningPrompt();
205
+ };
206
+
207
+ // Disable auto-compaction (sandbox handles it)
208
+ session.setAutoCompactionEnabled(false);
209
+ session.setAutoRetryEnabled(false);
210
+
211
+ // ── 5. Wire up remote events → Agent event pipeline ──
212
+ // Only emit LIVE events (from broadcast) to Agent → InteractiveMode.
213
+ // Replay events (from Durable Stream) are historical — skip them.
214
+ // Hydration is handled separately via session_hydration → agent.replaceMessages().
215
+ //
216
+ // Events that arrive before InteractiveMode is initialized are buffered
217
+ // and flushed after init (InteractiveMode registers listeners during init,
218
+ // so agent.emit() before that fires into the void).
219
+ let interactiveModeReady = false;
220
+ let eventBuffer: RpcEvent[] = [];
221
+ let seenMessageStart = false;
222
+ let seenAgentStart = false;
223
+
224
+ remote.onEvent((rpcEvent: RpcEvent) => {
225
+ const source = (rpcEvent as any)._source ?? 'unknown';
226
+ log(`Event received: ${rpcEvent.type} (source=${source})`);
227
+
228
+ // session_hydration is handled separately below — skip it here
229
+ if (rpcEvent.type === 'session_hydration') return;
230
+
231
+ // Skip duplicate agent_start if we already emitted a synthetic one
232
+ if (rpcEvent.type === 'agent_start' && syntheticAgentStartEmitted) {
233
+ syntheticAgentStartEmitted = false;
234
+ // Still update state from real event
235
+ updateAgentState(agent, rpcEvent);
236
+ return;
237
+ }
238
+
239
+ // Skip replay events — these are historical from Durable Stream
240
+ if (source === 'replay') {
241
+ log(`Skipping replay event: ${rpcEvent.type}`);
242
+ return;
243
+ }
244
+
245
+ // Track streaming lifecycle events so we can inject synthetics when
246
+ // we attach mid-stream (controller connected after agent already started).
247
+ if (rpcEvent.type === 'agent_start') seenAgentStart = true;
248
+ if (rpcEvent.type === 'agent_end') {
249
+ seenAgentStart = false;
250
+ seenMessageStart = false;
251
+ }
252
+ if (rpcEvent.type === 'message_start') seenMessageStart = true;
253
+ if (rpcEvent.type === 'message_end') seenMessageStart = false;
254
+
255
+ // Update agent internal state (mirrors Agent._runLoop behavior)
256
+ updateAgentState(agent, rpcEvent);
257
+
258
+ // Buffer events until InteractiveMode is ready to receive them
259
+ if (!interactiveModeReady) {
260
+ eventBuffer.push(rpcEvent);
261
+ log(`Buffered event: ${rpcEvent.type} (InteractiveMode not ready)`);
262
+ return;
263
+ }
264
+
265
+ // Mid-stream attach guard: if we receive message_update or message_end
266
+ // without having seen message_start, inject synthetic agent_start +
267
+ // message_start so InteractiveMode sets up its streaming component.
268
+ // Without this, the events are silently dropped because
269
+ // InteractiveMode.streamingComponent is null.
270
+ // This happens when the controller connects mid-stream or after the
271
+ // agent finishes — the broadcast of agent_start/message_start occurred
272
+ // before the controller WebSocket was registered.
273
+ if (
274
+ (rpcEvent.type === 'message_update' || rpcEvent.type === 'message_end') &&
275
+ !seenMessageStart
276
+ ) {
277
+ log(`Live ${rpcEvent.type} without prior message_start — injecting synthetics`);
278
+ if (!seenAgentStart) {
279
+ agent.emit({ type: 'agent_start', agentName: 'lead', timestamp: Date.now() } as any);
280
+ seenAgentStart = true;
281
+ }
282
+ agent.emit({
283
+ type: 'message_start',
284
+ message: {
285
+ role: 'assistant',
286
+ content: [],
287
+ api: 'anthropic-messages',
288
+ provider: 'anthropic',
289
+ model: 'remote',
290
+ usage: {
291
+ input: 0,
292
+ output: 0,
293
+ cacheRead: 0,
294
+ cacheWrite: 0,
295
+ totalTokens: 0,
296
+ cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
297
+ },
298
+ timestamp: Date.now(),
299
+ },
300
+ } as any);
301
+ seenMessageStart = true;
302
+ }
303
+
304
+ // Emit to subscribers — InteractiveMode.handleEvent processes this
305
+ agent.emit(rpcEvent);
306
+
307
+ // Resolve running prompt when agent finishes
308
+ if (rpcEvent.type === 'agent_end') {
309
+ resolveRunningPrompt();
310
+ }
311
+ });
312
+
313
+ // ── 6. Wire up UI handlers for extension dialogs from sandbox ──
314
+ remote.setUiHandler(async (request) => {
315
+ // TODO: Bridge to InteractiveMode's extension UI context
316
+ log(`UI request: ${request.method} (no handler yet)`);
317
+ const fireAndForget = ['notify', 'setStatus', 'setWidget', 'setTitle', 'set_editor_text'];
318
+ if (fireAndForget.includes(request.method)) return undefined;
319
+ return null;
320
+ });
321
+
322
+ // ── 7. Handle hydration (initial state from Hub) ──
323
+ // Hydration arrives as the message AFTER 'init' on the WebSocket.
324
+ // remote.connect() resolves on 'init', so hydration arrives in the next
325
+ // event loop tick. We use a promise to wait for it before creating
326
+ // InteractiveMode (which calls renderInitialMessages from SessionManager).
327
+ const sm = session.sessionManager;
328
+ let resolveHydration: () => void;
329
+ const hydrationReady = new Promise<void>((resolve) => {
330
+ resolveHydration = resolve;
331
+ });
332
+
333
+ remote.onEvent((event: RpcEvent) => {
334
+ if (event.type !== 'session_hydration') return;
335
+
336
+ const entries = (event as any).entries as
337
+ | Array<{
338
+ type: string;
339
+ content?: string | Array<{ type: string; text?: string }>;
340
+ role?: string;
341
+ timestamp?: number;
342
+ }>
343
+ | undefined;
344
+
345
+ // Extract task text from hydration (Hub includes session.sandbox?.task)
346
+ const hydrationTask = (event as any).task as string | undefined;
347
+
348
+ if (!entries?.length) {
349
+ log('Received session_hydration with no entries');
350
+ // Even with no entries, inject task as user message if available
351
+ if (hydrationTask) {
352
+ const taskMsg = {
353
+ role: 'user' as const,
354
+ content: [{ type: 'text' as const, text: hydrationTask }],
355
+ timestamp: Date.now(),
356
+ };
357
+ agent.replaceMessages([taskMsg]);
358
+ try {
359
+ sm.appendMessage(taskMsg as any);
360
+ } catch (err) {
361
+ log(`SM append task error: ${err}`);
362
+ }
363
+ log('Injected task as user message (no entries)');
364
+ }
365
+ resolveHydration!();
366
+ return;
367
+ }
368
+
369
+ log(`Hydrating ${entries.length} entries`);
370
+ const agentMessages: any[] = [];
371
+
372
+ // If we have a task and no user_prompt entry, inject the task as the first user message
373
+ const hasUserEntry = entries.some((e) => e.type === 'user_prompt' || e.role === 'user');
374
+ if (hydrationTask && !hasUserEntry) {
375
+ const taskMsg = {
376
+ role: 'user' as const,
377
+ content: [{ type: 'text' as const, text: hydrationTask }],
378
+ timestamp: Date.now(),
379
+ };
380
+ agentMessages.push(taskMsg);
381
+ try {
382
+ sm.appendMessage(taskMsg as any);
383
+ } catch (err) {
384
+ log(`SM append task error: ${err}`);
385
+ }
386
+ log('Injected task as user message');
387
+ }
388
+
389
+ for (const entry of entries) {
390
+ const text =
391
+ typeof entry.content === 'string'
392
+ ? entry.content
393
+ : Array.isArray(entry.content)
394
+ ? entry.content
395
+ .filter(
396
+ (c): c is { type: string; text: string } =>
397
+ c.type === 'text' && typeof c.text === 'string'
398
+ )
399
+ .map((c) => c.text)
400
+ .join('\n')
401
+ : '';
402
+
403
+ if (!text) continue;
404
+
405
+ // Hub conversation entries use type: 'message' for assistant, 'thinking' for thinking,
406
+ // 'task_result' for delegation results, 'turn' for turn markers, 'user_prompt' for user input.
407
+ // Only 'user_prompt' entries are user messages; everything else is assistant-side.
408
+ const isAssistant =
409
+ entry.role === 'assistant' ||
410
+ entry.type === 'message' ||
411
+ entry.type === 'thinking' ||
412
+ entry.type === 'task_result' ||
413
+ entry.type === 'assistant';
414
+ if (isAssistant) {
415
+ const msg = {
416
+ role: 'assistant' as const,
417
+ content: [{ type: 'text' as const, text }],
418
+ api: 'anthropic-messages',
419
+ provider: 'anthropic',
420
+ model: 'remote',
421
+ usage: {
422
+ input: 0,
423
+ output: 0,
424
+ cacheRead: 0,
425
+ cacheWrite: 0,
426
+ totalTokens: 0,
427
+ cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
428
+ },
429
+ stopReason: 'stop',
430
+ timestamp: entry.timestamp ?? Date.now(),
431
+ };
432
+ agentMessages.push(msg);
433
+ try {
434
+ sm.appendMessage(msg as any);
435
+ } catch (err) {
436
+ log(`SM append error: ${err}`);
437
+ }
438
+ } else {
439
+ const msg = {
440
+ role: 'user' as const,
441
+ content: [{ type: 'text' as const, text }],
442
+ timestamp: entry.timestamp ?? Date.now(),
443
+ };
444
+ agentMessages.push(msg);
445
+ try {
446
+ sm.appendMessage(msg as any);
447
+ } catch (err) {
448
+ log(`SM append error: ${err}`);
449
+ }
450
+ }
451
+ }
452
+
453
+ if (agentMessages.length > 0) {
454
+ agent.replaceMessages(agentMessages);
455
+ log(`Hydrated ${agentMessages.length} agent messages (+ session manager)`);
456
+ } else {
457
+ log('Hydration: 0 messages after filtering (all entries had empty text?)');
458
+ }
459
+
460
+ // Restore streaming state from hydration — fixes first-connect miss
461
+ const streamingState = (event as any).streamingState as
462
+ | {
463
+ isStreaming?: boolean;
464
+ activeTasks?: Array<{ taskId: string; agent: string }>;
465
+ }
466
+ | undefined;
467
+
468
+ if (streamingState?.isStreaming) {
469
+ agent._state.isStreaming = true;
470
+ hydrationStreamingDetected = true;
471
+ // Create runningPrompt so InteractiveMode knows we're busy
472
+ if (!agent.runningPrompt) {
473
+ const runPromise = new Promise<void>((resolve) => {
474
+ runningPromptResolve = resolve;
475
+ });
476
+ agent.runningPrompt = runPromise;
477
+ }
478
+ log(
479
+ `Hydration: session is streaming with ${streamingState.activeTasks?.length ?? 0} active tasks`
480
+ );
481
+ }
482
+
483
+ resolveHydration!();
484
+ });
485
+
486
+ // ── 8. NOW connect to Hub ──
487
+ // All handlers are registered, so hydration + replay events will be captured.
488
+ log('Connecting to Hub (handlers registered)...');
489
+ await remote.connect(hubWsUrl);
490
+ log('Connected to Hub as controller');
491
+
492
+ // Wait for hydration message (arrives right after init), with a timeout
493
+ // in case this is the first connection and there's nothing to hydrate.
494
+ const HYDRATION_TIMEOUT_MS = 2000;
495
+ await Promise.race([
496
+ hydrationReady,
497
+ new Promise<void>((resolve) =>
498
+ setTimeout(() => {
499
+ log('Hydration timeout — no session_hydration received');
500
+ resolve();
501
+ }, HYDRATION_TIMEOUT_MS)
502
+ ),
503
+ ]);
504
+ const smEntries = sm.getEntries?.() ?? [];
505
+ log(`SessionManager has ${smEntries.length} entries after hydration`);
506
+ log(`Post-hydration: SM has ${smEntries.length} entries, leafId=${sm.getLeafId?.() ?? 'N/A'}`);
507
+
508
+ // ── 9. Start InteractiveMode — full native Pi TUI ──
509
+ log('Creating InteractiveMode');
510
+ const interactive = new InteractiveMode(session);
511
+ log('InteractiveMode created, calling init...');
512
+ await interactive.init();
513
+
514
+ // Flush buffered events now that InteractiveMode is listening.
515
+ // If the session was already streaming when we connected (mid-stream attach),
516
+ // InteractiveMode needs agent_start + message_start to set up its streaming
517
+ // components. Without these, message_update events are silently dropped
518
+ // because InteractiveMode.streamingComponent is null.
519
+ interactiveModeReady = true;
520
+
521
+ if (hydrationStreamingDetected) {
522
+ // Immediately emit agent_start + message_start so InteractiveMode shows
523
+ // the streaming indicator right away, before any buffered events flush.
524
+ // This prevents the blank screen gap between connect and first event.
525
+ log('Hydration detected streaming — emitting immediate synthetics');
526
+ agent.emit({ type: 'agent_start', agentName: 'lead', timestamp: Date.now() } as any);
527
+ agent.emit({
528
+ type: 'message_start',
529
+ message: {
530
+ role: 'assistant',
531
+ content: [],
532
+ api: 'anthropic-messages',
533
+ provider: 'anthropic',
534
+ model: 'remote',
535
+ usage: {
536
+ input: 0,
537
+ output: 0,
538
+ cacheRead: 0,
539
+ cacheWrite: 0,
540
+ totalTokens: 0,
541
+ cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
542
+ },
543
+ timestamp: Date.now(),
544
+ },
545
+ } as any);
546
+ seenAgentStart = true;
547
+ seenMessageStart = true;
548
+
549
+ // Remove any agent_start/message_start from buffer since we already emitted them
550
+ eventBuffer = eventBuffer.filter(
551
+ (e) => e.type !== 'agent_start' && e.type !== 'message_start'
552
+ );
553
+ }
554
+
555
+ if (eventBuffer.length > 0) {
556
+ log(`Flushing ${eventBuffer.length} events: ${eventBuffer.map((e) => e.type).join(', ')}`);
557
+ for (const buffered of eventBuffer) {
558
+ agent.emit(buffered);
559
+ if (buffered.type === 'agent_end') {
560
+ resolveRunningPrompt();
561
+ }
562
+ }
563
+ }
564
+ eventBuffer = [];
565
+
566
+ log('InteractiveMode initialized, calling run...');
567
+
568
+ // Handle clean shutdown
569
+ const cleanup = () => {
570
+ remote.close();
571
+ interactive.stop();
572
+ };
573
+ process.on('SIGINT', cleanup);
574
+ process.on('SIGTERM', cleanup);
575
+
576
+ try {
577
+ await interactive.run();
578
+ } catch (err) {
579
+ log(`InteractiveMode.run() threw: ${err instanceof Error ? err.stack : String(err)}`);
580
+ throw err;
581
+ } finally {
582
+ remote.close();
583
+ log('Remote TUI exited');
584
+ }
585
+
586
+ // ── Helper: resolve the running prompt promise ──
587
+ function resolveRunningPrompt(): void {
588
+ syntheticAgentStartEmitted = false;
589
+ agent._state.isStreaming = false;
590
+ agent._state.streamMessage = null;
591
+ agent._state.pendingToolCalls = new Set();
592
+ if (runningPromptResolve) {
593
+ runningPromptResolve();
594
+ runningPromptResolve = null;
595
+ agent.runningPrompt = undefined;
596
+ }
597
+ }
598
+ }
599
+
600
+ // ══════════════════════════════════════════════
601
+ // Agent state synchronization
602
+ // Mirrors Agent._runLoop state updates (agent.js lines 317-352)
603
+ // ══════════════════════════════════════════════
604
+
605
+ function updateAgentState(agent: any, event: RpcEvent): void {
606
+ const state = agent._state;
607
+
608
+ switch (event.type) {
609
+ case 'agent_start':
610
+ state.isStreaming = true;
611
+ break;
612
+
613
+ case 'agent_end':
614
+ state.isStreaming = false;
615
+ state.streamMessage = null;
616
+ state.pendingToolCalls = new Set();
617
+ break;
618
+
619
+ case 'message_start':
620
+ state.streamMessage = (event as any).message;
621
+ state.isStreaming = true;
622
+ break;
623
+
624
+ case 'message_update':
625
+ state.streamMessage = (event as any).message;
626
+ break;
627
+
628
+ case 'message_end':
629
+ state.streamMessage = null;
630
+ state.messages = [...state.messages, (event as any).message];
631
+ break;
632
+
633
+ case 'tool_execution_start': {
634
+ const s = new Set(state.pendingToolCalls);
635
+ s.add((event as any).toolCallId);
636
+ state.pendingToolCalls = s;
637
+ break;
638
+ }
639
+
640
+ case 'tool_execution_end': {
641
+ const s = new Set(state.pendingToolCalls);
642
+ s.delete((event as any).toolCallId);
643
+ state.pendingToolCalls = s;
644
+ break;
645
+ }
646
+
647
+ case 'thinking_start':
648
+ state.isStreaming = true;
649
+ break;
650
+
651
+ case 'thinking_end':
652
+ break;
653
+
654
+ case 'tool_call':
655
+ state.isStreaming = true;
656
+ break;
657
+
658
+ case 'tool_result':
659
+ break;
660
+
661
+ case 'task_start':
662
+ state.isStreaming = true;
663
+ break;
664
+
665
+ case 'task_complete':
666
+ case 'task_error':
667
+ break;
668
+
669
+ case 'turn_end': {
670
+ const msg = (event as any).message;
671
+ if (msg?.role === 'assistant' && msg?.errorMessage) {
672
+ state.error = msg.errorMessage;
673
+ }
674
+ break;
675
+ }
676
+ }
677
+ }
678
+
679
+ // ══════════════════════════════════════════════
680
+ // Text extraction helpers
681
+ // ══════════════════════════════════════════════
682
+
683
+ function extractPromptText(input: any): string {
684
+ if (typeof input === 'string') return input;
685
+
686
+ if (Array.isArray(input)) {
687
+ return input.map(extractMessageText).filter(Boolean).join('\n');
688
+ }
689
+
690
+ return extractMessageText(input);
691
+ }
692
+
693
+ function extractMessageText(msg: any): string {
694
+ if (typeof msg === 'string') return msg;
695
+ if (!msg || typeof msg !== 'object') return '';
696
+
697
+ if (typeof msg.content === 'string') return msg.content;
698
+
699
+ if (Array.isArray(msg.content)) {
700
+ return msg.content
701
+ .filter((c: any) => c?.type === 'text' && typeof c.text === 'string')
702
+ .map((c: any) => c.text)
703
+ .join('\n');
704
+ }
705
+
706
+ return '';
707
+ }