@notionhq/custom-blocks-host 0.1.29 → 0.1.31

package/README.md

@@ -215,7 +215,6 @@ The returned handle exposes:
 - `setDataSources()`
 - `setCurrentUser()`
 - `refreshQuery()`
-- `post()`
 - `stop()`
 
 State updates are sent after `initResult.success`. Updates made earlier are held until

package/dist/createCustomBlockHost.d.ts

@@ -1,7 +1,6 @@
 import { type NotionContrastMode } from "./protocol/contrast.js";
 import type { NotionDataSourceId } from "./protocol/ids.js";
 import type { CustomBlockManifest } from "./protocol/manifest.js";
-import type { HostToSandboxMessage } from "./protocol/messages/hostToSandbox.js";
 import type { CustomBlockPage } from "./protocol/pages/page.js";
 import type { NotionParent } from "./protocol/parent.js";
 import type { NotionTheme } from "./protocol/theme.js";
@@ -42,7 +41,6 @@ export type CustomBlockHostOptions = {
 };
 export type CustomBlockHostHandle = {
     stop: () => void;
-    post: (message: HostToSandboxMessage) => void;
     setTheme: (theme: NotionTheme) => void;
     setContrastMode: (contrastMode: NotionContrastMode) => void;
     setParent: (parent: NotionParent) => void;

package/dist/createCustomBlockHost.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"createCustomBlockHost.d.ts","sourceRoot":"","sources":["../src/createCustomBlockHost.ts"],"names":[],"mappings":"AAAA,OAAO,EAEN,KAAK,kBAAkB,EACvB,MAAM,8CAA8C,CAAA;AAErD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAA;AACjF,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAA;AAavF,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,4DAA4D,CAAA;AAoBtG,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gDAAgD,CAAA;AACrF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4CAA4C,CAAA;AAE9E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2CAA2C,CAAA;AAC5E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gDAAgD,CAAA;AAIhF,OAAO,KAAK,EACX,iCAAiC,EACjC,2BAA2B,EAC3B,MAAM,sBAAsB,CAAA;AAE7B,OAAO,KAAK,EACX,sBAAsB,EACtB,uBAAuB,EACvB,2BAA2B,EAC3B,MAAM,qBAAqB,CAAA;AAc5B,MAAM,MAAM,sBAAsB,GAAG;IACpC,MAAM,EAAE,iBAAiB,CAAA;IACzB,aAAa,EAAE,MAAM,CAAA;IACrB;;;;OAIG;IACH,wBAAwB,CAAC,EAAE,MAAM,CAAA;IACjC,kBAAkB,CAAC,EAAE,MAAM,CAAA;IAC3B,qBAAqB,CAAC,EAAE,MAAM,CAAA;IAC9B,YAAY,EAAE,2BAA2B,CAAA;IACzC,QAAQ,EAAE,uBAAuB,CAAA;IACjC;;;OAGG;IACH,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,mBAAmB,KAAK,IAAI,CAAA;IACpD,gGAAgG;IAChG,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE;QACvB,QAAQ,EAAE,mBAAmB,CAAA;QAC7B,aAAa,CAAC,EAAE,MAAM,CAAA;KACtB,KAAK,IAAI,CAAA;IACV;;;;;OAKG;IACH,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,IAAI,CAAA;IACnD,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,2BAA2B,EAAE,OAAO,EAAE,OAAO,KAAK,IAAI,CAAA;CAC1E,CAAA;AAED,MAAM,MAAM,qBAAqB,GAAG;IACnC,IAAI,EAAE,MAAM,IAAI,CAAA;IAChB,IAAI,EAAE,CAAC,OAAO,EAAE,oBAAoB,KAAK,IAAI,CAAA;IAC7C,QAAQ,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAA;IACtC,eAAe,EAAE,CAAC,YAAY,EAAE,kBAAkB,KAAK,IAAI,CAAA;IAC3D,SAAS,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,IAAI,CAAA;IACzC,OAAO,EAAE,CAAC,IAAI,EAAE,eAAe,KAAK,IAAI,CAAA;IACxC,cAAc,EAAE,CAAC,WAAW,EAAE,iCAAiC,KAAK,IAAI,CAAA;IACxE,cAAc,EAAE,CAAC,WAAW,EAAE,UAAU,KAAK,IAAI,CAAA;IACjD,YAAY,EAAE,CAAC,IAAI,EAAE;QAAE,YAAY,EAAE,kBAAkB,CAAA;KAAE,KAAK,IAAI,CAAA;CAClE,CAAA;AAED,wBAAgB,qBAAqB,CACpC,OAAO,EAAE,sBAAsB,GAC7B,qBAAqB,CA+avB"}
+{"version":3,"file":"createCustomBlockHost.d.ts","sourceRoot":"","sources":["../src/createCustomBlockHost.ts"],"names":[],"mappings":"AAAA,OAAO,EAEN,KAAK,kBAAkB,EACvB,MAAM,8CAA8C,CAAA;AAErD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAA;AACjF,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAA;AAmBvF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gDAAgD,CAAA;AACrF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4CAA4C,CAAA;AAE9E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2CAA2C,CAAA;AAC5E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gDAAgD,CAAA;AAIhF,OAAO,KAAK,EACX,iCAAiC,EACjC,2BAA2B,EAC3B,MAAM,sBAAsB,CAAA;AAE7B,OAAO,KAAK,EACX,sBAAsB,EACtB,uBAAuB,EACvB,2BAA2B,EAC3B,MAAM,qBAAqB,CAAA;AAe5B,MAAM,MAAM,sBAAsB,GAAG;IACpC,MAAM,EAAE,iBAAiB,CAAA;IACzB,aAAa,EAAE,MAAM,CAAA;IACrB;;;;OAIG;IACH,wBAAwB,CAAC,EAAE,MAAM,CAAA;IACjC,kBAAkB,CAAC,EAAE,MAAM,CAAA;IAC3B,qBAAqB,CAAC,EAAE,MAAM,CAAA;IAC9B,YAAY,EAAE,2BAA2B,CAAA;IACzC,QAAQ,EAAE,uBAAuB,CAAA;IACjC;;;OAGG;IACH,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,mBAAmB,KAAK,IAAI,CAAA;IACpD,gGAAgG;IAChG,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE;QACvB,QAAQ,EAAE,mBAAmB,CAAA;QAC7B,aAAa,CAAC,EAAE,MAAM,CAAA;KACtB,KAAK,IAAI,CAAA;IACV;;;;;OAKG;IACH,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,IAAI,CAAA;IACnD,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,2BAA2B,EAAE,OAAO,EAAE,OAAO,KAAK,IAAI,CAAA;CAC1E,CAAA;AAED,MAAM,MAAM,qBAAqB,GAAG;IACnC,IAAI,EAAE,MAAM,IAAI,CAAA;IAChB,QAAQ,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAA;IACtC,eAAe,EAAE,CAAC,YAAY,EAAE,kBAAkB,KAAK,IAAI,CAAA;IAC3D,SAAS,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,IAAI,CAAA;IACzC,OAAO,EAAE,CAAC,IAAI,EAAE,eAAe,KAAK,IAAI,CAAA;IACxC,cAAc,EAAE,CAAC,WAAW,EAAE,iCAAiC,KAAK,IAAI,CAAA;IACxE,cAAc,EAAE,CAAC,WAAW,EAAE,UAAU,KAAK,IAAI,CAAA;IACjD,YAAY,EAAE,CAAC,IAAI,EAAE;QAAE,YAAY,EAAE,kBAAkB,CAAA;KAAE,KAAK,IAAI,CAAA;CAClE,CAAA;AAyBD,wBAAgB,qBAAqB,CACpC,OAAO,EAAE,sBAAsB,GAC7B,qBAAqB,CA4ZvB"}

package/dist/createCustomBlockHost.js

@@ -8,6 +8,7 @@ import { MIN_HOST_SUPPORTED_BRIDGE_PROTOCOL_VERSION } from "./protocolVersion.js
 import { QuerySubscriptions, } from "./queries/querySubscriptions.js";
 import { validateQueryDataSourceFilter } from "./queries/validateQueryDataSourceFilter.js";
 import { validateQueryDataSourceSorts } from "./queries/validateQueryDataSourceSorts.js";
+import { unreachable } from "./utils.js";
 // Bridge protocol version that introduced page: { type: "unavailable" }.
 // Earlier SDKs cannot represent this context, so the host reports context_unavailable.
 const BRIDGE_PROTOCOL_VERSION_INTRODUCING_UNAVAILABLE_PAGE = 5;
@@ -36,15 +37,63 @@ export function createCustomBlockHost(options) {
     function emit(direction, payload) {
         onLog?.(direction, payload);
     }
-    function post(message) {
-        if (initializationController.status === "stopped") {
-            // The parent page can stop this host. Do not send messages after the host stops.
-            return;
+    function postToSandbox(message) {
+        const status = initializationController.status;
+        switch (status) {
+            case "waitingForConnect":
+            case "waitingForInitResult":
+            case "initialized":
+            case "error":
+                break;
+            // A stopped host must not send messages, including pending responses.
+            // Return from the function to skip delivery below.
+            case "stopped":
+                return;
+            default:
+                return unreachable(status);
         }
         const sandboxMessage = normalizeHostToSandboxMessage(message, sandboxBridgeProtocolVersion);
         emit("out", sandboxMessage);
         iframe.contentWindow?.postMessage(sandboxMessage, sandboxOrigin);
     }
+    function sendInitMessage(message) {
+        postToSandbox(message);
+    }
+    function sendNack(message) {
+        postToSandbox(message);
+    }
+    function sendInitializedMessage(message) {
+        const status = initializationController.status;
+        switch (status) {
+            case "initialized":
+                postToSandbox(message);
+                return;
+            // Send state updates and handler results only after initialization succeeds.
+            case "waitingForConnect":
+            case "waitingForInitResult":
+            case "error":
+            case "stopped":
+                return;
+            default:
+                unreachable(status);
+        }
+    }
+    /**
+     * Invokes a one-shot handler and posts its result with the request's ID.
+     * Each result type is named `<request type>Result` by the bridge protocol.
+     */
+    async function respondToRequest(message, handler) {
+        if (handler === undefined) {
+            return;
+        }
+        const response = await handler(message);
+        const result = {
+            type: `${message.type}Result`,
+            requestId: message.requestId,
+            ...response,
+        };
+        sendInitializedMessage(result);
+    }
     function postQueryDataSourceResult(args) {
         const message = args.response.status === "error"
             ? {
@@ -60,7 +109,7 @@ export function createCustomBlockHost(options) {
                 items: args.response.items,
                 hasMore: args.response.hasMore,
             };
-        post(message);
+        sendInitializedMessage(message);
     }
     async function handleQuery(message, token = querySubscriptions.subscribe(message)) {
         const collectionSchema = getCollectionSchema(activeDataSources, message.dataSourceId);
@@ -83,111 +132,39 @@ export function createCustomBlockHost(options) {
             response,
         });
     }
-    async function handleCreatePage(message) {
-        if (handlers.createPage === undefined) {
-            return;
-        }
-        const response = await handlers.createPage(message);
-        const result = {
-            type: "createPageResult",
-            requestId: message.requestId,
-            ...response,
-        };
-        post(result);
-    }
-    async function handleGetPage(message) {
-        if (handlers.getPage === undefined) {
-            return;
-        }
-        const response = await handlers.getPage(message);
-        const result = {
-            type: "getPageResult",
-            requestId: message.requestId,
-            ...response,
-        };
-        post(result);
-    }
-    async function handleGetUser(message) {
-        if (handlers.getUser === undefined) {
-            return;
-        }
-        const response = await handlers.getUser(message);
-        const result = {
-            type: "getUserResult",
-            requestId: message.requestId,
-            ...response,
-        };
-        post(result);
-    }
-    async function handleListUsers(message) {
-        if (handlers.listUsers === undefined) {
-            return;
-        }
-        const response = await handlers.listUsers(message);
-        const result = {
-            type: "listUsersResult",
-            requestId: message.requestId,
-            ...response,
-        };
-        post(result);
-    }
-    async function handleOpenPage(message) {
-        if (handlers.openPage === undefined) {
-            return;
-        }
-        const response = await handlers.openPage(message);
-        const result = {
-            type: "openPageResult",
-            requestId: message.requestId,
-            ...response,
-        };
-        post(result);
-    }
-    async function handleUpdatePage(message) {
-        if (handlers.updatePage === undefined) {
-            return;
-        }
-        const response = await handlers.updatePage(message);
-        const result = {
-            type: "updatePageResult",
-            requestId: message.requestId,
-            ...response,
-        };
-        post(result);
-    }
     /**
      * Sends any pending host state changes to the sandbox. It is possible for these to change
      * between when the host sends `init` and when the sandbox responds with `initResult`. The host
-     * does send any pending state changes until hearing back from the sandbox.
+     * does not send pending state changes until the sandbox responds.
      */
     function flushPendingHostStateChanges() {
         if (pendingThemeChange) {
             pendingThemeChange = false;
-            post({ type: "themeChanged", theme });
+            sendInitializedMessage({ type: "themeChanged", theme });
         }
         if (pendingContrastModeChange) {
             pendingContrastModeChange = false;
-            post({ type: "contrastModeChanged", contrastMode });
+            sendInitializedMessage({ type: "contrastModeChanged", contrastMode });
         }
         if (pendingParentChange) {
             pendingParentChange = false;
-            post({ type: "parentChanged", parent });
+            sendInitializedMessage({ type: "parentChanged", parent });
         }
         if (pendingPageChange) {
             pendingPageChange = false;
-            post({ type: "pageChanged", page });
+            sendInitializedMessage({ type: "pageChanged", page });
         }
         if (pendingDataSourcesChange !== undefined) {
             const nextDataSources = pendingDataSourcesChange;
             pendingDataSourcesChange = undefined;
-            post({
+            sendInitializedMessage({
                 type: "dataSourcesChanged",
                 dataSources: nextDataSources,
             });
         }
         if (pendingCurrentUserChange) {
             pendingCurrentUserChange = false;
-            post({ type: "currentUserChanged", currentUser });
+            sendInitializedMessage({ type: "currentUserChanged", currentUser });
         }
     }
     const initializationController = createInitializationController({
@@ -232,7 +209,7 @@ export function createCustomBlockHost(options) {
                 dataSources: resolvedDataSources,
             };
         },
-        post,
+        sendInitMessage,
         warn: payload => emit("warn", payload),
         flushPendingHostStateChanges,
         minBridgeProtocolVersion,
@@ -251,19 +228,19 @@ export function createCustomBlockHost(options) {
         sandboxOrigin,
         getInitStatus: () => initializationController.status,
         emit,
-        post,
+        sendNack,
         onInvalidConnect: initializationController.handleInvalidConnect,
         onInvalidInitResult: initializationController.handleInvalidInitResult,
         onConnect: initializationController.handleConnect,
         onInitResult: initializationController.handleInitResult,
         onQueryDataSource: handleQuery,
         onUnsubscribeDataSourceQuery: message => querySubscriptions.unsubscribe(message.subscriptionId),
-        onCreatePage: handleCreatePage,
-        onGetPage: handleGetPage,
-        onOpenPage: handleOpenPage,
-        onGetUser: handleGetUser,
-        onListUsers: handleListUsers,
-        onUpdatePage: handleUpdatePage,
+        onCreatePage: message => respondToRequest(message, handlers.createPage),
+        onGetPage: message => respondToRequest(message, handlers.getPage),
+        onOpenPage: message => respondToRequest(message, handlers.openPage),
+        onGetUser: message => respondToRequest(message, handlers.getUser),
+        onListUsers: message => respondToRequest(message, handlers.listUsers),
+        onUpdatePage: message => respondToRequest(message, handlers.updatePage),
         onResize: message => handlers.resize?.(message),
     });
     window.addEventListener("message", onMessage);
@@ -281,21 +258,20 @@ export function createCustomBlockHost(options) {
     }
     return {
         stop,
-        post,
         setTheme(nextTheme) {
             theme = nextTheme;
-            if (initializationController.status !== "success") {
+            if (initializationController.status !== "initialized") {
                 if (initializationController.status === "waitingForInitResult") {
                     pendingThemeChange = true;
                 }
                 return;
             }
             const message = { type: "themeChanged", theme };
-            post(message);
+            sendInitializedMessage(message);
         },
         setContrastMode(nextContrastMode) {
             contrastMode = nextContrastMode;
-            if (initializationController.status !== "success") {
+            if (initializationController.status !== "initialized") {
                 if (initializationController.status === "waitingForInitResult") {
                     pendingContrastModeChange = true;
                 }
@@ -305,18 +281,18 @@ export function createCustomBlockHost(options) {
                 type: "contrastModeChanged",
                 contrastMode,
             };
-            post(message);
+            sendInitializedMessage(message);
         },
         setParent(nextParent) {
             parent = nextParent;
-            if (initializationController.status !== "success") {
+            if (initializationController.status !== "initialized") {
                 if (initializationController.status === "waitingForInitResult") {
                     pendingParentChange = true;
                 }
                 return;
             }
             const message = { type: "parentChanged", parent };
-            post(message);
+            sendInitializedMessage(message);
         },
         setPage(nextPage) {
             page = nextPage;
@@ -324,18 +300,18 @@ export function createCustomBlockHost(options) {
                 sandboxBridgeProtocolVersion <
                     BRIDGE_PROTOCOL_VERSION_INTRODUCING_UNAVAILABLE_PAGE &&
                 (initializationController.status === "waitingForInitResult" ||
-                    initializationController.status === "success")) {
+                    initializationController.status === "initialized")) {
                 stopWithFailure(initErrorForFailureReason("context_unavailable"));
                 return;
             }
-            if (initializationController.status !== "success") {
+            if (initializationController.status !== "initialized") {
                 if (initializationController.status === "waitingForInitResult") {
                     pendingPageChange = true;
                 }
                 return;
             }
             const message = { type: "pageChanged", page };
-            post(message);
+            sendInitializedMessage(message);
         },
         setDataSources(nextDataSources) {
             if (initializationController.hasConnectedSandbox &&
@@ -349,7 +325,7 @@ export function createCustomBlockHost(options) {
             }
             dataSources = nextDataSources;
             activeDataSources = nextDataSources;
-            if (initializationController.status !== "success") {
+            if (initializationController.status !== "initialized") {
                 if (initializationController.status === "waitingForInitResult") {
                     pendingDataSourcesChange = nextDataSources;
                 }
@@ -359,20 +335,20 @@ export function createCustomBlockHost(options) {
                 type: "dataSourcesChanged",
                 dataSources: nextDataSources,
             };
-            post(message);
+            sendInitializedMessage(message);
         },
         setCurrentUser(nextCurrentUser) {
             currentUser = nextCurrentUser;
-            if (initializationController.status !== "success") {
+            if (initializationController.status !== "initialized") {
                 if (initializationController.status === "waitingForInitResult") {
                     pendingCurrentUserChange = true;
                 }
                 return;
             }
-            post({ type: "currentUserChanged", currentUser });
+            sendInitializedMessage({ type: "currentUserChanged", currentUser });
         },
         refreshQuery({ dataSourceId }) {
-            if (initializationController.status !== "success") {
+            if (initializationController.status !== "initialized") {
                 return;
             }
             const refreshes = querySubscriptions.refresh(dataSourceId);

package/dist/lifecycle/initializationController.d.ts

@@ -2,8 +2,7 @@ import type { NotionContrastMode } from "../protocol/contrast.js";
 import type { NotionBlockId } from "../protocol/ids.js";
 import type { CustomBlockManifest } from "../protocol/manifest.js";
 import type { ConnectMessage } from "../protocol/messages/connect.js";
-import type { HostToSandboxMessage } from "../protocol/messages/hostToSandbox.js";
-import type { CustomBlockInitErrorInfo } from "../protocol/messages/init.js";
+import type { CustomBlockInitErrorInfo, InitMessage } from "../protocol/messages/init.js";
 import type { InitResultMessage } from "../protocol/messages/initResult.js";
 import type { CustomBlockPage } from "../protocol/pages/page.js";
 import type { NotionParent } from "../protocol/parent.js";
@@ -30,7 +29,7 @@ export type InitializationControllerOptions = {
     blockId: NotionBlockId;
     getState: () => InitializationState;
     prepareInit: (connect: ConnectSuccessMessage) => InitializationPreparation;
-    post: (message: HostToSandboxMessage) => void;
+    sendInitMessage: (message: InitMessage) => void;
     warn: (payload: unknown) => void;
     flushPendingHostStateChanges: () => void;
     minBridgeProtocolVersion: number;

package/dist/lifecycle/initializationController.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"initializationController.d.ts","sourceRoot":"","sources":["../../src/lifecycle/initializationController.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8CAA8C,CAAA;AACtF,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAA;AAC5E,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAA;AACvF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,4DAA4D,CAAA;AACtG,OAAO,KAAK,EACX,wBAAwB,EAExB,MAAM,mDAAmD,CAAA;AAC1D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yDAAyD,CAAA;AAChG,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gDAAgD,CAAA;AACrF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4CAA4C,CAAA;AAC9E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2CAA2C,CAAA;AAC5E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gDAAgD,CAAA;AAChF,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAA;AAGlE,OAAO,KAAK,EACX,qBAAqB,EACrB,iCAAiC,EACjC,UAAU,EACV,MAAM,YAAY,CAAA;AAKnB,KAAK,mBAAmB,GAAG;IAC1B,KAAK,EAAE,WAAW,CAAA;IAClB,YAAY,EAAE,kBAAkB,CAAA;IAChC,MAAM,EAAE,YAAY,CAAA;IACpB,IAAI,EAAE,eAAe,CAAA;IACrB,WAAW,EAAE,UAAU,CAAA;CACvB,CAAA;AAED,KAAK,yBAAyB,GAC3B;IACA,MAAM,EAAE,SAAS,CAAA;IACjB,QAAQ,EAAE,mBAAmB,CAAA;IAC7B,WAAW,EAAE,iCAAiC,CAAA;CAC7C,GACD;IACA,MAAM,EAAE,OAAO,CAAA;IACf,KAAK,EAAE,wBAAwB,CAAA;CAC9B,CAAA;AAEJ,MAAM,MAAM,+BAA+B,GAAG;IAC7C,OAAO,EAAE,aAAa,CAAA;IACtB,QAAQ,EAAE,MAAM,mBAAmB,CAAA;IACnC,WAAW,EAAE,CAAC,OAAO,EAAE,qBAAqB,KAAK,yBAAyB,CAAA;IAC1E,IAAI,EAAE,CAAC,OAAO,EAAE,oBAAoB,KAAK,IAAI,CAAA;IAC7C,IAAI,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAA;IAChC,4BAA4B,EAAE,MAAM,IAAI,CAAA;IACxC,wBAAwB,EAAE,MAAM,CAAA;IAChC,kBAAkB,CAAC,EAAE,MAAM,CAAA;IAC3B,qBAAqB,CAAC,EAAE,MAAM,CAAA;IAC9B,mBAAmB,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,IAAI,CAAA;IAC5D,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,CAAA;IACnD,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,IAAI,CAAA;CACnD,CAAA;AAED,MAAM,MAAM,wBAAwB,GAAG;IACtC,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAA;IAC3B,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAA;IACrC,aAAa,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,CAAA;IAChD,gBAAgB,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,CAAA;IACtD,oBAAoB,EAAE,CAAC,IAAI,EAAE;QAC5B,gBAAgB,EAAE,MAAM,GAAG,SAAS,CAAA;QACpC,KAAK,EAAE,wBAAwB,CAAA;QAC/B,qBAAqB,EAAE,MAAM,GAAG,SAAS,CAAA;KACzC,KAAK,SAAS,GAAG,MAAM,CAAA;IACxB,uBAAuB,EAAE,CAAC,IAAI,EAAE;QAC/B,gBAAgB,EAAE,MAAM,GAAG,SAAS,CAAA;QACpC,KAAK,EAAE,wBAAwB,CAAA;KAC/B,KAAK,IAAI,CAAA;IACV,YAAY,EAAE,MAAM,IAAI,CAAA;IACxB,IAAI,EAAE,MAAM,IAAI,CAAA;CAChB,CAAA;AAED,wBAAgB,8BAA8B,CAC7C,OAAO,EAAE,+BAA+B,GACtC,wBAAwB,CAmR1B"}
+{"version":3,"file":"initializationController.d.ts","sourceRoot":"","sources":["../../src/lifecycle/initializationController.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8CAA8C,CAAA;AACtF,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAA;AAC5E,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAA;AACvF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EACX,wBAAwB,EACxB,WAAW,EACX,MAAM,mDAAmD,CAAA;AAC1D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yDAAyD,CAAA;AAChG,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gDAAgD,CAAA;AACrF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4CAA4C,CAAA;AAC9E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2CAA2C,CAAA;AAC5E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gDAAgD,CAAA;AAChF,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAA;AAGlE,OAAO,KAAK,EACX,qBAAqB,EACrB,iCAAiC,EACjC,UAAU,EACV,MAAM,YAAY,CAAA;AAKnB,KAAK,mBAAmB,GAAG;IAC1B,KAAK,EAAE,WAAW,CAAA;IAClB,YAAY,EAAE,kBAAkB,CAAA;IAChC,MAAM,EAAE,YAAY,CAAA;IACpB,IAAI,EAAE,eAAe,CAAA;IACrB,WAAW,EAAE,UAAU,CAAA;CACvB,CAAA;AAED,KAAK,yBAAyB,GAC3B;IACA,MAAM,EAAE,SAAS,CAAA;IACjB,QAAQ,EAAE,mBAAmB,CAAA;IAC7B,WAAW,EAAE,iCAAiC,CAAA;CAC7C,GACD;IACA,MAAM,EAAE,OAAO,CAAA;IACf,KAAK,EAAE,wBAAwB,CAAA;CAC9B,CAAA;AAEJ,MAAM,MAAM,+BAA+B,GAAG;IAC7C,OAAO,EAAE,aAAa,CAAA;IACtB,QAAQ,EAAE,MAAM,mBAAmB,CAAA;IACnC,WAAW,EAAE,CAAC,OAAO,EAAE,qBAAqB,KAAK,yBAAyB,CAAA;IAC1E,eAAe,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,IAAI,CAAA;IAC/C,IAAI,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAA;IAChC,4BAA4B,EAAE,MAAM,IAAI,CAAA;IACxC,wBAAwB,EAAE,MAAM,CAAA;IAChC,kBAAkB,CAAC,EAAE,MAAM,CAAA;IAC3B,qBAAqB,CAAC,EAAE,MAAM,CAAA;IAC9B,mBAAmB,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,IAAI,CAAA;IAC5D,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,CAAA;IACnD,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,IAAI,CAAA;CACnD,CAAA;AAED,MAAM,MAAM,wBAAwB,GAAG;IACtC,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAA;IAC3B,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAA;IACrC,aAAa,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,CAAA;IAChD,gBAAgB,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,CAAA;IACtD,oBAAoB,EAAE,CAAC,IAAI,EAAE;QAC5B,gBAAgB,EAAE,MAAM,GAAG,SAAS,CAAA;QACpC,KAAK,EAAE,wBAAwB,CAAA;QAC/B,qBAAqB,EAAE,MAAM,GAAG,SAAS,CAAA;KACzC,KAAK,SAAS,GAAG,MAAM,CAAA;IACxB,uBAAuB,EAAE,CAAC,IAAI,EAAE;QAC/B,gBAAgB,EAAE,MAAM,GAAG,SAAS,CAAA;QACpC,KAAK,EAAE,wBAAwB,CAAA;KAC/B,KAAK,IAAI,CAAA;IACV,YAAY,EAAE,MAAM,IAAI,CAAA;IACxB,IAAI,EAAE,MAAM,IAAI,CAAA;CAChB,CAAA;AAED,wBAAgB,8BAA8B,CAC7C,OAAO,EAAE,+BAA+B,GACtC,wBAAwB,CAwR1B"}

package/dist/lifecycle/initializationController.js

@@ -3,7 +3,7 @@ import { isValidBridgeProtocolVersion } from "./protocolVersion.js";
 const DEFAULT_NO_CONNECT_TIMEOUT_MS = 5_000;
 const DEFAULT_NO_INIT_RESULT_TIMEOUT_MS = 5_000;
 export function createInitializationController(options) {
-    const { blockId, getState, prepareInit, post, warn, flushPendingHostStateChanges, minBridgeProtocolVersion, noConnectTimeoutMs = DEFAULT_NO_CONNECT_TIMEOUT_MS, noInitResultTimeoutMs = DEFAULT_NO_INIT_RESULT_TIMEOUT_MS, stopHostWithFailure, onInitResult, onFailure, } = options;
+    const { blockId, getState, prepareInit, sendInitMessage, warn, flushPendingHostStateChanges, minBridgeProtocolVersion, noConnectTimeoutMs = DEFAULT_NO_CONNECT_TIMEOUT_MS, noInitResultTimeoutMs = DEFAULT_NO_INIT_RESULT_TIMEOUT_MS, stopHostWithFailure, onInitResult, onFailure, } = options;
     let initStatus = "waitingForConnect";
     let noConnectTimeoutId;
     let noInitResultTimeoutId;
@@ -30,7 +30,7 @@ export function createInitializationController(options) {
         initStatus = "error";
         activeInitializationId = initializationId;
         clearNoConnectTimeout();
-        post({
+        sendInitMessage({
             type: "init",
             initializationId,
             status: "error",
@@ -38,11 +38,16 @@ export function createInitializationController(options) {
         });
         onFailure?.(error);
     }
-    function sendInit(connect) {
+    function prepareAndSendInit(connect) {
         if (initStatus !== "waitingForConnect") {
             return false;
         }
         const preparation = prepareInit(connect);
+        // The embedder can stop the host when it receives the manifest.
+        // If that happens, do not continue initialization or start its timer.
+        if (initStatus !== "waitingForConnect") {
+            return false;
+        }
         if (preparation.status === "error") {
             sendInitError({
                 initializationId: connect.initializationId,
@@ -67,7 +72,7 @@ export function createInitializationController(options) {
             dataSources: preparation.dataSources,
             currentUser: state.currentUser,
         };
-        post(message);
+        sendInitMessage(message);
         noInitResultTimeoutId = window.setTimeout(() => {
             if (initStatus !== "waitingForInitResult") {
                 return;
@@ -119,7 +124,7 @@ export function createInitializationController(options) {
             });
             return;
         }
-        if (sendInit(message)) {
+        if (prepareAndSendInit(message)) {
             hasConnectedSandbox = true;
         }
     }
@@ -204,7 +209,7 @@ export function createInitializationController(options) {
         }
         clearNoInitResultTimeout();
         if (message.status === "success") {
-            initStatus = "success";
+            initStatus = "initialized";
             flushPendingHostStateChanges();
         }
         else {

package/dist/lifecycle/types.d.ts

@@ -17,7 +17,7 @@ export type InitStatus =
 /** The host is waiting for a matching `initResult` after it sends `init`. */
  | "waitingForInitResult"
 /** The host is initialized and can send live updates. */
- | "success"
+ | "initialized"
 /** The host is in an error state after an initialization failure or timeout. */
  | "error"
 /** The host is stopped and no longer sends or accepts sandbox messages. */

package/dist/lifecycle/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/lifecycle/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8CAA8C,CAAA;AACtF,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,4DAA4D,CAAA;AAC1G,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAA;AAC5E,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAA;AACvF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gDAAgD,CAAA;AACrF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4CAA4C,CAAA;AAC9E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2CAA2C,CAAA;AAC5E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gDAAgD,CAAA;AAEhF,MAAM,MAAM,qBAAqB,GAAG,OAAO,CAC1C,cAAc,EACd;IAAE,MAAM,EAAE,SAAS,CAAA;CAAE,CACrB,CAAA;AAED,gEAAgE;AAChE,MAAM,MAAM,UAAU;AACrB,qEAAqE;AACnE,mBAAmB;AACrB,6EAA6E;GAC3E,sBAAsB;AACxB,yDAAyD;GACvD,SAAS;AACX,gFAAgF;GAC9E,OAAO;AACT,2EAA2E;GACzE,SAAS,CAAA;AAEZ,MAAM,MAAM,iCAAiC,GAAG;IAC/C,QAAQ,EAAE,wBAAwB,CAAA;CAClC,CAAA;AAED,MAAM,MAAM,iCAAiC,GAC1C,iCAAiC,GACjC,CAAC,CAAC,OAAO,EAAE,qBAAqB,KAAK,iCAAiC,CAAC,CAAA;AAE1E,MAAM,MAAM,2BAA2B,GAAG;IACzC,KAAK,EAAE,WAAW,CAAA;IAClB;;;OAGG;IACH,YAAY,CAAC,EAAE,kBAAkB,CAAA;IACjC,OAAO,EAAE,aAAa,CAAA;IACtB,MAAM,EAAE,YAAY,CAAA;IACpB,IAAI,EAAE,eAAe,CAAA;IACrB;;;OAGG;IACH,QAAQ,CAAC,EAAE,mBAAmB,CAAA;IAC9B,WAAW,EAAE,iCAAiC,CAAA;IAC9C,WAAW,EAAE,UAAU,CAAA;CACvB,CAAA"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/lifecycle/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8CAA8C,CAAA;AACtF,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,4DAA4D,CAAA;AAC1G,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAA;AAC5E,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAA;AACvF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gDAAgD,CAAA;AACrF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4CAA4C,CAAA;AAC9E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2CAA2C,CAAA;AAC5E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gDAAgD,CAAA;AAEhF,MAAM,MAAM,qBAAqB,GAAG,OAAO,CAC1C,cAAc,EACd;IAAE,MAAM,EAAE,SAAS,CAAA;CAAE,CACrB,CAAA;AAED,gEAAgE;AAChE,MAAM,MAAM,UAAU;AACrB,qEAAqE;AACnE,mBAAmB;AACrB,6EAA6E;GAC3E,sBAAsB;AACxB,yDAAyD;GACvD,aAAa;AACf,gFAAgF;GAC9E,OAAO;AACT,2EAA2E;GACzE,SAAS,CAAA;AAEZ,MAAM,MAAM,iCAAiC,GAAG;IAC/C,QAAQ,EAAE,wBAAwB,CAAA;CAClC,CAAA;AAED,MAAM,MAAM,iCAAiC,GAC1C,iCAAiC,GACjC,CAAC,CAAC,OAAO,EAAE,qBAAqB,KAAK,iCAAiC,CAAC,CAAA;AAE1E,MAAM,MAAM,2BAA2B,GAAG;IACzC,KAAK,EAAE,WAAW,CAAA;IAClB;;;OAGG;IACH,YAAY,CAAC,EAAE,kBAAkB,CAAA;IACjC,OAAO,EAAE,aAAa,CAAA;IACtB,MAAM,EAAE,YAAY,CAAA;IACpB,IAAI,EAAE,eAAe,CAAA;IACrB;;;OAGG;IACH,QAAQ,CAAC,EAAE,mBAAmB,CAAA;IAC9B,WAAW,EAAE,iCAAiC,CAAA;IAC9C,WAAW,EAAE,UAAU,CAAA;CACvB,CAAA"}

package/dist/messages/sandboxMessageRouter.d.ts

@@ -2,8 +2,8 @@ import type { ConnectMessage } from "../protocol/messages/connect.js";
 import type { CreatePageMessage } from "../protocol/messages/createPage.js";
 import type { GetPageMessage } from "../protocol/messages/getPage.js";
 import type { GetUserMessage } from "../protocol/messages/getUser.js";
-import type { HostToSandboxMessage } from "../protocol/messages/hostToSandbox.js";
 import type { InitResultMessage } from "../protocol/messages/initResult.js";
+import type { InvalidSandboxMessage } from "../protocol/messages/invalidSandboxMessage.js";
 import type { ListUsersMessage } from "../protocol/messages/listUsers.js";
 import type { OpenPageMessage } from "../protocol/messages/openPage.js";
 import type { QueryDataSourceMessage } from "../protocol/messages/queryDataSource.js";
@@ -18,7 +18,7 @@ export type SandboxMessageRouterOptions = {
     sandboxOrigin: string;
     getInitStatus: () => InitStatus;
     emit: (direction: CustomBlockHostLogDirection, payload: unknown) => void;
-    post: (message: HostToSandboxMessage) => void;
+    sendNack: (message: InvalidSandboxMessage) => void;
     onInvalidConnect: InitializationController["handleInvalidConnect"];
     onInvalidInitResult: InitializationController["handleInvalidInitResult"];
     onConnect: (message: ConnectMessage) => void;

package/dist/messages/sandboxMessageRouter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sandboxMessageRouter.d.ts","sourceRoot":"","sources":["../../src/messages/sandboxMessageRouter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yDAAyD,CAAA;AAChG,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,4DAA4D,CAAA;AACtG,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yDAAyD,CAAA;AAEhG,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wDAAwD,CAAA;AAC9F,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uDAAuD,CAAA;AAC5F,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,8DAA8D,CAAA;AAC1G,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qDAAqD,CAAA;AAExF,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,yEAAyE,CAAA;AAChI,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yDAAyD,CAAA;AAGhG,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,0CAA0C,CAAA;AACxF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAA;AAGvD,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,YAAY,CAAA;AAY7D,MAAM,MAAM,2BAA2B,GAAG;IACzC,MAAM,EAAE,iBAAiB,CAAA;IACzB,aAAa,EAAE,MAAM,CAAA;IACrB,aAAa,EAAE,MAAM,UAAU,CAAA;IAC/B,IAAI,EAAE,CAAC,SAAS,EAAE,2BAA2B,EAAE,OAAO,EAAE,OAAO,KAAK,IAAI,CAAA;IACxE,IAAI,EAAE,CAAC,OAAO,EAAE,oBAAoB,KAAK,IAAI,CAAA;IAC7C,gBAAgB,EAAE,wBAAwB,CAAC,sBAAsB,CAAC,CAAA;IAClE,mBAAmB,EAAE,wBAAwB,CAAC,yBAAyB,CAAC,CAAA;IACxE,SAAS,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,CAAA;IAC5C,YAAY,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,CAAA;IAClD,iBAAiB,EAAE,CAAC,OAAO,EAAE,sBAAsB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC5E,4BAA4B,EAAE,CAC7B,OAAO,EAAE,iCAAiC,KACtC,IAAI,CAAA;IACT,YAAY,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAClE,UAAU,EAAE,CAAC,OAAO,EAAE,eAAe,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC9D,SAAS,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC5D,SAAS,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC5D,WAAW,EAAE,CAAC,OAAO,EAAE,gBAAgB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAChE,YAAY,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAClE,QAAQ,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,IAAI,CAAA;CAC1C,CAAA;AAED,wBAAgB,0BAA0B,CACzC,OAAO,EAAE,2BAA2B,GAClC,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,CAsI/B"}
+{"version":3,"file":"sandboxMessageRouter.d.ts","sourceRoot":"","sources":["../../src/messages/sandboxMessageRouter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yDAAyD,CAAA;AAChG,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sDAAsD,CAAA;AAC1F,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yDAAyD,CAAA;AAChG,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oEAAoE,CAAA;AAE/G,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wDAAwD,CAAA;AAC9F,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uDAAuD,CAAA;AAC5F,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,8DAA8D,CAAA;AAC1G,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qDAAqD,CAAA;AAExF,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,yEAAyE,CAAA;AAChI,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yDAAyD,CAAA;AAGhG,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,0CAA0C,CAAA;AACxF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAA;AAGvD,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,YAAY,CAAA;AAY7D,MAAM,MAAM,2BAA2B,GAAG;IACzC,MAAM,EAAE,iBAAiB,CAAA;IACzB,aAAa,EAAE,MAAM,CAAA;IACrB,aAAa,EAAE,MAAM,UAAU,CAAA;IAC/B,IAAI,EAAE,CAAC,SAAS,EAAE,2BAA2B,EAAE,OAAO,EAAE,OAAO,KAAK,IAAI,CAAA;IACxE,QAAQ,EAAE,CAAC,OAAO,EAAE,qBAAqB,KAAK,IAAI,CAAA;IAClD,gBAAgB,EAAE,wBAAwB,CAAC,sBAAsB,CAAC,CAAA;IAClE,mBAAmB,EAAE,wBAAwB,CAAC,yBAAyB,CAAC,CAAA;IACxE,SAAS,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,CAAA;IAC5C,YAAY,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,CAAA;IAClD,iBAAiB,EAAE,CAAC,OAAO,EAAE,sBAAsB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC5E,4BAA4B,EAAE,CAC7B,OAAO,EAAE,iCAAiC,KACtC,IAAI,CAAA;IACT,YAAY,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAClE,UAAU,EAAE,CAAC,OAAO,EAAE,eAAe,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC9D,SAAS,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC5D,SAAS,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC5D,WAAW,EAAE,CAAC,OAAO,EAAE,gBAAgB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAChE,YAAY,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAClE,QAAQ,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,IAAI,CAAA;CAC1C,CAAA;AAED,wBAAgB,0BAA0B,CACzC,OAAO,EAAE,2BAA2B,GAClC,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,CAsI/B"}

package/dist/messages/sandboxMessageRouter.js

@@ -13,7 +13,7 @@ const bridgeProtocolVersionSchema = v.object({
     bridgeProtocolVersion: v.union([v.number(), v.nan()]),
 });
 export function createSandboxMessageRouter(options) {
-    const { iframe, sandboxOrigin, getInitStatus, emit, post, onInvalidConnect, onInvalidInitResult, onConnect, onInitResult, onQueryDataSource, onUnsubscribeDataSourceQuery, onCreatePage, onGetPage, onOpenPage, onGetUser, onListUsers, onUpdatePage, onResize, } = options;
+    const { iframe, sandboxOrigin, getInitStatus, emit, sendNack, onInvalidConnect, onInvalidInitResult, onConnect, onInitResult, onQueryDataSource, onUnsubscribeDataSourceQuery, onCreatePage, onGetPage, onOpenPage, onGetUser, onListUsers, onUpdatePage, onResize, } = options;
     function onInvalidSandboxMessage(data, issues) {
         emit("warn", {
             message: "Ignored unsupported sandbox message",
@@ -47,9 +47,9 @@ export function createSandboxMessageRouter(options) {
                 error: initErrorForFailureReason("invalid_init_result_payload"),
             });
         }
-        // Other messages get NACKed indicating they were invalid and skipped
-        if (response.nack !== undefined) {
-            post(response.nack);
+        // Other messages get NACKed indicating they were invalid and skipped.
+        if (response.nack !== undefined && getInitStatus() !== "stopped") {
+            sendNack(response.nack);
         }
     }
     return function onMessage(event) {
@@ -67,7 +67,7 @@ export function createSandboxMessageRouter(options) {
             return;
         }
         const message = parsed.output;
-        if (!isInitHandshakeMessage(message) && getInitStatus() !== "success") {
+        if (!isInitHandshakeMessage(message) && getInitStatus() !== "initialized") {
             emit("warn", {
                 message: `Ignored ${message.type} before initialization completed`,
                 payload: message,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@notionhq/custom-blocks-host",
-  "version": "0.1.29",
+  "version": "0.1.31",
   "description": "Reusable host runtime for Notion custom blocks.",
   "license": "MIT",
   "type": "module",

package/src/createCustomBlockHost.ts

@@ -6,27 +6,14 @@ import type { NotionCollectionSchema } from "@notionhq/custom-blocks-protocol/da
 import type { NotionDataSourceId } from "@notionhq/custom-blocks-protocol/ids.js"
 import type { CustomBlockManifest } from "@notionhq/custom-blocks-protocol/manifest.js"
 import type { ContrastModeChangedMessage } from "@notionhq/custom-blocks-protocol/messages/contrastModeChanged.js"
-import type { CreatePageMessage } from "@notionhq/custom-blocks-protocol/messages/createPage.js"
-import type { CreatePageResultMessage } from "@notionhq/custom-blocks-protocol/messages/createPageResult.js"
 import type { DataSourcesChangedMessage } from "@notionhq/custom-blocks-protocol/messages/dataSourcesChanged.js"
-import type {
-	GetPageMessage,
-	GetPageResultMessage,
-} from "@notionhq/custom-blocks-protocol/messages/getPage.js"
-import type {
-	GetUserMessage,
-	GetUserResultMessage,
-} from "@notionhq/custom-blocks-protocol/messages/getUser.js"
 import type { HostToSandboxMessage } from "@notionhq/custom-blocks-protocol/messages/hostToSandbox.js"
-import type { CustomBlockInitErrorCode } from "@notionhq/custom-blocks-protocol/messages/init.js"
-import type {
-	ListUsersMessage,
-	ListUsersResultMessage,
-} from "@notionhq/custom-blocks-protocol/messages/listUsers.js"
 import type {
-	OpenPageMessage,
-	OpenPageResultMessage,
-} from "@notionhq/custom-blocks-protocol/messages/openPage.js"
+	CustomBlockInitErrorCode,
+	InitMessage,
+} from "@notionhq/custom-blocks-protocol/messages/init.js"
+import type { InvalidSandboxMessage } from "@notionhq/custom-blocks-protocol/messages/invalidSandboxMessage.js"
+import type { BridgeMessagePayload } from "@notionhq/custom-blocks-protocol/messages/messagePayload.js"
 import type { PageChangedMessage } from "@notionhq/custom-blocks-protocol/messages/pageChanged.js"
 import type { ParentChangedMessage } from "@notionhq/custom-blocks-protocol/messages/parentChanged.js"
 import type { QueryDataSourceMessage } from "@notionhq/custom-blocks-protocol/messages/queryDataSource.js"
@@ -34,9 +21,8 @@ import type {
 	CustomBlockQueryDataSourceErrorInfo,
 	QueryDataSourceResultMessage,
 } from "@notionhq/custom-blocks-protocol/messages/queryDataSourceResult.js"
+import type { SandboxToHostMessage } from "@notionhq/custom-blocks-protocol/messages/sandboxToHost.js"
 import type { ThemeChangedMessage } from "@notionhq/custom-blocks-protocol/messages/themeChanged.js"
-import type { UpdatePageMessage } from "@notionhq/custom-blocks-protocol/messages/updatePage.js"
-import type { UpdatePageResultMessage } from "@notionhq/custom-blocks-protocol/messages/updatePageResult.js"
 import type { CustomBlockPage } from "@notionhq/custom-blocks-protocol/pages/page.js"
 import type { NotionParent } from "@notionhq/custom-blocks-protocol/parent.js"
 import { LATEST_BRIDGE_PROTOCOL_VERSION } from "@notionhq/custom-blocks-protocol/protocolVersion.js"
@@ -63,6 +49,7 @@ import {
 import type { CustomBlockHostQueryDataSourceResult } from "./queries/types.js"
 import { validateQueryDataSourceFilter } from "./queries/validateQueryDataSourceFilter.js"
 import { validateQueryDataSourceSorts } from "./queries/validateQueryDataSourceSorts.js"
+import { unreachable } from "./utils.js"
 
 // Bridge protocol version that introduced page: { type: "unavailable" }.
 // Earlier SDKs cannot represent this context, so the host reports context_unavailable.
@@ -103,7 +90,6 @@ export type CustomBlockHostOptions = {
 
 export type CustomBlockHostHandle = {
 	stop: () => void
-	post: (message: HostToSandboxMessage) => void
 	setTheme: (theme: NotionTheme) => void
 	setContrastMode: (contrastMode: NotionContrastMode) => void
 	setParent: (parent: NotionParent) => void
@@ -113,6 +99,29 @@ export type CustomBlockHostHandle = {
 	refreshQuery: (args: { dataSourceId: NotionDataSourceId }) => void
 }
 
+/**
+ * A request that expects one result correlated by `requestId`.
+ * Query subscriptions use `subscriptionId` and can receive repeated results.
+ */
+type OneShotRequest = Extract<SandboxToHostMessage, { requestId: string }>
+
+/** The matching host result message, named `<request type>Result` by the protocol. */
+type ResultForRequest<Request extends OneShotRequest> = Extract<
+	HostToSandboxMessage,
+	{ type: `${Request["type"]}Result` }
+>
+
+/** The result's success or error payload, without the bridge envelope fields. */
+type ResultPayloadForRequest<Request extends OneShotRequest> =
+	BridgeMessagePayload<ResultForRequest<Request>>
+
+/** Returns one result payload. `respondToRequest` adds the message type and request ID. */
+type OneShotHandler<Request extends OneShotRequest> = (
+	message: Request,
+) =>
+	| ResultPayloadForRequest<Request>
+	| Promise<ResultPayloadForRequest<Request>>
+
 export function createCustomBlockHost(
 	options: CustomBlockHostOptions,
 ): CustomBlockHostHandle {
@@ -158,10 +167,20 @@ export function createCustomBlockHost(
 		onLog?.(direction, payload)
 	}
 
-	function post(message: HostToSandboxMessage) {
-		if (initializationController.status === "stopped") {
-			// The parent page can stop this host. Do not send messages after the host stops.
-			return
+	function postToSandbox(message: HostToSandboxMessage) {
+		const status = initializationController.status
+		switch (status) {
+			case "waitingForConnect":
+			case "waitingForInitResult":
+			case "initialized":
+			case "error":
+				break
+			// A stopped host must not send messages, including pending responses.
+			// Return from the function to skip delivery below.
+			case "stopped":
+				return
+			default:
+				return unreachable(status)
 		}
 		const sandboxMessage = normalizeHostToSandboxMessage(
 			message,
@@ -171,6 +190,56 @@ export function createCustomBlockHost(
 		iframe.contentWindow?.postMessage(sandboxMessage, sandboxOrigin)
 	}
 
+	function sendInitMessage(message: InitMessage) {
+		postToSandbox(message)
+	}
+
+	function sendNack(message: InvalidSandboxMessage) {
+		postToSandbox(message)
+	}
+
+	function sendInitializedMessage(
+		message: Exclude<
+			HostToSandboxMessage,
+			{ type: "init" | "invalidSandboxMessage" }
+		>,
+	) {
+		const status = initializationController.status
+		switch (status) {
+			case "initialized":
+				postToSandbox(message)
+				return
+			// Send state updates and handler results only after initialization succeeds.
+			case "waitingForConnect":
+			case "waitingForInitResult":
+			case "error":
+			case "stopped":
+				return
+			default:
+				unreachable(status)
+		}
+	}
+
+	/**
+	 * Invokes a one-shot handler and posts its result with the request's ID.
+	 * Each result type is named `<request type>Result` by the bridge protocol.
+	 */
+	async function respondToRequest<Request extends OneShotRequest>(
+		message: Request,
+		handler: OneShotHandler<Request> | undefined,
+	) {
+		if (handler === undefined) {
+			return
+		}
+		const response = await handler(message)
+		const result = {
+			type: `${message.type}Result` as const,
+			requestId: message.requestId,
+			...response,
+		} as ResultForRequest<OneShotRequest>
+		sendInitializedMessage(result)
+	}
+
 	function postQueryDataSourceResult(args: {
 		subscriptionId: string
 		response: CustomBlockHostQueryDataSourceResult
@@ -190,7 +259,7 @@ export function createCustomBlockHost(
 						items: args.response.items,
 						hasMore: args.response.hasMore,
 					}
-		post(message)
+		sendInitializedMessage(message)
 	}
 
 	async function handleQuery(
@@ -221,117 +290,39 @@ export function createCustomBlockHost(
 		})
 	}
 
-	async function handleCreatePage(message: CreatePageMessage) {
-		if (handlers.createPage === undefined) {
-			return
-		}
-		const response = await handlers.createPage(message)
-		const result: CreatePageResultMessage = {
-			type: "createPageResult",
-			requestId: message.requestId,
-			...response,
-		}
-		post(result)
-	}
-
-	async function handleGetPage(message: GetPageMessage) {
-		if (handlers.getPage === undefined) {
-			return
-		}
-		const response = await handlers.getPage(message)
-		const result: GetPageResultMessage = {
-			type: "getPageResult",
-			requestId: message.requestId,
-			...response,
-		}
-		post(result)
-	}
-
-	async function handleGetUser(message: GetUserMessage) {
-		if (handlers.getUser === undefined) {
-			return
-		}
-		const response = await handlers.getUser(message)
-		const result: GetUserResultMessage = {
-			type: "getUserResult",
-			requestId: message.requestId,
-			...response,
-		}
-		post(result)
-	}
-
-	async function handleListUsers(message: ListUsersMessage) {
-		if (handlers.listUsers === undefined) {
-			return
-		}
-		const response = await handlers.listUsers(message)
-		const result: ListUsersResultMessage = {
-			type: "listUsersResult",
-			requestId: message.requestId,
-			...response,
-		}
-		post(result)
-	}
-
-	async function handleOpenPage(message: OpenPageMessage) {
-		if (handlers.openPage === undefined) {
-			return
-		}
-		const response = await handlers.openPage(message)
-		const result: OpenPageResultMessage = {
-			type: "openPageResult",
-			requestId: message.requestId,
-			...response,
-		}
-		post(result)
-	}
-
-	async function handleUpdatePage(message: UpdatePageMessage) {
-		if (handlers.updatePage === undefined) {
-			return
-		}
-		const response = await handlers.updatePage(message)
-		const result: UpdatePageResultMessage = {
-			type: "updatePageResult",
-			requestId: message.requestId,
-			...response,
-		}
-		post(result)
-	}
-
 	/**
 	 * Sends any pending host state changes to the sandbox. It is possible for these to change
 	 * between when the host sends `init` and when the sandbox responds with `initResult`. The host
-	 * does send any pending state changes until hearing back from the sandbox.
+	 * does not send pending state changes until the sandbox responds.
 	 */
 	function flushPendingHostStateChanges() {
 		if (pendingThemeChange) {
 			pendingThemeChange = false
-			post({ type: "themeChanged", theme })
+			sendInitializedMessage({ type: "themeChanged", theme })
 		}
 		if (pendingContrastModeChange) {
 			pendingContrastModeChange = false
-			post({ type: "contrastModeChanged", contrastMode })
+			sendInitializedMessage({ type: "contrastModeChanged", contrastMode })
 		}
 		if (pendingParentChange) {
 			pendingParentChange = false
-			post({ type: "parentChanged", parent })
+			sendInitializedMessage({ type: "parentChanged", parent })
 		}
 		if (pendingPageChange) {
 			pendingPageChange = false
-			post({ type: "pageChanged", page })
+			sendInitializedMessage({ type: "pageChanged", page })
 		}
 		if (pendingDataSourcesChange !== undefined) {
 			const nextDataSources = pendingDataSourcesChange
 			pendingDataSourcesChange = undefined
-			post({
+			sendInitializedMessage({
 				type: "dataSourcesChanged",
 				dataSources: nextDataSources,
 			})
 		}
 		if (pendingCurrentUserChange) {
 			pendingCurrentUserChange = false
-			post({ type: "currentUserChanged", currentUser })
+			sendInitializedMessage({ type: "currentUserChanged", currentUser })
 		}
 	}
 
@@ -383,7 +374,7 @@ export function createCustomBlockHost(
 				dataSources: resolvedDataSources,
 			}
 		},
-		post,
+		sendInitMessage,
 		warn: payload => emit("warn", payload),
 		flushPendingHostStateChanges,
 		minBridgeProtocolVersion,
@@ -403,7 +394,7 @@ export function createCustomBlockHost(
 		sandboxOrigin,
 		getInitStatus: () => initializationController.status,
 		emit,
-		post,
+		sendNack,
 		onInvalidConnect: initializationController.handleInvalidConnect,
 		onInvalidInitResult: initializationController.handleInvalidInitResult,
 		onConnect: initializationController.handleConnect,
@@ -411,12 +402,12 @@ export function createCustomBlockHost(
 		onQueryDataSource: handleQuery,
 		onUnsubscribeDataSourceQuery: message =>
 			querySubscriptions.unsubscribe(message.subscriptionId),
-		onCreatePage: handleCreatePage,
-		onGetPage: handleGetPage,
-		onOpenPage: handleOpenPage,
-		onGetUser: handleGetUser,
-		onListUsers: handleListUsers,
-		onUpdatePage: handleUpdatePage,
+		onCreatePage: message => respondToRequest(message, handlers.createPage),
+		onGetPage: message => respondToRequest(message, handlers.getPage),
+		onOpenPage: message => respondToRequest(message, handlers.openPage),
+		onGetUser: message => respondToRequest(message, handlers.getUser),
+		onListUsers: message => respondToRequest(message, handlers.listUsers),
+		onUpdatePage: message => respondToRequest(message, handlers.updatePage),
 		onResize: message => handlers.resize?.(message),
 	})
 
@@ -438,21 +429,20 @@ export function createCustomBlockHost(
 
 	return {
 		stop,
-		post,
 		setTheme(nextTheme) {
 			theme = nextTheme
-			if (initializationController.status !== "success") {
+			if (initializationController.status !== "initialized") {
 				if (initializationController.status === "waitingForInitResult") {
 					pendingThemeChange = true
 				}
 				return
 			}
 			const message: ThemeChangedMessage = { type: "themeChanged", theme }
-			post(message)
+			sendInitializedMessage(message)
 		},
 		setContrastMode(nextContrastMode) {
 			contrastMode = nextContrastMode
-			if (initializationController.status !== "success") {
+			if (initializationController.status !== "initialized") {
 				if (initializationController.status === "waitingForInitResult") {
 					pendingContrastModeChange = true
 				}
@@ -462,18 +452,18 @@ export function createCustomBlockHost(
 				type: "contrastModeChanged",
 				contrastMode,
 			}
-			post(message)
+			sendInitializedMessage(message)
 		},
 		setParent(nextParent) {
 			parent = nextParent
-			if (initializationController.status !== "success") {
+			if (initializationController.status !== "initialized") {
 				if (initializationController.status === "waitingForInitResult") {
 					pendingParentChange = true
 				}
 				return
 			}
 			const message: ParentChangedMessage = { type: "parentChanged", parent }
-			post(message)
+			sendInitializedMessage(message)
 		},
 		setPage(nextPage) {
 			page = nextPage
@@ -482,19 +472,19 @@ export function createCustomBlockHost(
 				sandboxBridgeProtocolVersion <
 					BRIDGE_PROTOCOL_VERSION_INTRODUCING_UNAVAILABLE_PAGE &&
 				(initializationController.status === "waitingForInitResult" ||
-					initializationController.status === "success")
+					initializationController.status === "initialized")
 			) {
 				stopWithFailure(initErrorForFailureReason("context_unavailable"))
 				return
 			}
-			if (initializationController.status !== "success") {
+			if (initializationController.status !== "initialized") {
 				if (initializationController.status === "waitingForInitResult") {
 					pendingPageChange = true
 				}
 				return
 			}
 			const message: PageChangedMessage = { type: "pageChanged", page }
-			post(message)
+			sendInitializedMessage(message)
 		},
 		setDataSources(nextDataSources) {
 			if (
@@ -510,7 +500,7 @@ export function createCustomBlockHost(
 			}
 			dataSources = nextDataSources
 			activeDataSources = nextDataSources
-			if (initializationController.status !== "success") {
+			if (initializationController.status !== "initialized") {
 				if (initializationController.status === "waitingForInitResult") {
 					pendingDataSourcesChange = nextDataSources
 				}
@@ -520,20 +510,20 @@ export function createCustomBlockHost(
 				type: "dataSourcesChanged",
 				dataSources: nextDataSources,
 			}
-			post(message)
+			sendInitializedMessage(message)
 		},
 		setCurrentUser(nextCurrentUser) {
 			currentUser = nextCurrentUser
-			if (initializationController.status !== "success") {
+			if (initializationController.status !== "initialized") {
 				if (initializationController.status === "waitingForInitResult") {
 					pendingCurrentUserChange = true
 				}
 				return
 			}
-			post({ type: "currentUserChanged", currentUser })
+			sendInitializedMessage({ type: "currentUserChanged", currentUser })
 		},
 		refreshQuery({ dataSourceId }) {
-			if (initializationController.status !== "success") {
+			if (initializationController.status !== "initialized") {
 				return
 			}
 			const refreshes = querySubscriptions.refresh(dataSourceId)

package/src/lifecycle/initializationController.ts

@@ -2,7 +2,6 @@ import type { NotionContrastMode } from "@notionhq/custom-blocks-protocol/contra
 import type { NotionBlockId } from "@notionhq/custom-blocks-protocol/ids.js"
 import type { CustomBlockManifest } from "@notionhq/custom-blocks-protocol/manifest.js"
 import type { ConnectMessage } from "@notionhq/custom-blocks-protocol/messages/connect.js"
-import type { HostToSandboxMessage } from "@notionhq/custom-blocks-protocol/messages/hostToSandbox.js"
 import type {
 	CustomBlockInitErrorInfo,
 	InitMessage,
@@ -47,7 +46,7 @@ export type InitializationControllerOptions = {
 	blockId: NotionBlockId
 	getState: () => InitializationState
 	prepareInit: (connect: ConnectSuccessMessage) => InitializationPreparation
-	post: (message: HostToSandboxMessage) => void
+	sendInitMessage: (message: InitMessage) => void
 	warn: (payload: unknown) => void
 	flushPendingHostStateChanges: () => void
 	minBridgeProtocolVersion: number
@@ -83,7 +82,7 @@ export function createInitializationController(
 		blockId,
 		getState,
 		prepareInit,
-		post,
+		sendInitMessage,
 		warn,
 		flushPendingHostStateChanges,
 		minBridgeProtocolVersion,
@@ -126,7 +125,7 @@ export function createInitializationController(
 		initStatus = "error"
 		activeInitializationId = initializationId
 		clearNoConnectTimeout()
-		post({
+		sendInitMessage({
 			type: "init",
 			initializationId,
 			status: "error",
@@ -135,11 +134,16 @@ export function createInitializationController(
 		onFailure?.(error)
 	}
 
-	function sendInit(connect: ConnectSuccessMessage): boolean {
+	function prepareAndSendInit(connect: ConnectSuccessMessage): boolean {
 		if (initStatus !== "waitingForConnect") {
 			return false
 		}
 		const preparation = prepareInit(connect)
+		// The embedder can stop the host when it receives the manifest.
+		// If that happens, do not continue initialization or start its timer.
+		if (initStatus !== "waitingForConnect") {
+			return false
+		}
 		if (preparation.status === "error") {
 			sendInitError({
 				initializationId: connect.initializationId,
@@ -164,7 +168,7 @@ export function createInitializationController(
 			dataSources: preparation.dataSources,
 			currentUser: state.currentUser,
 		}
-		post(message)
+		sendInitMessage(message)
 		noInitResultTimeoutId = window.setTimeout(() => {
 			if (initStatus !== "waitingForInitResult") {
 				return
@@ -223,7 +227,7 @@ export function createInitializationController(
 			return
 		}
 
-		if (sendInit(message)) {
+		if (prepareAndSendInit(message)) {
 			hasConnectedSandbox = true
 		}
 	}
@@ -324,7 +328,7 @@ export function createInitializationController(
 		}
 		clearNoInitResultTimeout()
 		if (message.status === "success") {
-			initStatus = "success"
+			initStatus = "initialized"
 			flushPendingHostStateChanges()
 		} else {
 			initStatus = "error"

package/src/lifecycle/types.ts

@@ -20,7 +20,7 @@ export type InitStatus =
 	/** The host is waiting for a matching `initResult` after it sends `init`. */
 	| "waitingForInitResult"
 	/** The host is initialized and can send live updates. */
-	| "success"
+	| "initialized"
 	/** The host is in an error state after an initialization failure or timeout. */
 	| "error"
 	/** The host is stopped and no longer sends or accepts sandbox messages. */

package/src/messages/sandboxMessageRouter.ts

@@ -2,8 +2,8 @@ import type { ConnectMessage } from "@notionhq/custom-blocks-protocol/messages/c
 import type { CreatePageMessage } from "@notionhq/custom-blocks-protocol/messages/createPage.js"
 import type { GetPageMessage } from "@notionhq/custom-blocks-protocol/messages/getPage.js"
 import type { GetUserMessage } from "@notionhq/custom-blocks-protocol/messages/getUser.js"
-import type { HostToSandboxMessage } from "@notionhq/custom-blocks-protocol/messages/hostToSandbox.js"
 import type { InitResultMessage } from "@notionhq/custom-blocks-protocol/messages/initResult.js"
+import type { InvalidSandboxMessage } from "@notionhq/custom-blocks-protocol/messages/invalidSandboxMessage.js"
 import { isInitHandshakeMessage } from "@notionhq/custom-blocks-protocol/messages/isInitHandshakeMessage.js"
 import type { ListUsersMessage } from "@notionhq/custom-blocks-protocol/messages/listUsers.js"
 import type { OpenPageMessage } from "@notionhq/custom-blocks-protocol/messages/openPage.js"
@@ -35,7 +35,7 @@ export type SandboxMessageRouterOptions = {
 	sandboxOrigin: string
 	getInitStatus: () => InitStatus
 	emit: (direction: CustomBlockHostLogDirection, payload: unknown) => void
-	post: (message: HostToSandboxMessage) => void
+	sendNack: (message: InvalidSandboxMessage) => void
 	onInvalidConnect: InitializationController["handleInvalidConnect"]
 	onInvalidInitResult: InitializationController["handleInvalidInitResult"]
 	onConnect: (message: ConnectMessage) => void
@@ -61,7 +61,7 @@ export function createSandboxMessageRouter(
 		sandboxOrigin,
 		getInitStatus,
 		emit,
-		post,
+		sendNack,
 		onInvalidConnect,
 		onInvalidInitResult,
 		onConnect,
@@ -116,9 +116,9 @@ export function createSandboxMessageRouter(
 			})
 		}
 
-		// Other messages get NACKed indicating they were invalid and skipped
-		if (response.nack !== undefined) {
-			post(response.nack)
+		// Other messages get NACKed indicating they were invalid and skipped.
+		if (response.nack !== undefined && getInitStatus() !== "stopped") {
+			sendNack(response.nack)
 		}
 	}
 
@@ -139,7 +139,7 @@ export function createSandboxMessageRouter(
 		}
 
 		const message = parsed.output
-		if (!isInitHandshakeMessage(message) && getInitStatus() !== "success") {
+		if (!isInitHandshakeMessage(message) && getInitStatus() !== "initialized") {
 			emit("warn", {
 				message: `Ignored ${message.type} before initialization completed`,
 				payload: message,