Addrefresh layer escape hatch (#319)

onlook-dev · Sep 11, 2024 · 22ebd61 · 22ebd61
1 parent da69356
commit 22ebd61
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/app/electron/preload/webview/api.ts b/app/electron/preload/webview/api.ts
@@ -3,6 +3,7 @@ import { getElementAtLoc, getElementWithSelector } from './elements';
 import { getInsertedElements, getInsertLocation } from './elements/insert';
 import { getMovedElements } from './elements/move';
 import { drag, endDrag, startDrag } from './elements/move/drag';
+import { processDom } from './dom';
 
 export function setApi() {
     contextBridge.exposeInMainWorld('api', {
@@ -14,5 +15,6 @@ export function setApi() {
         drag: drag,
         endDrag: endDrag,
         getMovedElements: getMovedElements,
+        processDom: processDom,
     });
 }
diff --git a/app/src/lib/editor/engine/index.ts b/app/src/lib/editor/engine/index.ts
@@ -118,4 +118,14 @@ export class EditorEngine {
         }
         webview.openDevTools();
     }
+
+    async refreshLayers() {
+        this.ast.clear();
+        const webviews = this.webviews.webviews;
+        if (webviews.size === 0) {
+            return;
+        }
+        const webview = Array.from(webviews.values())[0].webview;
+        webview.executeJavaScript('window.api?.processDom()');
+    }
 }
diff --git a/app/src/routes/project/RightClickMenu/index.tsx b/app/src/routes/project/RightClickMenu/index.tsx
@@ -33,6 +33,12 @@ export const RightClickMenu = observer(({ children }: RightClickMenuProps) => {
                 editorEngine.inspect();
             },
         },
+        {
+            label: 'Refresh layers',
+            action: () => {
+                editorEngine.refreshLayers();
+            },
+        },
     ];
 
     useEffect(() => {