Pick up name and parameters from exported functions

This makes sure that we get the name and params from exported
functions/classes/vars.

For export default the name is inferred from the file name.

Towards #424

Author: arv

lib/infer/finders.js

@@ -10,6 +10,10 @@ function findTarget(path) {
     path = path.node;
   }
 
+  if (t.isExportNamedDeclaration(path) || t.isExportDefaultDeclaration(path)) {
+    path = path.declaration;
+  }
+
   // var x = TARGET;
   if (t.isVariableDeclaration(path)) {
     return path.declarations[0].init;

lib/infer/name.js

@@ -33,12 +33,17 @@ module.exports = function () {
       }
     }
 
-    // The strategy here is to do a depth-first traversal of the AST,
-    // looking for nodes with a "name" property, with exceptions as needed.
-    // For example, name inference for a MemberExpression `foo.bar = baz` will
-    // infer the named based on the `property` of the MemberExpression (`bar`)
-    // rather than the `object` (`foo`).
     if (comment.context.ast) {
+      if (comment.context.ast.type === 'ExportDefaultDeclaration') {
+        comment.name = pathParse(comment.context.file).name;
+        return comment;
+      }
+
+      // The strategy here is to do a depth-first traversal of the AST,
+      // looking for nodes with a "name" property, with exceptions as needed.
+      // For example, name inference for a MemberExpression `foo.bar = baz` will
+      // infer the named based on the `property` of the MemberExpression (`bar`)
+      // rather than the `object` (`foo`).
       comment.context.ast.traverse({
         Identifier: function (path) {
           if (inferName(path, path.node)) {

test/lib/infer/name.js

@@ -123,5 +123,25 @@ test('inferName', function (t) {
     /** @module */
   }, '/path/inferred-from-file.js').name, 'inferred-from-file');
 
+  t.equal(evaluate(`
+    /** Test */
+    export function exported() {}
+    `).name, 'exported');
+
+  t.equal(evaluate(`
+    /** Test */
+    export default function exported() {}
+    `, '/path/inferred-from-file.js').name, 'inferred-from-file');
+
+  t.equal(evaluate(`
+    /** Test */
+    export var life = 42;
+    `).name, 'life');
+
+  t.equal(evaluate(`
+    /** Test */
+    export class Wizard {}
+    `).name, 'Wizard');
+
   t.end();
 });

test/lib/infer/params.js

@@ -0,0 +1,35 @@
+'use strict';
+
+var test = require('tap').test,
+  parse = require('../../../lib/parsers/javascript'),
+  inferParams = require('../../../lib/infer/params')();
+
+function toComment(fn, file) {
+  return parse({
+    file: file,
+    source: fn instanceof Function ? '(' + fn.toString() + ')' : fn
+  })[0];
+}
+
+function evaluate(fn, file) {
+  return inferParams(toComment(fn, file));
+}
+
+test('inferParams', function (t) {
+  t.deepEqual(evaluate(function () {
+    /** Test */
+    function f(x) {}
+  }).params, [{lineNumber: 3, name: 'x', title: 'param'}]);
+
+  t.deepEqual(evaluate(`
+    /** Test */
+    export function f(x) {}
+  `).params, [{lineNumber: 3, name: 'x', title: 'param'}]);
+
+  t.deepEqual(evaluate(`
+    /** Test */
+    export default function f(x) {}
+  `).params, [{lineNumber: 3, name: 'x', title: 'param'}]);
+
+  t.end();
+});