Make expressions full visitable (#657)

Added methods to visit and traverse all expression nodes
PebbleTemplates · Nov 27, 2023 · 99af0ff · 99af0ff
1 parent 1b2ccbf
commit 99af0ff
Show file tree

Hide file tree

Showing 10 changed files with 45 additions and 0 deletions.
diff --git a/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/ArrayExpression.java b/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/ArrayExpression.java
@@ -51,6 +51,10 @@ public List<?> evaluate(PebbleTemplateImpl self, EvaluationContextImpl context)
     return returnValues;
   }
 
+  public List<Expression<?>> getValues() {
+    return this.values;
+  }
+
   @Override
   public int getLineNumber() {
     return this.lineNumber;

diff --git a/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/BlockFunctionExpression.java b/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/BlockFunctionExpression.java
@@ -47,6 +47,10 @@ public void accept(NodeVisitor visitor) {
     visitor.visit(this);
   }
 
+  public Expression<?> getBlockNameExpression() {
+    return this.blockNameExpression;
+  }
+
   @Override
   public int getLineNumber() {
     return this.lineNumber;

diff --git a/.../src/main/java/io/pebbletemplates/pebble/node/expression/LiteralBigDecimalExpression.java b/.../src/main/java/io/pebbletemplates/pebble/node/expression/LiteralBigDecimalExpression.java
@@ -39,6 +39,10 @@ public int getLineNumber() {
     return this.lineNumber;
   }
 
+  public BigDecimal getValue() {
+    return this.value;
+  }
+
   @Override
   public String toString() {
     return this.value.toString();

diff --git a/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/LiteralBooleanExpression.java b/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/LiteralBooleanExpression.java
@@ -38,6 +38,10 @@ public int getLineNumber() {
     return this.lineNumber;
   }
 
+  public Boolean getValue() {
+    return this.value;
+  }
+
   @Override
   public String toString() {
     return this.value.toString();

diff --git a/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/LiteralDoubleExpression.java b/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/LiteralDoubleExpression.java
@@ -38,6 +38,10 @@ public int getLineNumber() {
     return this.lineNumber;
   }
 
+  public Double getValue() {
+    return this.value;
+  }
+
   @Override
   public String toString() {
     return this.value.toString();

diff --git a/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/LiteralIntegerExpression.java b/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/LiteralIntegerExpression.java
@@ -37,4 +37,13 @@ public int getLineNumber() {
     return this.lineNumber;
   }
 
+  public Integer getValue() {
+    return this.value;
+  }
+
+  @Override
+  public String toString() {
+    return this.value.toString();
+  }
+
 }
diff --git a/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/LiteralLongExpression.java b/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/LiteralLongExpression.java
@@ -37,6 +37,10 @@ public int getLineNumber() {
     return this.lineNumber;
   }
 
+  public Long getValue() {
+    return this.value;
+  }
+
   @Override
   public String toString() {
     return this.value.toString();

diff --git a/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/MapExpression.java b/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/MapExpression.java
@@ -57,6 +57,10 @@ public void accept(NodeVisitor visitor) {
     return returnEntries;
   }
 
+  public Map<Expression<?>, Expression<?>> getEntries() {
+    return this.entries;
+  }
+
   @Override
   public int getLineNumber() {
     return this.lineNumber;

diff --git a/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/ParentFunctionExpression.java b/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/ParentFunctionExpression.java
@@ -58,6 +58,10 @@ public void accept(NodeVisitor visitor) {
     visitor.visit(this);
   }
 
+  public String getBlockName() {
+    return this.blockName;
+  }
+
   @Override
   public int getLineNumber() {
     return this.lineNumber;

diff --git a/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/RenderableNodeExpression.java b/pebble/src/main/java/io/pebbletemplates/pebble/node/expression/RenderableNodeExpression.java
@@ -47,6 +47,10 @@ public Object evaluate(PebbleTemplateImpl self, EvaluationContextImpl context) {
     return writer.toString();
   }
 
+  public RenderableNode getNode() {
+    return this.node;
+  }
+
   @Override
   public int getLineNumber() {
     return this.lineNumber;