Add support for better formatting in toJson() method for the string and

lodash plugins.

Author: javadev

lodash-plugin/src/main/java/com/github/underscore/lodash/$.java

@@ -1163,8 +1163,74 @@ public static String trunc(final String string, final Integer length) {
         return result + omission;
     }
 
+    public static class JsonStringBuilder {
+        private final StringBuilder builder;
+        private int ident;
+        private boolean newString;
+        private boolean newKey;
+        private boolean newLine;
+
+        public JsonStringBuilder() {
+            builder = new StringBuilder();
+        }
+
+        public JsonStringBuilder append(final char character) {
+            switch (character) {
+                case '[':
+                case '{':
+                    fillSpaces(newLine);
+                    ident += 2;
+                    break;
+                case ']':
+                case '}':
+                    ident -= 2;
+                    builder.append("\n");
+                    fillSpaces(true);
+                    break;
+                case '\"':
+                    newString = !newString;
+                    fillSpaces(newString && !newKey);
+                    break;
+                default:
+                    break;
+            }
+            builder.append(character);
+            if (character == ',' || character == '[' || character == '{') {
+                builder.append("\n");
+                newLine = true;
+            } else {
+                newLine = false;
+            }
+            if (character == ':') {
+                builder.append(' ');
+                newKey = true;
+            } else {
+                newKey = false;
+            }
+            return this;
+        }
+
+        public JsonStringBuilder append(final String string) {
+            fillSpaces(!newString && !newKey);
+            builder.append(string);
+            return this;
+        }
+
+        private void fillSpaces(final boolean condition) {
+            if (condition) {
+                for (int index = 0; index < ident; index += 1) {
+                    builder.append(' ');
+                }
+            }
+        }
+
+        public String toString() {
+            return builder.toString();
+        }
+    }
+
     public static class JsonArray {
-        public static void writeJson(Collection collection, StringBuilder builder) {
+        public static void writeJson(Collection collection, JsonStringBuilder builder) {
             if (collection == null) {
                 builder.append(NULL);
                 return;
@@ -1192,7 +1258,7 @@ public static void writeJson(Collection collection, StringBuilder builder) {
             builder.append(']');
         }
 
-        public static void writeJson(byte[] array, StringBuilder builder) {
+        public static void writeJson(byte[] array, JsonStringBuilder builder) {
             if (array == null) {
                 builder.append(NULL);
             } else if (array.length == 0) {
@@ -1210,7 +1276,7 @@ public static void writeJson(byte[] array, StringBuilder builder) {
             }
         }
 
-        public static void writeJson(short[] array, StringBuilder builder) {
+        public static void writeJson(short[] array, JsonStringBuilder builder) {
             if (array == null) {
                 builder.append(NULL);
             } else if (array.length == 0) {
@@ -1228,7 +1294,7 @@ public static void writeJson(short[] array, StringBuilder builder) {
             }
         }
 
-        public static void writeJson(int[] array, StringBuilder builder) {
+        public static void writeJson(int[] array, JsonStringBuilder builder) {
             if (array == null) {
                 builder.append(NULL);
             } else if (array.length == 0) {
@@ -1246,7 +1312,7 @@ public static void writeJson(int[] array, StringBuilder builder) {
             }
         }
 
-        public static void writeJson(long[] array, StringBuilder builder) {
+        public static void writeJson(long[] array, JsonStringBuilder builder) {
             if (array == null) {
                 builder.append(NULL);
             } else if (array.length == 0) {
@@ -1264,7 +1330,7 @@ public static void writeJson(long[] array, StringBuilder builder) {
             }
         }
 
-        public static void writeJson(float[] array, StringBuilder builder) {
+        public static void writeJson(float[] array, JsonStringBuilder builder) {
             if (array == null) {
                 builder.append(NULL);
             } else if (array.length == 0) {
@@ -1282,7 +1348,7 @@ public static void writeJson(float[] array, StringBuilder builder) {
             }
         }
 
-        public static void writeJson(double[] array, StringBuilder builder) {
+        public static void writeJson(double[] array, JsonStringBuilder builder) {
             if (array == null) {
                 builder.append(NULL);
             } else if (array.length == 0) {
@@ -1300,7 +1366,7 @@ public static void writeJson(double[] array, StringBuilder builder) {
             }
         }
 
-        public static void writeJson(boolean[] array, StringBuilder builder) {
+        public static void writeJson(boolean[] array, JsonStringBuilder builder) {
             if (array == null) {
                 builder.append(NULL);
             } else if (array.length == 0) {
@@ -1318,25 +1384,25 @@ public static void writeJson(boolean[] array, StringBuilder builder) {
             }
         }
 
-        public static void writeJson(char[] array, StringBuilder builder) {
+        public static void writeJson(char[] array, JsonStringBuilder builder) {
             if (array == null) {
                 builder.append(NULL);
             } else if (array.length == 0) {
                 builder.append("[]");
             } else {
-                builder.append("[\"");
+                builder.append('[').append('\"');
                 builder.append(String.valueOf(array[0]));
 
                 for (int i = 1; i < array.length; i++) {
-                    builder.append("\",\"");
+                    builder.append('\"').append(',').append('\"');
                     builder.append(String.valueOf(array[i]));
                 }
 
-                builder.append("\"]");
+                builder.append('\"').append(']');
             }
         }
 
-        public static void writeJson(Object[] array, StringBuilder builder) {
+        public static void writeJson(Object[] array, JsonStringBuilder builder) {
             if (array == null) {
                 builder.append(NULL);
             } else if (array.length == 0) {
@@ -1356,7 +1422,7 @@ public static void writeJson(Object[] array, StringBuilder builder) {
     }
 
     public static class JsonObject {
-        public static void writeJson(Map map, StringBuilder builder) {
+        public static void writeJson(Map map, JsonStringBuilder builder) {
             if (map == null) {
                 builder.append(NULL);
                 return;
@@ -1384,7 +1450,7 @@ public static void writeJson(Map map, StringBuilder builder) {
     }
 
     public static class JsonValue {
-        public static void writeJson(Object value, StringBuilder builder) {
+        public static void writeJson(Object value, JsonStringBuilder builder) {
             if (value == null) {
                 builder.append(NULL);
             } else if (value instanceof String) {
@@ -1490,7 +1556,7 @@ static void escape(String s, StringBuilder sb) {
     }
 
     public static String toJson(Collection collection) {
-        final StringBuilder builder = new StringBuilder();
+        final JsonStringBuilder builder = new JsonStringBuilder();
 
         JsonArray.writeJson(collection, builder);
         return builder.toString();
@@ -1501,7 +1567,7 @@ public String toJson() {
     }
 
     public static String toJson(Map map) {
-        final StringBuilder builder = new StringBuilder();
+        final JsonStringBuilder builder = new JsonStringBuilder();
 
         JsonObject.writeJson(map, builder);
         return builder.toString();