fix: head transformer append scripts inside for Marko 5

marko-js · Jul 6, 2020 · cd66494 · cd66494
1 parent ed85dd3
commit cd66494
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/src/loader/head-transformer.ts b/src/loader/head-transformer.ts
@@ -19,15 +19,18 @@ function transformerMarko5(path, t): void {
     t.identifier("out"),
     t.identifier("___renderAssets")
   );
-  path.replaceWith(
-    t.markoScriptlet([
-      t.expressionStatement(
-        t.logicalExpression(
-          "&&",
-          renderAssetsMember,
-          t.callExpression(renderAssetsMember, [])
+  path
+    .get("body")
+    .pushContainer(
+      "body",
+      t.markoScriptlet([
+        t.expressionStatement(
+          t.logicalExpression(
+            "&&",
+            renderAssetsMember,
+            t.callExpression(renderAssetsMember, [])
+          )
         )
-      )
-    ])
-  );
+      ])
+    );
 }