feat(plugin-sfc): extract all script blocks

mdit-vue · Aug 24, 2022 · 8c39df6 · 8c39df6
1 parent cb702d1
commit 8c39df6
Show file tree

Hide file tree

Showing 3 changed files with 101 additions and 0 deletions.
diff --git a/packages/plugin-sfc/src/sfc-plugin.ts b/packages/plugin-sfc/src/sfc-plugin.ts
@@ -28,6 +28,7 @@ export const sfcPlugin: PluginWithOptions<SfcPluginOptions> = (
       template: null,
       script: null,
       scriptSetup: null,
+      scripts: [],
       styles: [],
       customBlocks: [],
     };
@@ -74,6 +75,7 @@ export const sfcPlugin: PluginWithOptions<SfcPluginOptions> = (
     // extract sfc blocks to `env.sfcBlocks`
     const sfcBlock = match.groups;
     if (sfcBlock.type === TAG_NAME_SCRIPT) {
+      env.sfcBlocks.scripts.push(sfcBlock);
       if (SCRIPT_SETUP_TAG_OPEN_REGEXP.test(sfcBlock.tagOpen)) {
         env.sfcBlocks.scriptSetup = sfcBlock;
       } else {

diff --git a/packages/plugin-sfc/src/types.ts b/packages/plugin-sfc/src/types.ts
@@ -41,10 +41,37 @@ export interface SfcBlock {
 }
 
 export interface MarkdownSfcBlocks {
+  /**
+   * The `<template>` block
+   */
   template: SfcBlock | null;
+
+  /**
+   * The common `<script>` block
+   */
   script: SfcBlock | null;
+
+  /**
+   * The `<script setup>` block
+   */
   scriptSetup: SfcBlock | null;
+
+  /**
+   * All `<script>` blocks.
+   *
+   * By default, SFC only allows one `<script>` block and one `<script setup>` block.
+   * However, some tools may support different types of `<script>`s, so we keep all of them here.
+   */
+  scripts: SfcBlock[];
+
+  /**
+   * All `<style>` blocks.
+   */
   styles: SfcBlock[];
+
+  /**
+   * All custom blocks.
+   */
   customBlocks: SfcBlock[];
 }
 

diff --git a/packages/plugin-sfc/tests/__snapshots__/sfc-plugin.spec.ts.snap b/packages/plugin-sfc/tests/__snapshots__/sfc-plugin.spec.ts.snap
@@ -55,6 +55,42 @@ const foo = 'scriptSetup'
     "tagOpen": "<script setup lang=\\"ts\\">",
     "type": "script",
   },
+  "scripts": [
+    {
+      "content": "<script setup lang=\\"ts\\">
+const foo = 'scriptSetup'
+</script>",
+      "contentStripped": "
+const foo = 'scriptSetup'
+",
+      "tagClose": "</script>",
+      "tagOpen": "<script setup lang=\\"ts\\">",
+      "type": "script",
+    },
+    {
+      "content": "<script>
+export default {
+  setup() {
+    return {
+      msg: 'script'
+    }
+  }
+}
+</script>",
+      "contentStripped": "
+export default {
+  setup() {
+    return {
+      msg: 'script'
+    }
+  }
+}
+",
+      "tagClose": "</script>",
+      "tagOpen": "<script>",
+      "type": "script",
+    },
+  ],
   "styles": [
     {
       "content": "<style lang=\\"stylus\\">
@@ -130,6 +166,42 @@ const foo = 'scriptSetup'
     "tagOpen": "<script setup lang=\\"ts\\">",
     "type": "script",
   },
+  "scripts": [
+    {
+      "content": "<script setup lang=\\"ts\\">
+const foo = 'scriptSetup'
+</script>",
+      "contentStripped": "
+const foo = 'scriptSetup'
+",
+      "tagClose": "</script>",
+      "tagOpen": "<script setup lang=\\"ts\\">",
+      "type": "script",
+    },
+    {
+      "content": "<script>
+export default {
+  setup() {
+    return {
+      msg: 'script'
+    }
+  }
+}
+</script>",
+      "contentStripped": "
+export default {
+  setup() {
+    return {
+      msg: 'script'
+    }
+  }
+}
+",
+      "tagClose": "</script>",
+      "tagOpen": "<script>",
+      "type": "script",
+    },
+  ],
   "styles": [
     {
       "content": "<style lang=\\"stylus\\">