@base44/vite-plugin 1.0.11 → 1.0.13

package/src/injections/page-height-bridge.ts

@@ -24,6 +24,12 @@ type IndexableCssRule = CSSRule & {
 
 type Debouncer = { trigger: () => void; cancel: () => void };
 
+export type PageHeightBridgeController = {
+  freezeVhUnits: (override?: number) => void;
+  measurePageHeight: (origin: string, settleMs?: number) => void;
+  teardown: () => void;
+};
+
 const FALLBACK_VHBASE: number = 900;
 const MIN_VHBASE: number = 400;
 const DEFAULT_SETTLE_MS: number = 2000;
@@ -45,34 +51,7 @@ export function setupPageHeightBridge(): () => void {
   if (window.self === window.top) return noop;
   started = true;
 
-  let vhCleanups: Array<() => void> | null = null;
-  let vhForceRun: (() => void) | null = null;
-  let pendingResponse: number | undefined;
-  let pendingOrigin: string = "*";
-
-  const freezeVhUnits = (override: number | undefined): void => {
-    if (vhCleanups) {
-      vhForceRun?.();
-      return;
-    }
-    const referenceVhBase: number = resolveReferenceVhBase(override);
-    vhCleanups = [];
-    vhForceRun = startVhNeutralizer(referenceVhBase, vhCleanups);
-  };
-
-  // Target the requester's origin so the height isn't broadcast to anyone
-  // who happens to embed us. Falls back to "*" when origin is unavailable
-  // (jsdom default, sandboxed iframes with `null` origin).
-  const measurePageHeight = (origin: string, settleMs: number): void => {
-    pendingOrigin = origin;
-    if (pendingResponse !== undefined) window.clearTimeout(pendingResponse);
-    pendingResponse = window.setTimeout((): void => {
-      requestAnimationFrame((): void => {
-        const height: number = measureContentHeight();
-        window.parent.postMessage({ type: "page-height-measured", height }, pendingOrigin);
-      });
-    }, settleMs);
-  };
+  const controller: PageHeightBridgeController = createPageHeightBridgeController();
 
   const onMessage = (event: MessageEvent): void => {
     const data: IncomingMessage | null = (event.data ?? null) as IncomingMessage | null;
@@ -81,7 +60,7 @@ export function setupPageHeightBridge(): () => void {
       case "freeze-vh-units": {
         const override: number | undefined =
           typeof data.referenceVhBase === "number" ? data.referenceVhBase : undefined;
-        freezeVhUnits(override);
+        controller.freezeVhUnits(override);
         return;
       }
       case "measure-page-height": {
@@ -89,7 +68,7 @@ export function setupPageHeightBridge(): () => void {
           typeof data.settleMs === "number" ? data.settleMs : DEFAULT_SETTLE_MS;
         const origin: string =
           event.origin && event.origin !== "null" ? event.origin : "*";
-        measurePageHeight(origin, settleMs);
+        controller.measurePageHeight(origin, settleMs);
         return;
       }
     }
@@ -103,12 +82,49 @@ export function setupPageHeightBridge(): () => void {
     torn = true;
     started = false;
     window.removeEventListener("message", onMessage);
-    if (vhCleanups) {
-      for (const c of vhCleanups) c();
-      vhCleanups = null;
-      vhForceRun = null;
-    }
-    if (pendingResponse !== undefined) window.clearTimeout(pendingResponse);
+    controller.teardown();
+  };
+}
+
+export function createPageHeightBridgeController(): PageHeightBridgeController {
+  let vhCleanups: Array<() => void> | null = null;
+  let vhForceRun: (() => void) | null = null;
+  let pendingResponse: number | undefined;
+  let pendingOrigin: string = "*";
+
+  return {
+    freezeVhUnits: (override: number | undefined): void => {
+      if (vhCleanups) {
+        vhForceRun?.();
+        return;
+      }
+      const referenceVhBase: number = resolveReferenceVhBase(override);
+      vhCleanups = [];
+      vhForceRun = startVhNeutralizer(referenceVhBase, vhCleanups);
+    },
+
+    // Target the requester's origin so the height isn't broadcast to anyone
+    // who happens to embed us. Falls back to "*" when origin is unavailable
+    // (jsdom default, sandboxed iframes with `null` origin).
+    measurePageHeight: (origin: string, settleMs: number = DEFAULT_SETTLE_MS): void => {
+      pendingOrigin = origin;
+      if (pendingResponse !== undefined) window.clearTimeout(pendingResponse);
+      pendingResponse = window.setTimeout((): void => {
+        requestAnimationFrame((): void => {
+          const height: number = measureContentHeight();
+          window.parent.postMessage({ type: "page-height-measured", height }, pendingOrigin);
+        });
+      }, settleMs);
+    },
+
+    teardown: (): void => {
+      if (vhCleanups) {
+        for (const c of vhCleanups) c();
+        vhCleanups = null;
+        vhForceRun = null;
+      }
+      if (pendingResponse !== undefined) window.clearTimeout(pendingResponse);
+    },
   };
 }
 

package/src/injections/visual-edit-agent.ts

@@ -3,10 +3,13 @@ import { createLayerController } from "./layer-dropdown/controller.js";
 import { LAYER_DROPDOWN_ATTR } from "./layer-dropdown/consts.js";
 import { createInlineEditController } from "../capabilities/inline-edit/index.js";
 import { THEME_FONT_PREVIEW_ID } from "../consts.js";
+import { createPageHeightBridgeController } from "./page-height-bridge.js";
 
 const REPOSITION_DELAY_MS = 50;
 
 export function setupVisualEditAgent() {
+  const pageHeightBridge = createPageHeightBridgeController();
+
   // State variables (replacing React useState/useRef)
   let isVisualEditMode = false;
   let isPopoverDragging = false;
@@ -638,6 +641,21 @@ export function setupVisualEditAgent() {
         }
         break;
 
+      case "freeze-vh-units":
+        pageHeightBridge.freezeVhUnits(
+          typeof message.referenceVhBase === "number"
+            ? message.referenceVhBase
+            : undefined
+        );
+        break;
+
+      case "measure-page-height":
+        pageHeightBridge.measurePageHeight(
+          event.origin && event.origin !== "null" ? event.origin : "*",
+          typeof message.settleMs === "number" ? message.settleMs : undefined
+        );
+        break;
+
       default:
         break;
     }
@@ -723,4 +741,4 @@ export function setupVisualEditAgent() {
 
   // Send ready message to parent
   window.parent.postMessage({ type: "visual-edit-agent-ready" }, "*");
-}
+}