From e74c13b55ba80a4454968d57f004cecf32efb731 Mon Sep 17 00:00:00 2001
From: Igor Aufricht <igor.aufricht@eligamo.com>
Date: Mon, 21 Oct 2024 14:05:55 +0200
Subject: [PATCH 1/2] Report React context exports

---
 src/only-export-components.test.ts | 17 +++++++++++++++++
 src/only-export-components.ts      | 20 ++++++++++++++++++--
 2 files changed, 35 insertions(+), 2 deletions(-)

diff --git a/src/only-export-components.test.ts b/src/only-export-components.test.ts
index 62ef3ee..8a2cc48 100755
--- a/src/only-export-components.test.ts
+++ b/src/only-export-components.test.ts
@@ -177,6 +177,18 @@ const valid = [
     name: "Allow connect from react-redux",
     code: "const MyComponent = () => {}; export default connect(() => ({}))(MyComponent);",
   },
+  {
+    name: "Two components, one of them with 'Context' in its name",
+    code: "export const MyComponent = () => {}; export const ChatContext = () => {};",
+  },
+  {
+    name: "Component & local React context",
+    code: "export const MyComponent = () => {}; const MyContext = createContext('test');",
+  },
+  {
+    name: "Only React context (",
+    code: "export const MyContext = createContext('test');",
+  },
 ];
 
 const invalid = [
@@ -273,6 +285,11 @@ const invalid = [
     code: 'const Foo = () => {}; export { Foo as "🍌"}',
     errorId: "localComponents",
   },
+  {
+    name: "Component and React Context",
+    code: "export const MyComponent = () => {}; export const MyContext = createContext('test');",
+    errorId: "reactContext",
+  },
 ];
 
 const it = (name: string, cases: Parameters<typeof ruleTester.run>[2]) => {
diff --git a/src/only-export-components.ts b/src/only-export-components.ts
index e71c6fe..24e1f40 100644
--- a/src/only-export-components.ts
+++ b/src/only-export-components.ts
@@ -13,7 +13,8 @@ export const onlyExportComponents: TSESLint.RuleModule<
   | "namedExport"
   | "anonymousExport"
   | "noExport"
-  | "localComponents",
+  | "localComponents"
+  | "reactContext",
   | []
   | [
       {
@@ -35,6 +36,8 @@ export const onlyExportComponents: TSESLint.RuleModule<
         "Fast refresh only works when a file only exports components. Move your component(s) to a separate file.",
       noExport:
         "Fast refresh only works when a file has exports. Move your component(s) to a separate file.",
+      reactContext:
+        "Fast refresh only works when a file only exports components. Move your React context(s) to a separate file.",
     },
     type: "problem",
     schema: [
@@ -56,7 +59,7 @@ export const onlyExportComponents: TSESLint.RuleModule<
       checkJS = false,
       allowExportNames,
     } = context.options[0] ?? {};
-    const filename = context.getFilename();
+    const filename = context.filename;
     // Skip tests & stories files
     if (
       filename.includes(".test.") ||
@@ -86,6 +89,7 @@ export const onlyExportComponents: TSESLint.RuleModule<
           | TSESTree.BindingName
           | TSESTree.StringLiteral
         )[] = [];
+        const reactContextExports: TSESTree.Identifier[] = [];
 
         const handleLocalIdentifier = (
           identifierNode: TSESTree.BindingName,
@@ -124,6 +128,15 @@ export const onlyExportComponents: TSESLint.RuleModule<
               nonComponentExports.push(identifierNode);
             }
           } else {
+            if (
+              init &&
+              init.type === "CallExpression" &&
+              init.callee.type === "Identifier" &&
+              init.callee.name === "createContext"
+            ) {
+              reactContextExports.push(identifierNode);
+              return;
+            }
             if (
               init &&
               // Switch to allowList?
@@ -263,6 +276,9 @@ export const onlyExportComponents: TSESLint.RuleModule<
             for (const node of nonComponentExports) {
               context.report({ messageId: "namedExport", node });
             }
+            for (const node of reactContextExports) {
+              context.report({ messageId: "reactContext", node });
+            }
           } else if (localComponents.length) {
             for (const node of localComponents) {
               context.report({ messageId: "localComponents", node });

From 3fbb451d70ef1b1d21f9f716cf504e82eee45bab Mon Sep 17 00:00:00 2001
From: Igor Aufricht <IgorAufricht@users.noreply.github.com>
Date: Mon, 21 Oct 2024 14:52:16 +0200
Subject: [PATCH 2/2] Fix typo
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: Arnaud Barré <arnaud.barre72@gmail.com>
---
 src/only-export-components.test.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/only-export-components.test.ts b/src/only-export-components.test.ts
index 8a2cc48..8ba94e8 100755
--- a/src/only-export-components.test.ts
+++ b/src/only-export-components.test.ts
@@ -186,7 +186,7 @@ const valid = [
     code: "export const MyComponent = () => {}; const MyContext = createContext('test');",
   },
   {
-    name: "Only React context (",
+    name: "Only React context",
     code: "export const MyContext = createContext('test');",
   },
 ];