refactor: centralise module script injection

fwouts · Aug 26, 2023 · 2b01334 · 2b01334
1 parent 0c5bac8
commit 2b01334
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 17 deletions.
diff --git a/core/src/vite/vite-manager.ts b/core/src/vite/vite-manager.ts
@@ -67,14 +67,19 @@ export class ViteManager {
         : null;
     return await viteServer.transformIndexHtml(
       url,
-      template.replace(/%([^%]+)%/gi, (matched) => {
-        switch (matched) {
-          case "%INIT_PREVIEW_BLOCK%":
-            return `
+      template.replace(
+        "%MODULE_SCRIPT%",
+        `
+    import { initListeners, initPreview } from "/__previewjs_internal__/index.ts";
+
+    initListeners();
+
+    import.meta.hot.accept();
+
     let latestPreviewableModule;
     let latestWrapperModule;
     let refresh;
-    
+
     import.meta.hot.accept(["/${componentPath}"], ([previewableModule]) => {
       if (previewableModule && refresh) {
         latestPreviewableModule = previewableModule;
@@ -120,11 +125,8 @@ export class ViteManager {
         });
       });
     });
-  `;
-          default:
-            throw new Error(`Unknown template key: ${matched}`);
-        }
-      })
+  `
+      )
     );
   }
 

diff --git a/iframe/preview/index.html b/iframe/preview/index.html
@@ -7,13 +7,7 @@
       window.module = undefined;
     </script>
     <script type="module">
-      import { initListeners, initPreview } from "/__previewjs_internal__/index.ts";
-
-      initListeners();
-
-      import.meta.hot.accept();
-
-      %INIT_PREVIEW_BLOCK%
+      %MODULE_SCRIPT%
     </script>
     <style>
       body {