@code-yeongyu/senpi 2026.7.31 → 2026.8.1

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-pump.js

@@ -0,0 +1,197 @@
+import { Buffer } from "node:buffer";
+import { createSessionUuid, } from "./session-registry.js";
+import { transitionToIdleSynced, transitionToTurnClaimed, transitionToTurnResultSeen, transitionToTurnSent, transitionToTurnStreaming, transitionToTurnWaiting, } from "./session-registry-state.js";
+export const DEFAULT_PRE_REPLAY_MAX_MESSAGES = 64;
+export const DEFAULT_PRE_REPLAY_MAX_BYTES = 256 * 1024;
+export const SESSION_TURN_ABORT_GRACE_MS = 1_000;
+export class ConcurrentSessionTurnAdmissionError extends Error {
+    constructor(sessionId) {
+        super(`Concurrent Claude SDK OAuth turn admission for session ${sessionId}`);
+        this.code = "claude_sdk_oauth_concurrent_turn_admission";
+        this.name = "ConcurrentSessionTurnAdmissionError";
+    }
+}
+export class SessionTurnAttributionError extends Error {
+    constructor(message) {
+        super(message);
+        this.code = "claude_sdk_oauth_turn_attribution";
+        this.name = "SessionTurnAttributionError";
+    }
+}
+function currentTurn(entry) {
+    return entry.activeTurn;
+}
+function isReplayFor(message, uuid) {
+    return message.type === "user" && "isReplay" in message && message.isReplay === true && message.uuid === uuid;
+}
+function isAutonomousResult(message) {
+    if (message.origin && message.origin.kind !== "human")
+        return true;
+    const wire = message;
+    return wire.parent_tool_use_id != null || wire.subagent_type != null || wire.isSynthetic === true;
+}
+function resultMatchesTurn(message, turn) {
+    if ("user_message_uuid" in message && message.user_message_uuid !== undefined) {
+        return message.user_message_uuid === turn.uuid;
+    }
+    return turn.claimed && !isAutonomousResult(message);
+}
+function deliver(entry, turn, message) {
+    if (entry.state === "TURN_CLAIMED")
+        transitionToTurnStreaming(entry);
+    turn.messages.push(message);
+    turn.onMessage?.(message);
+}
+function scheduleAbort(callback, delayMs) {
+    const timer = setTimeout(callback, delayMs);
+    timer.unref();
+    return () => clearTimeout(timer);
+}
+function removeAbortListener(turn) {
+    turn.signal?.removeEventListener("abort", turn.onAbort);
+    turn.cancelAbort?.();
+    turn.cancelAbort = undefined;
+}
+function failTurn(registry, entry, error) {
+    const turn = currentTurn(entry);
+    if (turn) {
+        removeAbortListener(turn);
+        entry.activeTurn = null;
+        turn.reject(error);
+    }
+    if (registry.isCurrentGeneration(entry.senpiSessionId, entry.generation)) {
+        registry.closeSession(entry.senpiSessionId, error.message);
+    }
+}
+function abortTurn(registry, entry, turn, error) {
+    if (currentTurn(entry) === turn && registry.isCurrentGeneration(entry.senpiSessionId, turn.generation)) {
+        failTurn(registry, entry, error);
+    }
+}
+function bufferBeforeReplay(registry, entry, turn, message) {
+    turn.preReplay.push(message);
+    turn.preReplayBytes += Buffer.byteLength(JSON.stringify(message));
+    if (turn.preReplay.length > turn.limits.maxMessages || turn.preReplayBytes > turn.limits.maxBytes) {
+        throw new SessionTurnAttributionError("Claude SDK OAuth pre-replay buffer overflow");
+    }
+    if (!registry.isCurrentGeneration(entry.senpiSessionId, turn.generation))
+        turn.preReplay.length = 0;
+}
+function claimTurn(entry, turn) {
+    turn.claimed = true;
+    transitionToTurnClaimed(entry);
+    for (const buffered of turn.preReplay)
+        deliver(entry, turn, buffered);
+    turn.preReplay.length = 0;
+    turn.preReplayBytes = 0;
+}
+function finishTurn(registry, entry, turn, message) {
+    if (!resultMatchesTurn(message, turn)) {
+        throw new SessionTurnAttributionError("Claude SDK OAuth result user_message_uuid did not match the active turn");
+    }
+    if (entry.state === "TURN_CLAIMED")
+        transitionToTurnStreaming(entry);
+    if (!turn.aborted)
+        deliver(entry, turn, message);
+    transitionToTurnResultSeen(entry);
+    removeAbortListener(turn);
+    entry.activeTurn = null;
+    if (turn.aborted)
+        registry.markTainted(entry.senpiSessionId, "abort");
+    else
+        transitionToIdleSynced(entry);
+    turn.resolve({ uuid: turn.uuid, messages: turn.messages, aborted: turn.aborted });
+}
+function handleMessage(registry, entry, message) {
+    const turn = currentTurn(entry);
+    if (!turn || !registry.isCurrentGeneration(entry.senpiSessionId, turn.generation))
+        return;
+    if (!turn.claimed) {
+        if (isReplayFor(message, turn.uuid))
+            claimTurn(entry, turn);
+        else if (message.type === "stream_event")
+            bufferBeforeReplay(registry, entry, turn, message);
+        else if (message.type === "result") {
+            throw new SessionTurnAttributionError("Claude SDK OAuth result arrived before replay claim");
+        }
+        return;
+    }
+    if (message.type === "user" && "isReplay" in message && message.isReplay === true)
+        return;
+    if (message.type === "result")
+        finishTurn(registry, entry, turn, message);
+    else
+        deliver(entry, turn, message);
+}
+async function runPump(registry, entry) {
+    const iterator = entry.query[Symbol.asyncIterator]();
+    try {
+        while (true) {
+            const { value, done } = await iterator.next();
+            if (done) {
+                failTurn(registry, entry, new Error("Claude SDK OAuth query ended before the active turn completed"));
+                return;
+            }
+            handleMessage(registry, entry, value);
+        }
+    }
+    catch (error) {
+        failTurn(registry, entry, error instanceof Error ? error : new Error(String(error)));
+    }
+}
+export function submitSessionTurn(registry, entry, request, limits = {
+    maxMessages: DEFAULT_PRE_REPLAY_MAX_MESSAGES,
+    maxBytes: DEFAULT_PRE_REPLAY_MAX_BYTES,
+}) {
+    if (currentTurn(entry))
+        throw new ConcurrentSessionTurnAdmissionError(entry.senpiSessionId);
+    if (entry.state === "STARTING")
+        transitionToIdleSynced(entry);
+    transitionToTurnWaiting(entry);
+    const uuid = createSessionUuid();
+    let turn;
+    const promise = new Promise((resolve, reject) => {
+        const onAbort = () => {
+            if (turn.aborted)
+                return;
+            turn.aborted = true;
+            turn.cancelAbort = (request.scheduleAbort ?? scheduleAbort)(() => abortTurn(registry, entry, turn, new Error("Claude SDK OAuth interrupted turn did not terminate")), SESSION_TURN_ABORT_GRACE_MS);
+            void entry.query.interrupt().catch((error) => {
+                const detail = error instanceof Error ? error.message : String(error);
+                abortTurn(registry, entry, turn, new Error(`Claude SDK OAuth query interrupt failed: ${detail}`));
+            });
+        };
+        turn = {
+            uuid,
+            generation: entry.generation,
+            messages: [],
+            preReplay: [],
+            preReplayBytes: 0,
+            claimed: false,
+            aborted: false,
+            onMessage: request.onMessage,
+            signal: request.signal,
+            onAbort,
+            resolve,
+            reject,
+            limits,
+        };
+    });
+    entry.activeTurn = turn;
+    if (!entry.pumpTask)
+        entry.pumpTask = runPump(registry, entry);
+    entry.inputController.push({
+        type: "user",
+        message: request.message,
+        parent_tool_use_id: null,
+        uuid,
+        session_id: entry.sdkSessionId,
+    });
+    transitionToTurnSent(entry);
+    if (request.signal?.aborted)
+        turn.onAbort();
+    else
+        request.signal?.addEventListener("abort", turn.onAbort, { once: true });
+    return promise;
+}
+//# sourceMappingURL=session-registry-pump.js.map

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-pump.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-registry-pump.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/claude-sdk-oauth/session-registry-pump.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,OAAO,EAGN,iBAAiB,GACjB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACN,sBAAsB,EACtB,uBAAuB,EACvB,0BAA0B,EAC1B,oBAAoB,EACpB,yBAAyB,EACzB,uBAAuB,GACvB,MAAM,6BAA6B,CAAC;AAErC,MAAM,CAAC,MAAM,+BAA+B,GAAG,EAAE,CAAC;AAClD,MAAM,CAAC,MAAM,4BAA4B,GAAG,GAAG,GAAG,IAAI,CAAC;AACvD,MAAM,CAAC,MAAM,2BAA2B,GAAG,KAAK,CAAC;AAqCjD,MAAM,OAAO,mCAAoC,SAAQ,KAAK;IAG7D,YAAY,SAAiB;QAC5B,KAAK,CAAC,0DAA0D,SAAS,EAAE,CAAC,CAAC;QAHrE,SAAI,GAAG,4CAA4C,CAAC;QAI5D,IAAI,CAAC,IAAI,GAAG,qCAAqC,CAAC;IACnD,CAAC;CACD;AAED,MAAM,OAAO,2BAA4B,SAAQ,KAAK;IAGrD,YAAY,OAAe;QAC1B,KAAK,CAAC,OAAO,CAAC,CAAC;QAHP,SAAI,GAAG,mCAAmC,CAAC;QAInD,IAAI,CAAC,IAAI,GAAG,6BAA6B,CAAC;IAC3C,CAAC;CACD;AAED,SAAS,WAAW,CAAC,KAAiC;IACrD,OAAO,KAAK,CAAC,UAA+B,CAAC;AAC9C,CAAC;AAED,SAAS,WAAW,CAAC,OAAmB,EAAE,IAAY;IACrD,OAAO,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,UAAU,IAAI,OAAO,IAAI,OAAO,CAAC,QAAQ,KAAK,IAAI,IAAI,OAAO,CAAC,IAAI,KAAK,IAAI,CAAC;AAC/G,CAAC;AAED,SAAS,kBAAkB,CAAC,OAAgD;IAC3E,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,IAAI,CAAC;IACnE,MAAM,IAAI,GAAG,OAIZ,CAAC;IACF,OAAO,IAAI,CAAC,kBAAkB,IAAI,IAAI,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,IAAI,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC;AACnG,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAgD,EAAE,IAAgB;IAC5F,IAAI,mBAAmB,IAAI,OAAO,IAAI,OAAO,CAAC,iBAAiB,KAAK,SAAS,EAAE,CAAC;QAC/E,OAAO,OAAO,CAAC,iBAAiB,KAAK,IAAI,CAAC,IAAI,CAAC;IAChD,CAAC;IACD,OAAO,IAAI,CAAC,OAAO,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;AACrD,CAAC;AAED,SAAS,OAAO,CAAC,KAAiC,EAAE,IAAgB,EAAE,OAAmB;IACxF,IAAI,KAAK,CAAC,KAAK,KAAK,cAAc;QAAE,yBAAyB,CAAC,KAAK,CAAC,CAAC;IACrE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5B,IAAI,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,aAAa,CAAC,QAAoB,EAAE,OAAe;IAC3D,MAAM,KAAK,GAAG,UAAU,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC5C,KAAK,CAAC,KAAK,EAAE,CAAC;IACd,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;AAClC,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAgB;IAC5C,IAAI,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACxD,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;IACrB,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;AAC9B,CAAC;AAED,SAAS,QAAQ,CAAC,QAAuC,EAAE,KAAiC,EAAE,KAAY;IACzG,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;IAChC,IAAI,IAAI,EAAE,CAAC;QACV,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAC1B,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACpB,CAAC;IACD,IAAI,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;QAC1E,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;AACF,CAAC;AAED,SAAS,SAAS,CACjB,QAAuC,EACvC,KAAiC,EACjC,IAAgB,EAChB,KAAY;IAEZ,IAAI,WAAW,CAAC,KAAK,CAAC,KAAK,IAAI,IAAI,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,cAAc,EAAE,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;QACxG,QAAQ,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IAClC,CAAC;AACF,CAAC;AAED,SAAS,kBAAkB,CAC1B,QAAuC,EACvC,KAAiC,EACjC,IAAgB,EAChB,OAAmB;IAEnB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC7B,IAAI,CAAC,cAAc,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;IAClE,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACnG,MAAM,IAAI,2BAA2B,CAAC,6CAA6C,CAAC,CAAC;IACtF,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,cAAc,EAAE,IAAI,CAAC,UAAU,CAAC;QAAE,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;AACrG,CAAC;AAED,SAAS,SAAS,CAAC,KAAiC,EAAE,IAAgB;IACrE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;IACpB,uBAAuB,CAAC,KAAK,CAAC,CAAC;IAC/B,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS;QAAE,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;IACtE,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;IAC1B,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;AACzB,CAAC;AAED,SAAS,UAAU,CAClB,QAAuC,EACvC,KAAiC,EACjC,IAAgB,EAChB,OAAgD;IAEhD,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC;QACvC,MAAM,IAAI,2BAA2B,CAAC,yEAAyE,CAAC,CAAC;IAClH,CAAC;IACD,IAAI,KAAK,CAAC,KAAK,KAAK,cAAc;QAAE,yBAAyB,CAAC,KAAK,CAAC,CAAC;IACrE,IAAI,CAAC,IAAI,CAAC,OAAO;QAAE,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACjD,0BAA0B,CAAC,KAAK,CAAC,CAAC;IAClC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAC1B,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC;IACxB,IAAI,IAAI,CAAC,OAAO;QAAE,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;;QACjE,sBAAsB,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;AACnF,CAAC;AAED,SAAS,aAAa,CACrB,QAAuC,EACvC,KAAiC,EACjC,OAAmB;IAEnB,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;IAChC,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,cAAc,EAAE,IAAI,CAAC,UAAU,CAAC;QAAE,OAAO;IAC1F,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACnB,IAAI,WAAW,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC;YAAE,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;aACvD,IAAI,OAAO,CAAC,IAAI,KAAK,cAAc;YAAE,kBAAkB,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;aACxF,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpC,MAAM,IAAI,2BAA2B,CAAC,qDAAqD,CAAC,CAAC;QAC9F,CAAC;QACD,OAAO;IACR,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,UAAU,IAAI,OAAO,IAAI,OAAO,CAAC,QAAQ,KAAK,IAAI;QAAE,OAAO;IAC1F,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ;QAAE,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;;QACrE,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AACpC,CAAC;AAED,KAAK,UAAU,OAAO,CAAC,QAAuC,EAAE,KAAiC;IAChG,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;IACrD,IAAI,CAAC;QACJ,OAAO,IAAI,EAAE,CAAC;YACb,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAC9C,IAAI,IAAI,EAAE,CAAC;gBACV,QAAQ,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC,CAAC;gBACtG,OAAO;YACR,CAAC;YACD,aAAa,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QACvC,CAAC;IACF,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,QAAQ,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACtF,CAAC;AACF,CAAC;AAED,MAAM,UAAU,iBAAiB,CAChC,QAAuC,EACvC,KAAiC,EACjC,OAA2B,EAC3B,MAAM,GAA0B;IAC/B,WAAW,EAAE,+BAA+B;IAC5C,QAAQ,EAAE,4BAA4B;CACtC;IAED,IAAI,WAAW,CAAC,KAAK,CAAC;QAAE,MAAM,IAAI,mCAAmC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAC5F,IAAI,KAAK,CAAC,KAAK,KAAK,UAAU;QAAE,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAC9D,uBAAuB,CAAC,KAAK,CAAC,CAAC;IAC/B,MAAM,IAAI,GAAG,iBAAiB,EAAE,CAAC;IACjC,IAAI,IAAiB,CAAC;IACtB,MAAM,OAAO,GAAG,IAAI,OAAO,CAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAClE,MAAM,OAAO,GAAG,GAAS,EAAE;YAC1B,IAAI,IAAI,CAAC,OAAO;gBAAE,OAAO;YACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,CAAC,WAAW,GAAG,CAAC,OAAO,CAAC,aAAa,IAAI,aAAa,CAAC,CAC1D,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC,EACxG,2BAA2B,CAC3B,CAAC;YACF,KAAK,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBACrD,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACtE,SAAS,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,KAAK,CAAC,4CAA4C,MAAM,EAAE,CAAC,CAAC,CAAC;YACnG,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC;QACF,IAAI,GAAG;YACN,IAAI;YACJ,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,QAAQ,EAAE,EAAE;YACZ,SAAS,EAAE,EAAE;YACb,cAAc,EAAE,CAAC;YACjB,OAAO,EAAE,KAAK;YACd,OAAO,EAAE,KAAK;YACd,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,OAAO;YACP,OAAO;YACP,MAAM;YACN,MAAM;SACN,CAAC;IACH,CAAC,CAAC,CAAC;IACH,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC;IACxB,IAAI,CAAC,KAAK,CAAC,QAAQ;QAAE,KAAK,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC/D,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC;QAC1B,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,kBAAkB,EAAE,IAAI;QACxB,IAAI;QACJ,UAAU,EAAE,KAAK,CAAC,YAAY;KAC9B,CAAC,CAAC;IACH,oBAAoB,CAAC,KAAK,CAAC,CAAC;IAC5B,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO;QAAE,IAAI,CAAC,OAAO,EAAE,CAAC;;QACvC,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7E,OAAO,OAAO,CAAC;AAChB,CAAC","sourcesContent":["import { Buffer } from \"node:buffer\";\nimport type { SDKMessage, SDKUserMessage } from \"./sdk-boundary.ts\";\nimport {\n\ttype ClaudeSdkOauthSessionEntry,\n\ttype ClaudeSdkOauthSessionRegistry,\n\tcreateSessionUuid,\n} from \"./session-registry.ts\";\nimport {\n\ttransitionToIdleSynced,\n\ttransitionToTurnClaimed,\n\ttransitionToTurnResultSeen,\n\ttransitionToTurnSent,\n\ttransitionToTurnStreaming,\n\ttransitionToTurnWaiting,\n} from \"./session-registry-state.ts\";\n\nexport const DEFAULT_PRE_REPLAY_MAX_MESSAGES = 64;\nexport const DEFAULT_PRE_REPLAY_MAX_BYTES = 256 * 1024;\nexport const SESSION_TURN_ABORT_GRACE_MS = 1_000;\n\nexport interface SessionTurnRequest {\n\tmessage: SDKUserMessage[\"message\"];\n\tsignal?: AbortSignal;\n\tonMessage?: (message: SDKMessage) => void;\n\tscheduleAbort?: (callback: () => void, delayMs: number) => () => void;\n}\n\nexport interface SessionTurnResult {\n\tuuid: string;\n\tmessages: SDKMessage[];\n\taborted: boolean;\n}\n\nexport interface PreReplayBufferLimits {\n\tmaxMessages: number;\n\tmaxBytes: number;\n}\n\ninterface ActiveTurn {\n\tuuid: string;\n\tgeneration: number;\n\tmessages: SDKMessage[];\n\tpreReplay: SDKMessage[];\n\tpreReplayBytes: number;\n\tclaimed: boolean;\n\taborted: boolean;\n\tonMessage?: (message: SDKMessage) => void;\n\tsignal?: AbortSignal;\n\tonAbort: () => void;\n\tcancelAbort?: () => void;\n\tresolve: (result: SessionTurnResult) => void;\n\treject: (error: Error) => void;\n\tlimits: PreReplayBufferLimits;\n}\n\nexport class ConcurrentSessionTurnAdmissionError extends Error {\n\treadonly code = \"claude_sdk_oauth_concurrent_turn_admission\";\n\n\tconstructor(sessionId: string) {\n\t\tsuper(`Concurrent Claude SDK OAuth turn admission for session ${sessionId}`);\n\t\tthis.name = \"ConcurrentSessionTurnAdmissionError\";\n\t}\n}\n\nexport class SessionTurnAttributionError extends Error {\n\treadonly code = \"claude_sdk_oauth_turn_attribution\";\n\n\tconstructor(message: string) {\n\t\tsuper(message);\n\t\tthis.name = \"SessionTurnAttributionError\";\n\t}\n}\n\nfunction currentTurn(entry: ClaudeSdkOauthSessionEntry): ActiveTurn | null {\n\treturn entry.activeTurn as ActiveTurn | null;\n}\n\nfunction isReplayFor(message: SDKMessage, uuid: string): boolean {\n\treturn message.type === \"user\" && \"isReplay\" in message && message.isReplay === true && message.uuid === uuid;\n}\n\nfunction isAutonomousResult(message: Extract<SDKMessage, { type: \"result\" }>): boolean {\n\tif (message.origin && message.origin.kind !== \"human\") return true;\n\tconst wire = message as SDKMessage & {\n\t\tparent_tool_use_id?: unknown;\n\t\tsubagent_type?: unknown;\n\t\tisSynthetic?: unknown;\n\t};\n\treturn wire.parent_tool_use_id != null || wire.subagent_type != null || wire.isSynthetic === true;\n}\n\nfunction resultMatchesTurn(message: Extract<SDKMessage, { type: \"result\" }>, turn: ActiveTurn): boolean {\n\tif (\"user_message_uuid\" in message && message.user_message_uuid !== undefined) {\n\t\treturn message.user_message_uuid === turn.uuid;\n\t}\n\treturn turn.claimed && !isAutonomousResult(message);\n}\n\nfunction deliver(entry: ClaudeSdkOauthSessionEntry, turn: ActiveTurn, message: SDKMessage): void {\n\tif (entry.state === \"TURN_CLAIMED\") transitionToTurnStreaming(entry);\n\tturn.messages.push(message);\n\tturn.onMessage?.(message);\n}\n\nfunction scheduleAbort(callback: () => void, delayMs: number): () => void {\n\tconst timer = setTimeout(callback, delayMs);\n\ttimer.unref();\n\treturn () => clearTimeout(timer);\n}\n\nfunction removeAbortListener(turn: ActiveTurn): void {\n\tturn.signal?.removeEventListener(\"abort\", turn.onAbort);\n\tturn.cancelAbort?.();\n\tturn.cancelAbort = undefined;\n}\n\nfunction failTurn(registry: ClaudeSdkOauthSessionRegistry, entry: ClaudeSdkOauthSessionEntry, error: Error): void {\n\tconst turn = currentTurn(entry);\n\tif (turn) {\n\t\tremoveAbortListener(turn);\n\t\tentry.activeTurn = null;\n\t\tturn.reject(error);\n\t}\n\tif (registry.isCurrentGeneration(entry.senpiSessionId, entry.generation)) {\n\t\tregistry.closeSession(entry.senpiSessionId, error.message);\n\t}\n}\n\nfunction abortTurn(\n\tregistry: ClaudeSdkOauthSessionRegistry,\n\tentry: ClaudeSdkOauthSessionEntry,\n\tturn: ActiveTurn,\n\terror: Error,\n): void {\n\tif (currentTurn(entry) === turn && registry.isCurrentGeneration(entry.senpiSessionId, turn.generation)) {\n\t\tfailTurn(registry, entry, error);\n\t}\n}\n\nfunction bufferBeforeReplay(\n\tregistry: ClaudeSdkOauthSessionRegistry,\n\tentry: ClaudeSdkOauthSessionEntry,\n\tturn: ActiveTurn,\n\tmessage: SDKMessage,\n): void {\n\tturn.preReplay.push(message);\n\tturn.preReplayBytes += Buffer.byteLength(JSON.stringify(message));\n\tif (turn.preReplay.length > turn.limits.maxMessages || turn.preReplayBytes > turn.limits.maxBytes) {\n\t\tthrow new SessionTurnAttributionError(\"Claude SDK OAuth pre-replay buffer overflow\");\n\t}\n\tif (!registry.isCurrentGeneration(entry.senpiSessionId, turn.generation)) turn.preReplay.length = 0;\n}\n\nfunction claimTurn(entry: ClaudeSdkOauthSessionEntry, turn: ActiveTurn): void {\n\tturn.claimed = true;\n\ttransitionToTurnClaimed(entry);\n\tfor (const buffered of turn.preReplay) deliver(entry, turn, buffered);\n\tturn.preReplay.length = 0;\n\tturn.preReplayBytes = 0;\n}\n\nfunction finishTurn(\n\tregistry: ClaudeSdkOauthSessionRegistry,\n\tentry: ClaudeSdkOauthSessionEntry,\n\tturn: ActiveTurn,\n\tmessage: Extract<SDKMessage, { type: \"result\" }>,\n): void {\n\tif (!resultMatchesTurn(message, turn)) {\n\t\tthrow new SessionTurnAttributionError(\"Claude SDK OAuth result user_message_uuid did not match the active turn\");\n\t}\n\tif (entry.state === \"TURN_CLAIMED\") transitionToTurnStreaming(entry);\n\tif (!turn.aborted) deliver(entry, turn, message);\n\ttransitionToTurnResultSeen(entry);\n\tremoveAbortListener(turn);\n\tentry.activeTurn = null;\n\tif (turn.aborted) registry.markTainted(entry.senpiSessionId, \"abort\");\n\telse transitionToIdleSynced(entry);\n\tturn.resolve({ uuid: turn.uuid, messages: turn.messages, aborted: turn.aborted });\n}\n\nfunction handleMessage(\n\tregistry: ClaudeSdkOauthSessionRegistry,\n\tentry: ClaudeSdkOauthSessionEntry,\n\tmessage: SDKMessage,\n): void {\n\tconst turn = currentTurn(entry);\n\tif (!turn || !registry.isCurrentGeneration(entry.senpiSessionId, turn.generation)) return;\n\tif (!turn.claimed) {\n\t\tif (isReplayFor(message, turn.uuid)) claimTurn(entry, turn);\n\t\telse if (message.type === \"stream_event\") bufferBeforeReplay(registry, entry, turn, message);\n\t\telse if (message.type === \"result\") {\n\t\t\tthrow new SessionTurnAttributionError(\"Claude SDK OAuth result arrived before replay claim\");\n\t\t}\n\t\treturn;\n\t}\n\tif (message.type === \"user\" && \"isReplay\" in message && message.isReplay === true) return;\n\tif (message.type === \"result\") finishTurn(registry, entry, turn, message);\n\telse deliver(entry, turn, message);\n}\n\nasync function runPump(registry: ClaudeSdkOauthSessionRegistry, entry: ClaudeSdkOauthSessionEntry): Promise<void> {\n\tconst iterator = entry.query[Symbol.asyncIterator]();\n\ttry {\n\t\twhile (true) {\n\t\t\tconst { value, done } = await iterator.next();\n\t\t\tif (done) {\n\t\t\t\tfailTurn(registry, entry, new Error(\"Claude SDK OAuth query ended before the active turn completed\"));\n\t\t\t\treturn;\n\t\t\t}\n\t\t\thandleMessage(registry, entry, value);\n\t\t}\n\t} catch (error) {\n\t\tfailTurn(registry, entry, error instanceof Error ? error : new Error(String(error)));\n\t}\n}\n\nexport function submitSessionTurn(\n\tregistry: ClaudeSdkOauthSessionRegistry,\n\tentry: ClaudeSdkOauthSessionEntry,\n\trequest: SessionTurnRequest,\n\tlimits: PreReplayBufferLimits = {\n\t\tmaxMessages: DEFAULT_PRE_REPLAY_MAX_MESSAGES,\n\t\tmaxBytes: DEFAULT_PRE_REPLAY_MAX_BYTES,\n\t},\n): Promise<SessionTurnResult> {\n\tif (currentTurn(entry)) throw new ConcurrentSessionTurnAdmissionError(entry.senpiSessionId);\n\tif (entry.state === \"STARTING\") transitionToIdleSynced(entry);\n\ttransitionToTurnWaiting(entry);\n\tconst uuid = createSessionUuid();\n\tlet turn!: ActiveTurn;\n\tconst promise = new Promise<SessionTurnResult>((resolve, reject) => {\n\t\tconst onAbort = (): void => {\n\t\t\tif (turn.aborted) return;\n\t\t\tturn.aborted = true;\n\t\t\tturn.cancelAbort = (request.scheduleAbort ?? scheduleAbort)(\n\t\t\t\t() => abortTurn(registry, entry, turn, new Error(\"Claude SDK OAuth interrupted turn did not terminate\")),\n\t\t\t\tSESSION_TURN_ABORT_GRACE_MS,\n\t\t\t);\n\t\t\tvoid entry.query.interrupt().catch((error: unknown) => {\n\t\t\t\tconst detail = error instanceof Error ? error.message : String(error);\n\t\t\t\tabortTurn(registry, entry, turn, new Error(`Claude SDK OAuth query interrupt failed: ${detail}`));\n\t\t\t});\n\t\t};\n\t\tturn = {\n\t\t\tuuid,\n\t\t\tgeneration: entry.generation,\n\t\t\tmessages: [],\n\t\t\tpreReplay: [],\n\t\t\tpreReplayBytes: 0,\n\t\t\tclaimed: false,\n\t\t\taborted: false,\n\t\t\tonMessage: request.onMessage,\n\t\t\tsignal: request.signal,\n\t\t\tonAbort,\n\t\t\tresolve,\n\t\t\treject,\n\t\t\tlimits,\n\t\t};\n\t});\n\tentry.activeTurn = turn;\n\tif (!entry.pumpTask) entry.pumpTask = runPump(registry, entry);\n\tentry.inputController.push({\n\t\ttype: \"user\",\n\t\tmessage: request.message,\n\t\tparent_tool_use_id: null,\n\t\tuuid,\n\t\tsession_id: entry.sdkSessionId,\n\t});\n\ttransitionToTurnSent(entry);\n\tif (request.signal?.aborted) turn.onAbort();\n\telse request.signal?.addEventListener(\"abort\", turn.onAbort, { once: true });\n\treturn promise;\n}\n"]}

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-state.d.ts

@@ -0,0 +1,18 @@
+export type ClaudeSdkOauthSessionState = "ABSENT" | "STARTING" | "IDLE_SYNCED" | "TURN_WAITING" | "TURN_SENT" | "TURN_CLAIMED" | "TURN_STREAMING" | "TURN_RESULT_SEEN" | "TAINTED" | "CLOSING" | "CLOSED" | "BROKEN";
+type StatefulSession = {
+    state: ClaudeSdkOauthSessionState;
+};
+export declare function transitionSessionState<T extends StatefulSession>(entry: T, next: ClaudeSdkOauthSessionState): T;
+export declare const transitionToStarting: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToIdleSynced: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToTurnWaiting: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToTurnSent: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToTurnClaimed: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToTurnStreaming: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToTurnResultSeen: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToTainted: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToClosing: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToClosed: <T extends StatefulSession>(entry: T) => T;
+export declare const transitionToBroken: <T extends StatefulSession>(entry: T) => T;
+export {};
+//# sourceMappingURL=session-registry-state.d.ts.map

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-state.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-registry-state.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/claude-sdk-oauth/session-registry-state.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,0BAA0B,GACnC,QAAQ,GACR,UAAU,GACV,aAAa,GACb,cAAc,GACd,WAAW,GACX,cAAc,GACd,gBAAgB,GAChB,kBAAkB,GAClB,SAAS,GACT,SAAS,GACT,QAAQ,GACR,QAAQ,CAAC;AAEZ,KAAK,eAAe,GAAG;IAAE,KAAK,EAAE,0BAA0B,CAAA;CAAE,CAAC;AAiB7D,wBAAgB,sBAAsB,CAAC,CAAC,SAAS,eAAe,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,0BAA0B,GAAG,CAAC,CAM/G;AAED,eAAO,MAAM,oBAAoB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CACjC,CAAC;AAC3C,eAAO,MAAM,sBAAsB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAChC,CAAC;AAC9C,eAAO,MAAM,uBAAuB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAChC,CAAC;AAC/C,eAAO,MAAM,oBAAoB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAChC,CAAC;AAC5C,eAAO,MAAM,uBAAuB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAChC,CAAC;AAC/C,eAAO,MAAM,yBAAyB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAChC,CAAC;AACjD,eAAO,MAAM,0BAA0B,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAC/B,CAAC;AACnD,eAAO,MAAM,mBAAmB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAA6C,CAAC;AACxH,eAAO,MAAM,mBAAmB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAA6C,CAAC;AACxH,eAAO,MAAM,kBAAkB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAA4C,CAAC;AACtH,eAAO,MAAM,kBAAkB,GAAI,CAAC,SAAS,eAAe,SAAS,CAAC,KAAG,CAA4C,CAAC"}

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-state.js

@@ -0,0 +1,33 @@
+const allowedTransitions = {
+    ABSENT: ["STARTING"],
+    STARTING: ["IDLE_SYNCED", "TAINTED", "CLOSING", "BROKEN"],
+    IDLE_SYNCED: ["TURN_WAITING", "TAINTED", "CLOSING", "BROKEN"],
+    TURN_WAITING: ["TURN_SENT", "TAINTED", "CLOSING", "BROKEN"],
+    TURN_SENT: ["TURN_CLAIMED", "TAINTED", "CLOSING", "BROKEN"],
+    TURN_CLAIMED: ["TURN_STREAMING", "TAINTED", "CLOSING", "BROKEN"],
+    TURN_STREAMING: ["TURN_RESULT_SEEN", "TAINTED", "CLOSING", "BROKEN"],
+    TURN_RESULT_SEEN: ["IDLE_SYNCED", "TAINTED", "CLOSING", "BROKEN"],
+    TAINTED: ["CLOSING", "BROKEN"],
+    CLOSING: ["CLOSED", "BROKEN"],
+    CLOSED: [],
+    BROKEN: ["CLOSING", "CLOSED"],
+};
+export function transitionSessionState(entry, next) {
+    if (!allowedTransitions[entry.state].includes(next)) {
+        throw new Error(`Illegal session state transition: ${entry.state} -> ${next}`);
+    }
+    entry.state = next;
+    return entry;
+}
+export const transitionToStarting = (entry) => transitionSessionState(entry, "STARTING");
+export const transitionToIdleSynced = (entry) => transitionSessionState(entry, "IDLE_SYNCED");
+export const transitionToTurnWaiting = (entry) => transitionSessionState(entry, "TURN_WAITING");
+export const transitionToTurnSent = (entry) => transitionSessionState(entry, "TURN_SENT");
+export const transitionToTurnClaimed = (entry) => transitionSessionState(entry, "TURN_CLAIMED");
+export const transitionToTurnStreaming = (entry) => transitionSessionState(entry, "TURN_STREAMING");
+export const transitionToTurnResultSeen = (entry) => transitionSessionState(entry, "TURN_RESULT_SEEN");
+export const transitionToTainted = (entry) => transitionSessionState(entry, "TAINTED");
+export const transitionToClosing = (entry) => transitionSessionState(entry, "CLOSING");
+export const transitionToClosed = (entry) => transitionSessionState(entry, "CLOSED");
+export const transitionToBroken = (entry) => transitionSessionState(entry, "BROKEN");
+//# sourceMappingURL=session-registry-state.js.map

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-state.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-registry-state.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/claude-sdk-oauth/session-registry-state.ts"],"names":[],"mappings":"AAgBA,MAAM,kBAAkB,GAAwF;IAC/G,MAAM,EAAE,CAAC,UAAU,CAAC;IACpB,QAAQ,EAAE,CAAC,aAAa,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC;IACzD,WAAW,EAAE,CAAC,cAAc,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC;IAC7D,YAAY,EAAE,CAAC,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC;IAC3D,SAAS,EAAE,CAAC,cAAc,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC;IAC3D,YAAY,EAAE,CAAC,gBAAgB,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC;IAChE,cAAc,EAAE,CAAC,kBAAkB,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC;IACpE,gBAAgB,EAAE,CAAC,aAAa,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC;IACjE,OAAO,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC;IAC9B,OAAO,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC7B,MAAM,EAAE,EAAE;IACV,MAAM,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC;CAC7B,CAAC;AAEF,MAAM,UAAU,sBAAsB,CAA4B,KAAQ,EAAE,IAAgC;IAC3G,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACrD,MAAM,IAAI,KAAK,CAAC,qCAAqC,KAAK,CAAC,KAAK,OAAO,IAAI,EAAE,CAAC,CAAC;IAChF,CAAC;IACD,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;IACnB,OAAO,KAAK,CAAC;AACd,CAAC;AAED,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CAC9E,sBAAsB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;AAC3C,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CAChF,sBAAsB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;AAC9C,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CACjF,sBAAsB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;AAC/C,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CAC9E,sBAAsB,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;AAC5C,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CACjF,sBAAsB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;AAC/C,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CACnF,sBAAsB,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC;AACjD,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAA4B,KAAQ,EAAK,EAAE,CACpF,sBAAsB,CAAC,KAAK,EAAE,kBAAkB,CAAC,CAAC;AACnD,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CAAC,sBAAsB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;AACxH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CAAC,sBAAsB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;AACxH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CAAC,sBAAsB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;AACtH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAA4B,KAAQ,EAAK,EAAE,CAAC,sBAAsB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC","sourcesContent":["export type ClaudeSdkOauthSessionState =\n\t| \"ABSENT\"\n\t| \"STARTING\"\n\t| \"IDLE_SYNCED\"\n\t| \"TURN_WAITING\"\n\t| \"TURN_SENT\"\n\t| \"TURN_CLAIMED\"\n\t| \"TURN_STREAMING\"\n\t| \"TURN_RESULT_SEEN\"\n\t| \"TAINTED\"\n\t| \"CLOSING\"\n\t| \"CLOSED\"\n\t| \"BROKEN\";\n\ntype StatefulSession = { state: ClaudeSdkOauthSessionState };\n\nconst allowedTransitions: Readonly<Record<ClaudeSdkOauthSessionState, readonly ClaudeSdkOauthSessionState[]>> = {\n\tABSENT: [\"STARTING\"],\n\tSTARTING: [\"IDLE_SYNCED\", \"TAINTED\", \"CLOSING\", \"BROKEN\"],\n\tIDLE_SYNCED: [\"TURN_WAITING\", \"TAINTED\", \"CLOSING\", \"BROKEN\"],\n\tTURN_WAITING: [\"TURN_SENT\", \"TAINTED\", \"CLOSING\", \"BROKEN\"],\n\tTURN_SENT: [\"TURN_CLAIMED\", \"TAINTED\", \"CLOSING\", \"BROKEN\"],\n\tTURN_CLAIMED: [\"TURN_STREAMING\", \"TAINTED\", \"CLOSING\", \"BROKEN\"],\n\tTURN_STREAMING: [\"TURN_RESULT_SEEN\", \"TAINTED\", \"CLOSING\", \"BROKEN\"],\n\tTURN_RESULT_SEEN: [\"IDLE_SYNCED\", \"TAINTED\", \"CLOSING\", \"BROKEN\"],\n\tTAINTED: [\"CLOSING\", \"BROKEN\"],\n\tCLOSING: [\"CLOSED\", \"BROKEN\"],\n\tCLOSED: [],\n\tBROKEN: [\"CLOSING\", \"CLOSED\"],\n};\n\nexport function transitionSessionState<T extends StatefulSession>(entry: T, next: ClaudeSdkOauthSessionState): T {\n\tif (!allowedTransitions[entry.state].includes(next)) {\n\t\tthrow new Error(`Illegal session state transition: ${entry.state} -> ${next}`);\n\t}\n\tentry.state = next;\n\treturn entry;\n}\n\nexport const transitionToStarting = <T extends StatefulSession>(entry: T): T =>\n\ttransitionSessionState(entry, \"STARTING\");\nexport const transitionToIdleSynced = <T extends StatefulSession>(entry: T): T =>\n\ttransitionSessionState(entry, \"IDLE_SYNCED\");\nexport const transitionToTurnWaiting = <T extends StatefulSession>(entry: T): T =>\n\ttransitionSessionState(entry, \"TURN_WAITING\");\nexport const transitionToTurnSent = <T extends StatefulSession>(entry: T): T =>\n\ttransitionSessionState(entry, \"TURN_SENT\");\nexport const transitionToTurnClaimed = <T extends StatefulSession>(entry: T): T =>\n\ttransitionSessionState(entry, \"TURN_CLAIMED\");\nexport const transitionToTurnStreaming = <T extends StatefulSession>(entry: T): T =>\n\ttransitionSessionState(entry, \"TURN_STREAMING\");\nexport const transitionToTurnResultSeen = <T extends StatefulSession>(entry: T): T =>\n\ttransitionSessionState(entry, \"TURN_RESULT_SEEN\");\nexport const transitionToTainted = <T extends StatefulSession>(entry: T): T => transitionSessionState(entry, \"TAINTED\");\nexport const transitionToClosing = <T extends StatefulSession>(entry: T): T => transitionSessionState(entry, \"CLOSING\");\nexport const transitionToClosed = <T extends StatefulSession>(entry: T): T => transitionSessionState(entry, \"CLOSED\");\nexport const transitionToBroken = <T extends StatefulSession>(entry: T): T => transitionSessionState(entry, \"BROKEN\");\n"]}

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-wiring.d.ts

@@ -0,0 +1,3 @@
+import type { ExtensionAPI } from "../../types.ts";
+export declare function registerSessionRegistry(pi: Pick<ExtensionAPI, "on">): void;
+//# sourceMappingURL=session-registry-wiring.d.ts.map

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-wiring.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-registry-wiring.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/claude-sdk-oauth/session-registry-wiring.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAuGnD,wBAAgB,uBAAuB,CAAC,EAAE,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,GAAG,IAAI,CAkD1E"}

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-wiring.js

@@ -0,0 +1,134 @@
+import { CLAUDE_SDK_OAUTH_PROVIDER_ID } from "./account-management.js";
+import { closeSession, getSession, markTainted, recordBranchInfo, } from "./session-registry.js";
+import { installAssistantProvenanceHooks, sessionSyncDigest, } from "./session-sync.js";
+const provenanceByMessages = new WeakMap();
+const provenanceByHashes = new WeakMap();
+const provenanceByEntry = new WeakMap();
+const stagedHashByEntry = new WeakMap();
+function assistantHash(message) {
+    return sessionSyncDigest({
+        role: message.role,
+        api: message.api,
+        provider: message.provider,
+        model: message.model,
+        content: message.content,
+    });
+}
+const assistantProvenanceHooks = {
+    captureMessages(context, messages) {
+        let sentCount = 0;
+        const provenance = [];
+        for (const message of context.messages) {
+            if (message.role === "assistant") {
+                provenance.push({ afterSentCount: sentCount, hash: assistantHash(message) });
+            }
+            else {
+                sentCount++;
+            }
+        }
+        provenanceByMessages.set(messages, provenance);
+    },
+    captureHashes(messages, hashes) {
+        const provenance = provenanceByMessages.get(messages);
+        if (provenance)
+            provenanceByHashes.set(hashes, provenance);
+    },
+    matches(entry, hashes, branchPrefix) {
+        const current = provenanceByHashes.get(hashes);
+        const resident = provenanceByEntry.get(entry) ?? [];
+        if (!current)
+            return false;
+        const expected = branchPrefix ? resident.filter((item) => item.afterSentCount < hashes.length) : resident;
+        return (current.length === expected.length &&
+            current.every((item, index) => item.afterSentCount === expected[index]?.afterSentCount && item.hash === expected[index]?.hash));
+    },
+    record(entry, hashes) {
+        const provenance = provenanceByHashes.get(hashes);
+        if (provenance)
+            provenanceByEntry.set(entry, provenance.map((item) => ({ ...item })));
+    },
+    prime(entry, previous, from) {
+        const provenance = (provenanceByEntry.get(previous) ?? []).filter((item) => item.afterSentCount <= from);
+        provenanceByEntry.set(entry, provenance);
+    },
+};
+function stageAssistantProvenance(entry, message) {
+    stagedHashByEntry.set(entry, assistantHash(message));
+}
+function commitAssistantProvenance(entry, message) {
+    const hash = assistantHash(message);
+    const stagedHash = stagedHashByEntry.get(entry);
+    stagedHashByEntry.delete(entry);
+    if (stagedHash !== hash)
+        return false;
+    const provenance = provenanceByEntry.get(entry) ?? [];
+    const latest = provenance.at(-1);
+    if (latest?.afterSentCount === entry.sentCount)
+        return latest.hash === hash;
+    provenanceByEntry.set(entry, [...provenance, { afterSentCount: entry.sentCount, hash }]);
+    return true;
+}
+function isResidentAssistant(message, modelId) {
+    return (message.api === CLAUDE_SDK_OAUTH_PROVIDER_ID &&
+        message.provider === CLAUDE_SDK_OAUTH_PROVIDER_ID &&
+        message.model === modelId &&
+        message.stopReason !== "error" &&
+        message.stopReason !== "aborted");
+}
+export function registerSessionRegistry(pi) {
+    installAssistantProvenanceHooks(assistantProvenanceHooks);
+    pi.on("session_compact", (_event, ctx) => {
+        markTainted(ctx.sessionManager.getSessionId(), "compaction");
+    });
+    pi.on("session_before_fork", (_event, ctx) => {
+        markTainted(ctx.sessionManager.getSessionId(), "fork");
+    });
+    pi.on("session_tree", (event, ctx) => {
+        if (event.oldLeafId === null || event.newLeafId === null)
+            return;
+        recordBranchInfo(ctx.sessionManager.getSessionId(), {
+            oldLeafId: event.oldLeafId,
+            newLeafId: event.newLeafId,
+        });
+    });
+    pi.on("model_select", (_event, ctx) => {
+        closeSession(ctx.sessionManager.getSessionId(), "model_selected");
+    });
+    pi.on("thinking_level_select", (_event, ctx) => {
+        closeSession(ctx.sessionManager.getSessionId(), "thinking_level_selected");
+    });
+    pi.on("message_start", (event, ctx) => {
+        if (event.message.role !== "assistant")
+            return;
+        const entry = getSession(ctx.sessionManager.getSessionId());
+        if (entry && isResidentAssistant(event.message, entry.modelId)) {
+            stageAssistantProvenance(entry, event.message);
+        }
+    });
+    pi.on("message_update", (event, ctx) => {
+        if (event.message.role !== "assistant")
+            return;
+        const entry = getSession(ctx.sessionManager.getSessionId());
+        if (entry && isResidentAssistant(event.message, entry.modelId)) {
+            stageAssistantProvenance(entry, event.message);
+        }
+    });
+    pi.on("message_end", (event, ctx) => {
+        if (event.message.role !== "assistant")
+            return;
+        const sessionId = ctx.sessionManager.getSessionId();
+        const entry = getSession(sessionId);
+        if (!entry)
+            return;
+        if (!isResidentAssistant(event.message, entry.modelId) || !commitAssistantProvenance(entry, event.message)) {
+            markTainted(sessionId, "assistant_provenance_unverified");
+        }
+    });
+    pi.on("session_shutdown", (event, ctx) => {
+        closeSession(ctx.sessionManager.getSessionId(), event.reason);
+    });
+    pi.on("session_extensions_removed", (_event, ctx) => {
+        closeSession(ctx.sessionManager.getSessionId(), "extensions_removed");
+    });
+}
+//# sourceMappingURL=session-registry-wiring.js.map

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry-wiring.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-registry-wiring.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/claude-sdk-oauth/session-registry-wiring.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,4BAA4B,EAAE,MAAM,yBAAyB,CAAC;AACvE,OAAO,EAEN,YAAY,EACZ,UAAU,EACV,WAAW,EACX,gBAAgB,GAChB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACN,+BAA+B,EAG/B,iBAAiB,GACjB,MAAM,mBAAmB,CAAC;AAI3B,MAAM,oBAAoB,GAAG,IAAI,OAAO,EAAiD,CAAC;AAC1F,MAAM,kBAAkB,GAAG,IAAI,OAAO,EAA4C,CAAC;AACnF,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAAqD,CAAC;AAC3F,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAAsC,CAAC;AAE5E,SAAS,aAAa,CAAC,OAAyB;IAC/C,OAAO,iBAAiB,CAAC;QACxB,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,OAAO,EAAE,OAAO,CAAC,OAAO;KACxB,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,wBAAwB,GAAoC;IACjE,eAAe,CAAC,OAAO,EAAE,QAAQ;QAChC,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,UAAU,GAA0B,EAAE,CAAC;QAC7C,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACxC,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAClC,UAAU,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC9E,CAAC;iBAAM,CAAC;gBACP,SAAS,EAAE,CAAC;YACb,CAAC;QACF,CAAC;QACD,oBAAoB,CAAC,GAAG,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IAChD,CAAC;IACD,aAAa,CAAC,QAAQ,EAAE,MAAM;QAC7B,MAAM,UAAU,GAAG,oBAAoB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,UAAU;YAAE,kBAAkB,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAC5D,CAAC;IACD,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY;QAClC,MAAM,OAAO,GAAG,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACpD,IAAI,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC;QAC3B,MAAM,QAAQ,GAAG,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC1G,OAAO,CACN,OAAO,CAAC,MAAM,KAAK,QAAQ,CAAC,MAAM;YAClC,OAAO,CAAC,KAAK,CACZ,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CACf,IAAI,CAAC,cAAc,KAAK,QAAQ,CAAC,KAAK,CAAC,EAAE,cAAc,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,KAAK,CAAC,EAAE,IAAI,CAC/F,CACD,CAAC;IACH,CAAC;IACD,MAAM,CAAC,KAAK,EAAE,MAAM;QACnB,MAAM,UAAU,GAAG,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAClD,IAAI,UAAU;YACb,iBAAiB,CAAC,GAAG,CACpB,KAAK,EACL,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CACvC,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI;QAC1B,MAAM,UAAU,GAAG,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC;QACzG,iBAAiB,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAC1C,CAAC;CACD,CAAC;AAEF,SAAS,wBAAwB,CAAC,KAAiC,EAAE,OAAyB;IAC7F,iBAAiB,CAAC,GAAG,CAAC,KAAK,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;AACtD,CAAC;AAED,SAAS,yBAAyB,CAAC,KAAiC,EAAE,OAAyB;IAC9F,MAAM,IAAI,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACpC,MAAM,UAAU,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAChD,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAChC,IAAI,UAAU,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC;IACtC,MAAM,UAAU,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IACtD,MAAM,MAAM,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACjC,IAAI,MAAM,EAAE,cAAc,KAAK,KAAK,CAAC,SAAS;QAAE,OAAO,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC;IAC5E,iBAAiB,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,GAAG,UAAU,EAAE,EAAE,cAAc,EAAE,KAAK,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACzF,OAAO,IAAI,CAAC;AACb,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAyB,EAAE,OAAe;IACtE,OAAO,CACN,OAAO,CAAC,GAAG,KAAK,4BAA4B;QAC5C,OAAO,CAAC,QAAQ,KAAK,4BAA4B;QACjD,OAAO,CAAC,KAAK,KAAK,OAAO;QACzB,OAAO,CAAC,UAAU,KAAK,OAAO;QAC9B,OAAO,CAAC,UAAU,KAAK,SAAS,CAChC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,EAA4B;IACnE,+BAA+B,CAAC,wBAAwB,CAAC,CAAC;IAC1D,EAAE,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QACxC,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE,YAAY,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,qBAAqB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QAC5C,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE,MAAM,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACpC,IAAI,KAAK,CAAC,SAAS,KAAK,IAAI,IAAI,KAAK,CAAC,SAAS,KAAK,IAAI;YAAE,OAAO;QACjE,gBAAgB,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE;YACnD,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,SAAS,EAAE,KAAK,CAAC,SAAS;SAC1B,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QACrC,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE,gBAAgB,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,uBAAuB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QAC9C,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE,yBAAyB,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACrC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW;YAAE,OAAO;QAC/C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC,CAAC;QAC5D,IAAI,KAAK,IAAI,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YAChE,wBAAwB,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC;IACF,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,gBAAgB,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACtC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW;YAAE,OAAO;QAC/C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC,CAAC;QAC5D,IAAI,KAAK,IAAI,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YAChE,wBAAwB,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC;IACF,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACnC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW;YAAE,OAAO;QAC/C,MAAM,SAAS,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QACpD,MAAM,KAAK,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,yBAAyB,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5G,WAAW,CAAC,SAAS,EAAE,iCAAiC,CAAC,CAAC;QAC3D,CAAC;IACF,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACxC,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,4BAA4B,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QACnD,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE,oBAAoB,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;AACJ,CAAC","sourcesContent":["import type { AssistantMessage } from \"@earendil-works/pi-ai\";\nimport type { ExtensionAPI } from \"../../types.ts\";\nimport { CLAUDE_SDK_OAUTH_PROVIDER_ID } from \"./account-management.ts\";\nimport {\n\ttype ClaudeSdkOauthSessionEntry,\n\tcloseSession,\n\tgetSession,\n\tmarkTainted,\n\trecordBranchInfo,\n} from \"./session-registry.ts\";\nimport {\n\tinstallAssistantProvenanceHooks,\n\ttype SentMessage,\n\ttype SessionAssistantProvenanceHooks,\n\tsessionSyncDigest,\n} from \"./session-sync.ts\";\n\ntype AssistantProvenance = { afterSentCount: number; hash: string };\n\nconst provenanceByMessages = new WeakMap<readonly SentMessage[], AssistantProvenance[]>();\nconst provenanceByHashes = new WeakMap<readonly string[], AssistantProvenance[]>();\nconst provenanceByEntry = new WeakMap<ClaudeSdkOauthSessionEntry, AssistantProvenance[]>();\nconst stagedHashByEntry = new WeakMap<ClaudeSdkOauthSessionEntry, string>();\n\nfunction assistantHash(message: AssistantMessage): string {\n\treturn sessionSyncDigest({\n\t\trole: message.role,\n\t\tapi: message.api,\n\t\tprovider: message.provider,\n\t\tmodel: message.model,\n\t\tcontent: message.content,\n\t});\n}\n\nconst assistantProvenanceHooks: SessionAssistantProvenanceHooks = {\n\tcaptureMessages(context, messages) {\n\t\tlet sentCount = 0;\n\t\tconst provenance: AssistantProvenance[] = [];\n\t\tfor (const message of context.messages) {\n\t\t\tif (message.role === \"assistant\") {\n\t\t\t\tprovenance.push({ afterSentCount: sentCount, hash: assistantHash(message) });\n\t\t\t} else {\n\t\t\t\tsentCount++;\n\t\t\t}\n\t\t}\n\t\tprovenanceByMessages.set(messages, provenance);\n\t},\n\tcaptureHashes(messages, hashes) {\n\t\tconst provenance = provenanceByMessages.get(messages);\n\t\tif (provenance) provenanceByHashes.set(hashes, provenance);\n\t},\n\tmatches(entry, hashes, branchPrefix) {\n\t\tconst current = provenanceByHashes.get(hashes);\n\t\tconst resident = provenanceByEntry.get(entry) ?? [];\n\t\tif (!current) return false;\n\t\tconst expected = branchPrefix ? resident.filter((item) => item.afterSentCount < hashes.length) : resident;\n\t\treturn (\n\t\t\tcurrent.length === expected.length &&\n\t\t\tcurrent.every(\n\t\t\t\t(item, index) =>\n\t\t\t\t\titem.afterSentCount === expected[index]?.afterSentCount && item.hash === expected[index]?.hash,\n\t\t\t)\n\t\t);\n\t},\n\trecord(entry, hashes) {\n\t\tconst provenance = provenanceByHashes.get(hashes);\n\t\tif (provenance)\n\t\t\tprovenanceByEntry.set(\n\t\t\t\tentry,\n\t\t\t\tprovenance.map((item) => ({ ...item })),\n\t\t\t);\n\t},\n\tprime(entry, previous, from) {\n\t\tconst provenance = (provenanceByEntry.get(previous) ?? []).filter((item) => item.afterSentCount <= from);\n\t\tprovenanceByEntry.set(entry, provenance);\n\t},\n};\n\nfunction stageAssistantProvenance(entry: ClaudeSdkOauthSessionEntry, message: AssistantMessage): void {\n\tstagedHashByEntry.set(entry, assistantHash(message));\n}\n\nfunction commitAssistantProvenance(entry: ClaudeSdkOauthSessionEntry, message: AssistantMessage): boolean {\n\tconst hash = assistantHash(message);\n\tconst stagedHash = stagedHashByEntry.get(entry);\n\tstagedHashByEntry.delete(entry);\n\tif (stagedHash !== hash) return false;\n\tconst provenance = provenanceByEntry.get(entry) ?? [];\n\tconst latest = provenance.at(-1);\n\tif (latest?.afterSentCount === entry.sentCount) return latest.hash === hash;\n\tprovenanceByEntry.set(entry, [...provenance, { afterSentCount: entry.sentCount, hash }]);\n\treturn true;\n}\n\nfunction isResidentAssistant(message: AssistantMessage, modelId: string): boolean {\n\treturn (\n\t\tmessage.api === CLAUDE_SDK_OAUTH_PROVIDER_ID &&\n\t\tmessage.provider === CLAUDE_SDK_OAUTH_PROVIDER_ID &&\n\t\tmessage.model === modelId &&\n\t\tmessage.stopReason !== \"error\" &&\n\t\tmessage.stopReason !== \"aborted\"\n\t);\n}\n\nexport function registerSessionRegistry(pi: Pick<ExtensionAPI, \"on\">): void {\n\tinstallAssistantProvenanceHooks(assistantProvenanceHooks);\n\tpi.on(\"session_compact\", (_event, ctx) => {\n\t\tmarkTainted(ctx.sessionManager.getSessionId(), \"compaction\");\n\t});\n\tpi.on(\"session_before_fork\", (_event, ctx) => {\n\t\tmarkTainted(ctx.sessionManager.getSessionId(), \"fork\");\n\t});\n\tpi.on(\"session_tree\", (event, ctx) => {\n\t\tif (event.oldLeafId === null || event.newLeafId === null) return;\n\t\trecordBranchInfo(ctx.sessionManager.getSessionId(), {\n\t\t\toldLeafId: event.oldLeafId,\n\t\t\tnewLeafId: event.newLeafId,\n\t\t});\n\t});\n\tpi.on(\"model_select\", (_event, ctx) => {\n\t\tcloseSession(ctx.sessionManager.getSessionId(), \"model_selected\");\n\t});\n\tpi.on(\"thinking_level_select\", (_event, ctx) => {\n\t\tcloseSession(ctx.sessionManager.getSessionId(), \"thinking_level_selected\");\n\t});\n\tpi.on(\"message_start\", (event, ctx) => {\n\t\tif (event.message.role !== \"assistant\") return;\n\t\tconst entry = getSession(ctx.sessionManager.getSessionId());\n\t\tif (entry && isResidentAssistant(event.message, entry.modelId)) {\n\t\t\tstageAssistantProvenance(entry, event.message);\n\t\t}\n\t});\n\tpi.on(\"message_update\", (event, ctx) => {\n\t\tif (event.message.role !== \"assistant\") return;\n\t\tconst entry = getSession(ctx.sessionManager.getSessionId());\n\t\tif (entry && isResidentAssistant(event.message, entry.modelId)) {\n\t\t\tstageAssistantProvenance(entry, event.message);\n\t\t}\n\t});\n\tpi.on(\"message_end\", (event, ctx) => {\n\t\tif (event.message.role !== \"assistant\") return;\n\t\tconst sessionId = ctx.sessionManager.getSessionId();\n\t\tconst entry = getSession(sessionId);\n\t\tif (!entry) return;\n\t\tif (!isResidentAssistant(event.message, entry.modelId) || !commitAssistantProvenance(entry, event.message)) {\n\t\t\tmarkTainted(sessionId, \"assistant_provenance_unverified\");\n\t\t}\n\t});\n\tpi.on(\"session_shutdown\", (event, ctx) => {\n\t\tcloseSession(ctx.sessionManager.getSessionId(), event.reason);\n\t});\n\tpi.on(\"session_extensions_removed\", (_event, ctx) => {\n\t\tcloseSession(ctx.sessionManager.getSessionId(), \"extensions_removed\");\n\t});\n}\n"]}

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry.d.ts

@@ -0,0 +1,85 @@
+import type { AccountSlot } from "./accounts.ts";
+import type { Options, SDKUserMessage, SdkQuery, SdkQueryHandle } from "./sdk-boundary.ts";
+import { type ClaudeSdkOauthSessionState } from "./session-registry-state.ts";
+export declare const SESSION_REGISTRY_IDLE_TTL_MS: number;
+export declare const SESSION_REGISTRY_MAX_ENTRIES = 32;
+export interface SessionInputController extends AsyncIterable<SDKUserMessage> {
+    push(message: SDKUserMessage): void;
+    close(): void;
+}
+export interface SessionRegistryReapHandle {
+    cancel(): void;
+    unref(): void;
+}
+export type SessionRegistryBoundary = {
+    now: () => number;
+    queryFactory: SdkQuery;
+    scheduleReap: (callback: () => void, delayMs: number) => SessionRegistryReapHandle;
+};
+export declare function overrideSessionRegistryBoundary(override: Partial<SessionRegistryBoundary>): void;
+export declare function resetSessionRegistryBoundary(): void;
+export type SessionBranchInfo = {
+    oldLeafId: string;
+    newLeafId: string;
+};
+export interface ClaudeSdkOauthSessionEntry {
+    senpiSessionId: string;
+    sdkSessionId: string;
+    generation: number;
+    accountName: string;
+    modelId: string;
+    toolsetHash: string;
+    systemPromptHash: string;
+    query: SdkQueryHandle;
+    inputController: SessionInputController;
+    pumpTask: Promise<void> | null;
+    state: ClaudeSdkOauthSessionState;
+    activeTurn: unknown | null;
+    syncedPrefixHash: string | null;
+    sentCount: number;
+    assistantUuidByIndex: Map<number, string>;
+    branchInfo: SessionBranchInfo | null;
+    taintedReason: string | null;
+    lastUsedAt: number;
+}
+export interface CreateSessionRegistryEntryInput {
+    senpiSessionId: string;
+    accountName: string;
+    modelId: string;
+    toolsetHash: string;
+    systemPromptHash: string;
+    options: Options;
+}
+export declare class SessionRegistryResourceLimitError extends Error {
+    readonly code = "session_registry_capacity";
+    constructor();
+}
+export declare function createSessionUuid(now?: number): `${string}-${string}-${string}-${string}-${string}`;
+export declare class ClaudeSdkOauthSessionRegistry {
+    private readonly entries;
+    private readonly generations;
+    private readonly scheduledReaps;
+    get size(): number;
+    get(senpiSessionId: string): ClaudeSdkOauthSessionEntry | undefined;
+    getOrCreate(input: CreateSessionRegistryEntryInput): ClaudeSdkOauthSessionEntry;
+    touch(entry: ClaudeSdkOauthSessionEntry): void;
+    closeSession(senpiSessionId: string, _reason: string): void;
+    markTainted(senpiSessionId: string, reason: string): void;
+    recordBranchInfo(senpiSessionId: string, info: SessionBranchInfo): void;
+    isCurrentGeneration(senpiSessionId: string, generation: number): boolean;
+    evictExpired(): void;
+    private recordUse;
+    private scheduleReap;
+    private reap;
+    private cancelReap;
+    private ensureCapacity;
+}
+export declare const sessionRegistry: ClaudeSdkOauthSessionRegistry;
+export declare function getOrCreateSession(input: CreateSessionRegistryEntryInput): ClaudeSdkOauthSessionEntry;
+export declare function getSession(senpiSessionId: string): ClaudeSdkOauthSessionEntry | undefined;
+export declare function closeSession(senpiSessionId: string, reason: string): void;
+export declare function markTainted(senpiSessionId: string, reason: string): void;
+export declare function recordBranchInfo(senpiSessionId: string, info: SessionBranchInfo): void;
+export declare function isCurrentGeneration(senpiSessionId: string, generation: number): boolean;
+export declare function isBoundAccountTokenExpiring(entry: Pick<ClaudeSdkOauthSessionEntry, "accountName">, accounts: readonly Pick<AccountSlot, "name" | "expires" | "source">[]): boolean;
+//# sourceMappingURL=session-registry.d.ts.map

package/dist/core/extensions/builtin/claude-sdk-oauth/session-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-registry.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/claude-sdk-oauth/session-registry.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAEjD,OAAO,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAE3F,OAAO,EACN,KAAK,0BAA0B,EAI/B,MAAM,6BAA6B,CAAC;AAErC,eAAO,MAAM,4BAA4B,QAAc,CAAC;AACxD,eAAO,MAAM,4BAA4B,KAAK,CAAC;AAE/C,MAAM,WAAW,sBAAuB,SAAQ,aAAa,CAAC,cAAc,CAAC;IAC5E,IAAI,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI,CAAC;IACpC,KAAK,IAAI,IAAI,CAAC;CACd;AAgCD,MAAM,WAAW,yBAAyB;IACzC,MAAM,IAAI,IAAI,CAAC;IACf,KAAK,IAAI,IAAI,CAAC;CACd;AAED,MAAM,MAAM,uBAAuB,GAAG;IACrC,GAAG,EAAE,MAAM,MAAM,CAAC;IAClB,YAAY,EAAE,QAAQ,CAAC;IACvB,YAAY,EAAE,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,OAAO,EAAE,MAAM,KAAK,yBAAyB,CAAC;CACnF,CAAC;AAYF,wBAAgB,+BAA+B,CAAC,QAAQ,EAAE,OAAO,CAAC,uBAAuB,CAAC,GAAG,IAAI,CAEhG;AAED,wBAAgB,4BAA4B,IAAI,IAAI,CAEnD;AAED,MAAM,MAAM,iBAAiB,GAAG;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,CAAC;AAEzE,MAAM,WAAW,0BAA0B;IAC1C,cAAc,EAAE,MAAM,CAAC;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,cAAc,CAAC;IACtB,eAAe,EAAE,sBAAsB,CAAC;IACxC,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC/B,KAAK,EAAE,0BAA0B,CAAC;IAClC,UAAU,EAAE,OAAO,GAAG,IAAI,CAAC;IAC3B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,UAAU,EAAE,iBAAiB,GAAG,IAAI,CAAC;IACrC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,UAAU,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,+BAA+B;IAC/C,cAAc,EAAE,MAAM,CAAC;IACvB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,OAAO,CAAC;CACjB;AAED,qBAAa,iCAAkC,SAAQ,KAAK;IAC3D,QAAQ,CAAC,IAAI,+BAA+B;IAE5C,cAKC;CACD;AAED,wBAAgB,iBAAiB,CAAC,GAAG,SAAa,GAAG,GAAG,MAAM,IAAI,MAAM,IAAI,MAAM,IAAI,MAAM,IAAI,MAAM,EAAE,CAiBvG;AAQD,qBAAa,6BAA6B;IACzC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiD;IACzE,OAAO,CAAC,QAAQ,CAAC,WAAW,CAA6B;IACzD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAoC;IAEnE,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED,GAAG,CAAC,cAAc,EAAE,MAAM,GAAG,0BAA0B,GAAG,SAAS,CAElE;IAED,WAAW,CAAC,KAAK,EAAE,+BAA+B,GAAG,0BAA0B,CAuD9E;IAED,KAAK,CAAC,KAAK,EAAE,0BAA0B,GAAG,IAAI,CAE7C;IAED,YAAY,CAAC,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAa1D;IAED,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAMxD;IAED,gBAAgB,CAAC,cAAc,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,GAAG,IAAI,CAKtE;IAED,mBAAmB,CAAC,cAAc,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAEvE;IAED,YAAY,IAAI,IAAI,CAOnB;IAED,OAAO,CAAC,SAAS;IAOjB,OAAO,CAAC,YAAY;IAUpB,OAAO,CAAC,IAAI;IAWZ,OAAO,CAAC,UAAU;IAOlB,OAAO,CAAC,cAAc;CAQtB;AAED,eAAO,MAAM,eAAe,+BAAsC,CAAC;AAEnE,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,+BAA+B,GAAG,0BAA0B,CAErG;AAED,wBAAgB,UAAU,CAAC,cAAc,EAAE,MAAM,GAAG,0BAA0B,GAAG,SAAS,CAEzF;AAED,wBAAgB,YAAY,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAEzE;AAED,wBAAgB,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAExE;AAED,wBAAgB,gBAAgB,CAAC,cAAc,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,GAAG,IAAI,CAEtF;AAED,wBAAgB,mBAAmB,CAAC,cAAc,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAEvF;AAED,wBAAgB,2BAA2B,CAC1C,KAAK,EAAE,IAAI,CAAC,0BAA0B,EAAE,aAAa,CAAC,EACtD,QAAQ,EAAE,SAAS,IAAI,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS,GAAG,QAAQ,CAAC,EAAE,GACnE,OAAO,CAKT"}