fix: fix walking program AST when multiple script blocks (#2703)

Author: ntnyq

lib/utils/index.js

@@ -1380,7 +1380,7 @@ module.exports = {
      * @param {any[]} args
      */
     function callVisitor(key, node, ...args) {
-      if (visitor[key] && inScriptSetup(node)) {
+      if (visitor[key] && (node.type === 'Program' || inScriptSetup(node))) {
         // @ts-expect-error
         visitor[key](node, ...args)
       }

tests/lib/rules/prefer-use-template-ref.js

@@ -263,6 +263,40 @@ tester.run('prefer-use-template-ref', rule, {
           })
         </script>
       `
+    },
+    {
+      filename: 'multiple-scripts-setup-first.vue',
+      code: `
+      <template>
+        <div ref="root" :data-a="A" />
+      </template>
+
+      <script setup>
+      import { useTemplateRef } from 'vue'
+      const root = useTemplateRef('root')
+      </script>
+
+      <script>
+      const A = 'foo'
+      </script>
+      `
+    },
+    {
+      filename: 'multiple-scripts-setup-last.vue',
+      code: `
+      <template>
+        <div ref="root" :data-a="A" />
+      </template>
+
+      <script>
+      const A = 'foo'
+      </script>
+
+      <script setup>
+      import { useTemplateRef } from 'vue'
+      const root = useTemplateRef('root')
+      </script>
+      `
     }
   ],
   invalid: [
@@ -420,6 +454,60 @@ tester.run('prefer-use-template-ref', rule, {
           column: 28
         }
       ]
+    },
+    {
+      filename: 'multiple-scripts-setup-first.vue',
+      code: `
+      <template>
+        <div ref="root" :data-a="A" />
+      </template>
+
+      <script setup>
+      import { ref } from 'vue'
+      const root = ref()
+      </script>
+
+      <script>
+      const A = 'foo'
+      </script>
+      `,
+      errors: [
+        {
+          messageId: 'preferUseTemplateRef',
+          data: {
+            name: 'ref'
+          },
+          line: 8,
+          column: 20
+        }
+      ]
+    },
+    {
+      filename: 'multiple-scripts-setup-last.vue',
+      code: `
+      <template>
+        <div ref="root" :data-a="A" />
+      </template>
+
+      <script>
+      const A = 'foo'
+      </script>
+
+      <script setup>
+      import { ref } from 'vue'
+      const root = ref()
+      </script>
+      `,
+      errors: [
+        {
+          messageId: 'preferUseTemplateRef',
+          data: {
+            name: 'ref'
+          },
+          line: 12,
+          column: 20
+        }
+      ]
     }
   ]
 })