feat: resolve referenced propTypes

Author: unstubbable

src/typings.ts

@@ -47,8 +47,7 @@ export function createTypings(moduleName: string|null, programAst: any, options:
   }
   componentNames.forEach(componentName => {
     const exportType = getComponentExportType(ast, componentName);
-    const propTypes = getPropTypesFromAssignment(ast, componentName) ||
-      getPropTypesFromStaticAttribute(ast, componentName);
+    const propTypes = getPropTypes(ast, componentName);
     if (exportType) {
       createExportedTypes(m, ast, componentName, reactComponentName, propTypes, propTypesName, exportType, options);
     }
@@ -344,6 +343,18 @@ function getComponentNamesByJsxInBody(ast: AstQuery): string[] {
   return [];
 }
 
+function getPropTypes(ast: AstQuery, componentName: string): any|undefined {
+  const propTypes = getPropTypesFromAssignment(ast, componentName) ||
+    getPropTypesFromStaticAttribute(ast, componentName);
+
+  const referencedComponentName = getReferencedPropTypesComponentName(ast, propTypes);
+  if (referencedComponentName) {
+    return getPropTypes(ast, referencedComponentName);
+  }
+
+  return propTypes;
+}
+
 function getPropTypesFromAssignment(ast: AstQuery, componentName: string): any|undefined {
   const res = ast.query(`
     //AssignmentExpression[
@@ -389,6 +400,20 @@ function getPropTypesFromStaticAttribute(ast: AstQuery, componentName: string):
   return undefined;
 }
 
+function getReferencedPropTypesComponentName(ast: AstQuery, propTypes: any|undefined): string|undefined {
+  if (propTypes) {
+    const propTypesReference = ast.querySubtree(propTypes, `
+      MemberExpression [
+        /:property Identifier[@name == 'propTypes']
+      ] /:object Identifier
+    `);
+    if (propTypesReference.length > 0) {
+      return propTypesReference[0].name;
+    }
+  }
+  return undefined;
+}
+
 function getComponentExportType(ast: AstQuery, componentName: string): dom.DeclarationFlags|undefined {
   if (componentName === '') {
     // case: unnamed default export

tests/parsing-test.ts

@@ -84,6 +84,9 @@ test('Parsing should create definition from file without propTypes', t => {
 test('Parsing should create definition from file with references in propTypes', t => {
   compare(t, 'component', 'references-in-proptypes.jsx', 'references-in-proptypes.d.ts');
 });
+test('Parsing should create definition from file with reference as propTypes', t => {
+  compare(t, 'component', 'reference-as-proptypes.jsx', 'reference-as-proptypes.d.ts');
+});
 test('Parsing should create definition from file with unnamed default export', t => {
   compare(t, 'path', 'unnamed-default-export.jsx', 'unnamed-default-export.d.ts');
 });

tests/reference-as-proptypes.d.ts

@@ -0,0 +1,11 @@
+declare module 'component' {
+  import {Component} from 'react';
+
+  export interface SomeComponentProps {
+    someString?: string;
+  }
+
+  export default class SomeComponent extends Component<SomeComponentProps, any> {
+    render(): JSX.Element;
+  }
+}

tests/reference-as-proptypes.jsx

@@ -0,0 +1,11 @@
+import React from 'react';
+
+class SomeInternalComponent extends React.Component {
+   static propTypes = {
+     someString: React.PropTypes.string
+   };
+}
+
+export default class SomeComponent extends React.Component {
+   static propTypes = SomeInternalComponent.propTypes;
+}