From baae46e0101cf01b1c06c25cdfe1f7f986338e72 Mon Sep 17 00:00:00 2001
From: Aaron Segal <asegal@hubspot.com>
Date: Tue, 7 Nov 2023 13:17:45 -0500
Subject: [PATCH] Using max output param to clamp outputs

---
 .../jinjava/interpret/JinjavaInterpreter.java | 38 +++++++++++++++++--
 1 file changed, 34 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/hubspot/jinjava/interpret/JinjavaInterpreter.java b/src/main/java/com/hubspot/jinjava/interpret/JinjavaInterpreter.java
index d58483865..bc5c51ee1 100644
--- a/src/main/java/com/hubspot/jinjava/interpret/JinjavaInterpreter.java
+++ b/src/main/java/com/hubspot/jinjava/interpret/JinjavaInterpreter.java
@@ -94,6 +94,7 @@ public class JinjavaInterpreter implements PyishSerializable {
   private final Set<Integer> errorSet = new HashSet<>();
 
   private static final int MAX_ERROR_SIZE = 100;
+  private static final long NO_LIMIT = -1;
 
   public JinjavaInterpreter(
     Jinjava application,
@@ -263,14 +264,38 @@ public String render(String template) {
   }
 
   /**
-   * Render the given root node, processing extend parents. Equivalent to render(root, true)
+   * Render the given root node, processing extend parents. Equivalent to render(root, true, -1)
    *
    * @param root
    *          node to render
    * @return rendered result
    */
   public String render(Node root) {
-    return render(root, true);
+    return render(root, true, NO_LIMIT);
+  }
+
+  /**
+   * Render the given root node with an option to process extend parents.
+   * Equivalent to render(root, processExtendRoots, -1).
+   * @param root
+   *          node to render
+   * @param processExtendRoots
+   * @return
+   */
+  public String render(Node root, boolean processExtendRoots) {
+    return render(root, processExtendRoots, NO_LIMIT);
+  }
+
+  /**
+   * Rendee the given root node to a certain limit, processing extend parents.
+   * @param root
+   *          node to render
+   * @param renderLimit
+   *          the number of characters in response text
+   * @return
+   */
+  public String render(Node root, long renderLimit) {
+    return render(root, true, renderLimit);
   }
 
   /**
@@ -280,10 +305,15 @@ public String render(Node root) {
    *          node to render
    * @param processExtendRoots
    *          if true, also render all extend parents
+   * @param renderLimit
+   *          the number of characters the result may contain
    * @return rendered result
    */
-  public String render(Node root, boolean processExtendRoots) {
-    OutputList output = new OutputList(config.getMaxOutputSize());
+  public String render(Node root, boolean processExtendRoots, long renderLimit) {
+    long maxOutput = (renderLimit == NO_LIMIT)
+      ? config.getMaxOutputSize()
+      : (Math.min(renderLimit, config.getMaxOutputSize()));
+    OutputList output = new OutputList(maxOutput);
 
     for (Node node : root.getChildren()) {
       lineNumber = node.getLineNumber();