add ES6 export declarations support

Author: Morlay

.gitignore

@@ -1 +1,2 @@
 /dist
+node_modules/

src/resolver/__tests__/findExportedReactCreateClassCall-test.js

@@ -103,4 +103,31 @@ describe('React documentation parser', () => {
 
     expect(parse(source)).toBeDefined();
   });
+
+  it('finds React.createClass in default exported', () => {
+    var source = [
+      'var React = require("React");',
+      'var Component = React.createClass({});',
+      'export default Component'
+    ].join('\n');
+
+    expect(parse(source)).toBeDefined();
+
+    source = [
+      'var React = require("React");',
+      'export default React.createClass({});'
+    ].join('\n');
+
+    expect(parse(source)).toBeDefined();
+  });
+
+  it('finds React.createClass in variable exported', () => {
+    var source = [
+      'var React = require("React");',
+      'export var Component = React.createClass({});'
+    ].join('\n');
+
+    expect(parse(source)).toBeDefined();
+  });
+
 });

src/resolver/findExportedReactCreateClassCall.js

@@ -47,6 +47,17 @@ function findExportedReactCreateClass(
     visitDoWhileStatement: ignore,
     visitForStatement: ignore,
     visitForInStatement: ignore,
+    visitExportDeclaration: function(path) {
+      path = resolveToValue(path.get('declaration'));
+      if (!isReactCreateClassCall(path)) {
+       return false
+      }
+      var resolvedPath = resolveToValue(path.get('arguments', 0));
+      if (types.ObjectExpression.check(resolvedPath.node)) {
+        definition = resolvedPath;
+      }
+      this.abort();
+    },
     visitAssignmentExpression: function(path) {
       // Ignore anything that is not `exports.X = ...;` or
       // `module.exports = ...;`

src/utils/__tests__/resolveToValue-test.js

@@ -86,4 +86,26 @@ describe('resolveToValue', () => {
       path.node
     )).toBe(true);
   });
+
+  it('resolves export variable declarator', () => {
+    var path = parse([
+      'export var Component = foo()'
+    ].join('\n'));
+
+    expect(
+      resolveToValue(path.parentPath.get('declaration')).node.callee.name
+    ).toBe('foo');
+  });
+
+  it('resolves export default declarator', () => {
+    var path = parse([
+      'var Component = foo();',
+      'export default Component;'
+    ].join('\n'));
+
+    expect(
+      resolveToValue(path.parentPath.get('declaration')).node.callee.name
+    ).toBe('foo');
+  });
+
 });

src/utils/resolveToValue.js

@@ -49,7 +49,13 @@ function buildMemberExpressionFromPattern(path: NodePath): ?Node {
  */
 function resolveToValue(path: NodePath): NodePath {
   var node = path.node;
-  if (types.AssignmentExpression.check(node)) {
+  if (types.VariableDeclaration.check(node)) {
+    return resolveToValue(path.get('declarations', 0))
+  } else if (types.VariableDeclarator.check(node)) {
+     if (node.init) {
+       return resolveToValue(path.get('init'));
+     }
+  } else if (types.AssignmentExpression.check(node)) {
     if (node.operator === '=') {
       return resolveToValue(node.get('right'));
     }