@notionhq/custom-blocks 0.1.34 → 0.1.36

package/README.md

@@ -111,7 +111,5 @@ No top-level navigation, `window.open`, or auth redirects. No direct network req
 
 ## Bridge protocol
 
-The bridge speaks a versioned `postMessage` protocol, but block authors should
-use this package's public APIs rather than send bridge messages directly.
-`@notionhq/custom-blocks-protocol` and `@notionhq/custom-blocks-host` are
-internal workspace packages and are not available to block authors.
+The SDK communicates with Notion through a versioned `postMessage` protocol.
+Use the SDK's public APIs and exported types. Do not send bridge messages directly.

package/dist/autoResize.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"autoResize.d.ts","sourceRoot":"","sources":["../../../home/runner/work/custom-blocks/custom-blocks/sdk/src/autoResize.ts"],"names":[],"mappings":"AAKA,wBAAgB,UAAU,CAAC,IAAI,EAAE;IAChC,MAAM,EAAE,WAAW,GAAG,IAAI,GAAG,SAAS,CAAA;CACtC,GAAG,MAAM,IAAI,CA4Eb"}
+{"version":3,"file":"autoResize.d.ts","sourceRoot":"","sources":["../src/autoResize.ts"],"names":[],"mappings":"AAKA,wBAAgB,UAAU,CAAC,IAAI,EAAE;IAChC,MAAM,EAAE,WAAW,GAAG,IAAI,GAAG,SAAS,CAAA;CACtC,GAAG,MAAM,IAAI,CA4Eb"}

package/dist/bridge/SandboxBridge.d.ts

@@ -48,6 +48,7 @@ export declare class SandboxBridge {
     private getMessageType;
     private notify;
     private handleMessage;
+    private receiveMessage;
     subscribe(listener: () => void): () => boolean;
     getHostState(): CustomBlockHostState;
     /**

package/dist/bridge/SandboxBridge.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SandboxBridge.d.ts","sourceRoot":"","sources":["../../../../home/runner/work/custom-blocks/custom-blocks/sdk/src/bridge/SandboxBridge.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACX,gBAAgB,EAEhB,MAAM,4DAA4D,CAAA;AACnE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAA;AAW3E,OAAO,EAEN,KAAK,WAAW,EAEhB,MAAM,mDAAmD,CAAA;AAc1D,OAAO,KAAK,EACX,cAAc,EACd,gBAAgB,EAChB,aAAa,EACb,aAAa,EACb,aAAa,EACb,eAAe,EACf,8BAA8B,EAC9B,YAAY,EACZ,cAAc,EACd,gBAAgB,EAChB,oBAAoB,EACpB,MAAM,aAAa,CAAA;AAUpB,OAAO,EACN,KAAK,oBAAoB,EAGzB,MAAM,gBAAgB,CAAA;AACvB,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAA;AAG3D;;;GAGG;AACH,MAAM,MAAM,eAAe,GAAG;IAC7B,SAAS,EAAE,MAAM,CAAA;IACjB,SAAS,EAAE,MAAM,GAAG,UAAU,CAAA;IAC9B,IAAI,EAAE,OAAO,CAAA;CACb,CAAA;AAID,qBAAa,aAAa;IACzB,OAAO,CAAC,SAAS,CAIhB;IACD,OAAO,CAAC,SAAS,CAAwB;IACzC,OAAO,CAAC,UAAU,CAAwB;IAC1C,OAAO,CAAC,mBAAmB,CAAwB;IACnD,OAAO,CAAC,aAAa,CAAI;IACzB,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAEjC;IACD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAE9B;IACD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAE9B;IACD,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAEhC;IACD,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAEjC;IACD,OAAO,CAAC,cAAc,CAAQ;IAC9B,OAAO,CAAC,eAAe,CAAQ;IAC/B,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,iBAAiB,CAAQ;IACjC,OAAO,CAAC,uBAAuB,CAAgB;IAC/C,OAAO,CAAC,wBAAwB,CAA+B;IAC/D,OAAO,CAAC,WAAW,CAAQ;IAC3B,OAAO,CAAC,WAAW,CAAQ;IAC3B,OAAO,CAAC,WAAW,CAA0B;IAC7C,OAAO,CAAC,UAAU,CAAuC;IACzD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAK3B;IAED,cAAc;IAQd,OAAO,CAAC,MAAM,CAAC,eAAe,CAAM;IAEpC,OAAO,CAAC,UAAU;IAclB,aAAa,IAAI,SAAS,eAAe,EAAE;IAI3C,qBAAqB,CAAC,QAAQ,EAAE,MAAM,IAAI;IAK1C,SAAS,CAAC,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB9C,WAAW,CAAC,cAAc,EAAE,kBAAkB;IAkC9C,OAAO,CAAC,UAAU;IAalB,OAAO,CAAC,UAAU;IAKlB,OAAO,CAAC,4BAA4B;IAQpC,OAAO,CAAC,cAAc;IAMtB,OAAO,CAAC,MAAM,CAIb;IAED,OAAO,CAAC,aAAa,CAgRpB;IAED,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI;IAK9B,YAAY,IAAI,oBAAoB;IAIpC;;;;;OAKG;IACH,YAAY,CAAC,OAAO,EAAE,WAAW;IAKjC,sBAAsB;IAmBtB,OAAO,CAAC,mBAAmB;IAmB3B,OAAO,CAAC,SAAS;IAkEjB;;;OAGG;IACH,OAAO,CAAC,kBAAkB;IAyB1B,8BAA8B,IAAI,MAAM;IAIxC,eAAe,CACd,cAAc,EAAE,MAAM,EACtB,GAAG,EAAE,MAAM,EACX,OAAO,GAAE,oBAAyB;IA+EnC,OAAO,CAAC,uBAAuB;IAwB/B,6BAA6B,CAAC,cAAc,EAAE,MAAM;IAepD,UAAU,CAAC,MAAM,EAAE,MAAM;IAazB,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA8B3D,OAAO,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;IAYrD,OAAO,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;IAYrD,SAAS,CAAC,IAAI,GAAE,aAAkB,GAAG,OAAO,CAAC,eAAe,CAAC;IAa7D,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA2C3D;;;;OAIG;IACH,oBAAoB,CAAC,IAAI,EAAE;QAC1B,UAAU,EAAE,gBAAgB,CAAA;QAC5B,MAAM,EAAE,YAAY,CAAA;QACpB,cAAc,EAAE,8BAA8B,CAAA;KAC9C,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAuB7B;;;;OAIG;IACH,OAAO,CAAC,uBAAuB;CAkE/B"}
+{"version":3,"file":"SandboxBridge.d.ts","sourceRoot":"","sources":["../../src/bridge/SandboxBridge.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACX,gBAAgB,EAEhB,MAAM,4DAA4D,CAAA;AACnE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAA;AAc3E,OAAO,EAEN,KAAK,WAAW,EAEhB,MAAM,mDAAmD,CAAA;AAe1D,OAAO,KAAK,EACX,cAAc,EACd,gBAAgB,EAChB,aAAa,EACb,aAAa,EACb,aAAa,EACb,eAAe,EACf,8BAA8B,EAC9B,YAAY,EACZ,cAAc,EACd,gBAAgB,EAChB,oBAAoB,EACpB,MAAM,aAAa,CAAA;AAUpB,OAAO,EACN,KAAK,oBAAoB,EAGzB,MAAM,gBAAgB,CAAA;AACvB,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAA;AAG3D;;;GAGG;AACH,MAAM,MAAM,eAAe,GAAG;IAC7B,SAAS,EAAE,MAAM,CAAA;IACjB,SAAS,EAAE,MAAM,GAAG,UAAU,CAAA;IAC9B,IAAI,EAAE,OAAO,CAAA;CACb,CAAA;AAaD,qBAAa,aAAa;IACzB,OAAO,CAAC,SAAS,CAIhB;IACD,OAAO,CAAC,SAAS,CAAwB;IACzC,OAAO,CAAC,UAAU,CAAwB;IAC1C,OAAO,CAAC,mBAAmB,CAAwB;IACnD,OAAO,CAAC,aAAa,CAAI;IACzB,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAEjC;IACD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAE9B;IACD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAE9B;IACD,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAEhC;IACD,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAEjC;IACD,OAAO,CAAC,cAAc,CAAQ;IAC9B,OAAO,CAAC,eAAe,CAAQ;IAC/B,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,iBAAiB,CAAQ;IACjC,OAAO,CAAC,uBAAuB,CAAgB;IAC/C,OAAO,CAAC,wBAAwB,CAA+B;IAC/D,OAAO,CAAC,WAAW,CAAQ;IAC3B,OAAO,CAAC,WAAW,CAAQ;IAC3B,OAAO,CAAC,WAAW,CAA0B;IAC7C,OAAO,CAAC,UAAU,CAAuC;IACzD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAK3B;IAED,cAAc;IAQd,OAAO,CAAC,MAAM,CAAC,eAAe,CAAM;IAEpC,OAAO,CAAC,UAAU;IAclB,aAAa,IAAI,SAAS,eAAe,EAAE;IAI3C,qBAAqB,CAAC,QAAQ,EAAE,MAAM,IAAI;IAK1C,SAAS,CAAC,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB9C,WAAW,CAAC,cAAc,EAAE,kBAAkB;IAkC9C,OAAO,CAAC,UAAU;IAalB,OAAO,CAAC,UAAU;IA+BlB,OAAO,CAAC,4BAA4B;IAQpC,OAAO,CAAC,cAAc;IAMtB,OAAO,CAAC,MAAM,CAIb;IAED,OAAO,CAAC,aAAa,CAKpB;IAED,OAAO,CAAC,cAAc;IA+QtB,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI;IAK9B,YAAY,IAAI,oBAAoB;IAIpC;;;;;OAKG;IACH,YAAY,CAAC,OAAO,EAAE,WAAW;IAKjC,sBAAsB;IAmBtB,OAAO,CAAC,mBAAmB;IAmB3B,OAAO,CAAC,SAAS;IAkEjB;;;OAGG;IACH,OAAO,CAAC,kBAAkB;IAyB1B,8BAA8B,IAAI,MAAM;IAIxC,eAAe,CACd,cAAc,EAAE,MAAM,EACtB,GAAG,EAAE,MAAM,EACX,OAAO,GAAE,oBAAyB;IAgFnC,OAAO,CAAC,uBAAuB;IAwB/B,6BAA6B,CAAC,cAAc,EAAE,MAAM;IAepD,UAAU,CAAC,MAAM,EAAE,MAAM;IAazB,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA8B3D,OAAO,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;IAYrD,OAAO,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;IAYrD,SAAS,CAAC,IAAI,GAAE,aAAkB,GAAG,OAAO,CAAC,eAAe,CAAC;IAa7D,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA2C3D;;;;OAIG;IACH,oBAAoB,CAAC,IAAI,EAAE;QAC1B,UAAU,EAAE,gBAAgB,CAAA;QAC5B,MAAM,EAAE,YAAY,CAAA;QACpB,cAAc,EAAE,8BAA8B,CAAA;KAC9C,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAuB7B;;;;OAIG;IACH,OAAO,CAAC,uBAAuB;CAkE/B"}

package/dist/bridge/SandboxBridge.js

@@ -1,9 +1,10 @@
 import { DEFAULT_CONTRAST_MODE } from "../protocol/contrast.js";
 import { readIncomingType } from "../protocol/incomingType.js";
-import { hostToSandboxMessageSchema } from "../protocol/messages/hostToSandbox.js";
+import { hostToSandboxMessageSchema, } from "../protocol/messages/hostToSandbox.js";
 import { CustomBlockInitializationError, initMessageSchema, } from "../protocol/messages/init.js";
 import { CUSTOM_BLOCK_BRIDGE_PROTOCOL_VERSION } from "../protocol/protocolVersion.js";
 import * as v from "valibot";
+import { globalPrimedResponsesCache } from "../testing.js";
 import { unreachable } from "../utils.js";
 import { CUSTOM_BLOCKS_SDK_VERSION } from "../version.js";
 import { NOTION_DARK_BACKGROUND_BASE, NOTION_LIGHT_BACKGROUND_BASE, } from "./appearance.js";
@@ -12,6 +13,14 @@ import { resolveDataSources } from "./dataSources/resolve.js";
 import { resolvePropertyWriteMapForDataSource } from "./dataSources/resolveProperty.js";
 import { createEmptyDataSourceQueryState, } from "./hostState.js";
 import { PendingRequests } from "./pendingRequests.js";
+const RESPONSE_TYPE_BY_REQUEST = new Map([
+    ["getPage", "getPageResult"],
+    ["createPage", "createPageResult"],
+    ["updatePage", "updatePageResult"],
+    ["getUser", "getUserResult"],
+    ["listUsers", "listUsersResult"],
+    ["queryDataSource", "queryDataSourceResult"],
+]);
 const INIT_RESULT_TIMER_FALLBACK_MS = 100;
 export class SandboxBridge {
     constructor() {
@@ -49,226 +58,7 @@ export class SandboxBridge {
             if (event.source !== window.parent) {
                 return;
             }
-            this.logMessage("received", event.data);
-            const parsed = v.safeParse(hostToSandboxMessageSchema, event.data);
-            if (!parsed.success) {
-                console.warn("[custom-blocks-sdk] ignoring malformed host message", parsed.issues);
-                const incomingType = readIncomingType(event.data);
-                if (!this.hasReceivedInit &&
-                    incomingType === "init" &&
-                    this.initializationId !== undefined &&
-                    readInitializationId(event.data) === this.initializationId) {
-                    this.hasReceivedInit = true;
-                    this.sendInitResultError(invalidInitPayloadError(parsed.issues));
-                    return;
-                }
-                // Loop guard: never NACK a NACK. Excludes both directions —
-                // `invalidSandboxMessage` is what the host normally sends, but
-                // a buggy host that echoes our own `invalidHostMessage` back
-                // would otherwise spin a NACK feedback loop.
-                if (incomingType !== "invalidSandboxMessage" &&
-                    incomingType !== "invalidHostMessage") {
-                    const nack = {
-                        type: "invalidHostMessage",
-                        reason: formatInvalidHostReason(incomingType, parsed.issues),
-                    };
-                    this.postToHost(nack);
-                }
-                return;
-            }
-            const message = parsed.output;
-            // If the host couldn't parse one of our outbound sandbox-to-host messages, it sends back
-            // an `invalidSandboxMessage` NACK with a human-readable `reason`. Log it for developer
-            // visibility and stop. We don't retry as the sandbox can't recover from a host-side parse
-            // failure on its own.
-            if (message.type === "invalidSandboxMessage") {
-                console.warn("[custom-blocks-sdk] host reported invalid sandbox message:", message.reason);
-                return;
-            }
-            // `init` is the only message valid before initialization. Handle it up
-            // front so every later case can assume `status === "initialized"`.
-            if (message.type === "init") {
-                if (message.initializationId !== this.initializationId) {
-                    console.warn(`[custom-blocks-sdk] ignoring init for unknown initializationId ${message.initializationId}`);
-                    return;
-                }
-                if (this.hasReceivedInit) {
-                    console.warn("[custom-blocks-sdk] ignoring duplicate init message");
-                    return;
-                }
-                this.hasReceivedInit = true;
-                this.applyInit(message, true);
-                return;
-            }
-            // Alias to keep TS's narrowed `InitializedHostState` across the switch
-            // below. Reading `this.hostState` repeatedly would re-widen it.
-            const hostState = this.hostState;
-            if (hostState.status !== "initialized") {
-                console.warn(`[custom-blocks-sdk] ignoring ${message.type} before init`);
-                return;
-            }
-            switch (message.type) {
-                case "themeChanged": {
-                    this.hostState = {
-                        ...hostState,
-                        theme: message.theme,
-                    };
-                    syncDocumentAppearance({
-                        theme: message.theme,
-                        contrastMode: hostState.contrastMode,
-                    });
-                    this.notify();
-                    return;
-                }
-                case "contrastModeChanged": {
-                    this.hostState = {
-                        ...hostState,
-                        contrastMode: message.contrastMode,
-                    };
-                    syncDocumentAppearance({
-                        theme: hostState.theme,
-                        contrastMode: message.contrastMode,
-                    });
-                    this.notify();
-                    return;
-                }
-                case "parentChanged": {
-                    this.hostState = {
-                        ...hostState,
-                        parent: message.parent,
-                    };
-                    this.notify();
-                    return;
-                }
-                case "pageChanged": {
-                    this.hostState = {
-                        ...hostState,
-                        page: message.page,
-                    };
-                    this.notify();
-                    return;
-                }
-                case "currentUserChanged": {
-                    this.hostState = {
-                        ...hostState,
-                        currentUser: message.currentUser,
-                    };
-                    this.notify();
-                    return;
-                }
-                case "dataSourcesChanged": {
-                    const nextBindings = message.dataSources.bindings;
-                    const dataSources = reuseDataSourcesForUnchangedBindings({
-                        previousDataSources: hostState.dataSources,
-                        previousBindings: this.latestDataSourceBindings,
-                        nextDataSources: this.isMockState
-                            ? resolveMockDataSources(nextBindings)
-                            : resolveDataSources({
-                                manifest: hostState.manifest,
-                                dataSourceBindings: nextBindings,
-                            }),
-                        nextBindings,
-                    });
-                    this.latestDataSourceBindings = nextBindings;
-                    // Drop cached query state for subscriptions whose key no longer exists.
-                    const nextKeys = new Set(dataSources.map(s => s.key));
-                    const prunedState = {};
-                    for (const [subscriptionId, state] of Object.entries(hostState.dataSourceState)) {
-                        if (nextKeys.has(state.dataSourceKey)) {
-                            prunedState[subscriptionId] = state;
-                        }
-                    }
-                    this.hostState = {
-                        ...hostState,
-                        dataSources,
-                        dataSourceState: prunedState,
-                    };
-                    this.notify();
-                    return;
-                }
-                case "createPageResult": {
-                    const result = message.status === "success"
-                        ? { status: "success", page: message.page }
-                        : { status: "error", error: message.error };
-                    if (!this.pendingCreatePage.resolve(message.requestId, result)) {
-                        console.warn(`[custom-blocks-sdk] createPageResult for unknown requestId ${message.requestId}`);
-                    }
-                    return;
-                }
-                case "getPageResult": {
-                    const result = message.status === "success"
-                        ? { status: "success", page: message.page }
-                        : { status: "error", error: message.error };
-                    if (!this.pendingGetPage.resolve(message.requestId, result)) {
-                        console.warn(`[custom-blocks-sdk] getPageResult for unknown requestId ${message.requestId}`);
-                    }
-                    return;
-                }
-                case "getUserResult": {
-                    const result = message.status === "success"
-                        ? { status: "success", user: message.user }
-                        : { status: "error", error: message.error };
-                    if (!this.pendingGetUser.resolve(message.requestId, result)) {
-                        console.warn(`[custom-blocks-sdk] getUserResult for unknown requestId ${message.requestId}`);
-                    }
-                    return;
-                }
-                case "listUsersResult": {
-                    const result = message.status === "success"
-                        ? { status: "success", list: message.list }
-                        : { status: "error", error: message.error };
-                    if (!this.pendingListUsers.resolve(message.requestId, result)) {
-                        console.warn(`[custom-blocks-sdk] listUsersResult for unknown requestId ${message.requestId}`);
-                    }
-                    return;
-                }
-                case "updatePageResult": {
-                    const result = message.status === "success"
-                        ? { status: "success", page: message.page }
-                        : { status: "error", error: message.error };
-                    if (!this.pendingUpdatePage.resolve(message.requestId, result)) {
-                        console.warn(`[custom-blocks-sdk] updatePageResult for unknown requestId ${message.requestId}`);
-                    }
-                    return;
-                }
-                case "queryDataSourceResult": {
-                    const currentState = hostState.dataSourceState[message.subscriptionId];
-                    if (currentState === undefined) {
-                        return;
-                    }
-                    const queryResult = message.status === "error"
-                        ? {
-                            items: [],
-                            hasMore: false,
-                            error: message.error,
-                        }
-                        : {
-                            items: message.items,
-                            hasMore: message.hasMore,
-                            error: undefined,
-                        };
-                    this.hostState = {
-                        ...hostState,
-                        dataSourceState: {
-                            ...hostState.dataSourceState,
-                            [message.subscriptionId]: {
-                                dataSourceKey: currentState.dataSourceKey,
-                                items: queryResult.items,
-                                isLoading: false,
-                                hasMore: queryResult.hasMore,
-                                error: queryResult.error,
-                                latestLimit: currentState.latestLimit,
-                                latestQueryIdentity: currentState.latestQueryIdentity,
-                            },
-                        },
-                    };
-                    this.notify();
-                    return;
-                }
-                default: {
-                    unreachable(message);
-                }
-            }
+            this.receiveMessage(event.data);
         };
     }
     startListening() {
@@ -361,6 +151,28 @@ export class SandboxBridge {
         this.sendToHost(message);
     }
     sendToHost(message) {
+        // Check if the message is a primed response
+        const messageType = readIncomingType(message);
+        if (messageType !== undefined &&
+            typeof message === "object" &&
+            message !== null) {
+            const responseType = RESPONSE_TYPE_BY_REQUEST.get(messageType);
+            const primed = responseType && globalPrimedResponsesCache.get(responseType);
+            if (responseType !== undefined && primed !== undefined) {
+                globalPrimedResponsesCache.delete(responseType);
+                const response = {
+                    ...primed,
+                    type: responseType,
+                    ...("requestId" in message ? { requestId: message.requestId } : {}),
+                    ...("subscriptionId" in message
+                        ? { subscriptionId: message.subscriptionId }
+                        : {}),
+                };
+                queueMicrotask(() => this.receiveMessage(response));
+                return;
+            }
+        }
+        // If no primed response is found, proceed as normal
         this.logMessage("sent", message);
         window.parent.postMessage(message, "*");
     }
@@ -376,6 +188,228 @@ export class SandboxBridge {
             ? message.type
             : undefined;
     }
+    receiveMessage(data) {
+        this.logMessage("received", data);
+        const parsed = v.safeParse(hostToSandboxMessageSchema, data);
+        if (!parsed.success) {
+            console.warn("[custom-blocks-sdk] ignoring malformed host message", parsed.issues);
+            const incomingType = readIncomingType(data);
+            if (!this.hasReceivedInit &&
+                incomingType === "init" &&
+                this.initializationId !== undefined &&
+                readInitializationId(data) === this.initializationId) {
+                this.hasReceivedInit = true;
+                this.sendInitResultError(invalidInitPayloadError(parsed.issues));
+                return;
+            }
+            // Loop guard: never NACK a NACK. Excludes both directions —
+            // `invalidSandboxMessage` is what the host normally sends, but
+            // a buggy host that echoes our own `invalidHostMessage` back
+            // would otherwise spin a NACK feedback loop.
+            if (incomingType !== "invalidSandboxMessage" &&
+                incomingType !== "invalidHostMessage") {
+                const nack = {
+                    type: "invalidHostMessage",
+                    reason: formatInvalidHostReason(incomingType, parsed.issues),
+                };
+                this.postToHost(nack);
+            }
+            return;
+        }
+        const message = parsed.output;
+        // If the host couldn't parse one of our outbound sandbox-to-host messages, it sends back
+        // an `invalidSandboxMessage` NACK with a human-readable `reason`. Log it for developer
+        // visibility and stop. We don't retry as the sandbox can't recover from a host-side parse
+        // failure on its own.
+        if (message.type === "invalidSandboxMessage") {
+            console.warn("[custom-blocks-sdk] host reported invalid sandbox message:", message.reason);
+            return;
+        }
+        // `init` is the only message valid before initialization. Handle it up
+        // front so every later case can assume `status === "initialized"`.
+        if (message.type === "init") {
+            if (message.initializationId !== this.initializationId) {
+                console.warn(`[custom-blocks-sdk] ignoring init for unknown initializationId ${message.initializationId}`);
+                return;
+            }
+            if (this.hasReceivedInit) {
+                console.warn("[custom-blocks-sdk] ignoring duplicate init message");
+                return;
+            }
+            this.hasReceivedInit = true;
+            this.applyInit(message, true);
+            return;
+        }
+        // Alias to keep TS's narrowed `InitializedHostState` across the switch
+        // below. Reading `this.hostState` repeatedly would re-widen it.
+        const hostState = this.hostState;
+        if (hostState.status !== "initialized") {
+            console.warn(`[custom-blocks-sdk] ignoring ${message.type} before init`);
+            return;
+        }
+        switch (message.type) {
+            case "themeChanged": {
+                this.hostState = {
+                    ...hostState,
+                    theme: message.theme,
+                };
+                syncDocumentAppearance({
+                    theme: message.theme,
+                    contrastMode: hostState.contrastMode,
+                });
+                this.notify();
+                return;
+            }
+            case "contrastModeChanged": {
+                this.hostState = {
+                    ...hostState,
+                    contrastMode: message.contrastMode,
+                };
+                syncDocumentAppearance({
+                    theme: hostState.theme,
+                    contrastMode: message.contrastMode,
+                });
+                this.notify();
+                return;
+            }
+            case "parentChanged": {
+                this.hostState = {
+                    ...hostState,
+                    parent: message.parent,
+                };
+                this.notify();
+                return;
+            }
+            case "pageChanged": {
+                this.hostState = {
+                    ...hostState,
+                    page: message.page,
+                };
+                this.notify();
+                return;
+            }
+            case "currentUserChanged": {
+                this.hostState = {
+                    ...hostState,
+                    currentUser: message.currentUser,
+                };
+                this.notify();
+                return;
+            }
+            case "dataSourcesChanged": {
+                const nextBindings = message.dataSources.bindings;
+                const dataSources = reuseDataSourcesForUnchangedBindings({
+                    previousDataSources: hostState.dataSources,
+                    previousBindings: this.latestDataSourceBindings,
+                    nextDataSources: this.isMockState
+                        ? resolveMockDataSources(nextBindings)
+                        : resolveDataSources({
+                            manifest: hostState.manifest,
+                            dataSourceBindings: nextBindings,
+                        }),
+                    nextBindings,
+                });
+                this.latestDataSourceBindings = nextBindings;
+                // Drop cached query state for subscriptions whose key no longer exists.
+                const nextKeys = new Set(dataSources.map(s => s.key));
+                const prunedState = {};
+                for (const [subscriptionId, state] of Object.entries(hostState.dataSourceState)) {
+                    if (nextKeys.has(state.dataSourceKey)) {
+                        prunedState[subscriptionId] = state;
+                    }
+                }
+                this.hostState = {
+                    ...hostState,
+                    dataSources,
+                    dataSourceState: prunedState,
+                };
+                this.notify();
+                return;
+            }
+            case "createPageResult": {
+                const result = message.status === "success"
+                    ? { status: "success", page: message.page }
+                    : { status: "error", error: message.error };
+                if (!this.pendingCreatePage.resolve(message.requestId, result)) {
+                    console.warn(`[custom-blocks-sdk] createPageResult for unknown requestId ${message.requestId}`);
+                }
+                return;
+            }
+            case "getPageResult": {
+                const result = message.status === "success"
+                    ? { status: "success", page: message.page }
+                    : { status: "error", error: message.error };
+                if (!this.pendingGetPage.resolve(message.requestId, result)) {
+                    console.warn(`[custom-blocks-sdk] getPageResult for unknown requestId ${message.requestId}`);
+                }
+                return;
+            }
+            case "getUserResult": {
+                const result = message.status === "success"
+                    ? { status: "success", user: message.user }
+                    : { status: "error", error: message.error };
+                if (!this.pendingGetUser.resolve(message.requestId, result)) {
+                    console.warn(`[custom-blocks-sdk] getUserResult for unknown requestId ${message.requestId}`);
+                }
+                return;
+            }
+            case "listUsersResult": {
+                const result = message.status === "success"
+                    ? { status: "success", list: message.list }
+                    : { status: "error", error: message.error };
+                if (!this.pendingListUsers.resolve(message.requestId, result)) {
+                    console.warn(`[custom-blocks-sdk] listUsersResult for unknown requestId ${message.requestId}`);
+                }
+                return;
+            }
+            case "updatePageResult": {
+                const result = message.status === "success"
+                    ? { status: "success", page: message.page }
+                    : { status: "error", error: message.error };
+                if (!this.pendingUpdatePage.resolve(message.requestId, result)) {
+                    console.warn(`[custom-blocks-sdk] updatePageResult for unknown requestId ${message.requestId}`);
+                }
+                return;
+            }
+            case "queryDataSourceResult": {
+                const currentState = hostState.dataSourceState[message.subscriptionId];
+                if (currentState === undefined) {
+                    return;
+                }
+                const queryResult = message.status === "error"
+                    ? {
+                        items: [],
+                        hasMore: false,
+                        error: message.error,
+                    }
+                    : {
+                        items: message.items,
+                        hasMore: message.hasMore,
+                        error: undefined,
+                    };
+                this.hostState = {
+                    ...hostState,
+                    dataSourceState: {
+                        ...hostState.dataSourceState,
+                        [message.subscriptionId]: {
+                            dataSourceKey: currentState.dataSourceKey,
+                            items: queryResult.items,
+                            isLoading: false,
+                            hasMore: queryResult.hasMore,
+                            error: queryResult.error,
+                            latestLimit: currentState.latestLimit,
+                            latestQueryIdentity: currentState.latestQueryIdentity,
+                        },
+                    },
+                };
+                this.notify();
+                return;
+            }
+            default: {
+                unreachable(message);
+            }
+        }
+    }
     subscribe(listener) {
         this.listeners.add(listener);
         return () => this.listeners.delete(listener);
@@ -580,6 +614,7 @@ export class SandboxBridge {
             dataSourceId: query.dataSourceId,
             limit: query.limit,
             ...(query.filter !== undefined ? { filter: query.filter } : {}),
+            ...(query.sorts !== undefined ? { sorts: query.sorts } : {}),
         };
         this.postToHost(outbound);
     }

package/dist/bridge/appearance.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"appearance.d.ts","sourceRoot":"","sources":["../../../../home/runner/work/custom-blocks/custom-blocks/sdk/src/bridge/appearance.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,eAAO,MAAM,2BAA2B,YAAY,CAAA;AACpD,eAAO,MAAM,4BAA4B,YAAY,CAAA"}
+{"version":3,"file":"appearance.d.ts","sourceRoot":"","sources":["../../src/bridge/appearance.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,eAAO,MAAM,2BAA2B,YAAY,CAAA;AACpD,eAAO,MAAM,4BAA4B,YAAY,CAAA"}

package/dist/bridge/dataSources/query.d.ts

@@ -1,11 +1,12 @@
 import type { NotionDataSource } from "../../protocol/dataSources/dataSource.js";
 import type { NotionDataSourceId } from "../../protocol/ids.js";
-import { type ResolvedCustomBlockDataSourceFilter } from "../../protocol/messages/queryDataSource.js";
+import { type ResolvedCustomBlockDataSourceFilter, type ResolvedCustomBlockDataSourceSort } from "../../protocol/messages/queryDataSource.js";
 import type { UseDataSourceOptions } from "../../types.js";
 export type ResolvedDataSourceQuery = {
     dataSourceId: NotionDataSourceId;
     limit: number;
     filter?: ResolvedCustomBlockDataSourceFilter;
+    sorts?: ResolvedCustomBlockDataSourceSort[];
     identity: string;
 };
 export type ResolveDataSourceQueryResult = {

package/dist/bridge/dataSources/query.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"query.d.ts","sourceRoot":"","sources":["../../../../../home/runner/work/custom-blocks/custom-blocks/sdk/src/bridge/dataSources/query.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,4DAA4D,CAAA;AAClG,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAA;AACjF,OAAO,EAON,KAAK,mCAAmC,EAExC,MAAM,8DAA8D,CAAA;AAErE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAA;AAM1D,MAAM,MAAM,uBAAuB,GAAG;IACrC,YAAY,EAAE,kBAAkB,CAAA;IAChC,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,CAAC,EAAE,mCAAmC,CAAA;IAC5C,QAAQ,EAAE,MAAM,CAAA;CAChB,CAAA;AAED,MAAM,MAAM,4BAA4B,GACrC;IAAE,MAAM,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAClC;IACA,MAAM,EAAE,IAAI,CAAA;IACZ,UAAU,EAAE,gBAAgB,CAAA;IAC5B,KAAK,EAAE,uBAAuB,CAAA;CAC7B,CAAA;AAEJ,wBAAgB,sBAAsB,CAAC,IAAI,EAAE;IAC5C,WAAW,EAAE,gBAAgB,EAAE,CAAA;IAC/B,GAAG,EAAE,MAAM,CAAA;IACX,OAAO,CAAC,EAAE,OAAO,CAAA;IACjB,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAA;CAChC,GAAG,4BAA4B,CAuD/B;AA6BD,wBAAgB,iCAAiC,CAChD,OAAO,EAAE,oBAAoB,GAAG,SAAS,GACvC,MAAM,CAMR"}
+{"version":3,"file":"query.d.ts","sourceRoot":"","sources":["../../../src/bridge/dataSources/query.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,4DAA4D,CAAA;AAElG,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAA;AACjF,OAAO,EAQN,KAAK,mCAAmC,EAExC,KAAK,iCAAiC,EACtC,MAAM,8DAA8D,CAAA;AAErE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAA;AAU1D,MAAM,MAAM,uBAAuB,GAAG;IACrC,YAAY,EAAE,kBAAkB,CAAA;IAChC,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,CAAC,EAAE,mCAAmC,CAAA;IAC5C,KAAK,CAAC,EAAE,iCAAiC,EAAE,CAAA;IAC3C,QAAQ,EAAE,MAAM,CAAA;CAChB,CAAA;AAED,MAAM,MAAM,4BAA4B,GACrC;IAAE,MAAM,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAClC;IACA,MAAM,EAAE,IAAI,CAAA;IACZ,UAAU,EAAE,gBAAgB,CAAA;IAC5B,KAAK,EAAE,uBAAuB,CAAA;CAC7B,CAAA;AAYJ,wBAAgB,sBAAsB,CAAC,IAAI,EAAE;IAC5C,WAAW,EAAE,gBAAgB,EAAE,CAAA;IAC/B,GAAG,EAAE,MAAM,CAAA;IACX,OAAO,CAAC,EAAE,OAAO,CAAA;IACjB,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAA;CAChC,GAAG,4BAA4B,CAmD/B;AAiFD,wBAAgB,iCAAiC,CAChD,OAAO,EAAE,oBAAoB,GAAG,SAAS,GACvC,MAAM,CAMR"}