@code-yeongyu/senpi 2026.8.1 → 2026.8.3

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

@@ -1 +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"]}
+{"version":3,"file":"session-registry-pump.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/claude-sdk-oauth/session-registry-pump.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAGN,iBAAiB,GACjB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACN,sBAAsB,EACtB,0BAA0B,EAC1B,oBAAoB,EACpB,yBAAyB,EACzB,uBAAuB,GACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,kBAAkB,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAKjH,OAAO,EAAE,2BAA2B,EAAE,MAAM,yBAAyB,CAAC;AAEtE,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;AASjD,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,SAAS,WAAW,CAAC,KAAiC;IACrD,OAAO,KAAK,CAAC,UAA+B,CAAC;AAC9C,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;;;;;GAKG;AACH,SAAS,SAAS,CACjB,QAAuC,EACvC,KAAiC,EACjC,IAAgB,EAChB,MAAa;IAEb,IAAI,WAAW,CAAC,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,CAAC,cAAc,EAAE,IAAI,CAAC,UAAU,CAAC;QAAE,OAAO;IAChH,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAC1B,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC;IACxB,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;IAC/D,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;AAC3E,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,CAAC,IAAI,CAAC,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,MAAM;QAAE,sBAAsB,CAAC,KAAK,CAAC,CAAC;;QACtG,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;IACpE,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,0EAA0E;IAC1E,0EAA0E;IAC1E,yEAAyE;IACzE,8BAA8B;IAC9B,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,OAAO,KAAK,MAAM,IAAI,OAAO,OAAO,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QACvG,IAAI,OAAO,CAAC,UAAU,KAAK,KAAK,CAAC,YAAY;YAAE,KAAK,CAAC,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC;IACxF,CAAC;IACD,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;iBACd,SAAS,EAAE;iBACX,IAAI,CAAC,CAAC,OAAgB,EAAE,EAAE;gBAC1B,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC;YACjC,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBACzB,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;QACL,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 type { SDKMessage, SDKUserMessage } from \"./sdk-boundary.ts\";\nimport { evaluateAbortOutcome } from \"./session-reattach.ts\";\nimport {\n\ttype ClaudeSdkOauthSessionEntry,\n\ttype ClaudeSdkOauthSessionRegistry,\n\tcreateSessionUuid,\n} from \"./session-registry.ts\";\nimport {\n\ttransitionToIdleSynced,\n\ttransitionToTurnResultSeen,\n\ttransitionToTurnSent,\n\ttransitionToTurnStreaming,\n\ttransitionToTurnWaiting,\n} from \"./session-registry-state.ts\";\nimport { bufferBeforeReplay, claimTurn, deliver, isReplayFor, resultMatchesTurn } from \"./session-turn-claim.ts\";\nimport type { ActiveTurn, PreReplayBufferLimits, SessionTurnResult } from \"./session-turn-types.ts\";\n\nexport type { SessionTurnResult } from \"./session-turn-types.ts\";\n\nimport { SessionTurnAttributionError } from \"./session-turn-types.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 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\nfunction currentTurn(entry: ClaudeSdkOauthSessionEntry): ActiveTurn | null {\n\treturn entry.activeTurn as ActiveTurn | null;\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\n/**\n * An interrupt that never produces a terminal result still ends the user's turn:\n * settle it as aborted and close only the query, so the binding survives and the\n * next turn reattaches. Rejecting here would drop the turn's continuity\n * observation and hand the following turn two.\n */\nfunction abortTurn(\n\tregistry: ClaudeSdkOauthSessionRegistry,\n\tentry: ClaudeSdkOauthSessionEntry,\n\tturn: ActiveTurn,\n\t_error: Error,\n): void {\n\tif (currentTurn(entry) !== turn || !registry.isCurrentGeneration(entry.senpiSessionId, turn.generation)) return;\n\tremoveAbortListener(turn);\n\tentry.activeTurn = null;\n\tregistry.closeSession(entry.senpiSessionId, \"abort_uncertain\");\n\tturn.resolve({ uuid: turn.uuid, messages: turn.messages, aborted: true });\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 || evaluateAbortOutcome(turn.interruptReceipt) === \"keep\") transitionToIdleSynced(entry);\n\telse registry.closeSession(entry.senpiSessionId, \"abort_uncertain\");\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\t// A forked query mints a NEW session id (forkSession: true + resume): the\n\t// init message carries it, and it must be persisted BEFORE any turn-state\n\t// guard — otherwise subsequent reattach targets the original session and\n\t// the fork's content is lost.\n\tif (message.type === \"system\" && message.subtype === \"init\" && typeof message.session_id === \"string\") {\n\t\tif (message.session_id !== entry.sdkSessionId) entry.sdkSessionId = message.session_id;\n\t}\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\n\t\t\t\t.interrupt()\n\t\t\t\t.then((receipt: unknown) => {\n\t\t\t\t\tturn.interruptReceipt = receipt;\n\t\t\t\t})\n\t\t\t\t.catch((error: unknown) => {\n\t\t\t\t\tconst detail = error instanceof Error ? error.message : String(error);\n\t\t\t\t\tabortTurn(registry, entry, turn, new Error(`Claude SDK OAuth query interrupt failed: ${detail}`));\n\t\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-wiring.d.ts.map

@@ -1 +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"}
+{"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;AA0BnD,wBAAgB,uBAAuB,CAAC,EAAE,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,GAAG,IAAI,CAqD1E"}

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

@@ -1,87 +1,26 @@
 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;
+import { AssistantCommitBoundary, isResidentAssistant, isTerminalFailure } from "./session-commit-boundary.js";
+import { bindingFromEntry, rememberBinding } from "./session-reattach.js";
+import { closeSession, getSession, recordBranchInfo, recordPendingFork, switchSessionModel, } from "./session-registry.js";
+const commitBoundary = new AssistantCommitBoundary();
+function keepBindingThenClose(sessionId, reason) {
+    const entry = getSession(sessionId);
+    if (entry)
+        rememberBinding(bindingFromEntry(entry, []));
+    closeSession(sessionId, reason);
 }
-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");
+function residentEntryFor(sessionId, message) {
+    const entry = getSession(sessionId);
+    if (!entry || !isResidentAssistant(message, entry.modelId))
+        return undefined;
+    return entry;
 }
 export function registerSessionRegistry(pi) {
-    installAssistantProvenanceHooks(assistantProvenanceHooks);
     pi.on("session_compact", (_event, ctx) => {
-        markTainted(ctx.sessionManager.getSessionId(), "compaction");
+        recordPendingFork(ctx.sessionManager.getSessionId(), "compaction");
     });
     pi.on("session_before_fork", (_event, ctx) => {
-        markTainted(ctx.sessionManager.getSessionId(), "fork");
+        recordPendingFork(ctx.sessionManager.getSessionId(), "fork");
     });
     pi.on("session_tree", (event, ctx) => {
         if (event.oldLeafId === null || event.newLeafId === null)
@@ -91,26 +30,25 @@ export function registerSessionRegistry(pi) {
             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")
+    pi.on("model_select", async (event, ctx) => {
+        const sessionId = ctx.sessionManager.getSessionId();
+        if (event.model?.provider !== CLAUDE_SDK_OAUTH_PROVIDER_ID) {
+            closeSession(sessionId, "model_selected");
             return;
-        const entry = getSession(ctx.sessionManager.getSessionId());
-        if (entry && isResidentAssistant(event.message, entry.modelId)) {
-            stageAssistantProvenance(entry, event.message);
         }
+        if (!(await switchSessionModel(sessionId, event.model.id))) {
+            keepBindingThenClose(sessionId, "model_selected");
+        }
+    });
+    pi.on("thinking_level_select", (_event, ctx) => {
+        keepBindingThenClose(ctx.sessionManager.getSessionId(), "thinking_level_selected");
     });
     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);
+        const sessionId = ctx.sessionManager.getSessionId();
+        if (residentEntryFor(sessionId, event.message)) {
+            commitBoundary.captureProviderFinal(sessionId, event.message);
         }
     });
     pi.on("message_end", (event, ctx) => {
@@ -120,8 +58,12 @@ export function registerSessionRegistry(pi) {
         const entry = getSession(sessionId);
         if (!entry)
             return;
-        if (!isResidentAssistant(event.message, entry.modelId) || !commitAssistantProvenance(entry, event.message)) {
-            markTainted(sessionId, "assistant_provenance_unverified");
+        if (isTerminalFailure(event.message)) {
+            commitBoundary.forget(sessionId);
+            return;
+        }
+        if (commitBoundary.commit(sessionId, event.message, entry.modelId) === "rewritten") {
+            recordPendingFork(sessionId, "assistant_rewritten");
         }
     });
     pi.on("session_shutdown", (event, ctx) => {

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

@@ -1 +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"]}
+{"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,EAAE,uBAAuB,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAC/G,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAC1E,OAAO,EACN,YAAY,EACZ,UAAU,EACV,gBAAgB,EAChB,iBAAiB,EACjB,kBAAkB,GAClB,MAAM,uBAAuB,CAAC;AAE/B,MAAM,cAAc,GAAG,IAAI,uBAAuB,EAAE,CAAC;AAErD,SAAS,oBAAoB,CAAC,SAAiB,EAAE,MAAc;IAC9D,MAAM,KAAK,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IACpC,IAAI,KAAK;QAAE,eAAe,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC;IACxD,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,gBAAgB,CAAC,SAAiB,EAAE,OAAyB;IACrE,MAAM,KAAK,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IACpC,IAAI,CAAC,KAAK,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC;QAAE,OAAO,SAAS,CAAC;IAC7E,OAAO,KAAK,CAAC;AACd,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,EAA4B;IACnE,EAAE,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QACxC,iBAAiB,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE,YAAY,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,qBAAqB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QAC5C,iBAAiB,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE,MAAM,CAAC,CAAC;IAC9D,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,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;QAC1C,MAAM,SAAS,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QACpD,IAAI,KAAK,CAAC,KAAK,EAAE,QAAQ,KAAK,4BAA4B,EAAE,CAAC;YAC5D,YAAY,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;YAC1C,OAAO;QACR,CAAC;QACD,IAAI,CAAC,CAAC,MAAM,kBAAkB,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YAC5D,oBAAoB,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;QACnD,CAAC;IACF,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,EAAE,CAAC,uBAAuB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QAC9C,oBAAoB,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,EAAE,yBAAyB,CAAC,CAAC;IACpF,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,SAAS,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QACpD,IAAI,gBAAgB,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YAChD,cAAc,CAAC,oBAAoB,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC/D,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,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YACtC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YACjC,OAAO;QACR,CAAC;QACD,IAAI,cAAc,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,WAAW,EAAE,CAAC;YACpF,iBAAiB,CAAC,SAAS,EAAE,qBAAqB,CAAC,CAAC;QACrD,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 { AssistantCommitBoundary, isResidentAssistant, isTerminalFailure } from \"./session-commit-boundary.ts\";\nimport { bindingFromEntry, rememberBinding } from \"./session-reattach.ts\";\nimport {\n\tcloseSession,\n\tgetSession,\n\trecordBranchInfo,\n\trecordPendingFork,\n\tswitchSessionModel,\n} from \"./session-registry.ts\";\n\nconst commitBoundary = new AssistantCommitBoundary();\n\nfunction keepBindingThenClose(sessionId: string, reason: string): void {\n\tconst entry = getSession(sessionId);\n\tif (entry) rememberBinding(bindingFromEntry(entry, []));\n\tcloseSession(sessionId, reason);\n}\n\nfunction residentEntryFor(sessionId: string, message: AssistantMessage) {\n\tconst entry = getSession(sessionId);\n\tif (!entry || !isResidentAssistant(message, entry.modelId)) return undefined;\n\treturn entry;\n}\n\nexport function registerSessionRegistry(pi: Pick<ExtensionAPI, \"on\">): void {\n\tpi.on(\"session_compact\", (_event, ctx) => {\n\t\trecordPendingFork(ctx.sessionManager.getSessionId(), \"compaction\");\n\t});\n\tpi.on(\"session_before_fork\", (_event, ctx) => {\n\t\trecordPendingFork(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\", async (event, ctx) => {\n\t\tconst sessionId = ctx.sessionManager.getSessionId();\n\t\tif (event.model?.provider !== CLAUDE_SDK_OAUTH_PROVIDER_ID) {\n\t\t\tcloseSession(sessionId, \"model_selected\");\n\t\t\treturn;\n\t\t}\n\t\tif (!(await switchSessionModel(sessionId, event.model.id))) {\n\t\t\tkeepBindingThenClose(sessionId, \"model_selected\");\n\t\t}\n\t});\n\tpi.on(\"thinking_level_select\", (_event, ctx) => {\n\t\tkeepBindingThenClose(ctx.sessionManager.getSessionId(), \"thinking_level_selected\");\n\t});\n\tpi.on(\"message_update\", (event, ctx) => {\n\t\tif (event.message.role !== \"assistant\") return;\n\t\tconst sessionId = ctx.sessionManager.getSessionId();\n\t\tif (residentEntryFor(sessionId, event.message)) {\n\t\t\tcommitBoundary.captureProviderFinal(sessionId, 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 (isTerminalFailure(event.message)) {\n\t\t\tcommitBoundary.forget(sessionId);\n\t\t\treturn;\n\t\t}\n\t\tif (commitBoundary.commit(sessionId, event.message, entry.modelId) === \"rewritten\") {\n\t\t\trecordPendingFork(sessionId, \"assistant_rewritten\");\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

@@ -1,5 +1,6 @@
 import type { AccountSlot } from "./accounts.ts";
 import type { Options, SDKUserMessage, SdkQuery, SdkQueryHandle } from "./sdk-boundary.ts";
+import { type SessionRegistryReapHandle } from "./session-reaper.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;
@@ -7,10 +8,6 @@ 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;
@@ -18,6 +15,7 @@ export type SessionRegistryBoundary = {
 };
 export declare function overrideSessionRegistryBoundary(override: Partial<SessionRegistryBoundary>): void;
 export declare function resetSessionRegistryBoundary(): void;
+export type { SessionRegistryReapHandle };
 export type SessionBranchInfo = {
     oldLeafId: string;
     newLeafId: string;
@@ -40,6 +38,8 @@ export interface ClaudeSdkOauthSessionEntry {
     assistantUuidByIndex: Map<number, string>;
     branchInfo: SessionBranchInfo | null;
     taintedReason: string | null;
+    /** Wave C seam: a divergence boundary recorded for the next continuity decision. */
+    pendingForkReason: string | null;
     lastUsedAt: number;
 }
 export interface CreateSessionRegistryEntryInput {
@@ -49,6 +49,10 @@ export interface CreateSessionRegistryEntryInput {
     toolsetHash: string;
     systemPromptHash: string;
     options: Options;
+    resume?: {
+        sdkSessionId: string;
+        atUuid?: string;
+    };
 }
 export declare class SessionRegistryResourceLimitError extends Error {
     readonly code = "session_registry_capacity";
@@ -58,27 +62,25 @@ export declare function createSessionUuid(now?: number): `${string}-${string}-${
 export declare class ClaudeSdkOauthSessionRegistry {
     private readonly entries;
     private readonly generations;
-    private readonly scheduledReaps;
+    private readonly reaper;
     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;
+    closeSession(senpiSessionId: string, reason: string): 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 isIdleExpired(entry: Pick<ClaudeSdkOauthSessionEntry, "lastUsedAt">): boolean;
 export declare function markTainted(senpiSessionId: string, reason: string): void;
+export declare function switchSessionModel(senpiSessionId: string, modelId: string): Promise<boolean>;
+export declare function recordPendingFork(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;

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

@@ -1 +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"}
+{"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;AAS3F,OAAO,EAAwB,KAAK,yBAAyB,EAAE,MAAM,qBAAqB,CAAC;AAC3F,OAAO,EAAE,KAAK,0BAA0B,EAA2C,MAAM,6BAA6B,CAAC;AAEvH,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,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,YAAY,EAAE,yBAAyB,EAAE,CAAC;AAE1C,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,oFAAoF;IACpF,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,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;IACjB,MAAM,CAAC,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CACnD;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;AAMD,qBAAa,6BAA6B;IACzC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiD;IACzE,OAAO,CAAC,QAAQ,CAAC,WAAW,CAA6B;IACzD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAUpB;IAEH,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,CA+D9E;IAED,KAAK,CAAC,KAAK,EAAE,0BAA0B,GAAG,IAAI,CAE7C;IAED,YAAY,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAczD;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,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,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,0BAA0B,EAAE,YAAY,CAAC,GAAG,OAAO,CAE5F;AAED,wBAAgB,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAExE;AAED,wBAAsB,kBAAkB,CAAC,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAElG;AAED,wBAAgB,iBAAiB,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAE9E;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

@@ -1,7 +1,10 @@
 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";
+import { annotateBranchInfo, annotatePendingFork, annotateTainted, switchEntryModel, } from "./session-entry-annotations.js";
+import { recordPendingCloseCause } from "./session-observability.js";
+import { SessionReapScheduler } from "./session-reaper.js";
+import { transitionToClosed, transitionToClosing } from "./session-registry-state.js";
 export const SESSION_REGISTRY_IDLE_TTL_MS = 30 * 60_000;
 export const SESSION_REGISTRY_MAX_ENTRIES = 32;
 class StreamingInputController {
@@ -85,7 +88,18 @@ export class ClaudeSdkOauthSessionRegistry {
     constructor() {
         this.entries = new Map();
         this.generations = new Map();
-        this.scheduledReaps = new Map();
+        this.reaper = new SessionReapScheduler({
+            now: () => activeSessionRegistryBoundary.now(),
+            scheduleTimer: (callback, delayMs) => activeSessionRegistryBoundary.scheduleReap(callback, delayMs),
+            idleTtlMs: SESSION_REGISTRY_IDLE_TTL_MS,
+            candidate: (senpiSessionId, generation) => {
+                const entry = this.entries.get(senpiSessionId);
+                if (!entry || entry.generation !== generation || !evictable(entry))
+                    return undefined;
+                return { generation: entry.generation, lastUsedAt: entry.lastUsedAt };
+            },
+            expire: (senpiSessionId) => this.closeSession(senpiSessionId, "idle_ttl"),
+        });
     }
     get size() {
         return this.entries.size;
@@ -107,18 +121,25 @@ export class ClaudeSdkOauthSessionRegistry {
         this.ensureCapacity();
         const now = activeSessionRegistryBoundary.now();
         const generation = (this.generations.get(input.senpiSessionId) ?? 0) + 1;
-        const sdkSessionId = createSessionUuid(now);
+        const sdkSessionId = input.resume?.sdkSessionId ?? createSessionUuid(now);
         const inputController = new StreamingInputController();
+        const lineageOptions = input.resume
+            ? {
+                resume: input.resume.sdkSessionId,
+                ...(input.resume.atUuid ? { resumeSessionAt: input.resume.atUuid, forkSession: true } : {}),
+            }
+            : { sessionId: sdkSessionId };
         const query = activeSessionRegistryBoundary.queryFactory({
             prompt: inputController,
             options: {
                 ...input.options,
-                sessionId: sdkSessionId,
+                ...lineageOptions,
                 extraArgs: { ...input.options.extraArgs, "replay-user-messages": "" },
             },
         });
+        const { resume: _resume, ...entryInput } = input;
         const target = {
-            ...input,
+            ...entryInput,
             sdkSessionId,
             generation,
             query,
@@ -131,6 +152,7 @@ export class ClaudeSdkOauthSessionRegistry {
             assistantUuidByIndex: new Map(),
             branchInfo: null,
             taintedReason: null,
+            pendingForkReason: null,
             lastUsedAt: now,
         };
         let entry;
@@ -153,11 +175,12 @@ export class ClaudeSdkOauthSessionRegistry {
     touch(entry) {
         this.recordUse(entry, entry.activeTurn === null && evictable(entry));
     }
-    closeSession(senpiSessionId, _reason) {
+    closeSession(senpiSessionId, reason) {
         const entry = this.entries.get(senpiSessionId);
         if (!entry)
             return;
-        this.cancelReap(entry);
+        recordPendingCloseCause(senpiSessionId, reason);
+        this.reaper.cancel(senpiSessionId, entry.generation);
         transitionToClosing(entry);
         entry.inputController.close();
         try {
@@ -165,26 +188,10 @@ export class ClaudeSdkOauthSessionRegistry {
         }
         finally {
             transitionToClosed(entry);
-            this.cancelReap(entry);
+            this.reaper.cancel(senpiSessionId, entry.generation);
             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;
     }
@@ -200,36 +207,9 @@ export class ClaudeSdkOauthSessionRegistry {
         if (this.entries.get(entry.senpiSessionId) !== entry)
             return;
         entry.lastUsedAt = activeSessionRegistryBoundary.now();
-        this.cancelReap(entry);
+        this.reaper.cancel(entry.senpiSessionId, entry.generation);
         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);
+            this.reaper.arm(entry.senpiSessionId, entry.generation);
     }
     ensureCapacity() {
         if (this.entries.size < SESSION_REGISTRY_MAX_ENTRIES)
@@ -252,11 +232,20 @@ export function getSession(senpiSessionId) {
 export function closeSession(senpiSessionId, reason) {
     sessionRegistry.closeSession(senpiSessionId, reason);
 }
+export function isIdleExpired(entry) {
+    return activeSessionRegistryBoundary.now() - entry.lastUsedAt >= SESSION_REGISTRY_IDLE_TTL_MS;
+}
 export function markTainted(senpiSessionId, reason) {
-    sessionRegistry.markTainted(senpiSessionId, reason);
+    annotateTainted(sessionRegistry, senpiSessionId, reason);
+}
+export async function switchSessionModel(senpiSessionId, modelId) {
+    return switchEntryModel(sessionRegistry, senpiSessionId, modelId);
+}
+export function recordPendingFork(senpiSessionId, reason) {
+    annotatePendingFork(sessionRegistry, senpiSessionId, reason);
 }
 export function recordBranchInfo(senpiSessionId, info) {
-    sessionRegistry.recordBranchInfo(senpiSessionId, info);
+    annotateBranchInfo(sessionRegistry, senpiSessionId, info);
 }
 export function isCurrentGeneration(senpiSessionId, generation) {
     return sessionRegistry.isCurrentGeneration(senpiSessionId, generation);