Added correctly handling exported variables which uses other props in its initializers

Fixes #7

Author: timocov

src/exports-symbol-tree.ts

@@ -51,6 +51,21 @@ export class ExportsSymbolTree {
 	}
 
 	private computeTreeForChildren(targetSymbolsSet: Set<ts.Symbol>, node: ts.Node, visitedSymbols: Set<ts.Symbol>): void {
+		// it's similar to handling ts.Block node - both Block and variable's initializer are part of _implementation_
+		// and we don't care about that implementation at all - we just only need to worry it's definition
+		// for functions it is arguments and return type
+		// for variables - the type of a variable
+		if (ts.isVariableDeclaration(node)) {
+			const typeChecker = this.program.getTypeChecker();
+			const variableType = typeChecker.getTypeAtLocation(node);
+			const variableTypeSymbol = variableType.getSymbol();
+			if (variableTypeSymbol !== undefined) {
+				targetSymbolsSet.add(variableTypeSymbol);
+			}
+
+			return;
+		}
+
 		ts.forEachChild(node, (childNode: ts.Node) => this.computeTreeForNode(targetSymbolsSet, childNode, visitedSymbols));
 	}
 

tests/test-cases/class-properties/input.ts

@@ -8,7 +8,5 @@ class TestProperties {
 const test1 = new TestProperties();
 test1.salad = 0;
 test1.dressing = 0;
-const totalSalad = test1.salad + 1;
-const totalDressing = test1.dressing + 0;
-
-export const temp = 1;
+export const totalSalad = test1.salad + 1;
+export const totalDressing = test1.dressing + 0;

tests/test-cases/class-properties/output.js

@@ -16,6 +16,5 @@ var TestProperties = /** @class */ (function () {
 var test1 = new TestProperties();
 test1._internal_salad = 0;
 test1._internal_dressing = 0;
-var totalSalad = test1._internal_salad + 1;
-var totalDressing = test1._internal_dressing + 0;
-exports.temp = 1;
+exports.totalSalad = test1._internal_salad + 1;
+exports.totalDressing = test1._internal_dressing + 0;