@kbediako/codex-orchestrator 0.1.38 → 0.2.1

package/dist/orchestrator/src/cli/control/controlState.js

@@ -1,3 +1,5 @@
+import { createHash } from 'node:crypto';
+const TRANSPORT_STATE_MAX_ENTRIES = 512;
 export class ControlStateStore {
     now;
     state;
@@ -7,18 +9,103 @@ export class ControlStateStore {
             run_id: options.runId,
             control_seq: options.controlSeq ?? 0,
             latest_action: options.latestAction ?? null,
-            feature_toggles: options.featureToggles ?? null
+            feature_toggles: options.featureToggles ?? null,
+            transport_mutation: normalizeTransportMutationState(options.transportMutation)
         };
     }
     updateAction(input) {
+        this.pruneTransportState();
+        const latest = this.state.latest_action ?? null;
+        const normalizedRequestId = input.requestId ?? null;
+        const normalizedIntentId = input.intentId ?? null;
+        if (input.transportContext) {
+            const replayEntry = this.findTransportIdempotencyReplay({
+                action: input.action,
+                transport: input.transportContext.transport,
+                requestId: normalizedRequestId,
+                intentId: normalizedIntentId
+            });
+            if (replayEntry) {
+                return {
+                    snapshot: this.snapshot(),
+                    idempotentReplay: true,
+                    replayEntry
+                };
+            }
+        }
+        const requestReplay = !input.transportContext &&
+            !!normalizedRequestId &&
+            latest?.action === input.action &&
+            latest.request_id === normalizedRequestId;
+        const intentReplay = !input.transportContext &&
+            !!normalizedIntentId &&
+            latest?.action === input.action &&
+            latest.intent_id === normalizedIntentId;
+        if (requestReplay || intentReplay) {
+            return { snapshot: this.snapshot(), idempotentReplay: true, replayEntry: null };
+        }
         this.state.control_seq += 1;
         this.state.latest_action = {
             request_id: input.requestId ?? null,
+            intent_id: input.intentId ?? null,
             requested_by: input.requestedBy,
             requested_at: this.now(),
             action: input.action,
-            reason: input.reason ?? null
+            reason: input.reason ?? null,
+            transport: input.transportContext?.transport ?? null,
+            actor_id: input.transportContext?.actorId ?? null,
+            actor_source: input.transportContext?.actorSource ?? null,
+            transport_principal: input.transportContext?.principal ?? null
         };
+        if (input.transportContext) {
+            this.recordTransportIdempotency({
+                action: input.action,
+                requestId: normalizedRequestId,
+                intentId: normalizedIntentId,
+                transport: input.transportContext.transport,
+                actorId: input.transportContext.actorId,
+                actorSource: input.transportContext.actorSource,
+                principal: input.transportContext.principal,
+                controlSeq: this.state.control_seq,
+                windowMs: input.transportContext.idempotencyWindowMs
+            });
+        }
+        return { snapshot: this.snapshot(), idempotentReplay: false, replayEntry: null };
+    }
+    isTransportNonceConsumed(nonce) {
+        this.pruneTransportState();
+        const digest = hashTransportNonce(nonce);
+        return (this.state.transport_mutation?.consumed_nonces.some((entry) => entry.nonce_sha256 === digest) ?? false);
+    }
+    consumeTransportNonce(input) {
+        this.pruneTransportState();
+        const state = this.ensureTransportMutationState();
+        const digest = hashTransportNonce(input.nonce);
+        if (state.consumed_nonces.some((entry) => entry.nonce_sha256 === digest)) {
+            return;
+        }
+        state.consumed_nonces.push({
+            nonce_sha256: digest,
+            action: input.action,
+            transport: input.transport,
+            request_id: input.requestId ?? null,
+            intent_id: input.intentId ?? null,
+            consumed_at: this.now(),
+            expires_at: input.expiresAt
+        });
+        state.consumed_nonces = trimExpiredOverflowEntries(state.consumed_nonces, TRANSPORT_STATE_MAX_ENTRIES, this.resolveNowMs());
+    }
+    rollbackTransportNonce(nonce) {
+        this.pruneTransportState();
+        const state = this.state.transport_mutation;
+        if (!state) {
+            return;
+        }
+        const digest = hashTransportNonce(nonce);
+        state.consumed_nonces = state.consumed_nonces.filter((entry) => entry.nonce_sha256 !== digest);
+        if (state.consumed_nonces.length === 0 && state.idempotency_index.length === 0) {
+            this.state.transport_mutation = null;
+        }
     }
     updateFeatureToggles(toggles) {
         const current = this.state.feature_toggles ?? {};
@@ -27,6 +114,104 @@ export class ControlStateStore {
     snapshot() {
         return structuredClone(this.state);
     }
+    setLatestAction(input) {
+        this.state.latest_action = {
+            request_id: input.requestId ?? null,
+            intent_id: input.intentId ?? null,
+            requested_by: input.requestedBy,
+            requested_at: this.now(),
+            action: input.action,
+            reason: input.reason ?? null
+        };
+    }
+    ensureTransportMutationState() {
+        if (!this.state.transport_mutation) {
+            this.state.transport_mutation = {
+                consumed_nonces: [],
+                idempotency_index: []
+            };
+        }
+        return this.state.transport_mutation;
+    }
+    pruneTransportState(nowMsInput) {
+        const state = this.state.transport_mutation;
+        if (!state) {
+            return;
+        }
+        const nowMs = this.resolveNowMs(nowMsInput);
+        state.consumed_nonces = state.consumed_nonces.filter((entry) => {
+            const expiresAt = Date.parse(entry.expires_at);
+            return Number.isFinite(expiresAt) && expiresAt > nowMs;
+        });
+        state.idempotency_index = state.idempotency_index.filter((entry) => {
+            const expiresAt = Date.parse(entry.expires_at);
+            return Number.isFinite(expiresAt) && expiresAt > nowMs;
+        });
+    }
+    findTransportIdempotencyReplay(input) {
+        const index = this.state.transport_mutation?.idempotency_index ?? [];
+        for (const entry of index) {
+            if (entry.action !== input.action || entry.transport !== input.transport) {
+                continue;
+            }
+            if (entry.key_type === 'request' && input.requestId && entry.key === input.requestId) {
+                return { ...entry };
+            }
+            if (entry.key_type === 'intent' && input.intentId && entry.key === input.intentId) {
+                return { ...entry };
+            }
+        }
+        return null;
+    }
+    recordTransportIdempotency(input) {
+        const state = this.ensureTransportMutationState();
+        const recordedAt = this.now();
+        const nowMs = Date.parse(recordedAt);
+        const effectiveNowMs = Number.isFinite(nowMs) ? nowMs : Date.now();
+        const expiresAt = new Date(effectiveNowMs + Math.max(1, input.windowMs)).toISOString();
+        if (input.requestId) {
+            state.idempotency_index = state.idempotency_index.filter((entry) => !(entry.key_type === 'request' && entry.key === input.requestId));
+            state.idempotency_index.push({
+                key_type: 'request',
+                key: input.requestId,
+                action: input.action,
+                transport: input.transport,
+                request_id: input.requestId,
+                intent_id: input.intentId,
+                actor_id: input.actorId,
+                actor_source: input.actorSource,
+                transport_principal: input.principal,
+                control_seq: input.controlSeq,
+                recorded_at: recordedAt,
+                expires_at: expiresAt
+            });
+        }
+        if (input.intentId) {
+            state.idempotency_index = state.idempotency_index.filter((entry) => !(entry.key_type === 'intent' && entry.key === input.intentId));
+            state.idempotency_index.push({
+                key_type: 'intent',
+                key: input.intentId,
+                action: input.action,
+                transport: input.transport,
+                request_id: input.requestId,
+                intent_id: input.intentId,
+                actor_id: input.actorId,
+                actor_source: input.actorSource,
+                transport_principal: input.principal,
+                control_seq: input.controlSeq,
+                recorded_at: recordedAt,
+                expires_at: expiresAt
+            });
+        }
+        state.idempotency_index = trimExpiredOverflowEntries(state.idempotency_index, TRANSPORT_STATE_MAX_ENTRIES, effectiveNowMs);
+    }
+    resolveNowMs(nowMsInput) {
+        if (Number.isFinite(nowMsInput) && typeof nowMsInput === 'number') {
+            return nowMsInput;
+        }
+        const fallbackNow = Date.parse(this.now());
+        return Number.isFinite(fallbackNow) ? fallbackNow : Date.now();
+    }
 }
 function mergeObjects(base, update) {
     const merged = { ...base };
@@ -44,3 +229,49 @@ function mergeObjects(base, update) {
     }
     return merged;
 }
+function normalizeTransportMutationState(input) {
+    if (!input) {
+        return null;
+    }
+    const consumed = Array.isArray(input.consumed_nonces)
+        ? input.consumed_nonces
+            .filter((entry) => Boolean(entry && typeof entry === 'object'))
+            .map((entry) => ({ ...entry }))
+        : [];
+    const index = Array.isArray(input.idempotency_index)
+        ? input.idempotency_index
+            .filter((entry) => Boolean(entry && typeof entry === 'object'))
+            .map((entry) => ({ ...entry }))
+        : [];
+    if (consumed.length === 0 && index.length === 0) {
+        return null;
+    }
+    return {
+        consumed_nonces: consumed,
+        idempotency_index: index
+    };
+}
+function hashTransportNonce(value) {
+    return createHash('sha256').update(value).digest('hex');
+}
+function trimExpiredOverflowEntries(entries, maxEntries, nowMs) {
+    if (entries.length <= maxEntries) {
+        return entries;
+    }
+    let overflow = entries.length - maxEntries;
+    const trimmed = entries.filter((entry) => {
+        if (overflow <= 0) {
+            return true;
+        }
+        const expiresAt = Date.parse(entry.expires_at);
+        if (!Number.isFinite(expiresAt) || expiresAt <= nowMs) {
+            overflow -= 1;
+            return false;
+        }
+        return true;
+    });
+    if (trimmed.length <= maxEntries) {
+        return trimmed;
+    }
+    return trimmed.slice(trimmed.length - maxEntries);
+}