@femtomc/mu-server 26.2.68 → 26.2.70

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.
@@ -1,7 +1,9 @@
1
- import { ApprovedCommandBroker, CommandContextResolver, MessagingOperatorRuntime, operatorExtensionPaths, PiMessagingOperatorBackend, } from "@femtomc/mu-agent";
2
- import { ControlPlaneCommandPipeline, ControlPlaneOutbox, ControlPlaneOutboxDispatcher, ControlPlaneRuntime, correlationFromCommandRecord, DiscordControlPlaneAdapter, getControlPlanePaths, SlackControlPlaneAdapter, TelegramControlPlaneAdapter, TelegramControlPlaneAdapterSpec, } from "@femtomc/mu-control-plane";
1
+ import { ControlPlaneCommandPipeline, ControlPlaneOutbox, ControlPlaneRuntime, DiscordControlPlaneAdapter, getControlPlanePaths, SlackControlPlaneAdapter, TelegramControlPlaneAdapterSpec, } from "@femtomc/mu-control-plane";
3
2
  import { DEFAULT_MU_CONFIG } from "./config.js";
4
3
  import { ControlPlaneRunSupervisor, } from "./run_supervisor.js";
4
+ import { buildMessagingOperatorRuntime, createOutboxDrainLoop, } from "./control_plane_bootstrap_helpers.js";
5
+ import { enqueueRunEventOutbox } from "./control_plane_run_outbox.js";
6
+ import { TelegramAdapterGenerationManager } from "./control_plane_telegram_generation.js";
5
7
  function generationTags(generation, component) {
6
8
  return {
7
9
  generation_id: generation.generation_id,
@@ -31,577 +33,6 @@ export function detectAdapters(config) {
31
33
  }
32
34
  return adapters;
33
35
  }
34
- const TELEGRAM_GENERATION_SUPERVISOR_ID = "telegram-adapter";
35
- const TELEGRAM_WARMUP_TIMEOUT_MS = 2_000;
36
- const TELEGRAM_DRAIN_TIMEOUT_MS = 5_000;
37
- function cloneControlPlaneConfig(config) {
38
- return JSON.parse(JSON.stringify(config));
39
- }
40
- function controlPlaneNonTelegramFingerprint(config) {
41
- return JSON.stringify({
42
- adapters: {
43
- slack: config.adapters.slack,
44
- discord: config.adapters.discord,
45
- gmail: config.adapters.gmail,
46
- },
47
- operator: config.operator,
48
- });
49
- }
50
- function telegramAdapterConfigFromControlPlane(config) {
51
- const webhookSecret = config.adapters.telegram.webhook_secret;
52
- if (!webhookSecret) {
53
- return null;
54
- }
55
- return {
56
- webhookSecret,
57
- botToken: config.adapters.telegram.bot_token,
58
- botUsername: config.adapters.telegram.bot_username,
59
- };
60
- }
61
- function applyTelegramAdapterConfig(base, telegram) {
62
- const next = cloneControlPlaneConfig(base);
63
- next.adapters.telegram.webhook_secret = telegram?.webhookSecret ?? null;
64
- next.adapters.telegram.bot_token = telegram?.botToken ?? null;
65
- next.adapters.telegram.bot_username = telegram?.botUsername ?? null;
66
- return next;
67
- }
68
- function cloneTelegramAdapterConfig(config) {
69
- return {
70
- webhookSecret: config.webhookSecret,
71
- botToken: config.botToken,
72
- botUsername: config.botUsername,
73
- };
74
- }
75
- function describeError(err) {
76
- if (err instanceof Error && err.message.trim().length > 0) {
77
- return err.message;
78
- }
79
- return String(err);
80
- }
81
- async function runWithTimeout(opts) {
82
- if (opts.timeoutMs <= 0) {
83
- return await opts.run();
84
- }
85
- return await new Promise((resolve, reject) => {
86
- let settled = false;
87
- const timer = setTimeout(() => {
88
- if (settled) {
89
- return;
90
- }
91
- settled = true;
92
- reject(new Error(opts.timeoutMessage));
93
- }, opts.timeoutMs);
94
- void opts
95
- .run()
96
- .then((value) => {
97
- if (settled) {
98
- return;
99
- }
100
- settled = true;
101
- clearTimeout(timer);
102
- resolve(value);
103
- })
104
- .catch((err) => {
105
- if (settled) {
106
- return;
107
- }
108
- settled = true;
109
- clearTimeout(timer);
110
- reject(err);
111
- });
112
- });
113
- }
114
- class TelegramAdapterGenerationManager {
115
- #pipeline;
116
- #outbox;
117
- #nowMs;
118
- #onOutboxEnqueued;
119
- #signalObserver;
120
- #hooks;
121
- #generationSeq = -1;
122
- #active = null;
123
- #previousConfig = null;
124
- #activeControlPlaneConfig;
125
- constructor(opts) {
126
- this.#pipeline = opts.pipeline;
127
- this.#outbox = opts.outbox;
128
- this.#nowMs = opts.nowMs ?? Date.now;
129
- this.#onOutboxEnqueued = opts.onOutboxEnqueued ?? null;
130
- this.#signalObserver = opts.signalObserver ?? null;
131
- this.#hooks = opts.hooks ?? null;
132
- this.#activeControlPlaneConfig = cloneControlPlaneConfig(opts.initialConfig);
133
- }
134
- #nextGeneration() {
135
- const nextSeq = this.#generationSeq + 1;
136
- return {
137
- generation_id: `${TELEGRAM_GENERATION_SUPERVISOR_ID}-gen-${nextSeq}`,
138
- generation_seq: nextSeq,
139
- };
140
- }
141
- #buildAdapter(config, opts) {
142
- return new TelegramControlPlaneAdapter({
143
- pipeline: this.#pipeline,
144
- outbox: this.#outbox,
145
- webhookSecret: config.webhookSecret,
146
- botUsername: config.botUsername,
147
- deferredIngress: true,
148
- onOutboxEnqueued: this.#onOutboxEnqueued ?? undefined,
149
- signalObserver: this.#signalObserver ?? undefined,
150
- acceptIngress: opts.acceptIngress,
151
- ingressDrainEnabled: opts.ingressDrainEnabled,
152
- nowMs: this.#nowMs,
153
- });
154
- }
155
- async initialize() {
156
- const initial = telegramAdapterConfigFromControlPlane(this.#activeControlPlaneConfig);
157
- if (!initial) {
158
- return;
159
- }
160
- const generation = this.#nextGeneration();
161
- const adapter = this.#buildAdapter(initial, {
162
- acceptIngress: true,
163
- ingressDrainEnabled: true,
164
- });
165
- await adapter.warmup();
166
- const health = await adapter.healthCheck();
167
- if (!health.ok) {
168
- await adapter.stop({ force: true, reason: "startup_health_gate_failed" });
169
- throw new Error(`telegram adapter warmup health failed: ${health.reason}`);
170
- }
171
- this.#active = {
172
- generation,
173
- config: cloneTelegramAdapterConfig(initial),
174
- adapter,
175
- };
176
- this.#generationSeq = generation.generation_seq;
177
- }
178
- hasActiveGeneration() {
179
- return this.#active != null;
180
- }
181
- activeGeneration() {
182
- return this.#active ? { ...this.#active.generation } : null;
183
- }
184
- activeBotToken() {
185
- return this.#active?.config.botToken ?? null;
186
- }
187
- activeAdapter() {
188
- return this.#active?.adapter ?? null;
189
- }
190
- canHandleConfig(nextConfig, reason) {
191
- if (reason === "rollback") {
192
- return true;
193
- }
194
- return (controlPlaneNonTelegramFingerprint(nextConfig) ===
195
- controlPlaneNonTelegramFingerprint(this.#activeControlPlaneConfig));
196
- }
197
- async #rollbackToPrevious(opts) {
198
- if (!opts.previous) {
199
- return { ok: false, error: "rollback_unavailable" };
200
- }
201
- try {
202
- opts.previous.adapter.activateIngress();
203
- this.#active = opts.previous;
204
- this.#previousConfig = cloneTelegramAdapterConfig(opts.failedRecord.config);
205
- await opts.failedRecord.adapter.stop({ force: true, reason: `rollback:${opts.reason}` });
206
- this.#activeControlPlaneConfig = applyTelegramAdapterConfig(this.#activeControlPlaneConfig, opts.previous.config);
207
- return { ok: true };
208
- }
209
- catch (err) {
210
- return { ok: false, error: describeError(err) };
211
- }
212
- }
213
- async reload(opts) {
214
- if (!this.canHandleConfig(opts.config, opts.reason)) {
215
- return {
216
- handled: false,
217
- ok: false,
218
- reason: opts.reason,
219
- route: "/webhooks/telegram",
220
- from_generation: this.#active?.generation ?? null,
221
- to_generation: null,
222
- active_generation: this.#active?.generation ?? null,
223
- warmup: null,
224
- cutover: null,
225
- drain: null,
226
- rollback: {
227
- requested: opts.reason === "rollback",
228
- trigger: null,
229
- attempted: false,
230
- ok: true,
231
- },
232
- };
233
- }
234
- const rollbackRequested = opts.reason === "rollback";
235
- let rollbackTrigger = rollbackRequested ? "manual" : null;
236
- let rollbackAttempted = false;
237
- let rollbackOk = true;
238
- let rollbackError;
239
- const fromGeneration = this.#active?.generation ?? null;
240
- const previousRecord = this.#active;
241
- const warmupTimeoutMs = Math.max(0, Math.trunc(opts.warmupTimeoutMs ?? TELEGRAM_WARMUP_TIMEOUT_MS));
242
- const drainTimeoutMs = Math.max(0, Math.trunc(opts.drainTimeoutMs ?? TELEGRAM_DRAIN_TIMEOUT_MS));
243
- const targetConfig = rollbackRequested
244
- ? this.#previousConfig
245
- : telegramAdapterConfigFromControlPlane(opts.config);
246
- if (rollbackRequested && !targetConfig) {
247
- return {
248
- handled: true,
249
- ok: false,
250
- reason: opts.reason,
251
- route: "/webhooks/telegram",
252
- from_generation: fromGeneration,
253
- to_generation: null,
254
- active_generation: fromGeneration,
255
- warmup: null,
256
- cutover: null,
257
- drain: null,
258
- rollback: {
259
- requested: true,
260
- trigger: "rollback_unavailable",
261
- attempted: false,
262
- ok: false,
263
- error: "rollback_unavailable",
264
- },
265
- error: "rollback_unavailable",
266
- };
267
- }
268
- if (!targetConfig && !previousRecord) {
269
- this.#activeControlPlaneConfig = cloneControlPlaneConfig(opts.config);
270
- return {
271
- handled: true,
272
- ok: true,
273
- reason: opts.reason,
274
- route: "/webhooks/telegram",
275
- from_generation: null,
276
- to_generation: null,
277
- active_generation: null,
278
- warmup: null,
279
- cutover: null,
280
- drain: null,
281
- rollback: {
282
- requested: rollbackRequested,
283
- trigger: rollbackTrigger,
284
- attempted: false,
285
- ok: true,
286
- },
287
- };
288
- }
289
- if (!targetConfig && previousRecord) {
290
- const drainStartedAtMs = Math.trunc(this.#nowMs());
291
- let forcedStop = false;
292
- let drainError;
293
- let drainTimedOut = false;
294
- try {
295
- previousRecord.adapter.beginDrain();
296
- if (this.#hooks?.onDrain) {
297
- await this.#hooks.onDrain({
298
- generation: previousRecord.generation,
299
- reason: opts.reason,
300
- timeout_ms: drainTimeoutMs,
301
- });
302
- }
303
- const drain = await runWithTimeout({
304
- timeoutMs: drainTimeoutMs,
305
- timeoutMessage: "telegram_drain_timeout",
306
- run: async () => await previousRecord.adapter.drain({ timeoutMs: drainTimeoutMs, reason: opts.reason }),
307
- });
308
- drainTimedOut = drain.timed_out;
309
- if (!drain.ok || drain.timed_out) {
310
- forcedStop = true;
311
- await previousRecord.adapter.stop({ force: true, reason: "disable_drain_timeout" });
312
- }
313
- else {
314
- await previousRecord.adapter.stop({ force: false, reason: "disable" });
315
- }
316
- }
317
- catch (err) {
318
- drainError = describeError(err);
319
- forcedStop = true;
320
- drainTimedOut = drainError.includes("timeout");
321
- await previousRecord.adapter.stop({ force: true, reason: "disable_drain_failed" });
322
- }
323
- this.#previousConfig = cloneTelegramAdapterConfig(previousRecord.config);
324
- this.#active = null;
325
- this.#activeControlPlaneConfig = applyTelegramAdapterConfig(this.#activeControlPlaneConfig, null);
326
- return {
327
- handled: true,
328
- ok: drainError == null,
329
- reason: opts.reason,
330
- route: "/webhooks/telegram",
331
- from_generation: fromGeneration,
332
- to_generation: null,
333
- active_generation: null,
334
- warmup: null,
335
- cutover: {
336
- ok: true,
337
- elapsed_ms: 0,
338
- },
339
- drain: {
340
- ok: drainError == null && !drainTimedOut,
341
- elapsed_ms: Math.max(0, Math.trunc(this.#nowMs()) - drainStartedAtMs),
342
- timed_out: drainTimedOut,
343
- forced_stop: forcedStop,
344
- ...(drainError ? { error: drainError } : {}),
345
- },
346
- rollback: {
347
- requested: rollbackRequested,
348
- trigger: rollbackTrigger,
349
- attempted: false,
350
- ok: true,
351
- },
352
- ...(drainError ? { error: drainError } : {}),
353
- };
354
- }
355
- const nextConfig = cloneTelegramAdapterConfig(targetConfig);
356
- const toGeneration = this.#nextGeneration();
357
- const nextAdapter = this.#buildAdapter(nextConfig, {
358
- acceptIngress: false,
359
- ingressDrainEnabled: false,
360
- });
361
- const nextRecord = {
362
- generation: toGeneration,
363
- config: nextConfig,
364
- adapter: nextAdapter,
365
- };
366
- const warmupStartedAtMs = Math.trunc(this.#nowMs());
367
- try {
368
- if (this.#hooks?.onWarmup) {
369
- await this.#hooks.onWarmup({ generation: toGeneration, reason: opts.reason });
370
- }
371
- await runWithTimeout({
372
- timeoutMs: warmupTimeoutMs,
373
- timeoutMessage: "telegram_warmup_timeout",
374
- run: async () => {
375
- await nextAdapter.warmup();
376
- const health = await nextAdapter.healthCheck();
377
- if (!health.ok) {
378
- throw new Error(`telegram_health_gate_failed:${health.reason}`);
379
- }
380
- },
381
- });
382
- }
383
- catch (err) {
384
- const error = describeError(err);
385
- rollbackTrigger = error.includes("health_gate") ? "health_gate_failed" : "warmup_failed";
386
- await nextAdapter.stop({ force: true, reason: "warmup_failed" });
387
- return {
388
- handled: true,
389
- ok: false,
390
- reason: opts.reason,
391
- route: "/webhooks/telegram",
392
- from_generation: fromGeneration,
393
- to_generation: toGeneration,
394
- active_generation: fromGeneration,
395
- warmup: {
396
- ok: false,
397
- elapsed_ms: Math.max(0, Math.trunc(this.#nowMs()) - warmupStartedAtMs),
398
- error,
399
- },
400
- cutover: null,
401
- drain: null,
402
- rollback: {
403
- requested: rollbackRequested,
404
- trigger: rollbackTrigger,
405
- attempted: false,
406
- ok: true,
407
- },
408
- error,
409
- };
410
- }
411
- const cutoverStartedAtMs = Math.trunc(this.#nowMs());
412
- try {
413
- if (this.#hooks?.onCutover) {
414
- await this.#hooks.onCutover({
415
- from_generation: fromGeneration,
416
- to_generation: toGeneration,
417
- reason: opts.reason,
418
- });
419
- }
420
- nextAdapter.activateIngress();
421
- if (previousRecord) {
422
- previousRecord.adapter.beginDrain();
423
- }
424
- this.#active = nextRecord;
425
- this.#generationSeq = toGeneration.generation_seq;
426
- const postCutoverHealth = await nextAdapter.healthCheck();
427
- if (!postCutoverHealth.ok) {
428
- throw new Error(`telegram_post_cutover_health_failed:${postCutoverHealth.reason}`);
429
- }
430
- }
431
- catch (err) {
432
- const error = describeError(err);
433
- rollbackTrigger = error.includes("post_cutover") ? "post_cutover_health_failed" : "cutover_failed";
434
- rollbackAttempted = true;
435
- const rollback = await this.#rollbackToPrevious({
436
- failedRecord: nextRecord,
437
- previous: previousRecord,
438
- reason: opts.reason,
439
- });
440
- rollbackOk = rollback.ok;
441
- rollbackError = rollback.error;
442
- if (!rollback.ok) {
443
- await nextAdapter.stop({ force: true, reason: "rollback_failed" });
444
- this.#active = previousRecord ?? null;
445
- this.#activeControlPlaneConfig = applyTelegramAdapterConfig(this.#activeControlPlaneConfig, previousRecord?.config ?? null);
446
- }
447
- return {
448
- handled: true,
449
- ok: false,
450
- reason: opts.reason,
451
- route: "/webhooks/telegram",
452
- from_generation: fromGeneration,
453
- to_generation: toGeneration,
454
- active_generation: this.#active?.generation ?? fromGeneration,
455
- warmup: {
456
- ok: true,
457
- elapsed_ms: Math.max(0, cutoverStartedAtMs - warmupStartedAtMs),
458
- },
459
- cutover: {
460
- ok: false,
461
- elapsed_ms: Math.max(0, Math.trunc(this.#nowMs()) - cutoverStartedAtMs),
462
- error,
463
- },
464
- drain: null,
465
- rollback: {
466
- requested: rollbackRequested,
467
- trigger: rollbackTrigger,
468
- attempted: rollbackAttempted,
469
- ok: rollbackOk,
470
- ...(rollbackError ? { error: rollbackError } : {}),
471
- },
472
- error,
473
- };
474
- }
475
- let drain = null;
476
- if (previousRecord) {
477
- const drainStartedAtMs = Math.trunc(this.#nowMs());
478
- let forcedStop = false;
479
- let drainTimedOut = false;
480
- let drainError;
481
- try {
482
- if (this.#hooks?.onDrain) {
483
- await this.#hooks.onDrain({
484
- generation: previousRecord.generation,
485
- reason: opts.reason,
486
- timeout_ms: drainTimeoutMs,
487
- });
488
- }
489
- const drained = await runWithTimeout({
490
- timeoutMs: drainTimeoutMs,
491
- timeoutMessage: "telegram_drain_timeout",
492
- run: async () => await previousRecord.adapter.drain({ timeoutMs: drainTimeoutMs, reason: opts.reason }),
493
- });
494
- drainTimedOut = drained.timed_out;
495
- if (!drained.ok || drained.timed_out) {
496
- forcedStop = true;
497
- await previousRecord.adapter.stop({ force: true, reason: "generation_drain_timeout" });
498
- }
499
- else {
500
- await previousRecord.adapter.stop({ force: false, reason: "generation_drained" });
501
- }
502
- }
503
- catch (err) {
504
- drainError = describeError(err);
505
- forcedStop = true;
506
- drainTimedOut = drainError.includes("timeout");
507
- await previousRecord.adapter.stop({ force: true, reason: "generation_drain_failed" });
508
- }
509
- drain = {
510
- ok: drainError == null && !drainTimedOut,
511
- elapsed_ms: Math.max(0, Math.trunc(this.#nowMs()) - drainStartedAtMs),
512
- timed_out: drainTimedOut,
513
- forced_stop: forcedStop,
514
- ...(drainError ? { error: drainError } : {}),
515
- };
516
- }
517
- this.#previousConfig = previousRecord ? cloneTelegramAdapterConfig(previousRecord.config) : this.#previousConfig;
518
- this.#activeControlPlaneConfig = applyTelegramAdapterConfig(this.#activeControlPlaneConfig, nextConfig);
519
- return {
520
- handled: true,
521
- ok: true,
522
- reason: opts.reason,
523
- route: "/webhooks/telegram",
524
- from_generation: fromGeneration,
525
- to_generation: toGeneration,
526
- active_generation: toGeneration,
527
- warmup: {
528
- ok: true,
529
- elapsed_ms: Math.max(0, cutoverStartedAtMs - warmupStartedAtMs),
530
- },
531
- cutover: {
532
- ok: true,
533
- elapsed_ms: Math.max(0, Math.trunc(this.#nowMs()) - cutoverStartedAtMs),
534
- },
535
- drain,
536
- rollback: {
537
- requested: rollbackRequested,
538
- trigger: rollbackTrigger,
539
- attempted: rollbackAttempted,
540
- ok: rollbackOk,
541
- ...(rollbackError ? { error: rollbackError } : {}),
542
- },
543
- };
544
- }
545
- async stop() {
546
- const active = this.#active;
547
- this.#active = null;
548
- if (!active) {
549
- return;
550
- }
551
- await active.adapter.stop({ force: true, reason: "shutdown" });
552
- }
553
- }
554
- function sha256Hex(input) {
555
- const hasher = new Bun.CryptoHasher("sha256");
556
- hasher.update(input);
557
- return hasher.digest("hex");
558
- }
559
- function outboxKindForRunEvent(kind) {
560
- switch (kind) {
561
- case "run_completed":
562
- return "result";
563
- case "run_failed":
564
- return "error";
565
- default:
566
- return "lifecycle";
567
- }
568
- }
569
- async function enqueueRunEventOutbox(opts) {
570
- const command = opts.event.command;
571
- if (!command) {
572
- return null;
573
- }
574
- const baseCorrelation = correlationFromCommandRecord(command);
575
- const correlation = {
576
- ...baseCorrelation,
577
- run_root_id: opts.event.run.root_issue_id ?? baseCorrelation.run_root_id,
578
- };
579
- const envelope = {
580
- v: 1,
581
- ts_ms: opts.nowMs,
582
- channel: command.channel,
583
- channel_tenant_id: command.channel_tenant_id,
584
- channel_conversation_id: command.channel_conversation_id,
585
- request_id: command.request_id,
586
- response_id: `resp-${sha256Hex(`run-event:${opts.event.run.job_id}:${opts.event.seq}:${opts.nowMs}`).slice(0, 20)}`,
587
- kind: outboxKindForRunEvent(opts.event.kind),
588
- body: opts.event.message,
589
- correlation,
590
- metadata: {
591
- async_run: true,
592
- run_event_kind: opts.event.kind,
593
- run_event_seq: opts.event.seq,
594
- run: opts.event.run,
595
- },
596
- };
597
- const decision = await opts.outbox.enqueue({
598
- dedupeKey: `run-event:${opts.event.run.job_id}:${opts.event.seq}`,
599
- envelope,
600
- nowMs: opts.nowMs,
601
- maxAttempts: 6,
602
- });
603
- return decision.record;
604
- }
605
36
  /**
606
37
  * Telegram supports a markdown dialect that uses single markers for emphasis.
607
38
  * Normalize the most common LLM/GitHub-style markers (`**bold**`, `__italic__`, headings)
@@ -664,26 +95,6 @@ async function postTelegramMessage(botToken, payload) {
664
95
  body: JSON.stringify(payload),
665
96
  });
666
97
  }
667
- const OUTBOX_DRAIN_INTERVAL_MS = 500;
668
- function buildMessagingOperatorRuntime(opts) {
669
- if (!opts.config.operator.enabled) {
670
- return null;
671
- }
672
- const backend = opts.backend ??
673
- new PiMessagingOperatorBackend({
674
- provider: opts.config.operator.provider ?? undefined,
675
- model: opts.config.operator.model ?? undefined,
676
- extensionPaths: operatorExtensionPaths,
677
- });
678
- return new MessagingOperatorRuntime({
679
- backend,
680
- broker: new ApprovedCommandBroker({
681
- runTriggersEnabled: opts.config.operator.run_triggers_enabled,
682
- contextResolver: new CommandContextResolver({ allowedRepoRoots: [opts.repoRoot] }),
683
- }),
684
- enabled: true,
685
- });
686
- }
687
98
  export async function bootstrapControlPlane(opts) {
688
99
  const controlPlaneConfig = opts.config ?? DEFAULT_MU_CONFIG.control_plane;
689
100
  const detected = detectAdapters(controlPlaneConfig);
@@ -709,7 +120,7 @@ export async function bootstrapControlPlane(opts) {
709
120
  const runtime = new ControlPlaneRuntime({ repoRoot: opts.repoRoot });
710
121
  let pipeline = null;
711
122
  let runSupervisor = null;
712
- let drainInterval = null;
123
+ let outboxDrainLoop = null;
713
124
  const adapterMap = new Map();
714
125
  try {
715
126
  await runtime.start();
@@ -767,30 +178,9 @@ export async function bootstrapControlPlane(opts) {
767
178
  };
768
179
  }
769
180
  const action = record.target_type;
770
- const hook = action === "reload"
771
- ? opts.sessionMutationHooks?.reload
772
- : opts.sessionMutationHooks?.update;
773
- if (!hook) {
774
- return {
775
- terminalState: "failed",
776
- errorCode: "session_lifecycle_unavailable",
777
- trace: {
778
- cliCommandKind: action,
779
- runRootId: null,
780
- },
781
- mutatingEvents: [
782
- {
783
- eventType: "session.lifecycle.command.failed",
784
- payload: {
785
- action,
786
- reason: "hook_missing",
787
- },
788
- },
789
- ],
790
- };
791
- }
181
+ const executeLifecycleAction = action === "reload" ? opts.sessionLifecycle.reload : opts.sessionLifecycle.update;
792
182
  try {
793
- const lifecycle = await hook();
183
+ const lifecycle = await executeLifecycleAction();
794
184
  if (!lifecycle.ok) {
795
185
  return {
796
186
  terminalState: "failed",
@@ -1064,38 +454,10 @@ export async function bootstrapControlPlane(opts) {
1064
454
  }
1065
455
  return undefined;
1066
456
  };
1067
- const dispatcher = new ControlPlaneOutboxDispatcher({ outbox, deliver });
1068
- let drainingOutbox = false;
1069
- let drainRequested = false;
1070
- const drainOutboxNow = async () => {
1071
- if (drainingOutbox) {
1072
- drainRequested = true;
1073
- return;
1074
- }
1075
- drainingOutbox = true;
1076
- try {
1077
- do {
1078
- drainRequested = false;
1079
- await dispatcher.drainDue();
1080
- } while (drainRequested);
1081
- }
1082
- catch {
1083
- // Swallow errors — the dispatcher handles retries internally.
1084
- }
1085
- finally {
1086
- drainingOutbox = false;
1087
- }
1088
- };
1089
- const scheduleOutboxDrain = () => {
1090
- queueMicrotask(() => {
1091
- void drainOutboxNow();
1092
- });
1093
- };
457
+ const outboxDrain = createOutboxDrainLoop({ outbox, deliver });
458
+ const scheduleOutboxDrain = outboxDrain.scheduleOutboxDrain;
1094
459
  scheduleOutboxDrainRef = scheduleOutboxDrain;
1095
- drainInterval = setInterval(() => {
1096
- scheduleOutboxDrain();
1097
- }, OUTBOX_DRAIN_INTERVAL_MS);
1098
- scheduleOutboxDrain();
460
+ outboxDrainLoop = outboxDrain;
1099
461
  return {
1100
462
  get activeAdapters() {
1101
463
  return [...adapterMap.values()].filter((entry) => entry.isActive()).map((v) => v.info);
@@ -1167,9 +529,9 @@ export async function bootstrapControlPlane(opts) {
1167
529
  return await pipeline.handleTerminalInbound(terminalOpts);
1168
530
  },
1169
531
  async stop() {
1170
- if (drainInterval) {
1171
- clearInterval(drainInterval);
1172
- drainInterval = null;
532
+ if (outboxDrainLoop) {
533
+ outboxDrainLoop.stop();
534
+ outboxDrainLoop = null;
1173
535
  }
1174
536
  for (const { adapter } of adapterMap.values()) {
1175
537
  try {
@@ -1190,9 +552,9 @@ export async function bootstrapControlPlane(opts) {
1190
552
  };
1191
553
  }
1192
554
  catch (err) {
1193
- if (drainInterval) {
1194
- clearInterval(drainInterval);
1195
- drainInterval = null;
555
+ if (outboxDrainLoop) {
556
+ outboxDrainLoop.stop();
557
+ outboxDrainLoop = null;
1196
558
  }
1197
559
  for (const { adapter } of adapterMap.values()) {
1198
560
  try {