From 6742120ab6457cd748b8c21217483bc2b1db7580 Mon Sep 17 00:00:00 2001
From: Graham <gpe@openrs2.dev>
Date: Sun, 25 Aug 2019 15:06:18 +0100
Subject: [PATCH] Simplify else if chains in POSTORDER

This prevents us from missing some nested cases.
---
 .../dev/openrs2/deob/ast/transform/IfElseTransformer.java     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IfElseTransformer.java b/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IfElseTransformer.java
index 73f69951..c2db141e 100644
--- a/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IfElseTransformer.java
+++ b/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IfElseTransformer.java
@@ -1,9 +1,11 @@
 package dev.openrs2.deob.ast.transform;
 
 import com.github.javaparser.ast.CompilationUnit;
+import com.github.javaparser.ast.Node;
 import com.github.javaparser.ast.stmt.IfStmt;
 import com.github.javaparser.ast.stmt.Statement;
 import dev.openrs2.deob.ast.util.ExprUtils;
+import dev.openrs2.deob.ast.util.NodeUtils;
 
 public final class IfElseTransformer extends Transformer {
 	private static boolean isIf(Statement stmt) {
@@ -64,7 +66,7 @@ public final class IfElseTransformer extends Transformer {
 			});
 		});
 
-		unit.findAll(IfStmt.class).forEach(stmt -> {
+		NodeUtils.walk(unit, Node.TreeTraversal.POSTORDER, IfStmt.class, stmt -> {
 			stmt.getElseStmt().ifPresent(elseStmt -> {
 				if (isIf(elseStmt)) {
 					stmt.setElseStmt(getIf(elseStmt));