@bbigbang/core 0.1.0

package/dist/services/droidMissionBroker.js

@@ -0,0 +1,124 @@
+import { randomUUID } from 'node:crypto';
+export class DroidMissionBroker {
+    pending = new Map();
+    pendingInspect = new Map();
+    active = new Map();
+    nodeRegistry;
+    timeoutMs;
+    constructor(params) {
+        this.nodeRegistry = params.nodeRegistry;
+        this.timeoutMs = params.timeoutMs ?? 15_000;
+    }
+    runMission(nodeId, mission, options = {}) {
+        const missionDir = mission.missionDir;
+        if (!missionDir) {
+            return Promise.reject(new Error('Mission has no missionDir configured.'));
+        }
+        const requestId = options.requestId ?? randomUUID();
+        options.onRequestCreated?.(requestId);
+        return new Promise((resolve, reject) => {
+            const timer = setTimeout(() => {
+                this.pending.delete(requestId);
+                reject(new Error('Droid mission start request timed out.'));
+            }, this.timeoutMs);
+            this.pending.set(requestId, { nodeId, missionId: mission.missionId, interaction: options.interaction === true, resolve, reject, timer });
+            const sent = this.nodeRegistry.send(nodeId, {
+                type: 'mission.run.request',
+                requestId,
+                missionId: mission.missionId,
+                title: mission.title,
+                prompt: options.promptOverride ?? (mission.description || mission.title),
+                continueSession: options.continueSession,
+                interaction: options.interaction,
+                workspaceRoot: mission.workspacePath,
+                missionDir,
+                modelMode: mission.modelMode,
+                orchestratorModel: mission.orchestratorModel,
+                workerModel: mission.workerModel,
+                validatorModel: mission.validatorModel,
+            });
+            if (!sent) {
+                clearTimeout(timer);
+                this.pending.delete(requestId);
+                reject(new Error('Agent node is offline.'));
+            }
+        });
+    }
+    inspectContinuation(nodeId, mission) {
+        const missionDir = mission.missionDir;
+        if (!missionDir) {
+            return Promise.reject(new Error('Mission has no missionDir configured.'));
+        }
+        const requestId = randomUUID();
+        return new Promise((resolve, reject) => {
+            const timer = setTimeout(() => {
+                this.pendingInspect.delete(requestId);
+                reject(new Error('Droid continuation readiness request timed out.'));
+            }, this.timeoutMs);
+            this.pendingInspect.set(requestId, { nodeId, missionId: mission.missionId, resolve, reject, timer });
+            const sent = this.nodeRegistry.send(nodeId, {
+                type: 'mission.continuation.inspect.request',
+                requestId,
+                missionId: mission.missionId,
+                workspaceRoot: mission.workspacePath,
+                missionDir,
+            });
+            if (!sent) {
+                clearTimeout(timer);
+                this.pendingInspect.delete(requestId);
+                reject(new Error('Agent node is offline.'));
+            }
+        });
+    }
+    hasActiveMission(missionId) {
+        for (const pending of this.pending.values()) {
+            if (pending.missionId === missionId)
+                return true;
+        }
+        for (const attempt of this.active.values()) {
+            if (attempt.missionId === missionId)
+                return true;
+        }
+        return false;
+    }
+    handleAccepted(msg, nodeId) {
+        const pending = this.pending.get(msg.requestId);
+        if (!pending || pending.nodeId !== nodeId || pending.missionId !== msg.missionId)
+            return null;
+        this.pending.delete(msg.requestId);
+        clearTimeout(pending.timer);
+        const attempt = { requestId: msg.requestId, nodeId, missionId: msg.missionId, interaction: pending.interaction };
+        this.active.set(msg.requestId, attempt);
+        pending.resolve(msg);
+        return attempt;
+    }
+    verifyEvent(msg, nodeId) {
+        const attempt = this.active.get(msg.requestId);
+        if (!attempt || attempt.nodeId !== nodeId || attempt.missionId !== msg.missionId)
+            return null;
+        return attempt;
+    }
+    handleEnd(msg, nodeId) {
+        const pending = this.pending.get(msg.requestId);
+        if (pending) {
+            this.pending.delete(msg.requestId);
+            clearTimeout(pending.timer);
+            pending.reject(new Error(msg.error ?? 'Droid mission did not start.'));
+            return null;
+        }
+        const attempt = this.active.get(msg.requestId);
+        if (!attempt || attempt.nodeId !== nodeId || attempt.missionId !== msg.missionId)
+            return null;
+        this.active.delete(msg.requestId);
+        return attempt;
+    }
+    handleContinuationInspectResponse(msg, nodeId) {
+        const pending = this.pendingInspect.get(msg.requestId);
+        if (!pending || pending.nodeId !== nodeId || pending.missionId !== msg.missionId)
+            return false;
+        this.pendingInspect.delete(msg.requestId);
+        clearTimeout(pending.timer);
+        pending.resolve(msg);
+        return true;
+    }
+}