chore: Refactor component name logic to a generic interface

Author: bebraw

README.md

@@ -47,6 +47,22 @@ propsParser: require('react-docgen-typescript').withCustomConfig('./tsconfig.jso
 
   Note: `children` without a doc comment will not be documented.
 
+- componentNameResolver:
+
+  ```typescript
+  (exp: ts.Symbol, source: ts.SourceFile) => string | undefined | null | false
+  ```
+
+  If a string is returned, then the component will use that name. Else it will fallback to the default logic of parser.
+
+**Styled components example:**
+
+```typescript
+componentNameResolver: (exp, source) => exp.getName() === 'StyledComponentClass' && getDefaultExportForFile(source);
+```
+
+> The parser exports `getDefaultExportForFile` helper through its public API.
+
 ## Example
 
 In the example folder you can see React Styleguidist integration.

src/__tests__/parser.ts

@@ -3,6 +3,7 @@ import * as fs from 'fs';
 import * as path from 'path';
 import * as ts from 'typescript';
 import {
+  getDefaultExportForFile,
   parse,
   PropFilter,
   withCustomConfig,
@@ -532,13 +533,6 @@ describe('parser', () => {
       assert.equal(parsed.displayName, 'StatelessDisplayNameDefaultExport');
     });
 
-    it('should be taken from filename for styled components', () => {
-      const [parsed] = parse(
-        fixturePath('StatelessDisplayNameStyledComponent')
-      );
-      assert.equal(parsed.displayName, 'StatelessDisplayNameStyledComponent');
-    });
-
     it('should be taken from stateful component `displayName` property (using default export)', () => {
       const [parsed] = parse(fixturePath('StatefulDisplayNameDefaultExport'));
       assert.equal(parsed.displayName, 'StatefulDisplayNameDefaultExport');
@@ -799,4 +793,29 @@ describe('parser', () => {
       assert.isTrue(programProviderInvoked);
     });
   });
+
+  describe('componentNameResolver', () => {
+    it('should override default behavior', () => {
+      const [parsed] = parse(
+        fixturePath('StatelessDisplayNameStyledComponent'),
+        {
+          componentNameResolver: (exp, source) => (
+            exp.getName() === 'StyledComponentClass' &&
+            getDefaultExportForFile(source)
+          )
+        }
+      );
+      assert.equal(parsed.displayName, 'StatelessDisplayNameStyledComponent');
+    });
+
+    it('should fallback to default behavior without a match', () => {
+      const [parsed] = parse(
+        fixturePath('StatelessDisplayNameStyledComponent'),
+        {
+          componentNameResolver: () => false
+        }
+      );
+      assert.equal(parsed.displayName, 'StatelessDisplayNameStyledComponent');
+    });
+  });
 });

src/parser.ts

@@ -41,8 +41,11 @@ export interface ParentType {
 
 export type PropFilter = (props: PropItem, component: Component) => boolean;
 
+export type ComponentNameResolver = (exp: ts.Symbol, source: ts.SourceFile) => string | undefined | null | false;
+
 export interface ParserOptions {
   propFilter?: StaticPropFilter | PropFilter;
+  componentNameResolver?: ComponentNameResolver;
 }
 
 export interface StaticPropFilter {
@@ -166,7 +169,8 @@ class Parser {
 
   public getComponentInfo(
     exp: ts.Symbol,
-    source: ts.SourceFile
+    source: ts.SourceFile,
+    componentNameResolver: ComponentNameResolver = () => undefined
   ): ComponentDoc | null {
     if (!!exp.declarations && exp.declarations.length === 0) {
       return null;
@@ -204,7 +208,8 @@ class Parser {
       this.extractPropsFromTypeIfStatefulComponent(type);
 
     if (propsType) {
-      const componentName = computeComponentName(exp, source);
+      const resolvedComponentName = componentNameResolver(exp, source);
+      const componentName = resolvedComponentName || computeComponentName(exp, source);
       const defaultProps = this.extractDefaultPropsFromComponent(exp, source);
       const props = this.getPropsInfo(propsType, defaultProps);
 
@@ -665,18 +670,21 @@ function computeComponentName(exp: ts.Symbol, source: ts.SourceFile) {
   if (
     exportName === 'default' ||
     exportName === '__function' ||
-    exportName === 'StyledComponentClass' ||
     exportName === 'StatelessComponent'
   ) {
-    // Default export for a file: named after file
+    return  getDefaultExportForFile(source);
+  } else {
+    return exportName;
+  }
+}
+
+// Default export for a file: named after file
+export function getDefaultExportForFile(source: ts.SourceFile) {
     const name = path.basename(source.fileName, path.extname(source.fileName));
 
     return name === 'index'
       ? path.basename(path.dirname(source.fileName))
       : name;
-  } else {
-    return exportName;
-  }
 }
 
 function getParentType(prop: ts.Symbol): ParentType | undefined {
@@ -746,7 +754,7 @@ function parseWithProgramProvider(
         docs,
         checker
           .getExportsOfModule(moduleSymbol)
-          .map(exp => parser.getComponentInfo(exp, sourceFile))
+          .map(exp => parser.getComponentInfo(exp, sourceFile, parserOpts.componentNameResolver))
           .filter((comp): comp is ComponentDoc => comp !== null)
           .filter((comp, index, comps) =>
             comps