Handle both script and script setup (#6)

* Handle both script and script setup

* Add test case for mixed script setup and script

---------

Co-authored-by: Bobby Wertman <bwertman@rapidcourt.com>

index.js

@@ -16,15 +16,16 @@ module.exports = function(content, options) {
 
   const result = compiler.parse(content, { sourceMap: false });
   const { styles, script, scriptSetup } = result.descriptor;
-  const usedScript = script || scriptSetup;
 
   const dependencies = [];
 
-  if (usedScript?.content) {
-    if (usedScript.attrs?.lang === 'ts') {
-      dependencies.push(...detectiveTypeScript(usedScript.content, options));
-    } else {
-      dependencies.push(...detectiveEs6(usedScript.content, options));
+  for (const usedScript of [script, scriptSetup]) {
+    if (usedScript?.content) {
+      if (usedScript.attrs?.lang === 'ts') {
+        dependencies.push(...detectiveTypeScript(usedScript.content, options));
+      } else {
+        dependencies.push(...detectiveEs6(usedScript.content, options));
+      }
     }
   }
 

test/test.js

@@ -46,6 +46,22 @@ import OtherComponent from "./OtherComponent.vue";
 <template>
 <OtherComponent />
 </template>
+`);
+    assert.equal(deps.length, 2);
+    assert.equal(deps[0], 'mylib');
+    assert.equal(deps[1], './OtherComponent.vue');
+  });
+
+  it('retrieves the dependencies of script block lang ts using both setup and normal syntax', () => {
+    const deps = detective(`<script lang="ts">
+import { foo, bar } from "mylib";
+</script>
+<script setup lang="ts">
+import OtherComponent from "./OtherComponent.vue";
+</script>
+<template>
+<OtherComponent />
+</template>
 `);
     assert.equal(deps.length, 2);
     assert.equal(deps[0], 'mylib');