From 4e575ea5e041f83485e0e528f3d665a1ff634004 Mon Sep 17 00:00:00 2001
From: Yosuke Ota <otameshiyo23@gmail.com>
Date: Tue, 6 Jul 2021 18:25:31 +0900
Subject: [PATCH] Update `vue/no-unused-refs` rule to support `setup()` and
 `<script setup>` (#1556)

---
 lib/rules/no-unused-refs.js       | 69 ++++++++++++++++++++++++-------
 tests/lib/rules/no-unused-refs.js | 31 ++++++++++++++
 2 files changed, 85 insertions(+), 15 deletions(-)

diff --git a/lib/rules/no-unused-refs.js b/lib/rules/no-unused-refs.js
index 0e9455f61..2c47bc035 100644
--- a/lib/rules/no-unused-refs.js
+++ b/lib/rules/no-unused-refs.js
@@ -180,25 +180,64 @@ module.exports = {
           reportUnusedRefs()
         }
       },
-      {
-        Identifier(id) {
-          if (hasUnknown) {
-            return
-          }
-          if (id.name !== '$refs') {
-            return
+      utils.compositingVisitors(
+        utils.isScriptSetup(context)
+          ? {
+              Program() {
+                const globalScope =
+                  context.getSourceCode().scopeManager.globalScope
+                if (!globalScope) {
+                  return
+                }
+                for (const variable of globalScope.variables) {
+                  if (variable.defs.length > 0) {
+                    usedRefs.add(variable.name)
+                  }
+                }
+                const moduleScope = globalScope.childScopes.find(
+                  (scope) => scope.type === 'module'
+                )
+                if (!moduleScope) {
+                  return
+                }
+                for (const variable of moduleScope.variables) {
+                  if (variable.defs.length > 0) {
+                    usedRefs.add(variable.name)
+                  }
+                }
+              }
+            }
+          : {},
+        utils.defineVueVisitor(context, {
+          onVueObjectEnter(node) {
+            for (const prop of utils.iterateProperties(
+              node,
+              new Set(['setup'])
+            )) {
+              usedRefs.add(prop.name)
+            }
           }
-          /** @type {Identifier | MemberExpression} */
-          let refsNode = id
-          if (id.parent.type === 'MemberExpression') {
-            if (id.parent.property === id) {
-              // `this.$refs.foo`
-              refsNode = id.parent
+        }),
+        {
+          Identifier(id) {
+            if (hasUnknown) {
+              return
+            }
+            if (id.name !== '$refs') {
+              return
             }
+            /** @type {Identifier | MemberExpression} */
+            let refsNode = id
+            if (id.parent.type === 'MemberExpression') {
+              if (id.parent.property === id) {
+                // `this.$refs.foo`
+                refsNode = id.parent
+              }
+            }
+            extractUsedForPattern(refsNode)
           }
-          extractUsedForPattern(refsNode)
         }
-      }
+      )
     )
   }
 }
diff --git a/tests/lib/rules/no-unused-refs.js b/tests/lib/rules/no-unused-refs.js
index 3deea4680..2547ef05d 100644
--- a/tests/lib/rules/no-unused-refs.js
+++ b/tests/lib/rules/no-unused-refs.js
@@ -284,6 +284,37 @@ tester.run('no-unused-refs', rule, {
       }
       </script>
       `
+    },
+    {
+      filename: 'test.vue',
+      code: `
+      <template>
+        <input ref="x" />
+      </template>
+      <script>
+      import {ref} from 'vue'
+      export default {
+        setup() {
+          const x = ref(null)
+          return {
+            x
+          }
+        }
+      }
+      </script>
+      `
+    },
+    {
+      filename: 'test.vue',
+      code: `
+      <template>
+        <input ref="x" />
+      </template>
+      <script setup>
+      import {ref} from 'vue'
+      const x = ref(null)
+      </script>
+      `
     }
   ],