From 669aea93b122a231383ec1665d4613c84a0f591e Mon Sep 17 00:00:00 2001
From: Joshua Amaju <princejosh0007@gmail.com>
Date: Tue, 16 Apr 2024 20:23:28 +0100
Subject: [PATCH] fix: invalid html error by vite

---
 .../core/src/vite/plugin-build-html/mod.ts    | 26 ++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/packages/core/src/vite/plugin-build-html/mod.ts b/packages/core/src/vite/plugin-build-html/mod.ts
index fe2ddf3..fb4ce80 100644
--- a/packages/core/src/vite/plugin-build-html/mod.ts
+++ b/packages/core/src/vite/plugin-build-html/mod.ts
@@ -26,6 +26,30 @@ function is_local_path(path: string) {
   );
 }
 
+const VITE_HTML_PLACEHOLDER = "<div data-obfuscation-placeholder></div>";
+
+const obfuscate: Plugin = {
+  name: "plugin-obfuscate",
+  /**
+   * we need to execute this before any vite html parser sees it and throws an error because of the
+   * svelte script at the beginning of the markup, to trick it into accepting the markup as valid HTML
+   */
+  transformIndexHtml: {
+    order: "pre",
+    handler(html) {
+      return `${VITE_HTML_PLACEHOLDER}\n${html}`;
+    },
+  },
+};
+
+const deobfuscate: Plugin = {
+  enforce: "post",
+  name: "plugin-deobfuscate",
+  transformIndexHtml(html) {
+    return html.replace(`${VITE_HTML_PLACEHOLDER}\n`, "");
+  },
+};
+
 /**
  * Builds svelte components as HTML files, processes images, stylesheets, script tags
  * and other asset types
@@ -197,13 +221,13 @@ export function plugin_build_html({
           logLevel: "silent",
           mode: "production",
           css: resolved_vite_config.css,
-          plugins: [...plugins, resolve],
           json: resolved_vite_config.json,
           base: resolved_vite_config.base,
           define: resolved_vite_config.define,
           esbuild: resolved_vite_config.esbuild,
           optimizeDeps: resolved_vite_config.optimizeDeps,
           envPrefix: resolved_vite_config.envPrefix as string[],
+          plugins: [...plugins, resolve, obfuscate, deobfuscate],
           build: {
             ssr: false,
             outDir: build_dir,