Fix VariableDeclaration visitors

ooflorent · ooflorent · commit e44ba306a097 · 2018-01-11T15:56:41.000+01:00
diff --git a/lib/Parser.js b/lib/Parser.js
@@ -1063,63 +1063,17 @@ class Parser extends Tapable {
 	}
 
 	prewalkVariableDeclaration(statement) {
-		if(statement.declarations)
-			this.prewalkVariableDeclarators(statement.declarations);
-	}
-
-	walkVariableDeclaration(statement) {
-		if(statement.declarations)
-			this.walkVariableDeclarators(statement.declarations);
-	}
-
-	prewalkClassDeclaration(statement) {
-		if(statement.id) {
-			this.scope.renames.set(statement.id.name, null);
-			this.scope.definitions.add(statement.id.name);
-		}
-	}
-
-	walkClassDeclaration(statement) {
-		this.walkClass(statement);
-	}
-
-	prewalkSwitchCases(switchCases) {
-		for(let index = 0, len = switchCases.length; index < len; index++) {
-			const switchCase = switchCases[index];
-			this.prewalkStatements(switchCase.consequent);
-		}
-	}
-
-	walkSwitchCases(switchCases) {
-		for(let index = 0, len = switchCases.length; index < len; index++) {
-			const switchCase = switchCases[index];
-
-			if(switchCase.test) {
-				this.walkExpression(switchCase.test);
-			}
-			this.walkStatements(switchCase.consequent);
-		}
-	}
-
-	walkCatchClause(catchClause) {
-		this.inScope([catchClause.param], () => {
-			this.prewalkStatement(catchClause.body);
-			this.walkStatement(catchClause.body);
-		});
-	}
-
-	prewalkVariableDeclarators(declarators) {
-		for(const declarator of declarators) {
+		const hookMap = statement.kind === "const" ? this.hooks.varDeclarationConst :
+			statement.kind === "let" ? this.hooks.varDeclarationLet :
+			this.hooks.varDeclarationVar;
+		for(const declarator of statement.declarations) {
 			switch(declarator.type) {
 				case "VariableDeclarator":
 					{
 						this.enterPattern(declarator.id, (name, decl) => {
-							const hookMap = declarator.kind === "const" ? this.hooks.varDeclarationConst :
-								declarator.kind === "let" ? this.hooks.varDeclarationLet :
-								this.hooks.varDeclarationVar;
-							const hook = hookMap.get(name);
+							let hook = hookMap.get(name);
 							if(hook === undefined || !hook.call(decl)) {
-								const hook = this.hooks.varDeclaration.get(name);
+								hook = this.hooks.varDeclaration.get(name);
 								if(hook === undefined || !hook.call(decl)) {
 									this.scope.renames.set(name, null);
 									this.scope.definitions.add(name);
@@ -1132,8 +1086,8 @@ class Parser extends Tapable {
 		}
 	}
 
-	walkVariableDeclarators(declarators) {
-		for(const declarator of declarators) {
+	walkVariableDeclaration(statement) {
+		for(const declarator of statement.declarations) {
 			switch(declarator.type) {
 				case "VariableDeclarator":
 					{
@@ -1159,6 +1113,42 @@ class Parser extends Tapable {
 		}
 	}
 
+	prewalkClassDeclaration(statement) {
+		if(statement.id) {
+			this.scope.renames.set(statement.id.name, null);
+			this.scope.definitions.add(statement.id.name);
+		}
+	}
+
+	walkClassDeclaration(statement) {
+		this.walkClass(statement);
+	}
+
+	prewalkSwitchCases(switchCases) {
+		for(let index = 0, len = switchCases.length; index < len; index++) {
+			const switchCase = switchCases[index];
+			this.prewalkStatements(switchCase.consequent);
+		}
+	}
+
+	walkSwitchCases(switchCases) {
+		for(let index = 0, len = switchCases.length; index < len; index++) {
+			const switchCase = switchCases[index];
+
+			if(switchCase.test) {
+				this.walkExpression(switchCase.test);
+			}
+			this.walkStatements(switchCase.consequent);
+		}
+	}
+
+	walkCatchClause(catchClause) {
+		this.inScope([catchClause.param], () => {
+			this.prewalkStatement(catchClause.body);
+			this.walkStatement(catchClause.body);
+		});
+	}
+
 	walkPattern(pattern) {
 		switch(pattern.type) {
 			case "ArrayPattern":
