Add support for whole program AST transforms

deob-ast/src/main/java/dev/openrs2/deob/ast/AstDeobfuscator.kt

@@ -59,20 +59,26 @@ class AstDeobfuscator(private val modules: List<Path>) {
 
         val printer = PrettyPrinter(printerConfig)
 
-        for (module in modules) {
+        val roots = modules.map { SourceRoot(it, config) }
-            val root = SourceRoot(module, config)
+        val units = mutableMapOf<String, CompilationUnit>()
 
+        for (root in roots) {
             val results = root.tryToParseParallelized()
             for (result in results) {
                 require(result.isSuccessful) { result }
             }
 
             for (unit in root.compilationUnits) {
-                for (transformer in TRANSFORMERS) {
+                val name = unit.primaryType.orElseThrow().fullyQualifiedName.orElseThrow()
-                    transformer.transform(unit)
+                units[name] = unit
-                }
             }
+        }
+
+        for (transformer in TRANSFORMERS) {
+            transformer.transform(units)
+        }
 
+        for (root in roots) {
             root.printer = Function<CompilationUnit, String>(printer::print)
             root.saveAll()
         }

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/AddSubTransformer.kt

@@ -12,7 +12,10 @@ import dev.openrs2.deob.ast.util.negate
 import dev.openrs2.deob.ast.util.walk
 
 class AddSubTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: BinaryExpr ->
             val op = expr.operator
             val left = expr.left

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/BinaryExprOrderTransformer.kt

@@ -6,7 +6,10 @@ import dev.openrs2.deob.ast.util.isString
 import dev.openrs2.deob.ast.util.walk
 
 class BinaryExprOrderTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: BinaryExpr ->
             val op = expr.operator.flip() ?: return@walk
 

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/BitMaskTransformer.kt

@@ -10,7 +10,10 @@ import dev.openrs2.deob.ast.util.isIntegerOrLongLiteral
 import dev.openrs2.deob.ast.util.walk
 
 class BitMaskTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: BinaryExpr ->
             val shiftOp = expr.operator
             val left = expr.left

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/ComplementTransformer.kt

@@ -12,7 +12,10 @@ import dev.openrs2.deob.ast.util.isIntegerOrLongLiteral
 import dev.openrs2.deob.ast.util.walk
 
 class ComplementTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: BinaryExpr ->
             val op = complement(expr.operator) ?: return@walk
 

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/EncloseTransformer.kt

@@ -71,7 +71,10 @@ class EncloseTransformer : Transformer() {
         }
     }
 
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: Expression ->
             when {
                 expr.isArrayAccessExpr -> {

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/ForLoopConditionTransformer.kt

@@ -7,7 +7,10 @@ import dev.openrs2.deob.ast.util.hasSideEffects
 import dev.openrs2.deob.ast.util.walk
 
 class ForLoopConditionTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { stmt: ForStmt ->
             stmt.compare.ifPresent { compare ->
                 if (!compare.isBinaryExpr) {

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IdentityTransformer.kt

@@ -7,7 +7,10 @@ import com.github.javaparser.ast.expr.UnaryExpr
 import dev.openrs2.deob.ast.util.walk
 
 class IdentityTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: BinaryExpr ->
             @Suppress("NON_EXHAUSTIVE_WHEN")
             when (expr.operator) {

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IfElseTransformer.kt

@@ -9,7 +9,10 @@ import dev.openrs2.deob.ast.util.not
 import dev.openrs2.deob.ast.util.walk
 
 class IfElseTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { stmt: IfStmt ->
             stmt.elseStmt.ifPresent { elseStmt: Statement ->
                 val condition = stmt.condition

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/IncrementTransformer.kt

@@ -7,7 +7,10 @@ import com.github.javaparser.ast.stmt.ForStmt
 import dev.openrs2.deob.ast.util.walk
 
 class IncrementTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { stmt: ExpressionStmt ->
             if (!stmt.expression.isUnaryExpr) {
                 return@walk

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/NegativeLiteralTransformer.kt

@@ -7,7 +7,10 @@ import dev.openrs2.deob.ast.util.negate
 import dev.openrs2.deob.ast.util.walk
 
 class NegativeLiteralTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: UnaryExpr ->
             val operand = expr.expression
             if (!operand.isIntegerOrLongLiteral()) {

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/NewInstanceTransformer.kt

@@ -5,7 +5,10 @@ import com.github.javaparser.ast.expr.MethodCallExpr
 import dev.openrs2.deob.ast.util.walk
 
 class NewInstanceTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: MethodCallExpr ->
             if (expr.nameAsString != "newInstance") {
                 return@walk

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/TernaryTransformer.kt

@@ -7,7 +7,10 @@ import dev.openrs2.deob.ast.util.not
 import dev.openrs2.deob.ast.util.walk
 
 class TernaryTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: ConditionalExpr ->
             val condition = expr.condition
             val notCondition = condition.not()

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/Transformer.kt

@@ -3,5 +3,25 @@ package dev.openrs2.deob.ast.transform
 import com.github.javaparser.ast.CompilationUnit
 
 abstract class Transformer {
-    abstract fun transform(unit: CompilationUnit)
+    fun transform(units: Map<String, CompilationUnit>) {
+        preTransform(units)
+
+        for (unit in units.values) {
+            transformUnit(units, unit)
+        }
+
+        postTransform(units)
+    }
+
+    protected open fun preTransform(units: Map<String, CompilationUnit>) {
+        // empty
+    }
+
+    protected open fun transformUnit(units: Map<String, CompilationUnit>, unit: CompilationUnit) {
+        // empty
+    }
+
+    protected open fun postTransform(units: Map<String, CompilationUnit>) {
+        // empty
+    }
 }

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/UnencloseTransformer.kt

@@ -5,7 +5,10 @@ import com.github.javaparser.ast.expr.EnclosedExpr
 import dev.openrs2.deob.ast.util.walk
 
 class UnencloseTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: EnclosedExpr ->
             expr.replace(expr.inner.clone())
         }

deob-ast/src/main/java/dev/openrs2/deob/ast/transform/ValueOfTransformer.kt

@@ -7,7 +7,10 @@ import com.github.javaparser.ast.expr.TypeExpr
 import dev.openrs2.deob.ast.util.walk
 
 class ValueOfTransformer : Transformer() {
-    override fun transform(unit: CompilationUnit) {
+    override fun transformUnit(
+        units: Map<String, CompilationUnit>,
+        unit: CompilationUnit
+    ) {
         unit.walk { expr: ObjectCreationExpr ->
             if (expr.type.isBoxedType) {
                 expr.replace(MethodCallExpr(TypeExpr(expr.type), "valueOf", expr.arguments))