@code-yeongyu/senpi 2026.7.30 → 2026.7.31-2

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"}

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

@@ -0,0 +1,270 @@
+import { randomBytes } from "node:crypto";
+import { EXPIRING_WITHIN_MS } from "./auth-lane.js";
+import { getSdkBoundary } from "./sdk-boundary.js";
+import { transitionToClosed, transitionToClosing, transitionToTainted, } from "./session-registry-state.js";
+export const SESSION_REGISTRY_IDLE_TTL_MS = 30 * 60_000;
+export const SESSION_REGISTRY_MAX_ENTRIES = 32;
+class StreamingInputController {
+    constructor() {
+        this.closed = false;
+        this.pending = [];
+        this.readers = [];
+    }
+    [Symbol.asyncIterator]() {
+        return this;
+    }
+    next() {
+        const message = this.pending.shift();
+        if (message)
+            return Promise.resolve({ value: message, done: false });
+        if (this.closed)
+            return Promise.resolve({ value: undefined, done: true });
+        return new Promise((resolve) => this.readers.push(resolve));
+    }
+    push(message) {
+        if (this.closed)
+            throw new Error("Cannot push to a closed session input controller");
+        const reader = this.readers.shift();
+        if (reader)
+            reader({ value: message, done: false });
+        else
+            this.pending.push(message);
+    }
+    close() {
+        if (this.closed)
+            return;
+        this.closed = true;
+        for (const reader of this.readers.splice(0))
+            reader({ value: undefined, done: true });
+    }
+}
+const defaultSessionRegistryBoundary = {
+    now: () => Date.now(),
+    queryFactory: (input) => getSdkBoundary().query(input),
+    scheduleReap: (callback, delayMs) => {
+        const timer = setTimeout(callback, delayMs);
+        return { cancel: () => clearTimeout(timer), unref: () => timer.unref() };
+    },
+};
+let activeSessionRegistryBoundary = defaultSessionRegistryBoundary;
+export function overrideSessionRegistryBoundary(override) {
+    activeSessionRegistryBoundary = { ...defaultSessionRegistryBoundary, ...override };
+}
+export function resetSessionRegistryBoundary() {
+    activeSessionRegistryBoundary = defaultSessionRegistryBoundary;
+}
+export class SessionRegistryResourceLimitError extends Error {
+    constructor() {
+        super(`Claude SDK OAuth session registry is at its ${SESSION_REGISTRY_MAX_ENTRIES}-entry limit with no idle session to evict`);
+        this.code = "session_registry_capacity";
+        this.name = "SessionRegistryResourceLimitError";
+    }
+}
+export function createSessionUuid(now = Date.now()) {
+    const ts = BigInt(Math.trunc(now));
+    const bytes = randomBytes(10);
+    const hex = [
+        ((ts >> 40n) & 0xffn).toString(16).padStart(2, "0"),
+        ((ts >> 32n) & 0xffn).toString(16).padStart(2, "0"),
+        ((ts >> 24n) & 0xffn).toString(16).padStart(2, "0"),
+        ((ts >> 16n) & 0xffn).toString(16).padStart(2, "0"),
+        ((ts >> 8n) & 0xffn).toString(16).padStart(2, "0"),
+        (ts & 0xffn).toString(16).padStart(2, "0"),
+        (0x70 | (bytes[0] & 0x0f)).toString(16).padStart(2, "0"),
+        bytes[1].toString(16).padStart(2, "0"),
+        (0x80 | (bytes[2] & 0x3f)).toString(16).padStart(2, "0"),
+        bytes[3].toString(16).padStart(2, "0"),
+        ...Array.from(bytes.slice(4), (byte) => byte.toString(16).padStart(2, "0")),
+    ].join("");
+    return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20, 32)}`;
+}
+function evictable(entry) {
+    return (entry.state === "IDLE_SYNCED" || entry.state === "TAINTED") && entry.activeTurn === null;
+}
+export class ClaudeSdkOauthSessionRegistry {
+    constructor() {
+        this.entries = new Map();
+        this.generations = new Map();
+        this.scheduledReaps = new Map();
+    }
+    get size() {
+        return this.entries.size;
+    }
+    get(senpiSessionId) {
+        return this.entries.get(senpiSessionId);
+    }
+    getOrCreate(input) {
+        const existing = this.entries.get(input.senpiSessionId);
+        if (existing) {
+            const now = activeSessionRegistryBoundary.now();
+            if (!evictable(existing) || now - existing.lastUsedAt < SESSION_REGISTRY_IDLE_TTL_MS) {
+                this.touch(existing);
+                return existing;
+            }
+            this.closeSession(existing.senpiSessionId, "idle_ttl");
+        }
+        this.evictExpired();
+        this.ensureCapacity();
+        const now = activeSessionRegistryBoundary.now();
+        const generation = (this.generations.get(input.senpiSessionId) ?? 0) + 1;
+        const sdkSessionId = createSessionUuid(now);
+        const inputController = new StreamingInputController();
+        const query = activeSessionRegistryBoundary.queryFactory({
+            prompt: inputController,
+            options: {
+                ...input.options,
+                sessionId: sdkSessionId,
+                extraArgs: { ...input.options.extraArgs, "replay-user-messages": "" },
+            },
+        });
+        const target = {
+            ...input,
+            sdkSessionId,
+            generation,
+            query,
+            inputController,
+            pumpTask: null,
+            state: "STARTING",
+            activeTurn: null,
+            syncedPrefixHash: null,
+            sentCount: 0,
+            assistantUuidByIndex: new Map(),
+            branchInfo: null,
+            taintedReason: null,
+            lastUsedAt: now,
+        };
+        let entry;
+        entry = new Proxy(target, {
+            set: (current, property, value) => {
+                if (this.entries.get(current.senpiSessionId) !== entry)
+                    return true;
+                const previousTurn = current.activeTurn;
+                Reflect.set(current, property, value);
+                if (property === "activeTurn" && previousTurn !== value) {
+                    this.recordUse(entry, value === null);
+                }
+                return true;
+            },
+        });
+        this.generations.set(input.senpiSessionId, generation);
+        this.entries.set(input.senpiSessionId, entry);
+        return entry;
+    }
+    touch(entry) {
+        this.recordUse(entry, entry.activeTurn === null && evictable(entry));
+    }
+    closeSession(senpiSessionId, _reason) {
+        const entry = this.entries.get(senpiSessionId);
+        if (!entry)
+            return;
+        this.cancelReap(entry);
+        transitionToClosing(entry);
+        entry.inputController.close();
+        try {
+            entry.query.close();
+        }
+        finally {
+            transitionToClosed(entry);
+            this.cancelReap(entry);
+            this.entries.delete(senpiSessionId);
+        }
+    }
+    markTainted(senpiSessionId, reason) {
+        const entry = this.entries.get(senpiSessionId);
+        if (!entry)
+            return;
+        entry.taintedReason = reason;
+        if (entry.state !== "TAINTED")
+            transitionToTainted(entry);
+        this.touch(entry);
+    }
+    recordBranchInfo(senpiSessionId, info) {
+        const entry = this.entries.get(senpiSessionId);
+        if (!entry)
+            return;
+        entry.branchInfo = { ...info };
+        this.touch(entry);
+    }
+    isCurrentGeneration(senpiSessionId, generation) {
+        return this.entries.get(senpiSessionId)?.generation === generation;
+    }
+    evictExpired() {
+        const now = activeSessionRegistryBoundary.now();
+        for (const entry of [...this.entries.values()]) {
+            if (evictable(entry) && now - entry.lastUsedAt >= SESSION_REGISTRY_IDLE_TTL_MS) {
+                this.closeSession(entry.senpiSessionId, "idle_ttl");
+            }
+        }
+    }
+    recordUse(entry, scheduleReap) {
+        if (this.entries.get(entry.senpiSessionId) !== entry)
+            return;
+        entry.lastUsedAt = activeSessionRegistryBoundary.now();
+        this.cancelReap(entry);
+        if (scheduleReap)
+            this.scheduleReap(entry);
+    }
+    scheduleReap(entry, delayMs = SESSION_REGISTRY_IDLE_TTL_MS) {
+        const token = Symbol("session-reap");
+        const handle = activeSessionRegistryBoundary.scheduleReap(() => this.reap(entry.senpiSessionId, entry.generation, token), delayMs);
+        this.scheduledReaps.set(entry.senpiSessionId, { generation: entry.generation, token, handle });
+        handle.unref();
+    }
+    reap(senpiSessionId, generation, token) {
+        const scheduled = this.scheduledReaps.get(senpiSessionId);
+        if (!scheduled || scheduled.generation !== generation || scheduled.token !== token)
+            return;
+        this.scheduledReaps.delete(senpiSessionId);
+        const entry = this.entries.get(senpiSessionId);
+        if (!entry || entry.generation !== generation || !evictable(entry))
+            return;
+        const remaining = entry.lastUsedAt + SESSION_REGISTRY_IDLE_TTL_MS - activeSessionRegistryBoundary.now();
+        if (remaining <= 0)
+            this.closeSession(senpiSessionId, "idle_ttl");
+        else
+            this.scheduleReap(entry, remaining);
+    }
+    cancelReap(entry) {
+        const scheduled = this.scheduledReaps.get(entry.senpiSessionId);
+        if (!scheduled || scheduled.generation !== entry.generation)
+            return;
+        scheduled.handle.cancel();
+        this.scheduledReaps.delete(entry.senpiSessionId);
+    }
+    ensureCapacity() {
+        if (this.entries.size < SESSION_REGISTRY_MAX_ENTRIES)
+            return;
+        const oldest = [...this.entries.values()]
+            .filter(evictable)
+            .sort((left, right) => left.lastUsedAt - right.lastUsedAt)[0];
+        if (!oldest)
+            throw new SessionRegistryResourceLimitError();
+        this.closeSession(oldest.senpiSessionId, "capacity");
+    }
+}
+export const sessionRegistry = new ClaudeSdkOauthSessionRegistry();
+export function getOrCreateSession(input) {
+    return sessionRegistry.getOrCreate(input);
+}
+export function getSession(senpiSessionId) {
+    return sessionRegistry.get(senpiSessionId);
+}
+export function closeSession(senpiSessionId, reason) {
+    sessionRegistry.closeSession(senpiSessionId, reason);
+}
+export function markTainted(senpiSessionId, reason) {
+    sessionRegistry.markTainted(senpiSessionId, reason);
+}
+export function recordBranchInfo(senpiSessionId, info) {
+    sessionRegistry.recordBranchInfo(senpiSessionId, info);
+}
+export function isCurrentGeneration(senpiSessionId, generation) {
+    return sessionRegistry.isCurrentGeneration(senpiSessionId, generation);
+}
+export function isBoundAccountTokenExpiring(entry, accounts) {
+    const account = accounts.find((candidate) => candidate.name === entry.accountName);
+    return account?.source !== "env" && account !== undefined
+        ? activeSessionRegistryBoundary.now() >= account.expires - EXPIRING_WITHIN_MS
+        : false;
+}
+//# sourceMappingURL=session-registry.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"session-registry.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/claude-sdk-oauth/session-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAEpD,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAEN,kBAAkB,EAClB,mBAAmB,EACnB,mBAAmB,GACnB,MAAM,6BAA6B,CAAC;AAErC,MAAM,CAAC,MAAM,4BAA4B,GAAG,EAAE,GAAG,MAAM,CAAC;AACxD,MAAM,CAAC,MAAM,4BAA4B,GAAG,EAAE,CAAC;AAO/C,MAAM,wBAAwB;IAA9B;QACS,WAAM,GAAG,KAAK,CAAC;QACN,YAAO,GAAqB,EAAE,CAAC;QAC/B,YAAO,GAA4D,EAAE,CAAC;IAyBxF,CAAC;IAvBA,CAAC,MAAM,CAAC,aAAa,CAAC;QACrB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrC,IAAI,OAAO;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACrE,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1E,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC7D,CAAC;IAED,IAAI,CAAC,OAAuB;QAC3B,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACrF,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACpC,IAAI,MAAM;YAAE,MAAM,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;;YAC/C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAED,KAAK;QACJ,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;YAAE,MAAM,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACvF,CAAC;CACD;AAaD,MAAM,8BAA8B,GAA4B;IAC/D,GAAG,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE;IACrB,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,cAAc,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC;IACtD,YAAY,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE;QACnC,MAAM,KAAK,GAAG,UAAU,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC5C,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;IAC1E,CAAC;CACD,CAAC;AACF,IAAI,6BAA6B,GAAG,8BAA8B,CAAC;AAEnE,MAAM,UAAU,+BAA+B,CAAC,QAA0C;IACzF,6BAA6B,GAAG,EAAE,GAAG,8BAA8B,EAAE,GAAG,QAAQ,EAAE,CAAC;AACpF,CAAC;AAED,MAAM,UAAU,4BAA4B;IAC3C,6BAA6B,GAAG,8BAA8B,CAAC;AAChE,CAAC;AAkCD,MAAM,OAAO,iCAAkC,SAAQ,KAAK;IAG3D;QACC,KAAK,CACJ,+CAA+C,4BAA4B,4CAA4C,CACvH,CAAC;QALM,SAAI,GAAG,2BAA2B,CAAC;QAM3C,IAAI,CAAC,IAAI,GAAG,mCAAmC,CAAC;IACjD,CAAC;CACD;AAED,MAAM,UAAU,iBAAiB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IACjD,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,WAAW,CAAC,EAAE,CAAC,CAAC;IAC9B,MAAM,GAAG,GAAG;QACX,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QACnD,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QACnD,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QACnD,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QACnD,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QAClD,CAAC,EAAE,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QAC1C,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,CAAE,GAAG,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QACzD,KAAK,CAAC,CAAC,CAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QACvC,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,CAAE,GAAG,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QACzD,KAAK,CAAC,CAAC,CAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC;QACvC,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;KAC3E,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACX,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;AAChH,CAAC;AAED,SAAS,SAAS,CAAC,KAAiC;IACnD,OAAO,CAAC,KAAK,CAAC,KAAK,KAAK,aAAa,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,IAAI,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC;AAClG,CAAC;AAID,MAAM,OAAO,6BAA6B;IAA1C;QACkB,YAAO,GAAG,IAAI,GAAG,EAAsC,CAAC;QACxD,gBAAW,GAAG,IAAI,GAAG,EAAkB,CAAC;QACxC,mBAAc,GAAG,IAAI,GAAG,EAAyB,CAAC;IA6JpE,CAAC;IA3JA,IAAI,IAAI;QACP,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC1B,CAAC;IAED,GAAG,CAAC,cAAsB;QACzB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IACzC,CAAC;IAED,WAAW,CAAC,KAAsC;QACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;QACxD,IAAI,QAAQ,EAAE,CAAC;YACd,MAAM,GAAG,GAAG,6BAA6B,CAAC,GAAG,EAAE,CAAC;YAChD,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,GAAG,GAAG,QAAQ,CAAC,UAAU,GAAG,4BAA4B,EAAE,CAAC;gBACtF,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACrB,OAAO,QAAQ,CAAC;YACjB,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,MAAM,GAAG,GAAG,6BAA6B,CAAC,GAAG,EAAE,CAAC;QAChD,MAAM,UAAU,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzE,MAAM,YAAY,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,eAAe,GAAG,IAAI,wBAAwB,EAAE,CAAC;QACvD,MAAM,KAAK,GAAG,6BAA6B,CAAC,YAAY,CAAC;YACxD,MAAM,EAAE,eAAe;YACvB,OAAO,EAAE;gBACR,GAAG,KAAK,CAAC,OAAO;gBAChB,SAAS,EAAE,YAAY;gBACvB,SAAS,EAAE,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,sBAAsB,EAAE,EAAE,EAAE;aACrE;SACD,CAAC,CAAC;QACH,MAAM,MAAM,GAA+B;YAC1C,GAAG,KAAK;YACR,YAAY;YACZ,UAAU;YACV,KAAK;YACL,eAAe;YACf,QAAQ,EAAE,IAAI;YACd,KAAK,EAAE,UAAU;YACjB,UAAU,EAAE,IAAI;YAChB,gBAAgB,EAAE,IAAI;YACtB,SAAS,EAAE,CAAC;YACZ,oBAAoB,EAAE,IAAI,GAAG,EAAE;YAC/B,UAAU,EAAE,IAAI;YAChB,aAAa,EAAE,IAAI;YACnB,UAAU,EAAE,GAAG;SACf,CAAC;QACF,IAAI,KAAkC,CAAC;QACvC,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,EAAE;YACzB,GAAG,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE;gBACjC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,KAAK;oBAAE,OAAO,IAAI,CAAC;gBACpE,MAAM,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;gBACtC,IAAI,QAAQ,KAAK,YAAY,IAAI,YAAY,KAAK,KAAK,EAAE,CAAC;oBACzD,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC,CAAC;gBACvC,CAAC;gBACD,OAAO,IAAI,CAAC;YACb,CAAC;SACD,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QACvD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;QAC9C,OAAO,KAAK,CAAC;IACd,CAAC;IAED,KAAK,CAAC,KAAiC;QACtC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,KAAK,IAAI,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,YAAY,CAAC,cAAsB,EAAE,OAAe;QACnD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvB,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC3B,KAAK,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC9B,IAAI,CAAC;YACJ,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;gBAAS,CAAC;YACV,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC1B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACvB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACrC,CAAC;IACF,CAAC;IAED,WAAW,CAAC,cAAsB,EAAE,MAAc;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,KAAK,CAAC,aAAa,GAAG,MAAM,CAAC;QAC7B,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;YAAE,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC1D,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACnB,CAAC;IAED,gBAAgB,CAAC,cAAsB,EAAE,IAAuB;QAC/D,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,KAAK,CAAC,UAAU,GAAG,EAAE,GAAG,IAAI,EAAE,CAAC;QAC/B,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACnB,CAAC;IAED,mBAAmB,CAAC,cAAsB,EAAE,UAAkB;QAC7D,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,UAAU,KAAK,UAAU,CAAC;IACpE,CAAC;IAED,YAAY;QACX,MAAM,GAAG,GAAG,6BAA6B,CAAC,GAAG,EAAE,CAAC;QAChD,KAAK,MAAM,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;YAChD,IAAI,SAAS,CAAC,KAAK,CAAC,IAAI,GAAG,GAAG,KAAK,CAAC,UAAU,IAAI,4BAA4B,EAAE,CAAC;gBAChF,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;YACrD,CAAC;QACF,CAAC;IACF,CAAC;IAEO,SAAS,CAAC,KAAiC,EAAE,YAAqB;QACzE,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,KAAK,KAAK;YAAE,OAAO;QAC7D,KAAK,CAAC,UAAU,GAAG,6BAA6B,CAAC,GAAG,EAAE,CAAC;QACvD,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvB,IAAI,YAAY;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IAC5C,CAAC;IAEO,YAAY,CAAC,KAAiC,EAAE,OAAO,GAAG,4BAA4B;QAC7F,MAAM,KAAK,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,6BAA6B,CAAC,YAAY,CACxD,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,EAC9D,OAAO,CACP,CAAC;QACF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/F,MAAM,CAAC,KAAK,EAAE,CAAC;IAChB,CAAC;IAEO,IAAI,CAAC,cAAsB,EAAE,UAAkB,EAAE,KAAa;QACrE,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,UAAU,KAAK,UAAU,IAAI,SAAS,CAAC,KAAK,KAAK,KAAK;YAAE,OAAO;QAC3F,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,UAAU,KAAK,UAAU,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YAAE,OAAO;QAC3E,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,GAAG,4BAA4B,GAAG,6BAA6B,CAAC,GAAG,EAAE,CAAC;QACxG,IAAI,SAAS,IAAI,CAAC;YAAE,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;;YAC7D,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAC1C,CAAC;IAEO,UAAU,CAAC,KAAiC;QACnD,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;QAChE,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,UAAU,KAAK,KAAK,CAAC,UAAU;YAAE,OAAO;QACpE,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QAC1B,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAClD,CAAC;IAEO,cAAc;QACrB,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,4BAA4B;YAAE,OAAO;QAC7D,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;aACvC,MAAM,CAAC,SAAS,CAAC;aACjB,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/D,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,iCAAiC,EAAE,CAAC;QAC3D,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;IACtD,CAAC;CACD;AAED,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,6BAA6B,EAAE,CAAC;AAEnE,MAAM,UAAU,kBAAkB,CAAC,KAAsC;IACxE,OAAO,eAAe,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,cAAsB;IAChD,OAAO,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,cAAsB,EAAE,MAAc;IAClE,eAAe,CAAC,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,cAAsB,EAAE,MAAc;IACjE,eAAe,CAAC,WAAW,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,cAAsB,EAAE,IAAuB;IAC/E,eAAe,CAAC,gBAAgB,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,cAAsB,EAAE,UAAkB;IAC7E,OAAO,eAAe,CAAC,mBAAmB,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,2BAA2B,CAC1C,KAAsD,EACtD,QAAqE;IAErE,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,KAAK,CAAC,WAAW,CAAC,CAAC;IACnF,OAAO,OAAO,EAAE,MAAM,KAAK,KAAK,IAAI,OAAO,KAAK,SAAS;QACxD,CAAC,CAAC,6BAA6B,CAAC,GAAG,EAAE,IAAI,OAAO,CAAC,OAAO,GAAG,kBAAkB;QAC7E,CAAC,CAAC,KAAK,CAAC;AACV,CAAC","sourcesContent":["import { randomBytes } from \"node:crypto\";\nimport type { AccountSlot } from \"./accounts.ts\";\nimport { EXPIRING_WITHIN_MS } from \"./auth-lane.ts\";\nimport type { Options, SDKUserMessage, SdkQuery, SdkQueryHandle } from \"./sdk-boundary.ts\";\nimport { getSdkBoundary } from \"./sdk-boundary.ts\";\nimport {\n\ttype ClaudeSdkOauthSessionState,\n\ttransitionToClosed,\n\ttransitionToClosing,\n\ttransitionToTainted,\n} from \"./session-registry-state.ts\";\n\nexport const SESSION_REGISTRY_IDLE_TTL_MS = 30 * 60_000;\nexport const SESSION_REGISTRY_MAX_ENTRIES = 32;\n\nexport interface SessionInputController extends AsyncIterable<SDKUserMessage> {\n\tpush(message: SDKUserMessage): void;\n\tclose(): void;\n}\n\nclass StreamingInputController implements SessionInputController, AsyncIterator<SDKUserMessage> {\n\tprivate closed = false;\n\tprivate readonly pending: SDKUserMessage[] = [];\n\tprivate readonly readers: Array<(result: IteratorResult<SDKUserMessage>) => void> = [];\n\n\t[Symbol.asyncIterator](): AsyncIterator<SDKUserMessage> {\n\t\treturn this;\n\t}\n\n\tnext(): Promise<IteratorResult<SDKUserMessage>> {\n\t\tconst message = this.pending.shift();\n\t\tif (message) return Promise.resolve({ value: message, done: false });\n\t\tif (this.closed) return Promise.resolve({ value: undefined, done: true });\n\t\treturn new Promise((resolve) => this.readers.push(resolve));\n\t}\n\n\tpush(message: SDKUserMessage): void {\n\t\tif (this.closed) throw new Error(\"Cannot push to a closed session input controller\");\n\t\tconst reader = this.readers.shift();\n\t\tif (reader) reader({ value: message, done: false });\n\t\telse this.pending.push(message);\n\t}\n\n\tclose(): void {\n\t\tif (this.closed) return;\n\t\tthis.closed = true;\n\t\tfor (const reader of this.readers.splice(0)) reader({ value: undefined, done: true });\n\t}\n}\n\nexport interface SessionRegistryReapHandle {\n\tcancel(): void;\n\tunref(): void;\n}\n\nexport type SessionRegistryBoundary = {\n\tnow: () => number;\n\tqueryFactory: SdkQuery;\n\tscheduleReap: (callback: () => void, delayMs: number) => SessionRegistryReapHandle;\n};\n\nconst defaultSessionRegistryBoundary: SessionRegistryBoundary = {\n\tnow: () => Date.now(),\n\tqueryFactory: (input) => getSdkBoundary().query(input),\n\tscheduleReap: (callback, delayMs) => {\n\t\tconst timer = setTimeout(callback, delayMs);\n\t\treturn { cancel: () => clearTimeout(timer), unref: () => timer.unref() };\n\t},\n};\nlet activeSessionRegistryBoundary = defaultSessionRegistryBoundary;\n\nexport function overrideSessionRegistryBoundary(override: Partial<SessionRegistryBoundary>): void {\n\tactiveSessionRegistryBoundary = { ...defaultSessionRegistryBoundary, ...override };\n}\n\nexport function resetSessionRegistryBoundary(): void {\n\tactiveSessionRegistryBoundary = defaultSessionRegistryBoundary;\n}\n\nexport type SessionBranchInfo = { oldLeafId: string; newLeafId: string };\n\nexport interface ClaudeSdkOauthSessionEntry {\n\tsenpiSessionId: string;\n\tsdkSessionId: string;\n\tgeneration: number;\n\taccountName: string;\n\tmodelId: string;\n\ttoolsetHash: string;\n\tsystemPromptHash: string;\n\tquery: SdkQueryHandle;\n\tinputController: SessionInputController;\n\tpumpTask: Promise<void> | null;\n\tstate: ClaudeSdkOauthSessionState;\n\tactiveTurn: unknown | null;\n\tsyncedPrefixHash: string | null;\n\tsentCount: number;\n\tassistantUuidByIndex: Map<number, string>;\n\tbranchInfo: SessionBranchInfo | null;\n\ttaintedReason: string | null;\n\tlastUsedAt: number;\n}\n\nexport interface CreateSessionRegistryEntryInput {\n\tsenpiSessionId: string;\n\taccountName: string;\n\tmodelId: string;\n\ttoolsetHash: string;\n\tsystemPromptHash: string;\n\toptions: Options;\n}\n\nexport class SessionRegistryResourceLimitError extends Error {\n\treadonly code = \"session_registry_capacity\";\n\n\tconstructor() {\n\t\tsuper(\n\t\t\t`Claude SDK OAuth session registry is at its ${SESSION_REGISTRY_MAX_ENTRIES}-entry limit with no idle session to evict`,\n\t\t);\n\t\tthis.name = \"SessionRegistryResourceLimitError\";\n\t}\n}\n\nexport function createSessionUuid(now = Date.now()): `${string}-${string}-${string}-${string}-${string}` {\n\tconst ts = BigInt(Math.trunc(now));\n\tconst bytes = randomBytes(10);\n\tconst hex = [\n\t\t((ts >> 40n) & 0xffn).toString(16).padStart(2, \"0\"),\n\t\t((ts >> 32n) & 0xffn).toString(16).padStart(2, \"0\"),\n\t\t((ts >> 24n) & 0xffn).toString(16).padStart(2, \"0\"),\n\t\t((ts >> 16n) & 0xffn).toString(16).padStart(2, \"0\"),\n\t\t((ts >> 8n) & 0xffn).toString(16).padStart(2, \"0\"),\n\t\t(ts & 0xffn).toString(16).padStart(2, \"0\"),\n\t\t(0x70 | (bytes[0]! & 0x0f)).toString(16).padStart(2, \"0\"),\n\t\tbytes[1]!.toString(16).padStart(2, \"0\"),\n\t\t(0x80 | (bytes[2]! & 0x3f)).toString(16).padStart(2, \"0\"),\n\t\tbytes[3]!.toString(16).padStart(2, \"0\"),\n\t\t...Array.from(bytes.slice(4), (byte) => byte.toString(16).padStart(2, \"0\")),\n\t].join(\"\");\n\treturn `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20, 32)}`;\n}\n\nfunction evictable(entry: ClaudeSdkOauthSessionEntry): boolean {\n\treturn (entry.state === \"IDLE_SYNCED\" || entry.state === \"TAINTED\") && entry.activeTurn === null;\n}\n\ntype ScheduledReap = { generation: number; token: symbol; handle: SessionRegistryReapHandle };\n\nexport class ClaudeSdkOauthSessionRegistry {\n\tprivate readonly entries = new Map<string, ClaudeSdkOauthSessionEntry>();\n\tprivate readonly generations = new Map<string, number>();\n\tprivate readonly scheduledReaps = new Map<string, ScheduledReap>();\n\n\tget size(): number {\n\t\treturn this.entries.size;\n\t}\n\n\tget(senpiSessionId: string): ClaudeSdkOauthSessionEntry | undefined {\n\t\treturn this.entries.get(senpiSessionId);\n\t}\n\n\tgetOrCreate(input: CreateSessionRegistryEntryInput): ClaudeSdkOauthSessionEntry {\n\t\tconst existing = this.entries.get(input.senpiSessionId);\n\t\tif (existing) {\n\t\t\tconst now = activeSessionRegistryBoundary.now();\n\t\t\tif (!evictable(existing) || now - existing.lastUsedAt < SESSION_REGISTRY_IDLE_TTL_MS) {\n\t\t\t\tthis.touch(existing);\n\t\t\t\treturn existing;\n\t\t\t}\n\t\t\tthis.closeSession(existing.senpiSessionId, \"idle_ttl\");\n\t\t}\n\t\tthis.evictExpired();\n\t\tthis.ensureCapacity();\n\t\tconst now = activeSessionRegistryBoundary.now();\n\t\tconst generation = (this.generations.get(input.senpiSessionId) ?? 0) + 1;\n\t\tconst sdkSessionId = createSessionUuid(now);\n\t\tconst inputController = new StreamingInputController();\n\t\tconst query = activeSessionRegistryBoundary.queryFactory({\n\t\t\tprompt: inputController,\n\t\t\toptions: {\n\t\t\t\t...input.options,\n\t\t\t\tsessionId: sdkSessionId,\n\t\t\t\textraArgs: { ...input.options.extraArgs, \"replay-user-messages\": \"\" },\n\t\t\t},\n\t\t});\n\t\tconst target: ClaudeSdkOauthSessionEntry = {\n\t\t\t...input,\n\t\t\tsdkSessionId,\n\t\t\tgeneration,\n\t\t\tquery,\n\t\t\tinputController,\n\t\t\tpumpTask: null,\n\t\t\tstate: \"STARTING\",\n\t\t\tactiveTurn: null,\n\t\t\tsyncedPrefixHash: null,\n\t\t\tsentCount: 0,\n\t\t\tassistantUuidByIndex: new Map(),\n\t\t\tbranchInfo: null,\n\t\t\ttaintedReason: null,\n\t\t\tlastUsedAt: now,\n\t\t};\n\t\tlet entry!: ClaudeSdkOauthSessionEntry;\n\t\tentry = new Proxy(target, {\n\t\t\tset: (current, property, value) => {\n\t\t\t\tif (this.entries.get(current.senpiSessionId) !== entry) return true;\n\t\t\t\tconst previousTurn = current.activeTurn;\n\t\t\t\tReflect.set(current, property, value);\n\t\t\t\tif (property === \"activeTurn\" && previousTurn !== value) {\n\t\t\t\t\tthis.recordUse(entry, value === null);\n\t\t\t\t}\n\t\t\t\treturn true;\n\t\t\t},\n\t\t});\n\t\tthis.generations.set(input.senpiSessionId, generation);\n\t\tthis.entries.set(input.senpiSessionId, entry);\n\t\treturn entry;\n\t}\n\n\ttouch(entry: ClaudeSdkOauthSessionEntry): void {\n\t\tthis.recordUse(entry, entry.activeTurn === null && evictable(entry));\n\t}\n\n\tcloseSession(senpiSessionId: string, _reason: string): void {\n\t\tconst entry = this.entries.get(senpiSessionId);\n\t\tif (!entry) return;\n\t\tthis.cancelReap(entry);\n\t\ttransitionToClosing(entry);\n\t\tentry.inputController.close();\n\t\ttry {\n\t\t\tentry.query.close();\n\t\t} finally {\n\t\t\ttransitionToClosed(entry);\n\t\t\tthis.cancelReap(entry);\n\t\t\tthis.entries.delete(senpiSessionId);\n\t\t}\n\t}\n\n\tmarkTainted(senpiSessionId: string, reason: string): void {\n\t\tconst entry = this.entries.get(senpiSessionId);\n\t\tif (!entry) return;\n\t\tentry.taintedReason = reason;\n\t\tif (entry.state !== \"TAINTED\") transitionToTainted(entry);\n\t\tthis.touch(entry);\n\t}\n\n\trecordBranchInfo(senpiSessionId: string, info: SessionBranchInfo): void {\n\t\tconst entry = this.entries.get(senpiSessionId);\n\t\tif (!entry) return;\n\t\tentry.branchInfo = { ...info };\n\t\tthis.touch(entry);\n\t}\n\n\tisCurrentGeneration(senpiSessionId: string, generation: number): boolean {\n\t\treturn this.entries.get(senpiSessionId)?.generation === generation;\n\t}\n\n\tevictExpired(): void {\n\t\tconst now = activeSessionRegistryBoundary.now();\n\t\tfor (const entry of [...this.entries.values()]) {\n\t\t\tif (evictable(entry) && now - entry.lastUsedAt >= SESSION_REGISTRY_IDLE_TTL_MS) {\n\t\t\t\tthis.closeSession(entry.senpiSessionId, \"idle_ttl\");\n\t\t\t}\n\t\t}\n\t}\n\n\tprivate recordUse(entry: ClaudeSdkOauthSessionEntry, scheduleReap: boolean): void {\n\t\tif (this.entries.get(entry.senpiSessionId) !== entry) return;\n\t\tentry.lastUsedAt = activeSessionRegistryBoundary.now();\n\t\tthis.cancelReap(entry);\n\t\tif (scheduleReap) this.scheduleReap(entry);\n\t}\n\n\tprivate scheduleReap(entry: ClaudeSdkOauthSessionEntry, delayMs = SESSION_REGISTRY_IDLE_TTL_MS): void {\n\t\tconst token = Symbol(\"session-reap\");\n\t\tconst handle = activeSessionRegistryBoundary.scheduleReap(\n\t\t\t() => this.reap(entry.senpiSessionId, entry.generation, token),\n\t\t\tdelayMs,\n\t\t);\n\t\tthis.scheduledReaps.set(entry.senpiSessionId, { generation: entry.generation, token, handle });\n\t\thandle.unref();\n\t}\n\n\tprivate reap(senpiSessionId: string, generation: number, token: symbol): void {\n\t\tconst scheduled = this.scheduledReaps.get(senpiSessionId);\n\t\tif (!scheduled || scheduled.generation !== generation || scheduled.token !== token) return;\n\t\tthis.scheduledReaps.delete(senpiSessionId);\n\t\tconst entry = this.entries.get(senpiSessionId);\n\t\tif (!entry || entry.generation !== generation || !evictable(entry)) return;\n\t\tconst remaining = entry.lastUsedAt + SESSION_REGISTRY_IDLE_TTL_MS - activeSessionRegistryBoundary.now();\n\t\tif (remaining <= 0) this.closeSession(senpiSessionId, \"idle_ttl\");\n\t\telse this.scheduleReap(entry, remaining);\n\t}\n\n\tprivate cancelReap(entry: ClaudeSdkOauthSessionEntry): void {\n\t\tconst scheduled = this.scheduledReaps.get(entry.senpiSessionId);\n\t\tif (!scheduled || scheduled.generation !== entry.generation) return;\n\t\tscheduled.handle.cancel();\n\t\tthis.scheduledReaps.delete(entry.senpiSessionId);\n\t}\n\n\tprivate ensureCapacity(): void {\n\t\tif (this.entries.size < SESSION_REGISTRY_MAX_ENTRIES) return;\n\t\tconst oldest = [...this.entries.values()]\n\t\t\t.filter(evictable)\n\t\t\t.sort((left, right) => left.lastUsedAt - right.lastUsedAt)[0];\n\t\tif (!oldest) throw new SessionRegistryResourceLimitError();\n\t\tthis.closeSession(oldest.senpiSessionId, \"capacity\");\n\t}\n}\n\nexport const sessionRegistry = new ClaudeSdkOauthSessionRegistry();\n\nexport function getOrCreateSession(input: CreateSessionRegistryEntryInput): ClaudeSdkOauthSessionEntry {\n\treturn sessionRegistry.getOrCreate(input);\n}\n\nexport function getSession(senpiSessionId: string): ClaudeSdkOauthSessionEntry | undefined {\n\treturn sessionRegistry.get(senpiSessionId);\n}\n\nexport function closeSession(senpiSessionId: string, reason: string): void {\n\tsessionRegistry.closeSession(senpiSessionId, reason);\n}\n\nexport function markTainted(senpiSessionId: string, reason: string): void {\n\tsessionRegistry.markTainted(senpiSessionId, reason);\n}\n\nexport function recordBranchInfo(senpiSessionId: string, info: SessionBranchInfo): void {\n\tsessionRegistry.recordBranchInfo(senpiSessionId, info);\n}\n\nexport function isCurrentGeneration(senpiSessionId: string, generation: number): boolean {\n\treturn sessionRegistry.isCurrentGeneration(senpiSessionId, generation);\n}\n\nexport function isBoundAccountTokenExpiring(\n\tentry: Pick<ClaudeSdkOauthSessionEntry, \"accountName\">,\n\taccounts: readonly Pick<AccountSlot, \"name\" | \"expires\" | \"source\">[],\n): boolean {\n\tconst account = accounts.find((candidate) => candidate.name === entry.accountName);\n\treturn account?.source !== \"env\" && account !== undefined\n\t\t? activeSessionRegistryBoundary.now() >= account.expires - EXPIRING_WITHIN_MS\n\t\t: false;\n}\n"]}

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

@@ -0,0 +1,17 @@
+import type { Api, Context, Model, SimpleStreamOptions } from "@earendil-works/pi-ai";
+import { queryWithAuthLane } from "./auth-lane.ts";
+import type { SDKMessage } from "./sdk-boundary.ts";
+import type { ClaudeSdkOauthProviderSettings } from "./settings.ts";
+export type ResidentSessionStreamInput = {
+    model: Model<Api>;
+    context: Context;
+    streamOptions: SimpleStreamOptions;
+    providerSettings: ClaudeSdkOauthProviderSettings;
+    pinnedAccount?: string;
+    buildOptions: Parameters<typeof queryWithAuthLane>[0]["buildOptions"];
+    customToolNameToSdk: ReadonlyMap<string, string>;
+    toolWatchNote?: string;
+    onResumeFallback: (error: unknown) => void;
+};
+export declare function residentSessionMessages(input: ResidentSessionStreamInput): AsyncIterable<SDKMessage>;
+//# sourceMappingURL=session-stream.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"session-stream.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/claude-sdk-oauth/session-stream.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AACtF,OAAO,EAAkC,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAEnF,OAAO,KAAK,EAAE,UAAU,EAAkB,MAAM,mBAAmB,CAAC;AAqBpE,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,eAAe,CAAC;AA6DpE,MAAM,MAAM,0BAA0B,GAAG;IACxC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IAClB,OAAO,EAAE,OAAO,CAAC;IACjB,aAAa,EAAE,mBAAmB,CAAC;IACnC,gBAAgB,EAAE,8BAA8B,CAAC;IACjD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,YAAY,EAAE,UAAU,CAAC,OAAO,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC;IACtE,mBAAmB,EAAE,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CAC3C,CAAC;AA4JF,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,0BAA0B,GAAG,aAAa,CAAC,UAAU,CAAC,CAUpG"}