refactor(prefer-use-template-ref): optimize array processing with flatMap

Author: 9romise

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

@@ -29,19 +29,20 @@ function convertDeclaratorToScriptRef(declarator) {
  * @returns {ScriptRef[]}
  * */
 function getScriptRefsFromSetupFunction(body) {
-  /** @type {VariableDeclaration[]} */
-  const variableDeclarations = body.filter(
-    (child) => child.type === 'VariableDeclaration'
-  )
-  const variableDeclarators = variableDeclarations.map(
-    (declaration) => declaration.declarations[0]
-  )
-  const refDeclarators = variableDeclarators.filter((declarator) =>
-    // @ts-ignore
-    ['ref', 'shallowRef'].includes(declarator.init?.callee?.name)
-  )
+  return body.flatMap((child) => {
+    if (child.type === 'VariableDeclaration') {
+      const declarator = child.declarations[0]
+
+      if (
+        declarator.init?.type === 'CallExpression' &&
+        declarator.init.callee?.type === 'Identifier' &&
+        ['ref', 'shallowRef'].includes(declarator.init?.callee?.name)
+      )
+        return [convertDeclaratorToScriptRef(declarator)]
+    }
 
-  return refDeclarators.map(convertDeclaratorToScriptRef)
+    return []
+  })
 }
 
 /** @type {import("eslint").Rule.RuleModule} */