From a4b9255f43175718124de9cb797b3490dc702776 Mon Sep 17 00:00:00 2001
From: mathuo <6710312+mathuo@users.noreply.github.com>
Date: Tue, 10 Dec 2024 20:57:53 +0000
Subject: [PATCH] bug: disable iframes within shadowdom during dnd

---
 packages/dockview-core/src/dom.ts             | 32 ++++++++++---
 .../react/dockview/demo-dockview/src/app.tsx  | 45 ++++++++++++++++++-
 .../demo-dockview/src/gridActions.tsx         | 23 +++++++++-
 .../demo-dockview/src/panelActions.tsx        |  4 +-
 4 files changed, 94 insertions(+), 10 deletions(-)

diff --git a/packages/dockview-core/src/dom.ts b/packages/dockview-core/src/dom.ts
index d61b1cacb..50114138e 100644
--- a/packages/dockview-core/src/dom.ts
+++ b/packages/dockview-core/src/dom.ts
@@ -112,8 +112,11 @@ export function isAncestor(
     return false;
 }
 
-export function getElementsByTagName(tag: string): HTMLElement[] {
-    return Array.prototype.slice.call(document.getElementsByTagName(tag), 0);
+export function getElementsByTagName(
+    tag: string,
+    document: ParentNode
+): HTMLElement[] {
+    return Array.prototype.slice.call(document.querySelectorAll(tag), 0);
 }
 
 export interface IFocusTracker extends IDisposable {
@@ -288,10 +291,29 @@ export function addTestId(element: HTMLElement, id: string): void {
     element.setAttribute('data-testid', id);
 }
 
-export function disableIframePointEvents() {
+export function disableIframePointEvents(rootNode: ParentNode = document) {
+    const includeShadowDom = true;
+
+    const shadowRoots = [];
+
+    if (includeShadowDom) {
+        const items = rootNode.querySelectorAll('*');
+
+        for (let i = 0; i < items.length; i++) {
+            const item = items[i];
+            if (item.shadowRoot) {
+                shadowRoots.push(item.shadowRoot);
+            }
+        }
+    }
+
     const iframes: HTMLElement[] = [
-        ...getElementsByTagName('iframe'),
-        ...getElementsByTagName('webview'),
+        ...getElementsByTagName('iframe', rootNode),
+        ...getElementsByTagName('webview', rootNode),
+        ...shadowRoots.flatMap((root) => [
+            ...getElementsByTagName('iframe', root),
+            ...getElementsByTagName('webview', root),
+        ]),
     ];
 
     const original = new WeakMap<HTMLElement, string>(); // don't hold onto HTMLElement references longer than required
diff --git a/packages/docs/sandboxes/react/dockview/demo-dockview/src/app.tsx b/packages/docs/sandboxes/react/dockview/demo-dockview/src/app.tsx
index 6595a3fce..7f1d8d8b4 100644
--- a/packages/docs/sandboxes/react/dockview/demo-dockview/src/app.tsx
+++ b/packages/docs/sandboxes/react/dockview/demo-dockview/src/app.tsx
@@ -7,6 +7,7 @@ import {
     DockviewApi,
 } from 'dockview';
 import * as React from 'react';
+import * as ReactDOM from 'react-dom/client';
 import './app.scss';
 import { defaultConfig } from './defaultLayout';
 import { GridActions } from './gridActions';
@@ -30,6 +31,20 @@ const Option = (props: {
     );
 };
 
+const ShadowIframe = (props: IDockviewPanelProps) => {
+    return (
+        <iframe
+            onMouseDown={() => {
+                if (!props.api.isActive) {
+                    props.api.setActive();
+                }
+            }}
+            style={{ border: 'none', width: '100%', height: '100%' }}
+            src="https://dockview.dev"
+        />
+    );
+};
+
 const components = {
     default: (props: IDockviewPanelProps) => {
         const isDebug = React.useContext(DebugContext);
@@ -105,6 +120,7 @@ const components = {
                     }
                 }}
                 style={{
+                    border: 'none',
                     width: '100%',
                     height: '100%',
                 }}
@@ -112,6 +128,33 @@ const components = {
             />
         );
     },
+    shadowDom: (props: IDockviewPanelProps) => {
+        const ref = React.useRef<HTMLDivElement>(null);
+
+        React.useEffect(() => {
+            if (!ref.current) {
+                return;
+            }
+
+            const shadow = ref.current.attachShadow({
+                mode: 'open',
+            });
+
+            const shadowRoot = document.createElement('div');
+            shadowRoot.style.height = '100%';
+            shadow.appendChild(shadowRoot);
+
+            const root = ReactDOM.createRoot(shadowRoot);
+
+            root.render(<ShadowIframe {...props} />);
+
+            return () => {
+                root.unmount();
+            };
+        }, []);
+
+        return <div style={{ height: '100%' }} ref={ref}></div>;
+    },
 };
 
 const headerComponents = {
@@ -208,7 +251,7 @@ const DockviewDemo = (props: { theme?: string }) => {
 
         event.api.onDidMaximizedGroupChange((event) => {
             addLogLine(
-                `Group Maximized Changed ${event.view.id} [${event.isMaximized}]`
+                `Group Maximized Changed ${event.group.id} [${event.isMaximized}]`
             );
         });
 
diff --git a/packages/docs/sandboxes/react/dockview/demo-dockview/src/gridActions.tsx b/packages/docs/sandboxes/react/dockview/demo-dockview/src/gridActions.tsx
index f1b2a0eeb..7087347b6 100644
--- a/packages/docs/sandboxes/react/dockview/demo-dockview/src/gridActions.tsx
+++ b/packages/docs/sandboxes/react/dockview/demo-dockview/src/gridActions.tsx
@@ -132,7 +132,10 @@ export const GridActions = (props: {
 
     const popover = usePopover();
 
-    const onAddPanel = (options?: { advanced: boolean }) => {
+    const onAddPanel = (options?: {
+        advanced: boolean;
+        component?: string;
+    }) => {
         if (options?.advanced) {
             popover.open(({ close }) => {
                 return <PanelBuilder api={props.api!} done={close} />;
@@ -140,7 +143,7 @@ export const GridActions = (props: {
         } else {
             props.api?.addPanel({
                 id: `id_${Date.now().toString()}`,
-                component: 'default',
+                component: options?.component ?? 'default',
                 title: `Tab ${nextId()}`,
                 renderer: 'always',
             });
@@ -170,6 +173,22 @@ export const GridActions = (props: {
                     <span className="material-symbols-outlined">tune</span>
                 </button>
             </div>
+            <div className="button-group">
+                <button
+                    className="text-button"
+                    onClick={() =>
+                        onAddPanel({ component: 'shadowDom', advanced: false })
+                    }
+                >
+                    Add Panel 2
+                </button>
+                <button
+                    className="demo-icon-button"
+                    onClick={() => onAddPanel({ advanced: true })}
+                >
+                    <span className="material-symbols-outlined">tune</span>
+                </button>
+            </div>
             <button className="text-button" onClick={onAddGroup}>
                 Add Group
             </button>
diff --git a/packages/docs/sandboxes/react/dockview/demo-dockview/src/panelActions.tsx b/packages/docs/sandboxes/react/dockview/demo-dockview/src/panelActions.tsx
index c81a63077..d12cd06ac 100644
--- a/packages/docs/sandboxes/react/dockview/demo-dockview/src/panelActions.tsx
+++ b/packages/docs/sandboxes/react/dockview/demo-dockview/src/panelActions.tsx
@@ -75,9 +75,9 @@ const PanelAction = (props: {
                         const panel = props.api.getPanel(props.panelId);
                         if (panel) {
                             props.api.addFloatingGroup(panel, {
+                                width: 400,
+                                height: 300,
                                 position: {
-                                    width: 400,
-                                    height: 300,
                                     bottom: 50,
                                     right: 50,
                                 },