Fix docblock handler for (default) exported named functions (fixes #51)

Author: fkling

src/handlers/__tests__/componentDocblockHandler-test.js

@@ -151,6 +151,21 @@ describe('componentDocblockHandler', () => {
     );
   });
 
+  describe('Stateless functions', () => {
+    test(
+      'function Component() {}',
+      src => lastStatement(src)
+    );
+    test(
+      'var Component = function () {};',
+      src => lastStatement(src).get('declarations', 0, 'init')
+    );
+    test(
+      'var Component = () => {}',
+      src => lastStatement(src).get('declarations', 0, 'init')
+    );
+  });
+
   describe('ES6 default exports', () => {
 
     describe('Default React.createClass export', () => {
@@ -182,6 +197,31 @@ describe('componentDocblockHandler', () => {
       );
     });
 
+    describe('Default stateless function export', () => {
+
+      describe('named function', () => {
+        test(
+          'export default function Component() {}',
+          src => lastStatement(src).get('declaration')
+        );
+      });
+
+      describe('anonymous function', () => {
+        test(
+          'export default function() {}',
+          src => lastStatement(src).get('declaration')
+        );
+      });
+
+      describe('arrow function', () => {
+        test(
+          'export default () => {}',
+          src => lastStatement(src).get('declaration')
+        );
+      });
+
+    });
+
   });
 
   describe('ES6 named exports', () => {
@@ -206,5 +246,30 @@ describe('componentDocblockHandler', () => {
       );
     });
 
+    describe('Named stateless function', () => {
+
+      describe('named function', () => {
+        test(
+          'export function Component() {}',
+          src => lastStatement(src).get('declaration')
+        );
+      });
+
+      describe('anonymous function', () => {
+        test(
+          'export var Component = function() {}',
+          src => lastStatement(src).get('declaration')
+        );
+      });
+
+      describe('arrow function', () => {
+        test(
+          'export var Component = () => {}',
+          src => lastStatement(src).get('declaration')
+        );
+      });
+
+    });
+
   });
 });

src/handlers/componentDocblockHandler.js

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