@code-yeongyu/senpi 2026.6.21 → 2026.6.30

package/dist/core/extensions/builtin/pi-codex-app-server/server-request-bridge.js

@@ -0,0 +1,169 @@
+import { createAdapterJsonRpcError } from "./error-mapper.js";
+import { createOpaqueServerRequestEnvelope, readAppServerRequestIds, readRequestId, readSecretQuestionIds, redactSecretAnswers, } from "./server-request-fields.js";
+export function createServerRequestBridge(options) {
+    return new DefaultServerRequestBridge(options);
+}
+class DefaultServerRequestBridge {
+    constructor(options) {
+        this.callbacksByExternalId = new Map();
+        this.externalCallbackIdByAppRequestId = new Map();
+        this.sequence = 0;
+        this.connectionId = options.connectionId;
+        this.capabilityFlags = options.capabilityFlags;
+        this.callbackTimeoutMs = options.callbackTimeoutMs;
+        this.nowMs = options.nowMs;
+        this.idMapper = options.idMapper;
+        this.sessionRegistry = options.sessionRegistry;
+        this.callbackClient = options.callbackClient;
+    }
+    deliver(input) {
+        if (!isSupportedCallbackMethod(input.method)) {
+            return invalidCallback(`Unsupported PR-008/PR-009 callback method: ${input.method}`);
+        }
+        const appRequestId = String(input.requestId);
+        const ids = readAppServerRequestIds(input.params);
+        const externalCallbackId = `callback-${appRequestId}`;
+        const timeoutAtMs = this.nowMs() + this.callbackTimeoutMs;
+        const registration = this.idMapper.registerServerRequest({
+            appRequestId,
+            externalCallbackId,
+            appThreadId: ids.appThreadId,
+            appTurnId: ids.appTurnId,
+            appItemId: ids.appItemId,
+            method: input.method,
+        });
+        if (registration.kind === "rejected")
+            return { kind: "adapter-error", error: registration.error };
+        const state = {
+            appRequestId,
+            externalCallbackId,
+            timeoutAtMs,
+            secretQuestionIds: readSecretQuestionIds(input.params),
+            request: {
+                kind: "opaque-request",
+                method: "appServer/request",
+                externalCallbackId,
+                timeoutAtMs,
+                envelope: createOpaqueServerRequestEnvelope({
+                    connectionId: this.connectionId,
+                    capabilityFlags: this.capabilityFlags,
+                    externalCallbackId,
+                    appRequestId,
+                    method: input.method,
+                    params: input.params,
+                    ids,
+                    sequence: this.nextSequence(),
+                    sessionRegistry: this.sessionRegistry,
+                }),
+            },
+            status: "pending",
+        };
+        this.callbacksByExternalId.set(externalCallbackId, state);
+        this.externalCallbackIdByAppRequestId.set(appRequestId, externalCallbackId);
+        return {
+            kind: "delivered",
+            request: state.request,
+        };
+    }
+    async respond(input) {
+        const state = this.readPendingCallback(input.externalCallbackId);
+        if (state.kind === "adapter-error")
+            return state;
+        await this.callbackClient.respond(state.callback.appRequestId, input.response);
+        state.callback.status = "forwarded";
+        return { kind: "forwarded" };
+    }
+    async reject(input) {
+        const state = this.readPendingCallback(input.externalCallbackId);
+        if (state.kind === "adapter-error")
+            return state;
+        await this.callbackClient.reject(state.callback.appRequestId, input.reason);
+        state.callback.status = "forwarded";
+        return { kind: "forwarded" };
+    }
+    replayPendingCallbacks() {
+        return [...this.callbacksByExternalId.values()]
+            .filter((state) => state.status === "pending")
+            .map((state) => state.request);
+    }
+    async rejectPendingCallbacks(reason) {
+        const rejected = [];
+        for (const state of this.callbacksByExternalId.values()) {
+            if (state.status !== "pending")
+                continue;
+            await this.callbackClient.reject(state.appRequestId, reason);
+            this.clearCallbackState(state);
+            rejected.push({ appRequestId: state.appRequestId, externalCallbackId: state.externalCallbackId });
+        }
+        return rejected;
+    }
+    async rejectTimedOutCallbacks(nowMs = this.nowMs()) {
+        const timedOut = [];
+        for (const state of this.callbacksByExternalId.values()) {
+            if ((state.status !== "pending" && state.status !== "timed-out") || state.timeoutAtMs > nowMs)
+                continue;
+            state.status = "timed-out";
+            await this.callbackClient.reject(state.appRequestId, "callback timed out");
+            this.clearCallbackState(state);
+            timedOut.push({ appRequestId: state.appRequestId, externalCallbackId: state.externalCallbackId });
+        }
+        return timedOut;
+    }
+    resolveFromNotification(input) {
+        if (input.method !== "serverRequest/resolved")
+            return { kind: "ignored" };
+        const appRequestId = readRequestId(input.params);
+        if (!appRequestId)
+            return { kind: "ignored" };
+        const externalCallbackId = this.externalCallbackIdByAppRequestId.get(appRequestId);
+        if (externalCallbackId) {
+            const state = this.callbacksByExternalId.get(externalCallbackId);
+            if (state)
+                this.clearCallbackState(state);
+        }
+        else {
+            this.idMapper.resolveServerRequest(appRequestId);
+        }
+        return { kind: "resolved", appRequestId };
+    }
+    redactCallbackResponse(externalCallbackId, response) {
+        const state = this.callbacksByExternalId.get(externalCallbackId);
+        if (!state)
+            return response;
+        return redactSecretAnswers(response, state.secretQuestionIds);
+    }
+    nextSequence() {
+        this.sequence += 1;
+        return this.sequence;
+    }
+    readPendingCallback(externalCallbackId) {
+        const callback = this.callbacksByExternalId.get(externalCallbackId);
+        if (callback?.status !== "pending") {
+            return invalidCallback(`Unknown, duplicate, or timed-out callback id: ${externalCallbackId}`);
+        }
+        return { kind: "callback", callback };
+    }
+    clearCallbackState(state) {
+        this.callbacksByExternalId.delete(state.externalCallbackId);
+        this.externalCallbackIdByAppRequestId.delete(state.appRequestId);
+        this.idMapper.resolveServerRequest(state.appRequestId);
+    }
+}
+const SUPPORTED_CALLBACK_METHODS = new Set([
+    "item/commandExecution/requestApproval",
+    "item/fileChange/requestApproval",
+    "item/tool/requestUserInput",
+    "mcpServer/elicitation/request",
+    "item/permissions/requestApproval",
+    "item/tool/call",
+]);
+function isSupportedCallbackMethod(method) {
+    return SUPPORTED_CALLBACK_METHODS.has(method);
+}
+function invalidCallback(message) {
+    return {
+        kind: "adapter-error",
+        error: createAdapterJsonRpcError({ adapterCode: "invalid-callback-state", message }),
+    };
+}
+//# sourceMappingURL=server-request-bridge.js.map

package/dist/core/extensions/builtin/pi-codex-app-server/server-request-bridge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server-request-bridge.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/pi-codex-app-server/server-request-bridge.ts"],"names":[],"mappings":"AAAA,OAAO,EAA4B,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAGxF,OAAO,EACN,iCAAiC,EACjC,uBAAuB,EACvB,aAAa,EACb,qBAAqB,EACrB,mBAAmB,GACnB,MAAM,4BAA4B,CAAC;AAsFpC,MAAM,UAAU,yBAAyB,CAAC,OAAmC;IAC5E,OAAO,IAAI,0BAA0B,CAAC,OAAO,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,0BAA0B;IAY/B,YAAY,OAAmC;QAJ9B,0BAAqB,GAAG,IAAI,GAAG,EAAyB,CAAC;QACzD,qCAAgC,GAAG,IAAI,GAAG,EAAkB,CAAC;QACtE,aAAQ,GAAG,CAAC,CAAC;QAGpB,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;QAC/C,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;QAC/C,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;IAC9C,CAAC;IAED,OAAO,CAAC,KAA4B;QACnC,IAAI,CAAC,yBAAyB,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,OAAO,eAAe,CAAC,8CAA8C,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QACtF,CAAC;QACD,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAC7C,MAAM,GAAG,GAAG,uBAAuB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAClD,MAAM,kBAAkB,GAAG,YAAY,YAAY,EAAE,CAAC;QACtD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAC1D,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC;YACxD,YAAY;YACZ,kBAAkB;YAClB,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,MAAM,EAAE,KAAK,CAAC,MAAM;SACpB,CAAC,CAAC;QACH,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,YAAY,CAAC,KAAK,EAAE,CAAC;QAElG,MAAM,KAAK,GAAkB;YAC5B,YAAY;YACZ,kBAAkB;YAClB,WAAW;YACX,iBAAiB,EAAE,qBAAqB,CAAC,KAAK,CAAC,MAAM,CAAC;YACtD,OAAO,EAAE;gBACR,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,mBAAmB;gBAC3B,kBAAkB;gBAClB,WAAW;gBACX,QAAQ,EAAE,iCAAiC,CAAC;oBAC3C,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,eAAe,EAAE,IAAI,CAAC,eAAe;oBACrC,kBAAkB;oBAClB,YAAY;oBACZ,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,GAAG;oBACH,QAAQ,EAAE,IAAI,CAAC,YAAY,EAAE;oBAC7B,eAAe,EAAE,IAAI,CAAC,eAAe;iBACrC,CAAC;aACF;YACD,MAAM,EAAE,SAAS;SACjB,CAAC;QACF,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;QAC1D,IAAI,CAAC,gCAAgC,CAAC,GAAG,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC;QAC5E,OAAO;YACN,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,KAAK,CAAC,OAAO;SACtB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,KAA4B;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACjE,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe;YAAE,OAAO,KAAK,CAAC;QACjD,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC/E,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,WAAW,CAAC;QACpC,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAA6B;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACjE,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe;YAAE,OAAO,KAAK,CAAC;QACjD,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5E,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,WAAW,CAAC;QACpC,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;IAC9B,CAAC;IAED,sBAAsB;QACrB,OAAO,CAAC,GAAG,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,CAAC;aAC7C,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,CAAC;aAC7C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,MAAc;QAC1C,MAAM,QAAQ,GAAuB,EAAE,CAAC;QACxC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YACzD,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS;gBAAE,SAAS;YACzC,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YAC7D,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC/B,QAAQ,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,kBAAkB,EAAE,KAAK,CAAC,kBAAkB,EAAE,CAAC,CAAC;QACnG,CAAC;QACD,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,uBAAuB,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE;QACjD,MAAM,QAAQ,GAAuB,EAAE,CAAC;QACxC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YACzD,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW,CAAC,IAAI,KAAK,CAAC,WAAW,GAAG,KAAK;gBAAE,SAAS;YACxG,KAAK,CAAC,MAAM,GAAG,WAAW,CAAC;YAC3B,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,oBAAoB,CAAC,CAAC;YAC3E,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC/B,QAAQ,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,kBAAkB,EAAE,KAAK,CAAC,kBAAkB,EAAE,CAAC,CAAC;QACnG,CAAC;QACD,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED,uBAAuB,CAAC,KAAiC;QACxD,IAAI,KAAK,CAAC,MAAM,KAAK,wBAAwB;YAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QAC1E,MAAM,YAAY,GAAG,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACjD,IAAI,CAAC,YAAY;YAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QAC9C,MAAM,kBAAkB,GAAG,IAAI,CAAC,gCAAgC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACnF,IAAI,kBAAkB,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;YACjE,IAAI,KAAK;gBAAE,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC;IAC3C,CAAC;IAED,sBAAsB,CAAC,kBAA0B,EAAE,QAAiB;QACnE,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QACjE,IAAI,CAAC,KAAK;YAAE,OAAO,QAAQ,CAAC;QAC5B,OAAO,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,iBAAiB,CAAC,CAAC;IAC/D,CAAC;IAEO,YAAY;QACnB,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC;IACtB,CAAC;IAEO,mBAAmB,CAC1B,kBAA0B;QAI1B,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QACpE,IAAI,QAAQ,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;YACpC,OAAO,eAAe,CAAC,iDAAiD,kBAAkB,EAAE,CAAC,CAAC;QAC/F,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;IACvC,CAAC;IAEO,kBAAkB,CAAC,KAAoB;QAC9C,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QAC5D,IAAI,CAAC,gCAAgC,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACjE,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACxD,CAAC;CACD;AAED,MAAM,0BAA0B,GAAwB,IAAI,GAAG,CAAC;IAC/D,uCAAuC;IACvC,iCAAiC;IACjC,4BAA4B;IAC5B,+BAA+B;IAC/B,kCAAkC;IAClC,gBAAgB;CACY,CAAC,CAAC;AAE/B,SAAS,yBAAyB,CAAC,MAAc;IAChD,OAAO,0BAA0B,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,eAAe,CAAC,OAAe;IACvC,OAAO;QACN,IAAI,EAAE,eAAe;QACrB,KAAK,EAAE,yBAAyB,CAAC,EAAE,WAAW,EAAE,wBAAwB,EAAE,OAAO,EAAE,CAAC;KACpF,CAAC;AACH,CAAC","sourcesContent":["import { type AdapterJsonRpcError, createAdapterJsonRpcError } from \"./error-mapper.ts\";\nimport type { IdMapper } from \"./id-mapper.ts\";\nimport type { OpaqueAppServerEnvelope } from \"./notification-projector.ts\";\nimport {\n\tcreateOpaqueServerRequestEnvelope,\n\treadAppServerRequestIds,\n\treadRequestId,\n\treadSecretQuestionIds,\n\tredactSecretAnswers,\n} from \"./server-request-fields.ts\";\nimport type { SessionRegistry } from \"./session-registry.ts\";\n\nexport interface AppServerCallbackClient {\n\trespond(appRequestId: string, response: unknown): Promise<void>;\n\treject(appRequestId: string, reason: string): Promise<void>;\n}\n\nexport interface ServerRequestBridgeOptions {\n\treadonly connectionId: string;\n\treadonly capabilityFlags: readonly string[];\n\treadonly callbackTimeoutMs: number;\n\treadonly nowMs: () => number;\n\treadonly idMapper: IdMapper;\n\treadonly sessionRegistry: SessionRegistry;\n\treadonly callbackClient: AppServerCallbackClient;\n}\n\nexport interface AppServerRequestInput {\n\treadonly method: string;\n\treadonly requestId: string | number;\n\treadonly params: unknown;\n}\n\nexport interface CallbackResponseInput {\n\treadonly externalCallbackId: string;\n\treadonly response: unknown;\n}\n\nexport interface CallbackRejectionInput {\n\treadonly externalCallbackId: string;\n\treadonly reason: string;\n}\n\nexport interface ServerRequestResolvedInput {\n\treadonly method: string;\n\treadonly params: unknown;\n}\n\nexport interface OpaqueServerRequestProjection {\n\treadonly kind: \"opaque-request\";\n\treadonly method: \"appServer/request\";\n\treadonly externalCallbackId: string;\n\treadonly timeoutAtMs: number;\n\treadonly envelope: OpaqueAppServerEnvelope;\n}\n\nexport type ServerRequestDeliveryResult =\n\t| { readonly kind: \"delivered\"; readonly request: OpaqueServerRequestProjection }\n\t| { readonly kind: \"adapter-error\"; readonly error: AdapterJsonRpcError };\n\nexport type CallbackForwardResult =\n\t| { readonly kind: \"forwarded\" }\n\t| { readonly kind: \"adapter-error\"; readonly error: AdapterJsonRpcError };\n\nexport type ServerRequestResolvedResult =\n\t| { readonly kind: \"resolved\"; readonly appRequestId: string }\n\t| { readonly kind: \"ignored\" };\n\nexport interface TimedOutCallback {\n\treadonly appRequestId: string;\n\treadonly externalCallbackId: string;\n}\n\nexport interface ServerRequestBridge {\n\tdeliver(input: AppServerRequestInput): ServerRequestDeliveryResult;\n\trespond(input: CallbackResponseInput): Promise<CallbackForwardResult>;\n\treject(input: CallbackRejectionInput): Promise<CallbackForwardResult>;\n\treplayPendingCallbacks(): readonly OpaqueServerRequestProjection[];\n\trejectPendingCallbacks(reason: string): Promise<readonly TimedOutCallback[]>;\n\trejectTimedOutCallbacks(nowMs?: number): Promise<readonly TimedOutCallback[]>;\n\tresolveFromNotification(input: ServerRequestResolvedInput): ServerRequestResolvedResult;\n\tredactCallbackResponse(externalCallbackId: string, response: unknown): unknown;\n}\n\ntype CallbackStatus = \"pending\" | \"forwarded\" | \"timed-out\";\n\ninterface CallbackState {\n\treadonly appRequestId: string;\n\treadonly externalCallbackId: string;\n\treadonly timeoutAtMs: number;\n\treadonly secretQuestionIds: ReadonlySet<string>;\n\treadonly request: OpaqueServerRequestProjection;\n\tstatus: CallbackStatus;\n}\n\nexport function createServerRequestBridge(options: ServerRequestBridgeOptions): ServerRequestBridge {\n\treturn new DefaultServerRequestBridge(options);\n}\n\nclass DefaultServerRequestBridge implements ServerRequestBridge {\n\tprivate readonly connectionId: string;\n\tprivate readonly capabilityFlags: readonly string[];\n\tprivate readonly callbackTimeoutMs: number;\n\tprivate readonly nowMs: () => number;\n\tprivate readonly idMapper: IdMapper;\n\tprivate readonly sessionRegistry: SessionRegistry;\n\tprivate readonly callbackClient: AppServerCallbackClient;\n\tprivate readonly callbacksByExternalId = new Map<string, CallbackState>();\n\tprivate readonly externalCallbackIdByAppRequestId = new Map<string, string>();\n\tprivate sequence = 0;\n\n\tconstructor(options: ServerRequestBridgeOptions) {\n\t\tthis.connectionId = options.connectionId;\n\t\tthis.capabilityFlags = options.capabilityFlags;\n\t\tthis.callbackTimeoutMs = options.callbackTimeoutMs;\n\t\tthis.nowMs = options.nowMs;\n\t\tthis.idMapper = options.idMapper;\n\t\tthis.sessionRegistry = options.sessionRegistry;\n\t\tthis.callbackClient = options.callbackClient;\n\t}\n\n\tdeliver(input: AppServerRequestInput): ServerRequestDeliveryResult {\n\t\tif (!isSupportedCallbackMethod(input.method)) {\n\t\t\treturn invalidCallback(`Unsupported PR-008/PR-009 callback method: ${input.method}`);\n\t\t}\n\t\tconst appRequestId = String(input.requestId);\n\t\tconst ids = readAppServerRequestIds(input.params);\n\t\tconst externalCallbackId = `callback-${appRequestId}`;\n\t\tconst timeoutAtMs = this.nowMs() + this.callbackTimeoutMs;\n\t\tconst registration = this.idMapper.registerServerRequest({\n\t\t\tappRequestId,\n\t\t\texternalCallbackId,\n\t\t\tappThreadId: ids.appThreadId,\n\t\t\tappTurnId: ids.appTurnId,\n\t\t\tappItemId: ids.appItemId,\n\t\t\tmethod: input.method,\n\t\t});\n\t\tif (registration.kind === \"rejected\") return { kind: \"adapter-error\", error: registration.error };\n\n\t\tconst state: CallbackState = {\n\t\t\tappRequestId,\n\t\t\texternalCallbackId,\n\t\t\ttimeoutAtMs,\n\t\t\tsecretQuestionIds: readSecretQuestionIds(input.params),\n\t\t\trequest: {\n\t\t\t\tkind: \"opaque-request\",\n\t\t\t\tmethod: \"appServer/request\",\n\t\t\t\texternalCallbackId,\n\t\t\t\ttimeoutAtMs,\n\t\t\t\tenvelope: createOpaqueServerRequestEnvelope({\n\t\t\t\t\tconnectionId: this.connectionId,\n\t\t\t\t\tcapabilityFlags: this.capabilityFlags,\n\t\t\t\t\texternalCallbackId,\n\t\t\t\t\tappRequestId,\n\t\t\t\t\tmethod: input.method,\n\t\t\t\t\tparams: input.params,\n\t\t\t\t\tids,\n\t\t\t\t\tsequence: this.nextSequence(),\n\t\t\t\t\tsessionRegistry: this.sessionRegistry,\n\t\t\t\t}),\n\t\t\t},\n\t\t\tstatus: \"pending\",\n\t\t};\n\t\tthis.callbacksByExternalId.set(externalCallbackId, state);\n\t\tthis.externalCallbackIdByAppRequestId.set(appRequestId, externalCallbackId);\n\t\treturn {\n\t\t\tkind: \"delivered\",\n\t\t\trequest: state.request,\n\t\t};\n\t}\n\n\tasync respond(input: CallbackResponseInput): Promise<CallbackForwardResult> {\n\t\tconst state = this.readPendingCallback(input.externalCallbackId);\n\t\tif (state.kind === \"adapter-error\") return state;\n\t\tawait this.callbackClient.respond(state.callback.appRequestId, input.response);\n\t\tstate.callback.status = \"forwarded\";\n\t\treturn { kind: \"forwarded\" };\n\t}\n\n\tasync reject(input: CallbackRejectionInput): Promise<CallbackForwardResult> {\n\t\tconst state = this.readPendingCallback(input.externalCallbackId);\n\t\tif (state.kind === \"adapter-error\") return state;\n\t\tawait this.callbackClient.reject(state.callback.appRequestId, input.reason);\n\t\tstate.callback.status = \"forwarded\";\n\t\treturn { kind: \"forwarded\" };\n\t}\n\n\treplayPendingCallbacks(): readonly OpaqueServerRequestProjection[] {\n\t\treturn [...this.callbacksByExternalId.values()]\n\t\t\t.filter((state) => state.status === \"pending\")\n\t\t\t.map((state) => state.request);\n\t}\n\n\tasync rejectPendingCallbacks(reason: string): Promise<readonly TimedOutCallback[]> {\n\t\tconst rejected: TimedOutCallback[] = [];\n\t\tfor (const state of this.callbacksByExternalId.values()) {\n\t\t\tif (state.status !== \"pending\") continue;\n\t\t\tawait this.callbackClient.reject(state.appRequestId, reason);\n\t\t\tthis.clearCallbackState(state);\n\t\t\trejected.push({ appRequestId: state.appRequestId, externalCallbackId: state.externalCallbackId });\n\t\t}\n\t\treturn rejected;\n\t}\n\n\tasync rejectTimedOutCallbacks(nowMs = this.nowMs()): Promise<readonly TimedOutCallback[]> {\n\t\tconst timedOut: TimedOutCallback[] = [];\n\t\tfor (const state of this.callbacksByExternalId.values()) {\n\t\t\tif ((state.status !== \"pending\" && state.status !== \"timed-out\") || state.timeoutAtMs > nowMs) continue;\n\t\t\tstate.status = \"timed-out\";\n\t\t\tawait this.callbackClient.reject(state.appRequestId, \"callback timed out\");\n\t\t\tthis.clearCallbackState(state);\n\t\t\ttimedOut.push({ appRequestId: state.appRequestId, externalCallbackId: state.externalCallbackId });\n\t\t}\n\t\treturn timedOut;\n\t}\n\n\tresolveFromNotification(input: ServerRequestResolvedInput): ServerRequestResolvedResult {\n\t\tif (input.method !== \"serverRequest/resolved\") return { kind: \"ignored\" };\n\t\tconst appRequestId = readRequestId(input.params);\n\t\tif (!appRequestId) return { kind: \"ignored\" };\n\t\tconst externalCallbackId = this.externalCallbackIdByAppRequestId.get(appRequestId);\n\t\tif (externalCallbackId) {\n\t\t\tconst state = this.callbacksByExternalId.get(externalCallbackId);\n\t\t\tif (state) this.clearCallbackState(state);\n\t\t} else {\n\t\t\tthis.idMapper.resolveServerRequest(appRequestId);\n\t\t}\n\t\treturn { kind: \"resolved\", appRequestId };\n\t}\n\n\tredactCallbackResponse(externalCallbackId: string, response: unknown): unknown {\n\t\tconst state = this.callbacksByExternalId.get(externalCallbackId);\n\t\tif (!state) return response;\n\t\treturn redactSecretAnswers(response, state.secretQuestionIds);\n\t}\n\n\tprivate nextSequence(): number {\n\t\tthis.sequence += 1;\n\t\treturn this.sequence;\n\t}\n\n\tprivate readPendingCallback(\n\t\texternalCallbackId: string,\n\t):\n\t\t| { readonly kind: \"callback\"; readonly callback: CallbackState }\n\t\t| { readonly kind: \"adapter-error\"; readonly error: AdapterJsonRpcError } {\n\t\tconst callback = this.callbacksByExternalId.get(externalCallbackId);\n\t\tif (callback?.status !== \"pending\") {\n\t\t\treturn invalidCallback(`Unknown, duplicate, or timed-out callback id: ${externalCallbackId}`);\n\t\t}\n\t\treturn { kind: \"callback\", callback };\n\t}\n\n\tprivate clearCallbackState(state: CallbackState): void {\n\t\tthis.callbacksByExternalId.delete(state.externalCallbackId);\n\t\tthis.externalCallbackIdByAppRequestId.delete(state.appRequestId);\n\t\tthis.idMapper.resolveServerRequest(state.appRequestId);\n\t}\n}\n\nconst SUPPORTED_CALLBACK_METHODS: ReadonlySet<string> = new Set([\n\t\"item/commandExecution/requestApproval\",\n\t\"item/fileChange/requestApproval\",\n\t\"item/tool/requestUserInput\",\n\t\"mcpServer/elicitation/request\",\n\t\"item/permissions/requestApproval\",\n\t\"item/tool/call\",\n] satisfies readonly string[]);\n\nfunction isSupportedCallbackMethod(method: string): boolean {\n\treturn SUPPORTED_CALLBACK_METHODS.has(method);\n}\n\nfunction invalidCallback(message: string): { readonly kind: \"adapter-error\"; readonly error: AdapterJsonRpcError } {\n\treturn {\n\t\tkind: \"adapter-error\",\n\t\terror: createAdapterJsonRpcError({ adapterCode: \"invalid-callback-state\", message }),\n\t};\n}\n"]}

package/dist/core/extensions/builtin/pi-codex-app-server/server-request-fields.d.ts

@@ -0,0 +1,24 @@
+import type { OpaqueAppServerEnvelope } from "./notification-projector.ts";
+import type { SessionRegistry } from "./session-registry.ts";
+export interface AppServerRequestIds {
+    readonly appThreadId: string | undefined;
+    readonly appTurnId: string | undefined;
+    readonly appItemId: string | undefined;
+}
+export interface OpaqueServerRequestEnvelopeInput {
+    readonly connectionId: string;
+    readonly capabilityFlags: readonly string[];
+    readonly externalCallbackId: string;
+    readonly appRequestId: string;
+    readonly method: string;
+    readonly params: unknown;
+    readonly ids: AppServerRequestIds;
+    readonly sequence: number;
+    readonly sessionRegistry: SessionRegistry;
+}
+export declare function readAppServerRequestIds(params: unknown): AppServerRequestIds;
+export declare function readRequestId(params: unknown): string | undefined;
+export declare function readSecretQuestionIds(params: unknown): ReadonlySet<string>;
+export declare function redactSecretAnswers(response: unknown, secretQuestionIds: ReadonlySet<string>): unknown;
+export declare function createOpaqueServerRequestEnvelope(input: OpaqueServerRequestEnvelopeInput): OpaqueAppServerEnvelope;
+//# sourceMappingURL=server-request-fields.d.ts.map

package/dist/core/extensions/builtin/pi-codex-app-server/server-request-fields.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server-request-fields.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/pi-codex-app-server/server-request-fields.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,6BAA6B,CAAC;AAE3E,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAE7D,MAAM,WAAW,mBAAmB;IACnC,QAAQ,CAAC,WAAW,EAAE,MAAM,GAAG,SAAS,CAAC;IACzC,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IACvC,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;CACvC;AAED,MAAM,WAAW,gCAAgC;IAChD,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,eAAe,EAAE,SAAS,MAAM,EAAE,CAAC;IAC5C,QAAQ,CAAC,kBAAkB,EAAE,MAAM,CAAC;IACpC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;IACzB,QAAQ,CAAC,GAAG,EAAE,mBAAmB,CAAC;IAClC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;CAC1C;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,OAAO,GAAG,mBAAmB,CAO5E;AAED,wBAAgB,aAAa,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAMjE;AAED,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC,CAU1E;AAED,wBAAgB,mBAAmB,CAAC,QAAQ,EAAE,OAAO,EAAE,iBAAiB,EAAE,WAAW,CAAC,MAAM,CAAC,GAAG,OAAO,CAStG;AAED,wBAAgB,iCAAiC,CAAC,KAAK,EAAE,gCAAgC,GAAG,uBAAuB,CAqBlH"}

package/dist/core/extensions/builtin/pi-codex-app-server/server-request-fields.js

@@ -0,0 +1,81 @@
+import { classifyAppServerSurface, PI_CODEX_APP_SERVER_PROTOCOL_VERSION } from "./protocol-core.js";
+export function readAppServerRequestIds(params) {
+    const record = isRecord(params) ? params : {};
+    return {
+        appThreadId: readString(record, "threadId") ?? readString(record, "thread_id"),
+        appTurnId: readString(record, "turnId") ?? readString(record, "turn_id"),
+        appItemId: readString(record, "itemId") ?? readString(record, "item_id"),
+    };
+}
+export function readRequestId(params) {
+    if (!isRecord(params))
+        return undefined;
+    const value = params.requestId ?? params.request_id ?? params.id;
+    if (typeof value === "string")
+        return value;
+    if (typeof value === "number")
+        return String(value);
+    return undefined;
+}
+export function readSecretQuestionIds(params) {
+    if (!isRecord(params) || !Array.isArray(params.questions))
+        return new Set();
+    const ids = [];
+    for (const question of params.questions) {
+        if (!isRecord(question))
+            continue;
+        if (question.isSecret !== true && question.is_secret !== true)
+            continue;
+        const id = readString(question, "id");
+        if (id)
+            ids.push(id);
+    }
+    return new Set(ids);
+}
+export function redactSecretAnswers(response, secretQuestionIds) {
+    if (secretQuestionIds.size === 0 || !isRecord(response))
+        return response;
+    const answers = response.answers;
+    if (!isRecord(answers))
+        return response;
+    const redactedAnswers = {};
+    for (const [questionId, answer] of Object.entries(answers)) {
+        redactedAnswers[questionId] = secretQuestionIds.has(questionId) ? redactAnswer(answer) : answer;
+    }
+    return { ...response, answers: redactedAnswers };
+}
+export function createOpaqueServerRequestEnvelope(input) {
+    const binding = input.ids.appThreadId ? input.sessionRegistry.getByAppThreadId(input.ids.appThreadId) : undefined;
+    return {
+        protocolVersion: PI_CODEX_APP_SERVER_PROTOCOL_VERSION,
+        connectionId: input.connectionId,
+        externalSessionId: binding?.externalSessionId,
+        externalRequestId: undefined,
+        externalMessageId: undefined,
+        externalCallbackId: input.externalCallbackId,
+        appThreadId: input.ids.appThreadId,
+        appSessionId: binding?.appSessionId,
+        appTurnId: input.ids.appTurnId,
+        appItemId: input.ids.appItemId,
+        appRequestId: input.appRequestId,
+        sequence: input.sequence,
+        streamClass: classifyAppServerSurface(input.method)?.streamClass ?? "lossless",
+        capabilityFlags: input.capabilityFlags,
+        originalMethod: input.method,
+        originalParams: input.params,
+        redactionClass: readSecretQuestionIds(input.params).size > 0 ? "secret-bearing" : "public-contract",
+    };
+}
+function redactAnswer(answer) {
+    if (!isRecord(answer) || !Array.isArray(answer.answers))
+        return answer;
+    return { ...answer, answers: answer.answers.map(() => "[REDACTED]") };
+}
+function readString(input, key) {
+    const value = input[key];
+    return typeof value === "string" ? value : undefined;
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+//# sourceMappingURL=server-request-fields.js.map

package/dist/core/extensions/builtin/pi-codex-app-server/server-request-fields.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server-request-fields.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/pi-codex-app-server/server-request-fields.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,wBAAwB,EAAE,oCAAoC,EAAE,MAAM,oBAAoB,CAAC;AAqBpG,MAAM,UAAU,uBAAuB,CAAC,MAAe;IACtD,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;IAC9C,OAAO;QACN,WAAW,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,EAAE,WAAW,CAAC;QAC9E,SAAS,EAAE,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,UAAU,CAAC,MAAM,EAAE,SAAS,CAAC;QACxE,SAAS,EAAE,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,UAAU,CAAC,MAAM,EAAE,SAAS,CAAC;KACxE,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,MAAe;IAC5C,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,SAAS,CAAC;IACxC,MAAM,KAAK,GAAG,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,EAAE,CAAC;IACjE,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACpD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,MAAe;IACpD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;QAAE,OAAO,IAAI,GAAG,EAAE,CAAC;IAC5E,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACzC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,SAAS;QAClC,IAAI,QAAQ,CAAC,QAAQ,KAAK,IAAI,IAAI,QAAQ,CAAC,SAAS,KAAK,IAAI;YAAE,SAAS;QACxE,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACtC,IAAI,EAAE;YAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACtB,CAAC;IACD,OAAO,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,QAAiB,EAAE,iBAAsC;IAC5F,IAAI,iBAAiB,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IACzE,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;IACjC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,QAAQ,CAAC;IACxC,MAAM,eAAe,GAA4B,EAAE,CAAC;IACpD,KAAK,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5D,eAAe,CAAC,UAAU,CAAC,GAAG,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IACjG,CAAC;IACD,OAAO,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,iCAAiC,CAAC,KAAuC;IACxF,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClH,OAAO;QACN,eAAe,EAAE,oCAAoC;QACrD,YAAY,EAAE,KAAK,CAAC,YAAY;QAChC,iBAAiB,EAAE,OAAO,EAAE,iBAAiB;QAC7C,iBAAiB,EAAE,SAAS;QAC5B,iBAAiB,EAAE,SAAS;QAC5B,kBAAkB,EAAE,KAAK,CAAC,kBAAkB;QAC5C,WAAW,EAAE,KAAK,CAAC,GAAG,CAAC,WAAW;QAClC,YAAY,EAAE,OAAO,EAAE,YAAY;QACnC,SAAS,EAAE,KAAK,CAAC,GAAG,CAAC,SAAS;QAC9B,SAAS,EAAE,KAAK,CAAC,GAAG,CAAC,SAAS;QAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;QAChC,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW,EAAE,wBAAwB,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,WAAW,IAAI,UAAU;QAC9E,eAAe,EAAE,KAAK,CAAC,eAAe;QACtC,cAAc,EAAE,KAAK,CAAC,MAAM;QAC5B,cAAc,EAAE,KAAK,CAAC,MAAM;QAC5B,cAAc,EAAE,qBAAqB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,iBAAiB;KACnG,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,MAAe;IACpC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;QAAE,OAAO,MAAM,CAAC;IACvE,OAAO,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;AACvE,CAAC;AAED,SAAS,UAAU,CAAC,KAAwC,EAAE,GAAW;IACxE,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;IACzB,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AACtD,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC/B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC7E,CAAC","sourcesContent":["import type { OpaqueAppServerEnvelope } from \"./notification-projector.ts\";\nimport { classifyAppServerSurface, PI_CODEX_APP_SERVER_PROTOCOL_VERSION } from \"./protocol-core.ts\";\nimport type { SessionRegistry } from \"./session-registry.ts\";\n\nexport interface AppServerRequestIds {\n\treadonly appThreadId: string | undefined;\n\treadonly appTurnId: string | undefined;\n\treadonly appItemId: string | undefined;\n}\n\nexport interface OpaqueServerRequestEnvelopeInput {\n\treadonly connectionId: string;\n\treadonly capabilityFlags: readonly string[];\n\treadonly externalCallbackId: string;\n\treadonly appRequestId: string;\n\treadonly method: string;\n\treadonly params: unknown;\n\treadonly ids: AppServerRequestIds;\n\treadonly sequence: number;\n\treadonly sessionRegistry: SessionRegistry;\n}\n\nexport function readAppServerRequestIds(params: unknown): AppServerRequestIds {\n\tconst record = isRecord(params) ? params : {};\n\treturn {\n\t\tappThreadId: readString(record, \"threadId\") ?? readString(record, \"thread_id\"),\n\t\tappTurnId: readString(record, \"turnId\") ?? readString(record, \"turn_id\"),\n\t\tappItemId: readString(record, \"itemId\") ?? readString(record, \"item_id\"),\n\t};\n}\n\nexport function readRequestId(params: unknown): string | undefined {\n\tif (!isRecord(params)) return undefined;\n\tconst value = params.requestId ?? params.request_id ?? params.id;\n\tif (typeof value === \"string\") return value;\n\tif (typeof value === \"number\") return String(value);\n\treturn undefined;\n}\n\nexport function readSecretQuestionIds(params: unknown): ReadonlySet<string> {\n\tif (!isRecord(params) || !Array.isArray(params.questions)) return new Set();\n\tconst ids: string[] = [];\n\tfor (const question of params.questions) {\n\t\tif (!isRecord(question)) continue;\n\t\tif (question.isSecret !== true && question.is_secret !== true) continue;\n\t\tconst id = readString(question, \"id\");\n\t\tif (id) ids.push(id);\n\t}\n\treturn new Set(ids);\n}\n\nexport function redactSecretAnswers(response: unknown, secretQuestionIds: ReadonlySet<string>): unknown {\n\tif (secretQuestionIds.size === 0 || !isRecord(response)) return response;\n\tconst answers = response.answers;\n\tif (!isRecord(answers)) return response;\n\tconst redactedAnswers: Record<string, unknown> = {};\n\tfor (const [questionId, answer] of Object.entries(answers)) {\n\t\tredactedAnswers[questionId] = secretQuestionIds.has(questionId) ? redactAnswer(answer) : answer;\n\t}\n\treturn { ...response, answers: redactedAnswers };\n}\n\nexport function createOpaqueServerRequestEnvelope(input: OpaqueServerRequestEnvelopeInput): OpaqueAppServerEnvelope {\n\tconst binding = input.ids.appThreadId ? input.sessionRegistry.getByAppThreadId(input.ids.appThreadId) : undefined;\n\treturn {\n\t\tprotocolVersion: PI_CODEX_APP_SERVER_PROTOCOL_VERSION,\n\t\tconnectionId: input.connectionId,\n\t\texternalSessionId: binding?.externalSessionId,\n\t\texternalRequestId: undefined,\n\t\texternalMessageId: undefined,\n\t\texternalCallbackId: input.externalCallbackId,\n\t\tappThreadId: input.ids.appThreadId,\n\t\tappSessionId: binding?.appSessionId,\n\t\tappTurnId: input.ids.appTurnId,\n\t\tappItemId: input.ids.appItemId,\n\t\tappRequestId: input.appRequestId,\n\t\tsequence: input.sequence,\n\t\tstreamClass: classifyAppServerSurface(input.method)?.streamClass ?? \"lossless\",\n\t\tcapabilityFlags: input.capabilityFlags,\n\t\toriginalMethod: input.method,\n\t\toriginalParams: input.params,\n\t\tredactionClass: readSecretQuestionIds(input.params).size > 0 ? \"secret-bearing\" : \"public-contract\",\n\t};\n}\n\nfunction redactAnswer(answer: unknown): unknown {\n\tif (!isRecord(answer) || !Array.isArray(answer.answers)) return answer;\n\treturn { ...answer, answers: answer.answers.map(() => \"[REDACTED]\") };\n}\n\nfunction readString(input: Readonly<Record<string, unknown>>, key: string): string | undefined {\n\tconst value = input[key];\n\treturn typeof value === \"string\" ? value : undefined;\n}\n\nfunction isRecord(value: unknown): value is Readonly<Record<string, unknown>> {\n\treturn typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n"]}

package/dist/core/extensions/builtin/pi-codex-app-server/session-registry.d.ts

@@ -0,0 +1,42 @@
+import { type AdapterJsonRpcError } from "./error-mapper.ts";
+export interface SessionBindingInput {
+    readonly externalSessionId: string;
+    readonly appThreadId: string;
+    readonly appSessionId: string;
+}
+export interface ReplayCursor {
+    readonly lastCompletedTurnId?: string;
+    readonly lastProjectedItemId?: string;
+    readonly lastLosslessSequence?: number;
+}
+export interface SessionBinding {
+    readonly externalSessionId: string;
+    readonly appThreadId: string;
+    readonly appSessionId: string;
+    readonly tombstoned: boolean;
+    readonly replayCursor: ReplayCursor;
+}
+export type SessionBindResult = {
+    readonly kind: "bound";
+    readonly binding: SessionBinding;
+} | {
+    readonly kind: "rejected";
+    readonly error: AdapterJsonRpcError;
+};
+export type SessionLookupResult = {
+    readonly kind: "active";
+    readonly binding: SessionBinding;
+} | {
+    readonly kind: "rejected";
+    readonly error: AdapterJsonRpcError;
+};
+export interface SessionRegistry {
+    bindSession(input: SessionBindingInput): SessionBindResult;
+    getByExternalSessionId(externalSessionId: string): SessionBinding | undefined;
+    getByAppThreadId(appThreadId: string): SessionBinding | undefined;
+    requireActiveSession(externalSessionId: string): SessionLookupResult;
+    tombstoneExternalSession(externalSessionId: string): SessionLookupResult;
+    updateReplayCursor(externalSessionId: string, replayCursor: ReplayCursor): SessionLookupResult;
+}
+export declare function createSessionRegistry(): SessionRegistry;
+//# sourceMappingURL=session-registry.d.ts.map

package/dist/core/extensions/builtin/pi-codex-app-server/session-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-registry.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/pi-codex-app-server/session-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,mBAAmB,EAA6B,MAAM,mBAAmB,CAAC;AAExF,MAAM,WAAW,mBAAmB;IACnC,QAAQ,CAAC,iBAAiB,EAAE,MAAM,CAAC;IACnC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,YAAY;IAC5B,QAAQ,CAAC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IACtC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IACtC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,MAAM,CAAC;CACvC;AAED,MAAM,WAAW,cAAc;IAC9B,QAAQ,CAAC,iBAAiB,EAAE,MAAM,CAAC;IACnC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC;IAC7B,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;CACpC;AAED,MAAM,MAAM,iBAAiB,GAC1B;IAAE,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IAAC,QAAQ,CAAC,OAAO,EAAE,cAAc,CAAA;CAAE,GAC5D;IAAE,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;IAAC,QAAQ,CAAC,KAAK,EAAE,mBAAmB,CAAA;CAAE,CAAC;AAEtE,MAAM,MAAM,mBAAmB,GAC5B;IAAE,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;IAAC,QAAQ,CAAC,OAAO,EAAE,cAAc,CAAA;CAAE,GAC7D;IAAE,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;IAAC,QAAQ,CAAC,KAAK,EAAE,mBAAmB,CAAA;CAAE,CAAC;AAEtE,MAAM,WAAW,eAAe;IAC/B,WAAW,CAAC,KAAK,EAAE,mBAAmB,GAAG,iBAAiB,CAAC;IAC3D,sBAAsB,CAAC,iBAAiB,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS,CAAC;IAC9E,gBAAgB,CAAC,WAAW,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS,CAAC;IAClE,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,GAAG,mBAAmB,CAAC;IACrE,wBAAwB,CAAC,iBAAiB,EAAE,MAAM,GAAG,mBAAmB,CAAC;IACzE,kBAAkB,CAAC,iBAAiB,EAAE,MAAM,EAAE,YAAY,EAAE,YAAY,GAAG,mBAAmB,CAAC;CAC/F;AAED,wBAAgB,qBAAqB,IAAI,eAAe,CAEvD"}

package/dist/core/extensions/builtin/pi-codex-app-server/session-registry.js

@@ -0,0 +1,89 @@
+import { createAdapterJsonRpcError } from "./error-mapper.js";
+export function createSessionRegistry() {
+    return new InMemorySessionRegistry();
+}
+class InMemorySessionRegistry {
+    constructor() {
+        this.byExternalSessionId = new Map();
+        this.externalSessionIdByAppThreadId = new Map();
+    }
+    bindSession(input) {
+        if (this.byExternalSessionId.has(input.externalSessionId)) {
+            return rejectDuplicate(`Duplicate external session id: ${input.externalSessionId}`);
+        }
+        const existingExternalSessionId = this.externalSessionIdByAppThreadId.get(input.appThreadId);
+        if (existingExternalSessionId && existingExternalSessionId !== input.externalSessionId) {
+            return rejectDuplicate(`Duplicate app-server thread id: ${input.appThreadId}`);
+        }
+        const binding = {
+            externalSessionId: input.externalSessionId,
+            appThreadId: input.appThreadId,
+            appSessionId: input.appSessionId,
+            tombstoned: false,
+            replayCursor: {},
+        };
+        this.byExternalSessionId.set(binding.externalSessionId, binding);
+        this.externalSessionIdByAppThreadId.set(binding.appThreadId, binding.externalSessionId);
+        return { kind: "bound", binding };
+    }
+    getByExternalSessionId(externalSessionId) {
+        return this.byExternalSessionId.get(externalSessionId);
+    }
+    getByAppThreadId(appThreadId) {
+        const externalSessionId = this.externalSessionIdByAppThreadId.get(appThreadId);
+        if (!externalSessionId)
+            return undefined;
+        return this.byExternalSessionId.get(externalSessionId);
+    }
+    requireActiveSession(externalSessionId) {
+        const binding = this.byExternalSessionId.get(externalSessionId);
+        if (!binding) {
+            return rejectSession(`Unknown external session: ${externalSessionId}`);
+        }
+        if (binding.tombstoned) {
+            return rejectSession(`External session is tombstoned: ${externalSessionId}`);
+        }
+        return { kind: "active", binding };
+    }
+    tombstoneExternalSession(externalSessionId) {
+        const lookup = this.requireActiveSession(externalSessionId);
+        if (lookup.kind === "rejected")
+            return lookup;
+        const tombstoned = {
+            ...lookup.binding,
+            tombstoned: true,
+        };
+        this.byExternalSessionId.set(externalSessionId, tombstoned);
+        return { kind: "active", binding: tombstoned };
+    }
+    updateReplayCursor(externalSessionId, replayCursor) {
+        const lookup = this.requireActiveSession(externalSessionId);
+        if (lookup.kind === "rejected")
+            return lookup;
+        const updated = {
+            ...lookup.binding,
+            replayCursor,
+        };
+        this.byExternalSessionId.set(externalSessionId, updated);
+        return { kind: "active", binding: updated };
+    }
+}
+function rejectSession(message) {
+    return {
+        kind: "rejected",
+        error: createAdapterJsonRpcError({
+            adapterCode: "invalid-session-state",
+            message,
+        }),
+    };
+}
+function rejectDuplicate(message) {
+    return {
+        kind: "rejected",
+        error: createAdapterJsonRpcError({
+            adapterCode: "duplicate-routing-id",
+            message,
+        }),
+    };
+}
+//# sourceMappingURL=session-registry.js.map

package/dist/core/extensions/builtin/pi-codex-app-server/session-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-registry.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/pi-codex-app-server/session-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAA4B,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAuCxF,MAAM,UAAU,qBAAqB;IACpC,OAAO,IAAI,uBAAuB,EAAE,CAAC;AACtC,CAAC;AAED,MAAM,uBAAuB;IAA7B;QACkB,wBAAmB,GAAG,IAAI,GAAG,EAA0B,CAAC;QACxD,mCAA8B,GAAG,IAAI,GAAG,EAAkB,CAAC;IAgE7E,CAAC;IA9DA,WAAW,CAAC,KAA0B;QACrC,IAAI,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC3D,OAAO,eAAe,CAAC,kCAAkC,KAAK,CAAC,iBAAiB,EAAE,CAAC,CAAC;QACrF,CAAC;QACD,MAAM,yBAAyB,GAAG,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC7F,IAAI,yBAAyB,IAAI,yBAAyB,KAAK,KAAK,CAAC,iBAAiB,EAAE,CAAC;YACxF,OAAO,eAAe,CAAC,mCAAmC,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;QAChF,CAAC;QACD,MAAM,OAAO,GAAmB;YAC/B,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;YAC1C,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,UAAU,EAAE,KAAK;YACjB,YAAY,EAAE,EAAE;SAChB,CAAC;QACF,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;QACjE,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,CAAC,iBAAiB,CAAC,CAAC;QACxF,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;IACnC,CAAC;IAED,sBAAsB,CAAC,iBAAyB;QAC/C,OAAO,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IACxD,CAAC;IAED,gBAAgB,CAAC,WAAmB;QACnC,MAAM,iBAAiB,GAAG,IAAI,CAAC,8BAA8B,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC/E,IAAI,CAAC,iBAAiB;YAAE,OAAO,SAAS,CAAC;QACzC,OAAO,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IACxD,CAAC;IAED,oBAAoB,CAAC,iBAAyB;QAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;QAChE,IAAI,CAAC,OAAO,EAAE,CAAC;YACd,OAAO,aAAa,CAAC,6BAA6B,iBAAiB,EAAE,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACxB,OAAO,aAAa,CAAC,mCAAmC,iBAAiB,EAAE,CAAC,CAAC;QAC9E,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpC,CAAC;IAED,wBAAwB,CAAC,iBAAyB;QACjD,MAAM,MAAM,GAAG,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;QAC5D,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,MAAM,CAAC;QAC9C,MAAM,UAAU,GAAmB;YAClC,GAAG,MAAM,CAAC,OAAO;YACjB,UAAU,EAAE,IAAI;SAChB,CAAC;QACF,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,iBAAiB,EAAE,UAAU,CAAC,CAAC;QAC5D,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;IAChD,CAAC;IAED,kBAAkB,CAAC,iBAAyB,EAAE,YAA0B;QACvE,MAAM,MAAM,GAAG,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;QAC5D,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,MAAM,CAAC;QAC9C,MAAM,OAAO,GAAmB;YAC/B,GAAG,MAAM,CAAC,OAAO;YACjB,YAAY;SACZ,CAAC;QACF,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;QACzD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;IAC7C,CAAC;CACD;AAED,SAAS,aAAa,CAAC,OAAe;IACrC,OAAO;QACN,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,yBAAyB,CAAC;YAChC,WAAW,EAAE,uBAAuB;YACpC,OAAO;SACP,CAAC;KACF,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,OAAe;IACvC,OAAO;QACN,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,yBAAyB,CAAC;YAChC,WAAW,EAAE,sBAAsB;YACnC,OAAO;SACP,CAAC;KACF,CAAC;AACH,CAAC","sourcesContent":["import { type AdapterJsonRpcError, createAdapterJsonRpcError } from \"./error-mapper.ts\";\n\nexport interface SessionBindingInput {\n\treadonly externalSessionId: string;\n\treadonly appThreadId: string;\n\treadonly appSessionId: string;\n}\n\nexport interface ReplayCursor {\n\treadonly lastCompletedTurnId?: string;\n\treadonly lastProjectedItemId?: string;\n\treadonly lastLosslessSequence?: number;\n}\n\nexport interface SessionBinding {\n\treadonly externalSessionId: string;\n\treadonly appThreadId: string;\n\treadonly appSessionId: string;\n\treadonly tombstoned: boolean;\n\treadonly replayCursor: ReplayCursor;\n}\n\nexport type SessionBindResult =\n\t| { readonly kind: \"bound\"; readonly binding: SessionBinding }\n\t| { readonly kind: \"rejected\"; readonly error: AdapterJsonRpcError };\n\nexport type SessionLookupResult =\n\t| { readonly kind: \"active\"; readonly binding: SessionBinding }\n\t| { readonly kind: \"rejected\"; readonly error: AdapterJsonRpcError };\n\nexport interface SessionRegistry {\n\tbindSession(input: SessionBindingInput): SessionBindResult;\n\tgetByExternalSessionId(externalSessionId: string): SessionBinding | undefined;\n\tgetByAppThreadId(appThreadId: string): SessionBinding | undefined;\n\trequireActiveSession(externalSessionId: string): SessionLookupResult;\n\ttombstoneExternalSession(externalSessionId: string): SessionLookupResult;\n\tupdateReplayCursor(externalSessionId: string, replayCursor: ReplayCursor): SessionLookupResult;\n}\n\nexport function createSessionRegistry(): SessionRegistry {\n\treturn new InMemorySessionRegistry();\n}\n\nclass InMemorySessionRegistry implements SessionRegistry {\n\tprivate readonly byExternalSessionId = new Map<string, SessionBinding>();\n\tprivate readonly externalSessionIdByAppThreadId = new Map<string, string>();\n\n\tbindSession(input: SessionBindingInput): SessionBindResult {\n\t\tif (this.byExternalSessionId.has(input.externalSessionId)) {\n\t\t\treturn rejectDuplicate(`Duplicate external session id: ${input.externalSessionId}`);\n\t\t}\n\t\tconst existingExternalSessionId = this.externalSessionIdByAppThreadId.get(input.appThreadId);\n\t\tif (existingExternalSessionId && existingExternalSessionId !== input.externalSessionId) {\n\t\t\treturn rejectDuplicate(`Duplicate app-server thread id: ${input.appThreadId}`);\n\t\t}\n\t\tconst binding: SessionBinding = {\n\t\t\texternalSessionId: input.externalSessionId,\n\t\t\tappThreadId: input.appThreadId,\n\t\t\tappSessionId: input.appSessionId,\n\t\t\ttombstoned: false,\n\t\t\treplayCursor: {},\n\t\t};\n\t\tthis.byExternalSessionId.set(binding.externalSessionId, binding);\n\t\tthis.externalSessionIdByAppThreadId.set(binding.appThreadId, binding.externalSessionId);\n\t\treturn { kind: \"bound\", binding };\n\t}\n\n\tgetByExternalSessionId(externalSessionId: string): SessionBinding | undefined {\n\t\treturn this.byExternalSessionId.get(externalSessionId);\n\t}\n\n\tgetByAppThreadId(appThreadId: string): SessionBinding | undefined {\n\t\tconst externalSessionId = this.externalSessionIdByAppThreadId.get(appThreadId);\n\t\tif (!externalSessionId) return undefined;\n\t\treturn this.byExternalSessionId.get(externalSessionId);\n\t}\n\n\trequireActiveSession(externalSessionId: string): SessionLookupResult {\n\t\tconst binding = this.byExternalSessionId.get(externalSessionId);\n\t\tif (!binding) {\n\t\t\treturn rejectSession(`Unknown external session: ${externalSessionId}`);\n\t\t}\n\t\tif (binding.tombstoned) {\n\t\t\treturn rejectSession(`External session is tombstoned: ${externalSessionId}`);\n\t\t}\n\t\treturn { kind: \"active\", binding };\n\t}\n\n\ttombstoneExternalSession(externalSessionId: string): SessionLookupResult {\n\t\tconst lookup = this.requireActiveSession(externalSessionId);\n\t\tif (lookup.kind === \"rejected\") return lookup;\n\t\tconst tombstoned: SessionBinding = {\n\t\t\t...lookup.binding,\n\t\t\ttombstoned: true,\n\t\t};\n\t\tthis.byExternalSessionId.set(externalSessionId, tombstoned);\n\t\treturn { kind: \"active\", binding: tombstoned };\n\t}\n\n\tupdateReplayCursor(externalSessionId: string, replayCursor: ReplayCursor): SessionLookupResult {\n\t\tconst lookup = this.requireActiveSession(externalSessionId);\n\t\tif (lookup.kind === \"rejected\") return lookup;\n\t\tconst updated: SessionBinding = {\n\t\t\t...lookup.binding,\n\t\t\treplayCursor,\n\t\t};\n\t\tthis.byExternalSessionId.set(externalSessionId, updated);\n\t\treturn { kind: \"active\", binding: updated };\n\t}\n}\n\nfunction rejectSession(message: string): SessionLookupResult {\n\treturn {\n\t\tkind: \"rejected\",\n\t\terror: createAdapterJsonRpcError({\n\t\t\tadapterCode: \"invalid-session-state\",\n\t\t\tmessage,\n\t\t}),\n\t};\n}\n\nfunction rejectDuplicate(message: string): SessionBindResult {\n\treturn {\n\t\tkind: \"rejected\",\n\t\terror: createAdapterJsonRpcError({\n\t\t\tadapterCode: \"duplicate-routing-id\",\n\t\t\tmessage,\n\t\t}),\n\t};\n}\n"]}

package/dist/core/extensions/builtin/pi-codex-app-server/stream-backpressure.d.ts

@@ -0,0 +1,36 @@
+import type { AppServerJsonRpcError } from "./error-mapper.ts";
+import type { NotificationProjection } from "./notification-projector.ts";
+export type StreamBackpressureEvent = Exclude<NotificationProjection, {
+    readonly kind: "skipped";
+}>;
+export interface LagStreamEvent {
+    readonly kind: "lag";
+    readonly method: "lag";
+    readonly connectionId: string;
+    readonly sequence: number;
+    readonly streamClass: "control";
+    readonly droppedProgressEvents: number;
+    readonly nextLosslessSequence: number;
+}
+export type StreamBackpressureOutput = StreamBackpressureEvent | LagStreamEvent;
+export interface StreamBackpressureOptions {
+    readonly connectionId: string;
+    readonly bestEffortQueueLimit: number;
+}
+export interface StreamBackpressureStats {
+    readonly droppedProgressEvents: number;
+    readonly emittedLagMarkers: number;
+    readonly queuedEvents: number;
+}
+export interface AppServerOverloadOptions {
+    readonly retryAfterMs?: number;
+}
+export interface StreamBackpressureController {
+    enqueue(event: StreamBackpressureEvent): void;
+    drainAll(): readonly StreamBackpressureOutput[];
+    flushTerminal(): readonly StreamBackpressureOutput[];
+    stats(): StreamBackpressureStats;
+}
+export declare function createStreamBackpressureController(options: StreamBackpressureOptions): StreamBackpressureController;
+export declare function createAppServerOverloadError(options?: AppServerOverloadOptions): AppServerJsonRpcError;
+//# sourceMappingURL=stream-backpressure.d.ts.map

package/dist/core/extensions/builtin/pi-codex-app-server/stream-backpressure.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-backpressure.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/pi-codex-app-server/stream-backpressure.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,mBAAmB,CAAC;AAC/D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAE1E,MAAM,MAAM,uBAAuB,GAAG,OAAO,CAAC,sBAAsB,EAAE;IAAE,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAA;CAAE,CAAC,CAAC;AAEpG,MAAM,WAAW,cAAc;IAC9B,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC;IACrB,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC;IACvB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,WAAW,EAAE,SAAS,CAAC;IAChC,QAAQ,CAAC,qBAAqB,EAAE,MAAM,CAAC;IACvC,QAAQ,CAAC,oBAAoB,EAAE,MAAM,CAAC;CACtC;AAED,MAAM,MAAM,wBAAwB,GAAG,uBAAuB,GAAG,cAAc,CAAC;AAEhF,MAAM,WAAW,yBAAyB;IACzC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,oBAAoB,EAAE,MAAM,CAAC;CACtC;AAED,MAAM,WAAW,uBAAuB;IACvC,QAAQ,CAAC,qBAAqB,EAAE,MAAM,CAAC;IACvC,QAAQ,CAAC,iBAAiB,EAAE,MAAM,CAAC;IACnC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,wBAAwB;IACxC,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,WAAW,4BAA4B;IAC5C,OAAO,CAAC,KAAK,EAAE,uBAAuB,GAAG,IAAI,CAAC;IAC9C,QAAQ,IAAI,SAAS,wBAAwB,EAAE,CAAC;IAChD,aAAa,IAAI,SAAS,wBAAwB,EAAE,CAAC;IACrD,KAAK,IAAI,uBAAuB,CAAC;CACjC;AAED,wBAAgB,kCAAkC,CAAC,OAAO,EAAE,yBAAyB,GAAG,4BAA4B,CAEnH;AAED,wBAAgB,4BAA4B,CAAC,OAAO,GAAE,wBAA6B,GAAG,qBAAqB,CAS1G"}

package/dist/core/extensions/builtin/pi-codex-app-server/stream-backpressure.js

@@ -0,0 +1,94 @@
+export function createStreamBackpressureController(options) {
+    return new DefaultStreamBackpressureController(options);
+}
+export function createAppServerOverloadError(options = {}) {
+    return {
+        code: -32001,
+        message: "app-server overloaded",
+        data: {
+            retryable: true,
+            ...(options.retryAfterMs === undefined ? {} : { retryAfterMs: options.retryAfterMs }),
+        },
+    };
+}
+class DefaultStreamBackpressureController {
+    constructor(options) {
+        this.queue = [];
+        this.droppedProgressEvents = 0;
+        this.emittedLagMarkers = 0;
+        this.pendingDroppedProgressEvents = 0;
+        this.connectionId = options.connectionId;
+        this.bestEffortQueueLimit = Math.max(0, options.bestEffortQueueLimit);
+    }
+    enqueue(event) {
+        if (readStreamClass(event) === "best-effort") {
+            this.enqueueBestEffort(event);
+            return;
+        }
+        this.enqueueLossless(event);
+    }
+    drainAll() {
+        return this.drainQueuedEvents();
+    }
+    flushTerminal() {
+        return this.drainQueuedEvents();
+    }
+    stats() {
+        return {
+            droppedProgressEvents: this.droppedProgressEvents,
+            emittedLagMarkers: this.emittedLagMarkers,
+            queuedEvents: this.queue.length,
+        };
+    }
+    enqueueBestEffort(event) {
+        if (this.countQueuedBestEffortEvents() >= this.bestEffortQueueLimit) {
+            this.droppedProgressEvents += 1;
+            this.pendingDroppedProgressEvents += 1;
+            this.pendingLagSequence = readSequence(event);
+            return;
+        }
+        this.queue.push(event);
+    }
+    enqueueLossless(event) {
+        if (this.pendingDroppedProgressEvents > 0) {
+            const nextLosslessSequence = readSequence(event);
+            this.queue.push(this.createLagEvent(this.pendingLagSequence ?? nextLosslessSequence - 1, nextLosslessSequence));
+            this.pendingDroppedProgressEvents = 0;
+            this.pendingLagSequence = undefined;
+            this.emittedLagMarkers += 1;
+        }
+        this.queue.push(event);
+    }
+    countQueuedBestEffortEvents() {
+        return this.queue.filter((event) => readStreamClass(event) === "best-effort").length;
+    }
+    createLagEvent(sequence, nextLosslessSequence) {
+        return {
+            kind: "lag",
+            method: "lag",
+            connectionId: this.connectionId,
+            sequence,
+            streamClass: "control",
+            droppedProgressEvents: this.pendingDroppedProgressEvents,
+            nextLosslessSequence,
+        };
+    }
+    drainQueuedEvents() {
+        const drained = [...this.queue];
+        this.queue.length = 0;
+        return drained;
+    }
+}
+function readStreamClass(event) {
+    if (event.kind === "lag")
+        return event.streamClass;
+    if (event.kind === "semantic")
+        return event.streamClass;
+    return event.envelope.streamClass;
+}
+function readSequence(event) {
+    if (event.kind === "semantic")
+        return event.sequence;
+    return event.envelope.sequence;
+}
+//# sourceMappingURL=stream-backpressure.js.map