Supports React.forwardRef and React.memo [publish] (#46)

Author: SukkaW

.github/workflows/ci.yml

@@ -0,0 +1,13 @@
+name: CI
+on:
+  pull_request:
+    branches:
+      - main
+jobs:
+  ci:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: oven-sh/setup-bun@v2
+      - run: bun install
+      - run: bun ci

.github/workflows/publish.yml

@@ -8,8 +8,8 @@ jobs:
     runs-on: ubuntu-latest
     if: ${{ contains(github.event.head_commit.message, '[publish]') }}
     steps:
-      - uses: actions/checkout@v3
-      - uses: oven-sh/setup-bun@v1
+      - uses: actions/checkout@v4
+      - uses: oven-sh/setup-bun@v2
       - run: bun install
       - run: bun ci
       - uses: ArnaudBarre/npm-publish@v1

CHANGELOG.md

@@ -1,8 +1,12 @@
 # Changelog
 
+## 0.4.10
+
+- Support `function Foo() {}; export default React.memo(Foo)` (#46) (thanks @SukkaW!)
+
 ## 0.4.9
 
-- Support `function Foo() {}; export default memo(Foo)` (fixes #44)
+- Support `function Foo() {}; export default memo(Foo)` (fixes #44) (thanks @SukkaW!)
 
 ## 0.4.8
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "eslint-plugin-react-refresh",
-  "version": "0.4.9",
+  "version": "0.4.10",
   "type": "module",
   "license": "MIT",
   "scripts": {

src/only-export-components.test.ts

@@ -89,14 +89,26 @@ const valid = [
     name: "export default memo function",
     code: "export default memo(function Foo () {});",
   },
+  {
+    name: "export default React.memo function",
+    code: "export default React.memo(function Foo () {});",
+  },
   {
     name: "export default memo function assignment",
     code: "const Foo = () => {}; export default memo(Foo);",
   },
+  {
+    name: "export default React.memo function assignment",
+    code: "const Foo = () => {}; export default React.memo(Foo);",
+  },
   {
     name: "export default memo function declaration",
     code: "function Foo() {}; export default memo(Foo);",
   },
+  {
+    name: "export default React.memo function declaration",
+    code: "function Foo() {}; export default React.memo(Foo);",
+  },
   {
     name: "export type *",
     code: "export type * from './module';",

src/only-export-components.ts

@@ -157,10 +157,20 @@ export const onlyExportComponents: TSESLint.RuleModule<
               handleExportIdentifier(node.id, true);
             }
           } else if (node.type === "CallExpression") {
-            if (
-              node.callee.type !== "Identifier" ||
-              !reactHOCs.has(node.callee.name)
-            ) {
+            // we rule out non HoC first
+            if (node.callee.type !== "Identifier") {
+              // export default React.memo(function Foo() {})
+              // export default Preact.memo(function Foo() {})
+              if (
+                node.callee.type === "MemberExpression" &&
+                node.callee.property.type === "Identifier" &&
+                reactHOCs.has(node.callee.property.name)
+              ) {
+                mayHaveReactExport = true;
+              } else {
+                context.report({ messageId: "anonymousExport", node });
+              }
+            } else if (!reactHOCs.has(node.callee.name)) {
               // we rule out non HoC first
               context.report({ messageId: "anonymousExport", node });
             } else if (