Fix handling of IFrame src (#31)

* Fix handling of IFrame src * simplify
TileDB-Inc · Dec 11, 2024 · bab0f4e · bab0f4e
1 parent 782b8be
commit bab0f4e
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/demo/src/main.js b/demo/src/main.js
@@ -121,14 +121,11 @@ const modeRadios = document.querySelectorAll('input[name="mode"]');
 modeRadios.forEach(radio => {
   radio.addEventListener('change', e => {
     const isNotebookView = e.target.value === 'notebook';
-    const currentUrl = new URL(iframe.src);
+    let currentUrl = new URL(iframe.src);
     const isLite = currentUrl.pathname.includes('lite');
 
     if (isLite) {
-      currentUrl.pathname = isNotebookView
-        ? '/lite/notebooks/index.html'
-        : '/lite/lab';
-      currentUrl.search = isNotebookView ? '?path=example.ipynb' : '';
+      currentUrl = `./lite/${isNotebookView ? 'notebooks/index.html?path=example.ipynb' : 'lab'}`;
     } else {
       currentUrl.pathname = isNotebookView
         ? '/notebooks/example.ipynb'