feat: only count function body vars

searchfe · Apr 29, 2019 · 06fe03e · 06fe03e
1 parent 1c6e70d
commit 06fe03e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/emitter.ts b/src/emitter.ts
@@ -1764,7 +1764,8 @@ export function emitFile(
             || (node.kind === ts.SyntaxKind.MethodDeclaration && !ts.isClassDeclaration(node.parent))
         ) {
             const wrappedNode = createWrappedNode(node, {typeChecker});
-            const identifiers = wrappedNode.getDescendantsOfKind(ts.SyntaxKind.Identifier);
+            const body = wrappedNode.getBody();
+            const identifiers = body.getKind() === ts.SyntaxKind.Identifier ? [body as tsMorph.Identifier] : body.getDescendantsOfKind(ts.SyntaxKind.Identifier);
             const inheritedVariables: tsMorph.Identifier[] = [];
             const nodeStart = node.getStart();
             const nodeEnd = node.getEnd();

diff --git a/test/features/inheritedVariables.ts b/test/features/inheritedVariables.ts
@@ -21,6 +21,11 @@ let f = function () {
 }
 
 let obj = {
+    /**
+     * aaa
+     *
+     * @ssr
+     */
     a: function () {
         return '123' + b;
     },
-Original file line number
+Diff line change
@@ Expand Up / @@ -21,6 +21,11 @@ let f = function () { @@
     }
     let obj = {
+        /**
+         * aaa
+         *
+         * @ssr
+         */
         a: function () {
             return '123' + b;
         },
@@ Expand Down @@