@measured/puck-plugin-heading-analyzer 0.14.2-canary.58b08a9 → 0.14.2-canary.717ccb9

package/dist/index.js

@@ -45907,7 +45907,10 @@ var require_dist = __commonJS({
       return collected;
     };
     var getStyleSheet = (el) => {
-      return Array.from(document.styleSheets).find((ss) => ss.ownerNode === el);
+      return Array.from(document.styleSheets).find((ss) => {
+        const ownerNode = ss.ownerNode;
+        return ownerNode.href === el.href;
+      });
     };
     var getStyles2 = (styleSheet) => {
       if (styleSheet) {
@@ -45929,25 +45932,16 @@ var require_dist = __commonJS({
       onStylesLoaded = () => null
     }) => {
       const { document: doc, window: win } = (0, import_react_frame_component.useFrame)();
-      (0, import_react22.useLayoutEffect)(() => {
+      (0, import_react22.useEffect)(() => {
         if (!win || !doc) {
           return () => {
           };
         }
-        const elements = [];
+        let elements = [];
+        const hashes = {};
         const lookupEl = (el) => elements.findIndex((elementMap) => elementMap.original === el);
-        const addEl = (el, onLoad = () => {
+        const mirrorEl = (el, onLoad = () => {
         }) => __async2(void 0, null, function* () {
-          const index = lookupEl(el);
-          if (index > -1) {
-            if (debug)
-              console.log(
-                `Tried to add an element that was already mirrored. Updating instead...`
-              );
-            elements[index].mirror.innerText = el.innerText;
-            onLoad();
-            return;
-          }
           let mirror;
           if (el.nodeName === "LINK") {
             mirror = document.createElement("style");
@@ -45964,7 +45958,6 @@ var require_dist = __commonJS({
                   `Tried to load styles for link element, but couldn't find them. Skipping...`
                 );
               }
-              onLoad();
               return;
             }
             mirror.innerHTML = styles;
@@ -45972,11 +45965,28 @@ var require_dist = __commonJS({
           } else {
             mirror = el.cloneNode(true);
           }
+          mirror.onload = onLoad;
+          return mirror;
+        });
+        const addEl = (el, onLoad = () => {
+        }) => __async2(void 0, null, function* () {
+          const index = lookupEl(el);
+          if (index > -1) {
+            if (debug)
+              console.log(
+                `Tried to add an element that was already mirrored. Updating instead...`
+              );
+            elements[index].mirror.innerText = el.innerText;
+            onLoad();
+            return;
+          }
+          const mirror = yield mirrorEl(el, onLoad);
+          if (!mirror) {
+            onLoad();
+            return;
+          }
           const elHash = (0, import_object_hash.default)(mirror.outerHTML);
-          const existingHashes = collectStyles(doc).map(
-            (existingStyle) => existingStyle !== mirror && (0, import_object_hash.default)(existingStyle.outerHTML)
-          ).filter(Boolean);
-          if (existingHashes.indexOf(elHash) > -1) {
+          if (hashes[elHash]) {
             if (debug)
               console.log(
                 `iframe already contains element that is being mirrored. Skipping...`
@@ -45984,6 +45994,7 @@ var require_dist = __commonJS({
             onLoad();
             return;
           }
+          hashes[elHash] = true;
           mirror.onload = onLoad;
           doc.head.append(mirror);
           elements.push({ original: el, mirror });
@@ -45991,6 +46002,7 @@ var require_dist = __commonJS({
             console.log(`Added style node ${el.outerHTML}`);
         });
         const removeEl = (el) => {
+          var _a3, _b;
           const index = lookupEl(el);
           if (index === -1) {
             if (debug)
@@ -45999,7 +46011,9 @@ var require_dist = __commonJS({
               );
             return;
           }
-          elements[index].mirror.remove();
+          const elHash = (0, import_object_hash.default)(el.outerHTML);
+          (_b = (_a3 = elements[index]) == null ? void 0 : _a3.mirror) == null ? void 0 : _b.remove();
+          delete hashes[elHash];
           if (debug)
             console.log(`Removed style node ${el.outerHTML}`);
         };
@@ -46027,18 +46041,35 @@ var require_dist = __commonJS({
         });
         const parentDocument = win.parent.document;
         const collectedStyles = collectStyles(parentDocument);
-        let mountedCounter = 0;
-        collectedStyles.forEach((styleNode) => {
-          defer(
-            () => addEl(styleNode, () => {
-              mountedCounter += 1;
-              if (mountedCounter === collectedStyles.length) {
-                onStylesLoaded();
+        const hrefs = [];
+        Promise.all(
+          collectedStyles.map((styleNode, i2) => __async2(void 0, null, function* () {
+            if (styleNode.nodeName === "LINK") {
+              const linkHref = styleNode.href;
+              if (hrefs.indexOf(linkHref) > -1) {
+                return;
               }
-            })
+              hrefs.push(linkHref);
+            }
+            const mirror = yield mirrorEl(styleNode);
+            if (!mirror)
+              return;
+            elements.push({ original: styleNode, mirror });
+            return mirror;
+          }))
+        ).then((mirrorStyles) => {
+          const filtered = mirrorStyles.filter(
+            (el) => typeof el !== "undefined"
           );
+          doc.head.innerHTML = "";
+          doc.head.append(...filtered);
+          observer.observe(parentDocument.head, { childList: true, subtree: true });
+          filtered.forEach((el) => {
+            const elHash = (0, import_object_hash.default)(el.outerHTML);
+            hashes[elHash] = true;
+          });
+          onStylesLoaded();
         });
-        observer.observe(parentDocument.head, { childList: true, subtree: true });
         return () => {
           observer.disconnect();
         };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@measured/puck-plugin-heading-analyzer",
-  "version": "0.14.2-canary.58b08a9",
+  "version": "0.14.2-canary.717ccb9",
   "author": "Measured Corporation Ltd <hello@measured.co>",
   "repository": "measuredco/puck",
   "bugs": "https://github.com/measuredco/puck/issues",
@@ -18,7 +18,7 @@
     "dist"
   ],
   "devDependencies": {
-    "@measured/puck": "^0.14.2-canary.58b08a9",
+    "@measured/puck": "^0.14.2-canary.717ccb9",
     "@types/react": "^18.2.0",
     "@types/react-dom": "^18.2.0",
     "eslint": "^7.32.0",