(fix) deal with multiple script tags

packages/svelte2tsx/src/svelte2tsx.ts

@@ -191,20 +191,43 @@ function processSvelteTemplate(str: MagicString): TemplateProcessResult {
         }
     };
 
-    let scriptTag: Node = null;
-    let moduleScriptTag: Node = null;
-    const handleScriptTag = (node: Node) => {
-        if (
-            node.attributes &&
-            node.attributes.find(
-                (a) => a.name == 'context' && a.value.length == 1 && a.value[0].raw == 'module',
-            )
-        ) {
-            moduleScriptTag = node;
-        } else {
-            scriptTag = node;
+    // All script tags, no matter at what level, are listed within the root children.
+    // To get the top level scripts, filter out all those that are part of children's children.
+    // Those have another type ('Element' with name 'script').
+    const scriptTags = (<Node[]>htmlxAst.children).filter((child) => child.type === 'Script');
+    let topLevelScripts = scriptTags;
+    const handleScriptTag = (node: Node, parent: Node) => {
+        if (parent !== htmlxAst && node.name === 'script') {
+            topLevelScripts = topLevelScripts.filter(
+                (tag) => tag.start !== node.start || tag.end !== node.end,
+            );
         }
     };
+    const getTopLevelScriptTags = () => {
+        let scriptTag: Node = null;
+        let moduleScriptTag: Node = null;
+        // should be 2 at most, one each, so using forEach is safe
+        topLevelScripts.forEach((tag) => {
+            if (
+                tag.attributes &&
+                tag.attributes.find(
+                    (a) => a.name == 'context' && a.value.length == 1 && a.value[0].raw == 'module',
+                )
+            ) {
+                moduleScriptTag = tag;
+            } else {
+                scriptTag = tag;
+            }
+        });
+        return { scriptTag, moduleScriptTag };
+    };
+    const blankOtherScriptTags = () => {
+        scriptTags
+            .filter((tag) => !topLevelScripts.includes(tag))
+            .forEach((tag) => {
+                str.remove(tag.start, tag.end);
+            });
+    };
 
     const slots = new Map<string, Map<string, string>>();
     const handleSlot = (node: Node) => {
@@ -245,8 +268,8 @@ function processSvelteTemplate(str: MagicString): TemplateProcessResult {
             case 'Style':
                 handleStyleTag(node);
                 break;
-            case 'Script':
-                handleScriptTag(node);
+            case 'Element':
+                handleScriptTag(node, parent);
                 break;
             case 'BlockStatement':
                 enterBlockStatement();
@@ -290,6 +313,10 @@ function processSvelteTemplate(str: MagicString): TemplateProcessResult {
 
     convertHtmlxToJsx(str, htmlxAst, onHtmlxWalk, onHtmlxLeave);
 
+    // resolve scripts
+    const { scriptTag, moduleScriptTag } = getTopLevelScriptTags();
+    blankOtherScriptTags();
+
     //resolve stores
     pendingStoreResolutions.map(resolveStore);
 

packages/svelte2tsx/test/svelte2tsx/samples/script-inside-head-after-toplevel-script/expected.tsx

@@ -0,0 +1,20 @@
+<></>;function render() {
+
+  let b = 'top level';
+;
+<><div>
+
+</div>
+
+
+
+<sveltehead>
+  <link rel="stylesheet" href="/lib/jodit.es2018.min.css" />
+
+</sveltehead></>
+return { props: {}, slots: {} }}
+
+export default class {
+    $$prop_def = __sveltets_partial(render().props)
+    $$slot_def = render().slots
+}

packages/svelte2tsx/test/svelte2tsx/samples/script-inside-head-after-toplevel-script/input.svelte

@@ -0,0 +1,14 @@
+<div>
+  <script>let a = 'not top level';</script>
+</div>
+
+<script>
+  let b = 'top level';
+</script>
+
+<svelte:head>
+  <link rel="stylesheet" href="/lib/jodit.es2018.min.css" />
+  <script src="/lib/jodit.es2018.min.js">
+
+  </script>
+</svelte:head>

packages/svelte2tsx/test/svelte2tsx/samples/self-closing-component/expected.tsx

@@ -1,7 +1,8 @@
 <></>;import Test from './Test.svelte';
 function render() {
 
-
+
+let a = 'b';
 ;
 <><Test b="6" />
 </>

packages/svelte2tsx/test/svelte2tsx/samples/self-closing-component/input.svelte

@@ -1,4 +1,5 @@
 <Test b="6" />
 <script>
-import Test from './Test.svelte'; 
+import Test from './Test.svelte';
+let a = 'b';
 </script>