Find correct docblock for definitions in export statements

Author: fkling

src/handlers/__tests__/componentDocblockHandler-test.js

@@ -95,8 +95,53 @@ describe('componentDocblockHandler', () => {
 
   describe('ClassExpression', () => {
     test(
-      'var Compoent = class {};',
+      'var Component = class {};',
       src => lastStatement(src).get('declarations', 0, 'init')
     );
   });
+
+  describe('ES6 default exports', () => {
+
+    describe('Default React.createClass export', () => {
+      test(
+        'export default React.createClass({});',
+        src => lastStatement(src).get('declaration', 'arguments', 0)
+      );
+    });
+
+    describe('Default class declaration export', () => {
+      test(
+        'export default class Component {}',
+        src => lastStatement(src).get('declaration')
+      );
+    });
+
+    describe('Default class expression export', () => {
+      test(
+        'export default class {}',
+        src => lastStatement(src).get('declaration')
+      );
+    });
+
+  });
+
+  describe('ES6 named exports', () => {
+
+    describe('Named React.createClass export', () => {
+      test(
+        'export var Component = React.createClass({});',
+        src => lastStatement(src).get(
+          'declaration', 'declarations', '0', 'init', 'arguments', 0
+        )
+      );
+    });
+
+    describe('Named class declaration export', () => {
+      test(
+        'export class Component {}',
+        src => lastStatement(src).get('declaration')
+      );
+    });
+
+  });
 });

src/handlers/componentDocblockHandler.js

@@ -30,6 +30,16 @@ export default function componentDocblockHandler(
     path = path.parent;
   }
   if (path) {
+    // Class declarations are statements but can be part of default
+    // export declarations
+    if (types.ClassDeclaration.check(path.node) &&
+        types.ExportDefaultDeclaration.check(path.parentPath.node)) {
+      path = path.parentPath;
+    }
+    // If the parent is an export statement, we have to traverse one more up
+    if (types.ExportNamedDeclaration.check(path.parentPath.node)) {
+      path = path.parentPath;
+    }
     description = getDocblock(path);
   }
   if (description == null) {