@fitlab-ai/agent-infra 0.11.10 → 0.11.12

package/dist/lib/agent-clients/adapters/codex-lifecycle/store.js

@@ -1,120 +1,101 @@
 import crypto from 'node:crypto';
-import fs from 'node:fs';
 import path from 'node:path';
+import { mutateDelegationAdapterEvidence, mutateDelegationAdapterEvidenceWithinTaskLock, readRun } from '../../../task/orchestration.js';
+import { resolveTaskRef } from '../../../task/resolve-ref.js';
 import { createCodexLifecycleState, expireCodexLifecycleState, reduceCodexLifecycleEvent } from './evidence.js';
-import { resolveAgentRuntimeStoreRoot } from '../../../runtime/agent-runtime.js';
-const LOCK_RETRY_MS = 10;
-const LOCK_TIMEOUT_MS = 1_000;
-const LOCK_STALE_MS = 30_000;
-function digest(value) {
-    return crypto.createHash('sha256').update(value).digest('hex');
+function readCodexLifecycleActivationEvidence(receipt) {
+    const state = storedEvidence(receipt.adapterEvidence?.codex);
+    return state.activationEvidence && typeof state.activationEvidence === 'object'
+        ? state.activationEvidence
+        : null;
 }
-function readRecord(file) {
-    const value = JSON.parse(fs.readFileSync(file, 'utf8'));
-    if (value.schemaVersion !== 1
-        || !Number.isSafeInteger(value.revision)
-        || value.revision < 1
-        || !value.state
-        || value.state.schemaVersion !== 1
-        || (value.spawnObservedAt != null && (typeof value.spawnObservedAt !== 'string'
-            || !Number.isFinite(Date.parse(value.spawnObservedAt))))) {
-        throw new Error(`Codex lifecycle record '${path.basename(file)}' is invalid`);
-    }
-    return Object.freeze({ ...value, spawnObservedAt: value.spawnObservedAt ?? null });
+const NON_PERSISTENT_FAILURES = new Set([
+    'CODEX_EVIDENCE_IDENTITY_MISMATCH',
+    'CODEX_EVIDENCE_PARENT_MISMATCH',
+    'CODEX_EVIDENCE_REPLAY_CONFLICT'
+]);
+function recordKey(sessionId, turnId, toolUseId) {
+    return crypto.createHash('sha256').update(`${sessionId}\0${turnId}\0${toolUseId}`).digest('hex');
 }
-function recordFiles(root) {
-    if (!fs.existsSync(root))
-        return [];
-    return fs.readdirSync(root)
-        .filter((name) => /^[a-f0-9]{64}\.json$/.test(name))
-        .map((name) => path.join(root, name));
+function isRecord(value) {
+    return typeof value === 'object' && value !== null && !Array.isArray(value);
 }
-function hasActiveCodexLifecycleEvidence(root, query) {
-    return recordFiles(root).map(readRecord).some((record) => {
-        const evidence = record.state.startEvidence;
-        return record.consumer === null
-            && ['start-ready', 'observed-terminal', 'stop-ready'].includes(record.state.status)
-            && evidence?.nativeAgent === query.nativeAgent
-            && evidence.hookDefinitionHash === query.hookDefinitionHash;
-    });
+function storedEvidence(value) {
+    if (!isRecord(value) || !isRecord(value.records))
+        return Object.freeze({ records: Object.freeze({}) });
+    return value;
 }
-function writeRecord(file, record, expectedRevision) {
-    const temp = `${file}.${process.pid}.${crypto.randomUUID()}.tmp`;
-    fs.writeFileSync(temp, `${JSON.stringify(record, null, 2)}\n`, {
-        encoding: 'utf8',
-        flag: 'wx',
-        mode: 0o600
-    });
-    try {
-        const actualRevision = fs.existsSync(file) ? readRecord(file).revision : 0;
-        if (actualRevision !== expectedRevision) {
-            throw new Error(`Codex lifecycle revision changed from ${expectedRevision} to ${actualRevision}`);
-        }
-        fs.renameSync(temp, file);
-        fs.chmodSync(file, 0o600);
-    }
-    finally {
-        if (fs.existsSync(temp))
-            fs.unlinkSync(temp);
+function receipts(taskDir) {
+    const run = readRun(taskDir);
+    return run ? Object.freeze([...run.receipts, ...(run.pendingDelegation ? [run.pendingDelegation] : [])]) : [];
+}
+function allRecords(taskDir) {
+    return Object.freeze(receipts(taskDir).flatMap((receipt) => {
+        const state = storedEvidence(receipt.adapterEvidence?.codex);
+        return Object.values(state.records);
+    }));
+}
+function lifecyclePath(taskDir) {
+    return path.join(taskDir, '.runtime', 'orchestration.json');
+}
+function assertCurrentCanApply(current, binding) {
+    if (current.revision > 0 && binding && JSON.stringify(current.taskBinding) !== JSON.stringify(binding)) {
+        throw new Error('Codex lifecycle task binding does not match the stored spawn');
     }
+    if (current.consumer)
+        throw new Error(`Codex lifecycle evidence was already consumed by '${current.consumer}'`);
+}
+function hasActiveCodexLifecycleEvidence(taskDir, query) {
+    return allRecords(taskDir).some((record) => {
+        return record.consumer === null
+            && !['invalid', 'expired'].includes(record.state.status)
+            && record.state.spawn?.hookDefinitionHash === query.hookDefinitionHash
+            && (!query.identity || record.state.spawn?.sessionId === query.identity.sessionId
+                && record.state.spawn.turnId === query.identity.turnId
+                && record.state.spawn.toolUseId === query.identity.toolUseId);
+    });
 }
 function createCodexLifecycleStore(options) {
-    const root = options.root ?? resolveAgentRuntimeStoreRoot({ store: 'lifecycle' });
+    if (!/^TASK-[0-9]{8}-[0-9]{6}$/u.test(options.taskId)) {
+        throw new Error('Codex lifecycle store task id is invalid');
+    }
+    const repoRoot = options.repoRoot ?? process.cwd();
+    const resolved = resolveTaskRef(options.taskId, { repoRoot });
+    if (!resolved.ok)
+        throw new Error(`${resolved.code}: ${resolved.message}`);
+    const taskDir = resolved.taskDir;
     const now = options.now ?? (() => new Date().toISOString());
-    fs.mkdirSync(root, { recursive: true, mode: 0o700 });
-    fs.chmodSync(root, 0o700);
-    function withWriteLock(operation) {
-        const lock = path.join(root, '.write.lock');
-        const deadline = Date.now() + LOCK_TIMEOUT_MS;
-        let descriptor = null;
-        while (descriptor === null) {
-            try {
-                descriptor = fs.openSync(lock, 'wx', 0o600);
-            }
-            catch (error) {
-                if (error.code !== 'EEXIST')
-                    throw error;
-                try {
-                    if (Date.now() - fs.statSync(lock).mtimeMs > LOCK_STALE_MS) {
-                        fs.unlinkSync(lock);
-                        continue;
-                    }
-                }
-                catch (lockError) {
-                    if (lockError.code === 'ENOENT')
-                        continue;
-                    throw lockError;
-                }
-                if (Date.now() >= deadline)
-                    throw new Error('Codex lifecycle store is busy');
-                Atomics.wait(new Int32Array(new SharedArrayBuffer(4)), 0, 0, LOCK_RETRY_MS);
-            }
-        }
-        try {
-            return operation();
-        }
-        finally {
-            fs.closeSync(descriptor);
-            fs.unlinkSync(lock);
+    function currentBinding() {
+        const run = readRun(taskDir);
+        const receipt = run?.pendingDelegation;
+        if (!run || !receipt || receipt.client !== 'codex') {
+            throw new Error('Codex lifecycle evidence has no current task delegation');
         }
+        return Object.freeze({ taskId: receipt.taskId, runId: receipt.runId, receiptId: receipt.id });
     }
     function findByChild(childThreadId) {
-        return recordFiles(root).filter((file) => {
-            const child = readRecord(file).state.child;
-            return child?.childThreadId === childThreadId;
-        });
+        return allRecords(taskDir).filter((record) => record.state.child?.childThreadId === childThreadId);
     }
     function locate(event) {
         if (event.type === 'hook-spawn') {
-            return path.join(root, `${digest(`${event.sessionId}\0${event.turnId}\0${event.toolUseId}`)}.json`);
+            const key = recordKey(event.sessionId, event.turnId, event.toolUseId);
+            const existing = allRecords(taskDir).find((record) => record.spawnObservedAt && record.state.spawn
+                && recordKey(record.state.spawn.sessionId, record.state.spawn.turnId, record.state.spawn.toolUseId) === key);
+            return existing ?? Object.freeze({
+                schemaVersion: 2,
+                revision: 0,
+                taskBinding: event.taskBinding ?? null,
+                state: createCodexLifecycleState(options.cliVersion),
+                consumer: null,
+                consumedAt: null,
+                spawnObservedAt: null,
+                updatedAt: now()
+            });
         }
         if (event.type === 'hook-child') {
-            const matches = recordFiles(root).filter((file) => {
-                const state = readRecord(file).state;
-                return state.spawn?.sessionId === event.parentThreadId
-                    && state.spawn.nativeAgent === event.nativeAgent
-                    && (!state.child || state.child.childThreadId === event.childThreadId);
-            });
+            const matches = allRecords(taskDir).filter((record) => record.state.spawn?.sessionId === event.parentThreadId
+                && record.state.spawn.nativeAgent === event.nativeAgent
+                && (!record.state.child || record.state.child.childThreadId === event.childThreadId));
             if (matches.length !== 1) {
                 throw new Error(matches.length === 0
                     ? 'Codex lifecycle parent session and agent correlation was not found'
@@ -130,150 +111,181 @@ function createCodexLifecycleStore(options) {
         }
         return matches[0];
     }
+    function update(binding, key, mutate, taskLockHeld = false) {
+        let result = null;
+        const updateState = taskLockHeld ? mutateDelegationAdapterEvidenceWithinTaskLock : mutateDelegationAdapterEvidence;
+        updateState(options.taskId, binding.receiptId, 'codex', (value) => {
+            const currentState = storedEvidence(value);
+            const current = currentState.records[key] ?? null;
+            result = mutate(current);
+            return Object.freeze({
+                ...currentState,
+                records: Object.freeze({ ...currentState.records, [key]: result })
+            });
+        }, { repoRoot, now });
+        if (!result)
+            throw new Error('Codex lifecycle record was not updated');
+        return result;
+    }
     function apply(event) {
-        return withWriteLock(() => {
-            const file = locate(event);
-            const observedAt = now();
-            const current = fs.existsSync(file)
-                ? readRecord(file)
-                : Object.freeze({
-                    schemaVersion: 1,
-                    revision: 0,
-                    state: createCodexLifecycleState(options.cliVersion),
-                    consumer: null,
-                    consumedAt: null,
-                    spawnObservedAt: null,
-                    updatedAt: observedAt
-                });
-            if (current.consumer)
-                throw new Error(`Codex lifecycle evidence was already consumed by '${current.consumer}'`);
-            const nextState = reduceCodexLifecycleEvent(current.state, event);
-            const next = Object.freeze({
-                schemaVersion: 1,
-                revision: current.revision + 1,
-                state: nextState,
+        const existing = event.type === 'hook-spawn' ? null : locate(event);
+        const binding = event.type === 'hook-spawn' ? event.taskBinding ?? currentBinding() : existing?.taskBinding;
+        if (!binding)
+            throw new Error('Codex lifecycle event has no task receipt binding');
+        const key = event.type === 'hook-spawn'
+            ? recordKey(event.sessionId, event.turnId, event.toolUseId)
+            : existing?.state.spawn
+                ? recordKey(existing.state.spawn.sessionId, existing.state.spawn.turnId, existing.state.spawn.toolUseId)
+                : '';
+        const observedAt = now();
+        const next = update(binding, key, (current) => {
+            const base = current ?? Object.freeze({
+                schemaVersion: 2,
+                revision: 0,
+                taskBinding: binding,
+                state: createCodexLifecycleState(options.cliVersion),
                 consumer: null,
                 consumedAt: null,
-                spawnObservedAt: event.type === 'hook-spawn' && current.revision === 0
-                    ? observedAt
-                    : current.spawnObservedAt,
+                spawnObservedAt: null,
+                updatedAt: observedAt
+            });
+            assertCurrentCanApply(base, binding);
+            const state = reduceCodexLifecycleEvent(base.state, event);
+            if (state === base.state)
+                return base;
+            if (state.status === 'invalid' && state.error && NON_PERSISTENT_FAILURES.has(state.error.code)) {
+                throw new Error(`${state.error.code}: ${state.error.message}`);
+            }
+            return Object.freeze({
+                ...base,
+                revision: base.revision + 1,
+                state,
+                spawnObservedAt: event.type === 'hook-spawn' && base.revision === 0 ? observedAt : base.spawnObservedAt,
                 updatedAt: observedAt
             });
-            writeRecord(file, next, current.revision);
-            return Object.freeze({ path: file, revision: next.revision, state: next.state });
         });
+        return Object.freeze({ path: lifecyclePath(taskDir), revision: next.revision, state: next.state });
     }
-    function applyToSpawn(identity, event) {
-        return withWriteLock(() => {
-            const file = path.join(root, `${digest(`${identity.sessionId}\0${identity.turnId}\0${identity.toolUseId}`)}.json`);
-            if (!fs.existsSync(file))
-                throw new Error('Codex lifecycle spawn identity was not found');
-            const current = readRecord(file);
-            if (current.consumer)
-                throw new Error(`Codex lifecycle evidence was already consumed by '${current.consumer}'`);
-            if (current.state.spawn?.sessionId !== identity.sessionId
-                || current.state.spawn.turnId !== identity.turnId
-                || current.state.spawn.toolUseId !== identity.toolUseId)
+    function applyToSpawn(identity, event, expectedRevision) {
+        const key = recordKey(identity.sessionId, identity.turnId, identity.toolUseId);
+        const current = allRecords(taskDir).find((record) => record.state.spawn
+            && recordKey(record.state.spawn.sessionId, record.state.spawn.turnId, record.state.spawn.toolUseId) === key);
+        if (!current)
+            throw new Error('Codex lifecycle spawn identity was not found');
+        const next = update(identity.taskBinding, key, (latest) => {
+            if (!latest || JSON.stringify(latest.taskBinding) !== JSON.stringify(identity.taskBinding)) {
+                throw new Error('Codex lifecycle task binding does not match the stored spawn');
+            }
+            if (latest.revision !== expectedRevision) {
+                throw new Error(`Codex lifecycle revision changed from ${expectedRevision} to ${latest.revision}`);
+            }
+            if (latest.consumer)
+                throw new Error(`Codex lifecycle evidence was already consumed by '${latest.consumer}'`);
+            if (latest.state.spawn?.sessionId !== identity.sessionId
+                || latest.state.spawn.turnId !== identity.turnId
+                || latest.state.spawn.toolUseId !== identity.toolUseId
+                || latest.state.spawn.nativeAgent !== event.nativeAgent
+                || event.sessionId !== identity.sessionId
+                || event.parentThreadId !== identity.sessionId) {
                 throw new Error('Codex lifecycle spawn identity does not match the stored event');
-            const nextState = reduceCodexLifecycleEvent(current.state, event);
-            const next = Object.freeze({
-                ...current,
-                revision: current.revision + 1,
-                state: nextState,
-                updatedAt: now()
-            });
-            writeRecord(file, next, current.revision);
-            return Object.freeze({ path: file, revision: next.revision, state: next.state });
+            }
+            const state = reduceCodexLifecycleEvent(latest.state, event);
+            if (state.status === 'invalid' && state.error && NON_PERSISTENT_FAILURES.has(state.error.code)) {
+                throw new Error(`${state.error.code}: ${state.error.message}`);
+            }
+            return state === latest.state ? latest : Object.freeze({ ...latest, revision: latest.revision + 1, state, updatedAt: now() });
         });
+        return Object.freeze({ path: lifecyclePath(taskDir), revision: next.revision, state: next.state });
     }
     function findByParent(parentThreadId) {
-        const matches = recordFiles(root)
-            .map(readRecord)
-            .filter((record) => record.state.startEvidence?.parentThreadId === parentThreadId
+        const matches = allRecords(taskDir).filter((record) => record.state.startEvidence?.parentThreadId === parentThreadId
             && ['start-ready', 'observed-terminal', 'stop-ready'].includes(record.state.status));
         const unconsumed = matches.filter((record) => !record.consumer);
         if (unconsumed.length > 1)
             throw new Error(`Codex lifecycle parent '${parentThreadId}' has ambiguous active children`);
-        if (unconsumed.length === 1)
-            return Object.freeze(unconsumed);
-        return Object.freeze(matches
-            .filter((record) => record.consumer && record.state.status === 'stop-ready')
-            .sort((left, right) => right.updatedAt.localeCompare(left.updatedAt)));
+        return Object.freeze(unconsumed.length === 1 ? unconsumed : matches.filter((record) => record.consumer));
+    }
+    function findByTaskBinding(binding) {
+        return Object.freeze(allRecords(taskDir).filter((record) => record.taskBinding?.runId === binding.runId
+            && record.taskBinding.receiptId === binding.receiptId));
     }
     function read(childThreadId) {
         const matches = findByChild(childThreadId);
-        if (matches.length === 0)
+        if (matches.length !== 1)
             throw new Error(`Codex lifecycle child '${childThreadId}' was not found uniquely`);
-        if (matches.length > 1)
-            throw new Error(`Codex lifecycle child '${childThreadId}' is ambiguous`);
-        return readRecord(matches[0]);
+        return matches[0];
     }
-    function consumeInternal(childThreadId, consumer, expectedHookDefinitionHash) {
+    function consume(childThreadId, consumer, expectedHookDefinitionHash, expectedTaskBinding) {
         if (!consumer.trim())
             throw new Error('Codex lifecycle consumer is required');
-        return withWriteLock(() => {
-            const matches = findByChild(childThreadId);
-            if (matches.length !== 1)
-                throw new Error(`Codex lifecycle child '${childThreadId}' was not found uniquely`);
-            const file = matches[0];
-            const current = readRecord(file);
-            if (current.consumer) {
-                if (current.consumer !== consumer) {
-                    throw new Error(`Codex lifecycle evidence was already consumed by '${current.consumer}'`);
-                }
-                if (expectedHookDefinitionHash
-                    && current.state.startEvidence?.hookDefinitionHash !== expectedHookDefinitionHash)
-                    throw new Error('Codex lifecycle hook definition hash is stale');
-                return current;
+        const current = read(childThreadId);
+        const binding = expectedTaskBinding ?? current.taskBinding;
+        if (!binding)
+            throw new Error('Codex lifecycle task receipt binding is missing');
+        const spawn = current.state.spawn;
+        if (!spawn)
+            throw new Error('Codex lifecycle spawn evidence is missing');
+        const key = recordKey(spawn.sessionId, spawn.turnId, spawn.toolUseId);
+        return update(binding, key, (latest) => {
+            if (!latest || expectedTaskBinding && JSON.stringify(latest.taskBinding) !== JSON.stringify(expectedTaskBinding)) {
+                throw new Error('Codex lifecycle task binding does not match the expected receipt');
+            }
+            if (latest.consumer && latest.consumer !== consumer) {
+                throw new Error(`Codex lifecycle evidence was already consumed by '${latest.consumer}'`);
             }
-            if (current.state.status !== 'stop-ready')
+            if (latest.state.status !== 'stop-ready')
                 throw new Error('Codex lifecycle evidence is not stop-ready');
-            if (expectedHookDefinitionHash
-                && current.state.startEvidence?.hookDefinitionHash !== expectedHookDefinitionHash)
+            if (expectedHookDefinitionHash && latest.state.startEvidence?.hookDefinitionHash !== expectedHookDefinitionHash) {
                 throw new Error('Codex lifecycle hook definition hash is stale');
-            const next = Object.freeze({
-                ...current,
-                revision: current.revision + 1,
-                consumer,
-                consumedAt: now(),
-                updatedAt: now()
-            });
-            writeRecord(file, next, current.revision);
-            return next;
+            }
+            if (latest.consumer)
+                return latest;
+            return Object.freeze({ ...latest, revision: latest.revision + 1, consumer, consumedAt: now(), updatedAt: now() });
         });
     }
-    function consume(childThreadId, consumer, expectedHookDefinitionHash) {
+    function consumeWithinTaskLock(childThreadId, consumer, expectedHookDefinitionHash, expectedTaskBinding) {
         if (!consumer.trim())
             throw new Error('Codex lifecycle consumer is required');
-        return consumeInternal(childThreadId, consumer, expectedHookDefinitionHash);
+        const current = read(childThreadId);
+        const binding = expectedTaskBinding ?? current.taskBinding;
+        if (!binding || !current.state.spawn)
+            throw new Error('Codex lifecycle task receipt binding is missing');
+        const key = recordKey(current.state.spawn.sessionId, current.state.spawn.turnId, current.state.spawn.toolUseId);
+        return update(binding, key, (latest) => {
+            if (!latest || expectedTaskBinding && JSON.stringify(latest.taskBinding) !== JSON.stringify(expectedTaskBinding)) {
+                throw new Error('Codex lifecycle task binding does not match the expected receipt');
+            }
+            if (latest.consumer && latest.consumer !== consumer) {
+                throw new Error(`Codex lifecycle evidence was already consumed by '${latest.consumer}'`);
+            }
+            if (latest.state.status !== 'stop-ready')
+                throw new Error('Codex lifecycle evidence is not stop-ready');
+            if (expectedHookDefinitionHash && latest.state.startEvidence?.hookDefinitionHash !== expectedHookDefinitionHash) {
+                throw new Error('Codex lifecycle hook definition hash is stale');
+            }
+            return latest.consumer ? latest : Object.freeze({ ...latest, revision: latest.revision + 1, consumer, consumedAt: now(), updatedAt: now() });
+        }, true);
     }
     function expireBefore(cutoff) {
-        return withWriteLock(() => {
-            let changed = 0;
-            for (const file of recordFiles(root)) {
-                const current = readRecord(file);
-                if (current.updatedAt >= cutoff)
-                    continue;
-                if (current.consumer || ['invalid', 'expired', 'stop-ready'].includes(current.state.status)) {
-                    fs.unlinkSync(file);
-                    changed += 1;
-                    continue;
-                }
-                const expiredState = expireCodexLifecycleState(current.state);
-                if (expiredState === current.state)
-                    continue;
-                writeRecord(file, Object.freeze({
-                    ...current,
-                    revision: current.revision + 1,
-                    state: expiredState,
-                    updatedAt: now()
-                }), current.revision);
+        let changed = 0;
+        for (const record of allRecords(taskDir)) {
+            if (record.updatedAt >= cutoff || !record.taskBinding?.receiptId || !record.state.spawn)
+                continue;
+            const key = recordKey(record.state.spawn.sessionId, record.state.spawn.turnId, record.state.spawn.toolUseId);
+            update(record.taskBinding, key, (latest) => {
+                if (!latest || latest.updatedAt >= cutoff || latest.consumer
+                    || ['invalid', 'expired', 'stop-ready'].includes(latest.state.status))
+                    return latest ?? record;
+                const state = expireCodexLifecycleState(latest.state);
+                if (state === latest.state)
+                    return latest;
                 changed += 1;
-            }
-            return changed;
-        });
+                return Object.freeze({ ...latest, revision: latest.revision + 1, state, updatedAt: now() });
+            });
+        }
+        return changed;
     }
-    return Object.freeze({ root, apply, applyToSpawn, consume, expireBefore, findByParent, read });
+    return Object.freeze({ taskId: options.taskId, root: path.join(taskDir, '.runtime'), apply, applyToSpawn, consume, consumeWithinTaskLock, expireBefore, findByParent, findByTaskBinding, read });
 }
-export { createCodexLifecycleStore, hasActiveCodexLifecycleEvidence };
+export { createCodexLifecycleStore, hasActiveCodexLifecycleEvidence, readCodexLifecycleActivationEvidence };
 //# sourceMappingURL=store.js.map