Support memo default export function components (fixes #27)

ArnaudBarre · Oct 24, 2023 · 8763332 · 8763332
1 parent 8254172
commit 8763332
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+## Unreleased
+
+- Support memo default export function components (fixes #27)
+
 ## 0.4.3
 
 - Add warning for TS enums exports

diff --git a/src/only-export-components.test.ts b/src/only-export-components.test.ts
@@ -81,6 +81,10 @@ const valid = [
     name: "Direct export default AF",
     code: "export default function foo () {};",
   },
+  {
+    name: "export default memo function",
+    code: "export default memo(function Foo () {});",
+  },
   {
     name: "export type *",
     code: "export type * from './module';",
@@ -146,6 +150,11 @@ const invalid = [
     code: "export default () => {};",
     errorId: "anonymousExport",
   },
+  {
+    name: "export default anonymous memo AF",
+    code: "export default memo(() => {});",
+    errorId: "anonymousExport",
+  },
   {
     name: "Export default anonymous function",
     code: "export default function () {};",

diff --git a/src/only-export-components.ts b/src/only-export-components.ts
@@ -126,7 +126,16 @@ export const onlyExportComponents: TSESLint.RuleModule<
               handleExportIdentifier(node.id, true);
             }
           } else if (node.type === "CallExpression") {
-            context.report({ messageId: "anonymousExport", node });
+            if (
+              node.callee.type === "Identifier" &&
+              reactHOCs.includes(node.callee.name) &&
+              node.arguments[0]?.type === "FunctionExpression" &&
+              node.arguments[0].id
+            ) {
+              handleExportIdentifier(node.arguments[0].id, true);
+            } else {
+              context.report({ messageId: "anonymousExport", node });
+            }
           } else if (node.type === "TSEnumDeclaration") {
             nonComponentExports.push(node.id);
           }
@@ -196,12 +205,13 @@ export const onlyExportComponents: TSESLint.RuleModule<
   },
 };
 
+const reactHOCs = ["memo", "forwardRef"];
 const canBeReactFunctionComponent = (init: TSESTree.Expression | null) => {
   if (!init) return false;
   if (init.type === "ArrowFunctionExpression") return true;
   if (init.type === "CallExpression") {
     if (init.callee.type === "Identifier") {
-      return ["memo", "forwardRef"].includes(init.callee.name);
+      return reactHOCs.includes(init.callee.name);
     }
   }
   return false;