Make findTarget return VariableDeclarator

This is in preparation for inferring types for const/let/var.

Author: arv

lib/infer/finders.js

@@ -22,9 +22,9 @@ function findTarget(path) {
     path = path.declaration;
   }
 
-  // var x = TARGET;
+  // var x = init;
   if (t.isVariableDeclaration(path)) {
-    return path.declarations[0].init;
+    return path.declarations[0];
   }
 
   // foo.x = TARGET

lib/infer/params.js

@@ -162,6 +162,12 @@ function inferParams() {
   return shouldSkipInference(function inferParams(comment) {
     var node = finders.findTarget(comment.context.ast);
 
+    // In case of `/** */ var x = function () {}` findTarget returns
+    // the declarator.
+    if (t.isVariableDeclarator(node)) {
+      node = node.init;
+    }
+
     if (!t.isFunction(node)) {
       return comment;
     }

test/lib/infer/params.js

@@ -21,6 +21,22 @@ test('inferParams', function (t) {
     function f(x) {}
   }).params, [{lineNumber: 3, name: 'x', title: 'param'}]);
 
+  t.deepEqual(evaluate(function () {
+    /** Test */
+    var f = function (x) {};
+  }).params, [{lineNumber: 3, name: 'x', title: 'param'}]);
+
+  t.deepEqual(evaluate('/** Test */ var f = (x) => {}').params,
+    [{lineNumber: 1, name: 'x', title: 'param'}]);
+
+  t.deepEqual(evaluate(function () {
+    var x = 1,
+      g = function (y) {},
+      /** Test */
+      f = function (x) {};
+  }).params, [{lineNumber: 5, name: 'x', title: 'param'}]);
+
+
   t.deepEqual(evaluate('/** Test */ export function f(x) {}').params,
     [{lineNumber: 1, name: 'x', title: 'param'}]);