Add support for higher order components with any arbitrary depth.

src/index.js

@@ -81,6 +81,15 @@ function isExported(path, className, t){
     'ExportNamedDeclaration'
   ];
 
+  function findMostRightHandArgument(args = []) {
+    const arg = args[0]
+    if (t.isIdentifier(arg)) {
+      return arg.name
+    } else if(t.isCallExpression(arg)) {
+      return findMostRightHandArgument(arg.arguments)
+    }
+  }
+
   if(path.parentPath.node &&
      types.some(type => {return path.parentPath.node.type === type;})) {
     return true;
@@ -93,7 +102,12 @@ function isExported(path, className, t){
        path.node.specifiers.length) {
       return className === path.node.specifiers[0].exported.name;
     } else if(path.node.type === 'ExportDefaultDeclaration') {
-      return className === path.node.declaration.name;
+      const decl = path.node.declaration
+      if (t.isCallExpression(decl)) {
+        return className === findMostRightHandArgument(decl.arguments);
+      } else {
+        return className === path.node.declaration.name;
+      }
     }
     return false;
   });

test/fixtures/hoc/actual.js

@@ -0,0 +1,11 @@
+class Component extends React.Component {
+  render() { return null }
+}
+
+Component.propTypes = {
+  children: React.PropTypes.string.isRequired,
+  onClick: React.PropTypes.func,
+  style: React.PropTypes.object,
+}
+
+export default withHoc()(deeperHoc(Component))

test/fixtures/hoc/expected.js

@@ -0,0 +1,77 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+
+var _createClass = function () { function defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; }();
+
+function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
+
+function _possibleConstructorReturn(self, call) { if (!self) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return call && (typeof call === "object" || typeof call === "function") ? call : self; }
+
+function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function, not " + typeof superClass); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, enumerable: false, writable: true, configurable: true } }); if (superClass) Object.setPrototypeOf ? Object.setPrototypeOf(subClass, superClass) : subClass.__proto__ = superClass; }
+
+var Component = function (_React$Component) {
+  _inherits(Component, _React$Component);
+
+  function Component() {
+    _classCallCheck(this, Component);
+
+    return _possibleConstructorReturn(this, (Component.__proto__ || Object.getPrototypeOf(Component)).apply(this, arguments));
+  }
+
+  _createClass(Component, [{
+    key: "render",
+    value: function render() {
+      return null;
+    }
+  }]);
+
+  return Component;
+}(React.Component);
+
+Component.propTypes = {
+  children: React.PropTypes.string.isRequired,
+  onClick: React.PropTypes.func,
+  style: React.PropTypes.object
+};
+
+exports.default = withHoc()(deeperHoc(Component));
+Component.__docgenInfo = {
+  "description": "",
+  "props": {
+    "children": {
+      "type": {
+        "name": "custom",
+        "raw": "React.PropTypes.string.isRequired"
+      },
+      "required": false,
+      "description": ""
+    },
+    "onClick": {
+      "type": {
+        "name": "custom",
+        "raw": "React.PropTypes.func"
+      },
+      "required": false,
+      "description": ""
+    },
+    "style": {
+      "type": {
+        "name": "custom",
+        "raw": "React.PropTypes.object"
+      },
+      "required": false,
+      "description": ""
+    }
+  }
+};
+
+if (typeof STORYBOOK_REACT_CLASSES !== "undefined") {
+  STORYBOOK_REACT_CLASSES["test/fixtures/hoc/actual.js"] = {
+    name: "Component",
+    docgenInfo: Component.__docgenInfo,
+    path: "test/fixtures/hoc/actual.js"
+  };
+}