unplugin · xiaoxiangmoe · Aug 1, 2023 · Jun 12, 2023 · Jun 12, 2023 · Jun 13, 2023
diff --git a/src/core/transform.ts b/src/core/transform.ts
@@ -6,7 +6,7 @@ import { transformScriptSetup } from './transformScriptSetup'
 import { transformSfcRefSugar } from './transformSfcRefSugar'
 import { resolveOptions } from './options'
 
-const scriptSetupRE = /<script\s(.*\s)?setup(\s.*)?>/
+export const scriptSetupRE = /<script\s+(.*\s+)?setup(\s+.*)?\s*>/
 
 export function shouldTransform(code: string, id: string, options?: ScriptSetupTransformOptions): boolean {
   // avoid transforming twice

diff --git a/test/transform_filter.test.ts b/test/transform_filter.test.ts
@@ -0,0 +1,75 @@
+import { } from 'node:test'
+import { describe, expect, it } from 'vitest'
+import { scriptSetupRE } from '../src/core'
+
+describe('transform filter', () => {
+  describe('look for what needs to be converted by regular ', () => {
+    const cases: string[] = [
+      `<script  lang="tsx"
+      setup>
+      import HelloWorld from './HelloWorld.vue'
+      </setup>`,
+      '<script  lang="ts" setup>',
+      `<script 
+       lang="ts"
+        setup>
+        import HelloWorld from './HelloWorld.vue'
+        </script>`,
+        `<script 
+        setup>
+        import HelloWorld from './HelloWorld.vue'
+        </script>`,
+        `<script setup>
+        import HelloWorld from './HelloWorld.vue'
+        </script>`,
+        `<script setup lang="tsx"      >
+        import HelloWorld from './HelloWorld.vue'
+        </script>
+        `,
+        `<script setup 
+        lang="ts">
+        import HelloWorld from './HelloWorld.vue'
+        </script>
+        `,
+        `<script 
+        setup 
+                lang="ts">
+        import HelloWorld from './HelloWorld.vue'
+        </script>
+                `,
+       `<script setup 
+                lang="ts">
+         import HelloWorld from './HelloWorld.vue'
+        </script>`,
+    ]
+
+    for (const input of cases) {
+      it(input, () => {
+        expect(scriptSetupRE.test(input)).toEqual(true)
+      })
+    }
+  })
+
+  describe('filter what is not needed by regular ', () => {
+    const cases: string[] = [
+      `<scriptlang="ts"
+      setup>
+      import HelloWorld from './HelloWorld.vue'
+      </script>
+      `,
+      `<script lang="ts">
+      import HelloWorld from './HelloWorld.vue'
+      </script>`,
+      `<script>
+      import HelloWorld from './HelloWorld.vue'
+      </script>
+      `,
+    ]
+
+    for (const input of cases) {
+      it(input, () => {
+        expect(scriptSetupRE.test(input)).toEqual(false)
+      })
+    }
+  })
+})