Put codeToRunBeforeInit into begin of <head> tag to ensure it is ru…

…n before any user code [Publish]
JetBrains · Feb 16, 2024 · e3e60de · e3e60de
1 parent 56581a5
commit e3e60de
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/lib/websandbox.ts b/lib/websandbox.ts
@@ -114,8 +114,15 @@ class Websandbox {
   }
 
   _prepareFrameContent(options: SandboxOptions) {
-    let frameContent = options.frameContent
-      ?.replace('<head>', `<head>\n<script>${CompiledFrameScript}</script>`) ?? '';
+    let frameContent = options.frameContent ?? '';
+
+    if (options.codeToRunBeforeInit) {
+      frameContent = frameContent
+        .replace('</head>', `<script>${options.codeToRunBeforeInit}</script>\n</head>`);
+    }
+
+    frameContent = frameContent
+      .replace('<head>', `<head>\n<script>${CompiledFrameScript}</script>`) ?? '';
 
     if (options.initialStyles) {
       frameContent = frameContent
@@ -127,10 +134,6 @@ class Websandbox {
         .replace('<head>', `<head>\n<base href="${options.baseUrl}"/>`);
     }
 
-    if (options.codeToRunBeforeInit) {
-      frameContent = frameContent
-        .replace('</head>', `<script>${options.codeToRunBeforeInit}</script>\n</head>`);
-    }
     return frameContent;
   }