Merge pull request #143 from usdAG/feat/json-beautifier

JSON Beautifier

src/main/java/de/usd/cstchef/Utils.java

@@ -59,6 +59,7 @@
 import de.usd.cstchef.operations.conditional.StringContains;
 import de.usd.cstchef.operations.conditional.StringMatch;
 import de.usd.cstchef.operations.dataformat.FromBase64;
+import de.usd.cstchef.operations.dataformat.JsonBeautifier;
 import de.usd.cstchef.operations.dataformat.FromHex;
 import de.usd.cstchef.operations.dataformat.HtmlDecode;
 import de.usd.cstchef.operations.dataformat.HtmlEncode;
@@ -321,7 +322,7 @@ public static Class<? extends Operation>[] getOperationsDev() {
                 TimestampOffset.class, TimestampToDateTime.class, ToBase64.class, ToHex.class, UnixTimestamp.class,
                 UrlDecode.class, UrlEncode.class,
                 Whirlpool.class, WriteFile.class, XmlFullSignature.class, XmlMultiSignature.class,
-                Xor.class, SoapMultiSignature.class, Luhn.class, Concatenate.class
+                Xor.class, SoapMultiSignature.class, Luhn.class, Concatenate.class, JsonBeautifier.class
         };
     }
 

src/main/java/de/usd/cstchef/operations/dataformat/JsonBeautifier.java

@@ -0,0 +1,24 @@
+package de.usd.cstchef.operations.dataformat;
+
+import burp.api.montoya.core.ByteArray;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.SerializationFeature;
+import de.usd.cstchef.Utils;
+import de.usd.cstchef.operations.Operation;
+import de.usd.cstchef.operations.OperationCategory;
+
+@Operation.OperationInfos(name = "JSON Beautifier", category = OperationCategory.DATAFORMAT, description = "Format JSON Data.")
+public class JsonBeautifier extends Operation {
+	@Override
+	protected ByteArray perform(ByteArray input, Utils.MessageType messageType) throws Exception {
+		try {
+			ObjectMapper objectMapper = new ObjectMapper();
+			objectMapper.configure(SerializationFeature.ORDER_MAP_ENTRIES_BY_KEYS, true);
+			JsonNode jsonNode = objectMapper.readTree(input.toString());
+			return factory.createByteArray(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(jsonNode));
+		} catch (Exception e2) {
+			return input;
+		}
+	}
+}