Add createContextMethods option

src/only-export-components.test.ts

@@ -189,6 +189,21 @@ const valid = [
     name: "Only React context",
     code: "export const MyContext = createContext('test');",
   },
+  {
+    name: "Component and local React Context from other module",
+    code: "export const MyComponent = () => {}; const MyContext = createMyContext('test');",
+    errorId: "reactContext",
+    options: [
+      {
+        createContextMethods: ["createMyContext"],
+      },
+    ],
+  },
+  {
+    name: "Component and React Context from other module without options",
+    code: "export const MyComponent = () => {}; export const MyContext = createMyContext('test');",
+    errorId: "reactContext",
+  },
 ];
 
 const invalid = [
@@ -295,6 +310,16 @@ const invalid = [
     code: "export const MyComponent = () => {}; export const MyContext = React.createContext('test');",
     errorId: "reactContext",
   },
+  {
+    name: "Component and React Context from other module",
+    code: "export const MyComponent = () => {}; export const MyContext = createMyContext('test');",
+    errorId: "reactContext",
+    options: [
+      {
+        createContextMethods: ["createMyContext"],
+      },
+    ],
+  },
 ];
 
 const it = (name: string, cases: Parameters<typeof ruleTester.run>[2]) => {

src/only-export-components.ts

@@ -21,6 +21,7 @@ export const onlyExportComponents: TSESLint.RuleModule<
         allowConstantExport?: boolean;
         checkJS?: boolean;
         allowExportNames?: string[];
+        createContextMethods?: string[];
       },
     ]
 > = {
@@ -47,6 +48,7 @@ export const onlyExportComponents: TSESLint.RuleModule<
           allowConstantExport: { type: "boolean" },
           checkJS: { type: "boolean" },
           allowExportNames: { type: "array", items: { type: "string" } },
+          createContextMethods: { type: "array", items: { type: "string" } },
         },
         additionalProperties: false,
       },
@@ -58,6 +60,7 @@ export const onlyExportComponents: TSESLint.RuleModule<
       allowConstantExport = false,
       checkJS = false,
       allowExportNames,
+      createContextMethods = [],
     } = context.options[0] ?? {};
     const filename = context.filename;
     // Skip tests & stories files
@@ -79,6 +82,11 @@ export const onlyExportComponents: TSESLint.RuleModule<
       ? new Set(allowExportNames)
       : undefined;
 
+    const createContextMethodsSet = new Set([
+      ...createContextMethods,
+      "createContext",
+    ]);
+
     return {
       Program(program) {
         let hasExports = false;
@@ -133,10 +141,10 @@ export const onlyExportComponents: TSESLint.RuleModule<
               init.type === "CallExpression" &&
               // createContext || React.createContext
               ((init.callee.type === "Identifier" &&
-                init.callee.name === "createContext") ||
+                createContextMethodsSet.has(init.callee.name)) ||
                 (init.callee.type === "MemberExpression" &&
                   init.callee.property.type === "Identifier" &&
-                  init.callee.property.name === "createContext"))
+                  createContextMethodsSet.has(init.callee.property.name)))
             ) {
               reactContextExports.push(identifierNode);
               return;