feat: Added arrayOf type props

Author: KnisterPeter

Diff for: index.ts

@@ -103,9 +103,16 @@ function isNode(obj: any): boolean {
 }
 
 function getReactPropTypeFromExpression(node: any): any {
-  if (node.object.type == 'MemberExpression'
+  if (node.type == 'MemberExpression' && node.object.type == 'MemberExpression'
       && node.object.object.name == 'React' && node.object.property.name == 'PropTypes') {
     return node.property;
+  } else if (node.type == 'CallExpression') {
+    if (getReactPropTypeFromExpression(node.callee).name == 'arrayOf') {
+      return {
+        name: 'array',
+        arrayType: getReactPropTypeFromExpression(node.arguments[0])
+      };
+    }
   }
   return 'undefined';
 }
@@ -136,7 +143,8 @@ export function getTypeFromPropType(node: any): IProperty {
         result.type = 'any';
         break;
       case 'array':
-        result.type = 'any[]';
+        let arrayType: any = type.arrayType || {name: 'any'};
+        result.type = arrayType.name + '[]';
         break;
       case 'bool':
         result.type = 'boolean';

Diff for: tests/parse-prop-types-test.ts

@@ -52,6 +52,7 @@ describe('The PropType parser', () => {
       type: 'MemberExpression',
       loc: {},
       object: {
+        type: 'MemberExpression',
         object: reactPropTypesMemberExpression,
         property: {
           name: 'func'
@@ -120,4 +121,31 @@ describe('The PropType parser', () => {
     };
     assert.deepEqual(getTypeFromPropType(ast), {type: 'React.ReactElement<any>', optional: true});
   });
+  it('should return number[] for arrayOf(React.PropTypes.number) prop types', () => {
+    const ast: any = {
+      type: 'CallExpression',
+      loc: {},
+      callee: {
+        type: 'MemberExpression',
+        loc: {},
+        object: reactPropTypesMemberExpression,
+        property: {
+          name: 'arrayOf'
+        }
+      },
+      arguments: [
+        {
+          type: 'MemberExpression',
+          loc: {},
+          object: reactPropTypesMemberExpression,
+          property: {
+            name: 'number'
+          }
+        }
+      ]
+    };
+    const result: any = getTypeFromPropType(ast);
+    assert.equal(result.type, 'number[]');
+    assert.equal(result.optional, true);
+  });
 });

Diff for: tests/simple-component.d.ts

@@ -12,8 +12,10 @@ declare module 'simple-component' {
 		optionalString?: string;
 		optionalNode?: React.ReactNode;
 		optionalElement?: React.ReactElement<any>;
+		optionalArrayOf?: number[];
 		requiredFunc: (...args: any[]) => any;
 		requiredAny: any;
+		requiredArrayOf: string[];
 	}
 
 	export default class SimpleComponent extends React.Component<SimpleComponentProps, any> {

Diff for: tests/simple-component.jsx

@@ -19,14 +19,15 @@ export default class SimpleComponent extends React.Component {
 		//	React.PropTypes.number,
 		//	React.PropTypes.instanceOf(Message)
 		//]),
-		//optionalArrayOf: React.PropTypes.arrayOf(React.PropTypes.number),
+		optionalArrayOf: React.PropTypes.arrayOf(React.PropTypes.number),
 		//optionalObjectOf: React.PropTypes.objectOf(React.PropTypes.number),
 		//optionalObjectWithShape: React.PropTypes.shape({
 		//	color: React.PropTypes.string,
 		//	fontSize: React.PropTypes.number
 		//}),
 		requiredFunc: React.PropTypes.func.isRequired,
 		requiredAny: React.PropTypes.any.isRequired,
+    requiredArrayOf: React.PropTypes.arrayOf(React.PropTypes.string).isRequired,
 	};
 
 	render() {