Map exprs without skipping the first one

SomeRanDev · Feb 11, 2024 · f8112f6 · f8112f6
1 parent 593d87b
commit f8112f6
Showing 1 changed file with 10 additions and 13 deletions.
diff --git a/src/reflaxe/compiler/RepeatVariableFixer.hx b/src/reflaxe/compiler/RepeatVariableFixer.hx
@@ -112,23 +112,20 @@ class RepeatVariableFixer {
 	}
 
 	function handleExpression(expr: TypedExpr): TypedExpr {
-		function mapSubExprs(subExpr: TypedExpr) {
-			switch(subExpr.expr) {
-				case TBlock(_): {
-					return handleBlock(subExpr);
-				}
-				case TLocal(tvar): {
-					final replacement = varReplacement(tvar.id);
-					if(replacement != null) {
-						return subExpr.copy(TLocal(replacement));
-					}
+		switch(expr.expr) {
+			case TBlock(_): {
+				return handleBlock(expr);
+			}
+			case TLocal(tvar): {
+				final replacement = varReplacement(tvar.id);
+				if(replacement != null) {
+					return expr.copy(TLocal(replacement));
 				}
-				case _:
 			}
-			return haxe.macro.TypedExprTools.map(subExpr, mapSubExprs);
+			case _:
 		}
 
-		return haxe.macro.TypedExprTools.map(expr, mapSubExprs);
+		return haxe.macro.TypedExprTools.map(expr, handleExpression);
 	}
 
 	function handleBlock(subExpr: TypedExpr): TypedExpr {