fix(editor): fix captureDOM for <slot> elements

seanchas116 · seanchas116 · commit 1584b9de0302 · 2022-06-29T20:10:52.000+09:00
diff --git a/packages/editor/src/util/CaptureDOM.ts b/packages/editor/src/util/CaptureDOM.ts
@@ -95,15 +95,26 @@ function getRenderableAST(node: Node): hast.Element | hast.Text | undefined {
     }
     attributes.style = styleLines.join("");
 
-    const childNodes = element.shadowRoot
-      ? element.shadowRoot.childNodes
-      : element.tagName === "SLOT"
-      ? (element as HTMLSlotElement).assignedNodes()
-      : element.childNodes;
+    const childNodes = getActualChildNodes(element);
 
     const children = compact(Array.from(childNodes).map(getRenderableAST));
     const isSVG = element.namespaceURI === "http://www.w3.org/2000/svg";
 
     return (isSVG ? s : h)(element.tagName.toLowerCase(), attributes, children);
   }
 }
+
+function getActualChildNodes(element: Element): Node[] {
+  if (element.shadowRoot) {
+    return [...element.shadowRoot.childNodes];
+  }
+
+  if (element.tagName === "SLOT") {
+    const assignedNodes = (element as HTMLSlotElement).assignedNodes();
+    if (assignedNodes.length) {
+      return assignedNodes;
+    }
+  }
+
+  return [...element.childNodes];
+}
