fix: rename injectCode to injectCodeFunction and improve CSS injection logic in vite.config.ts

MickaelAustoni · MickaelAustoni · commit 2f0a2b9384ee · 2025-11-10T00:37:40.000+01:00
diff --git a/vite.config.ts b/vite.config.ts
@@ -44,10 +44,21 @@ const config = () =>
         jsAssetsFilterFunction: (outputChunk) => {
           return outputChunk.fileName.includes("ThemeContext");
         },
-        // Inject CSS at the beginning of <head> for lower specificity
-        // This allows users to easily override Treege styles
-        injectCode: (cssCode) => {
-          return `try{if(typeof document!=="undefined"){var style=document.createElement("style");style.id="treege-styles";style.appendChild(document.createTextNode(${JSON.stringify(cssCode)}));document.head.insertBefore(style,document.head.firstChild);}}catch(e){console.error("vite-plugin-css-injected-by-js",e);}`;
+        injectCodeFunction: (cssCode: string) => {
+          // Inject CSS at the beginning of <head> for lower specificity
+          // This allows users to easily override Treege styles
+          const doc = (globalThis as any).document;
+
+          try {
+            if (typeof doc !== "undefined") {
+              const style = doc.createElement("style");
+              style.id = "treege-styles";
+              style.appendChild(doc.createTextNode(cssCode));
+              doc.head.insertBefore(style, doc.head.firstChild);
+            }
+          } catch (e) {
+            console.error("vite-plugin-css-injected-by-js", e);
+          }
         },
       }),
     ],
