Add support for objectOf

Fixes #74

Author: fkling

src/utils/__tests__/getPropType-test.js

@@ -85,7 +85,6 @@ describe('getPropType', () => {
       value: [{name: 'custom', raw: 'foo'}],
     });
 
-    // custom type
     expect(getPropType(expression('instanceOf(Foo)'))).toEqual({
       name: 'instanceOf',
       value: 'Foo',
@@ -96,6 +95,11 @@ describe('getPropType', () => {
       value: {name: 'string'},
     });
 
+    expect(getPropType(expression('objectOf(string)'))).toEqual({
+      name: 'objectOf',
+      value: {name: 'string'},
+    });
+
     expect(getPropType(expression('shape({foo: string, bar: bool})'))).toEqual({
       name: 'shape',
       value: {

src/utils/getPropType.js

@@ -67,6 +67,19 @@ function getPropTypeArrayOf(argumentPath) {
   return type;
 }
 
+function getPropTypeObjectOf(argumentPath) {
+  var type: PropTypeDescriptor = {name: 'objectOf'};
+  var subType = getPropType(argumentPath);
+
+  if (subType.name === 'unknown') {
+    type.value = printValue(argumentPath);
+    type.computed = true;
+  } else {
+    type.value = subType;
+  }
+  return type;
+}
+
 function getPropTypeShape(argumentPath) {
   var type: PropTypeDescriptor = {name: 'shape', value: 'unknown'};
   if (!types.ObjectExpression.check(argumentPath.node)) {
@@ -116,6 +129,7 @@ var propTypes = {
   oneOfType: getPropTypeOneOfType,
   instanceOf: getPropTypeInstanceOf,
   arrayOf: getPropTypeArrayOf,
+  objectOf: getPropTypeObjectOf,
   shape: getPropTypeShape,
 };