Extract static function isNoteTagOrExprChar

HubSpot · Oct 27, 2023 · 160c254 · 160c254
1 parent 1a8d43a
commit 160c254
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 15 deletions.
diff --git a/src/main/java/com/hubspot/jinjava/tree/output/OutputList.java b/src/main/java/com/hubspot/jinjava/tree/output/OutputList.java
@@ -71,16 +71,7 @@ private String joinNodesWithoutAddingExpressions(
     LengthLimitingStringBuilder val,
     TokenScannerSymbols tokenScannerSymbols
   ) {
-    @SuppressWarnings("StringBufferReplaceableByString")
-    String separator = new StringBuilder()
-      .append('\n')
-      .append(tokenScannerSymbols.getPrefixChar())
-      .append(tokenScannerSymbols.getNoteChar())
-      .append(tokenScannerSymbols.getTrimChar())
-      .append(' ')
-      .append(tokenScannerSymbols.getNoteChar())
-      .append(tokenScannerSymbols.getExprEndChar())
-      .toString();
+    String separator = getWhitespaceSeparator(tokenScannerSymbols);
     String prev = null;
     String cur;
     for (OutputNode node : nodes) {
@@ -93,11 +84,7 @@ private String joinNodesWithoutAddingExpressions(
         ) {
           if (
             cur.length() > 0 &&
-            (
-              cur.charAt(0) == tokenScannerSymbols.getTag() ||
-              cur.charAt(0) == tokenScannerSymbols.getExprStartChar() ||
-              cur.charAt(0) == tokenScannerSymbols.getNoteChar()
-            )
+            TokenScannerSymbols.isNoteTagOrExprChar(tokenScannerSymbols, cur.charAt(0))
           ) {
             val.append(separator);
           }
@@ -115,6 +102,20 @@ private String joinNodesWithoutAddingExpressions(
     return val.toString();
   }
 
+  private static String getWhitespaceSeparator(TokenScannerSymbols tokenScannerSymbols) {
+    @SuppressWarnings("StringBufferReplaceableByString")
+    String separator = new StringBuilder()
+      .append('\n')
+      .append(tokenScannerSymbols.getPrefixChar())
+      .append(tokenScannerSymbols.getNoteChar())
+      .append(tokenScannerSymbols.getTrimChar())
+      .append(' ')
+      .append(tokenScannerSymbols.getNoteChar())
+      .append(tokenScannerSymbols.getExprEndChar())
+      .toString();
+    return separator;
+  }
+
   private String joinNodes(LengthLimitingStringBuilder val) {
     for (OutputNode node : nodes) {
       try {

diff --git a/src/main/java/com/hubspot/jinjava/tree/parse/TokenScannerSymbols.java b/src/main/java/com/hubspot/jinjava/tree/parse/TokenScannerSymbols.java
@@ -122,4 +122,10 @@ public String getClosingComment() {
     }
     return closingComment;
   }
+
+  public static boolean isNoteTagOrExprChar(TokenScannerSymbols symbols, char c) {
+    return (
+      c == symbols.getNote() || c == symbols.getTag() || c == symbols.getExprStartChar()
+    );
+  }
 }