[Refactor] display-name: consolidate identical visitors

HenryBrown0 · Feb 25, 2023 · 8f24366 · 8f24366
1 parent a684ae1
commit 8f24366
Showing 1 changed file with 2 additions and 28 deletions.
diff --git a/lib/rules/display-name.js b/lib/rules/display-name.js
@@ -198,7 +198,7 @@ module.exports = {
         markDisplayNameAsDeclared(component.node.type === 'TSAsExpression' ? component.node.expression : component.node);
       },
 
-      FunctionExpression(node) {
+      'FunctionExpression, FunctionDeclaration, ArrowFunctionExpression'(node) {
         if (ignoreTranspilerName || !hasTranspilerName(node)) {
           return;
         }
@@ -207,40 +207,14 @@ module.exports = {
         }
       },
 
-      FunctionDeclaration(node) {
-        if (ignoreTranspilerName || !hasTranspilerName(node)) {
-          return;
-        }
-        if (components.get(node)) {
-          markDisplayNameAsDeclared(node);
-        }
-      },
-
-      ArrowFunctionExpression(node) {
-        if (ignoreTranspilerName || !hasTranspilerName(node)) {
-          return;
-        }
-
-        if (components.get(node)) {
-          markDisplayNameAsDeclared(node);
-        }
-      },
-
       MethodDefinition(node) {
         if (!propsUtil.isDisplayNameDeclaration(node.key)) {
           return;
         }
         markDisplayNameAsDeclared(node);
       },
 
-      ClassExpression(node) {
-        if (ignoreTranspilerName || !hasTranspilerName(node)) {
-          return;
-        }
-        markDisplayNameAsDeclared(node);
-      },
-
-      ClassDeclaration(node) {
+      'ClassExpression, ClassDeclaration'(node) {
         if (ignoreTranspilerName || !hasTranspilerName(node)) {
           return;
         }