Merge pull request #445 from KnisterPeter/support-prop-types

feat: support prop-types repository

Author: KnisterPeter

package-lock.json

@@ -41,6 +41,7 @@
     "@types/babel-generator": "^6.25.0",
     "@types/diff": "^3.2.2",
     "@types/node": "^8.0.50",
+    "@types/prop-types": "15.5.2",
     "@types/react": "^16.0.21",
     "ava": "^0.23.0",
     "babel-core": "^6.26.0",
@@ -52,6 +53,7 @@
     "cz-conventional-changelog": "2.1.0",
     "diff": "3.4.0",
     "nyc": "11.3.0",
+    "prop-types": "15.6.0",
     "react": "^16.0.0",
     "rimraf": "2.6.2",
     "source-map-support": "0.5.0",

package.json

@@ -204,7 +204,7 @@ function getReactComponentName(ast: AstQuery): string|undefined {
 }
 
 function getPropTypesName(ast: AstQuery): string|undefined {
-  const res = ast.query(`
+  let res = ast.query(`
     // ImportDeclaration[
       /:source StringLiteral[@value == 'react']
     ]
@@ -216,6 +216,18 @@ function getPropTypesName(ast: AstQuery): string|undefined {
   if (res.length > 0) {
     return res[0].name;
   }
+  res = ast.query(`
+    // ImportDeclaration[
+      /:source StringLiteral[@value == 'prop-types']
+    ]
+    /:specifiers *[
+      / Identifier[@name == 'PropTypes']
+    ]
+    /:local Identifier
+  `);
+  if (res.length > 0) {
+    return res[0].name;
+  }
   return undefined;
 }
 

src/typings.ts

@@ -96,3 +96,6 @@ test('Parsing should create definition from file with named export specifiers',
 test('Parsing should create preact definition', t => {
   compare(t, 'path', 'preact-definition.jsx', 'preact-definition.d.ts', {}, 'preact');
 });
+test('Parsing should suppport props-types repo', t => {
+  compare(t, 'path', 'prop-types.jsx', 'prop-types.d.ts', {});
+});

tests/parsing-test.ts

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

tests/prop-types.d.ts

@@ -0,0 +1,15 @@
+import {Component} from 'react';
+import * as PropTypes from 'prop-types';
+
+export default class extends Component {
+
+    static propTypes = {
+      optionalString: PropTypes.string
+    };
+
+    render() {
+        return (
+            <div></div>
+        );
+    }
+}