Merge pull request #433 from TrevorBurnham/trevorburnham/export-raw-expression

feat: Export raw expression when shouldIncludeExpression is set

Author: pvasek

src/__tests__/parser.ts

@@ -1513,13 +1513,26 @@ describe('parser', () => {
       it('should be disabled by default', () => {
         const [parsed] = parse(fixturePath('StatelessDisplayName'));
         assert.equal(parsed.expression, undefined);
+        assert.equal(parsed.expression, parsed.rootExpression);
       });
 
       it('should cause the parser to return the component expression when set to true', () => {
         const [parsed] = parse(fixturePath('StatelessDisplayName'), {
           shouldIncludeExpression: true
         });
         assert.equal(parsed.expression!.name, 'Stateless');
+        assert.equal(parsed.expression, parsed.rootExpression);
+      });
+
+      it('should cause the parser to return the root expression when set to true', () => {
+        const [parsed] = parse(
+          fixturePath('FunctionDeclarationAsDefaultExportWithMemo'),
+          {
+            shouldIncludeExpression: true
+          }
+        );
+        assert.equal(parsed.expression!.name, 'Jumbotron');
+        assert.equal(parsed.rootExpression!.name, 'default');
       });
     });
   });

src/parser.ts

@@ -15,6 +15,7 @@ export interface StringIndexedObject<T> {
 
 export interface ComponentDoc {
   expression?: ts.Symbol;
+  rootExpression?: ts.Symbol;
   displayName: string;
   filePath: string;
   description: string;
@@ -395,6 +396,7 @@ export class Parser {
 
     if (result !== null && this.shouldIncludeExpression) {
       result.expression = rootExp;
+      result.rootExpression = exp;
     }
 
     return result;