Support 'memo(forwardRef(...'

Author: mfazekas

src/handlers/__tests__/__snapshots__/defaultPropsHandler-test.ts.snap

@@ -561,6 +561,17 @@ Object {
 }
 `;
 
+exports[`defaultPropsHandler forwardRef resolves default props when memo used 1`] = `
+Object {
+  "foo": Object {
+    "defaultValue": Object {
+      "computed": false,
+      "value": "'bar'",
+    },
+  },
+}
+`;
+
 exports[`defaultPropsHandler forwardRef resolves defaultProps 1`] = `
 Object {
   "foo": Object {

src/handlers/__tests__/componentDocblockHandler-test.ts

@@ -320,6 +320,22 @@ describe('componentDocblockHandler', () => {
       );
     });
 
+    describe('inline implementation with memo', () => {
+      test(`
+        React.memo(React.forwardRef((props, ref) => {}));
+        import React from "react";`, src =>
+        beforeLastStatement(src).get('expression'));
+
+      testImports(
+        `
+        export default React.memo(React.forwardRef((props, ref) => {}));
+        import React from 'react';`,
+        src => beforeLastStatement(src).get('declaration'),
+        'RefComponent',
+        useDefault,
+      );
+    });
+
     describe('out of line implementation', () => {
       test(`let Component = (props, ref) => {};
         React.forwardRef(Component);

src/handlers/__tests__/defaultPropsHandler-test.ts

@@ -573,6 +573,19 @@ describe('defaultPropsHandler', () => {
       expect(documentation.descriptors).toMatchSnapshot();
     });
 
+    it('resolves default props when memo used', () => {
+      const src = `
+        import React, {memo, forwardRef} from 'react';
+        memo(forwardRef(({ foo = 'bar' }, ref) => <div ref={ref}>{foo}</div>));
+      `;
+      defaultPropsHandler(
+        documentation,
+        parse(src).get('body', 1, 'expression'),
+        noopImporter,
+      );
+      expect(documentation.descriptors).toMatchSnapshot();
+    });
+
     it('resolves imported default props in the parameters', () => {
       const src = `
         import baz from 'baz';

src/handlers/defaultPropsHandler.ts

@@ -9,6 +9,7 @@ import isReactForwardRefCall from '../utils/isReactForwardRefCall';
 import type Documentation from '../Documentation';
 import type { Importer } from '../parse';
 import type { NodePath } from 'ast-types/lib/node-path';
+import isReactBuiltinCall from '../utils/isReactBuiltinCall';
 
 function getDefaultValue(path: NodePath, importer: Importer) {
   let node = path.node;
@@ -46,12 +47,21 @@ function getStatelessPropsPath(
   componentDefinition: NodePath,
   importer: Importer,
 ): NodePath {
-  const value = resolveToValue(componentDefinition, importer);
-  if (isReactForwardRefCall(value, importer)) {
-    const inner = resolveToValue(value.get('arguments', 0), importer);
-    return inner.get('params', 0);
-  }
-  return value.get('params', 0);
+  let actPath = componentDefinition;
+  let changed = false;
+  do {
+    changed = false;
+    const value = resolveToValue(actPath, importer);
+    if (
+      isReactBuiltinCall(value, 'memo', importer) ||
+      isReactForwardRefCall(value, importer)
+    ) {
+      actPath = resolveToValue(actPath.get('arguments', 0), importer);
+      changed = true;
+    }
+  } while (changed);
+
+  return actPath.get('params', 0);
 }
 
 function getDefaultPropsPath(