Flatten Variable Derivation Chains

Author: rcosta358

liquidjava-verifier/pom.xml

@@ -11,7 +11,7 @@
 
 	<groupId>io.github.liquid-java</groupId>
 	<artifactId>liquidjava-verifier</artifactId>
-	<version>0.0.8</version>
+	<version>0.0.10</version>
 	<name>liquidjava-verifier</name>
 	<description>LiquidJava Verifier</description>
 	<url>https://github.com/liquid-java/liquidjava</url>

liquidjava-verifier/src/main/java/liquidjava/rj_language/opt/ConstantPropagation.java

@@ -47,8 +47,7 @@ private static ValDerivationNode propagateRecursive(Expression exp, Map<String,
                 DerivationNode previousOrigin = varOrigins.get(name);
 
                 // preserve origin if value came from previous derivation
-                DerivationNode origin = previousOrigin != null ? new VarDerivationNode(name, previousOrigin)
-                        : new VarDerivationNode(name);
+                DerivationNode origin = flattenVarOrigin(name, previousOrigin);
                 return new ValDerivationNode(value.clone(), origin);
             }
 
@@ -138,4 +137,15 @@ private static void extractVarOrigins(ValDerivationNode node, Map<String, Deriva
             extractVarOrigins(valOrigin, varOrigins);
         }
     }
+
+    /**
+     * Flattens variable derivations to avoid redundancy by collapsing var -> var chains
+     * If the origin is a VarDerivationNode, we use its variable name and origin instead
+     */
+    private static VarDerivationNode flattenVarOrigin(String varName, DerivationNode origin) {
+        if (origin instanceof VarDerivationNode varOrigin) {
+            return flattenVarOrigin(varOrigin.getVar(), varOrigin.getOrigin()); // recursively flatten
+        }
+        return origin != null ? new VarDerivationNode(varName, origin) : new VarDerivationNode(varName);
+    }
 }