Version: 3.2 Update

src/main/java/hae/utils/project/model/HaeFileContent.java

@@ -0,0 +1,67 @@
+package hae.utils.project.model;
+
+import burp.api.montoya.MontoyaApi;
+
+import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+public class HaeFileContent {
+    private final MontoyaApi api;
+    private String host;
+    private final Map<String, List<String>> dataMap;
+    private final Map<String, Map<String, String>> httpMap;
+
+    public HaeFileContent(MontoyaApi api) {
+        this.api = api;
+        this.dataMap = new HashMap<>();
+        this.httpMap = new HashMap<>();
+    }
+
+    public String getHost() {
+        return host;
+    }
+
+    public Map<String, List<String>> getDataMap() {
+        return dataMap;
+    }
+
+    public Map<String, Map<String, String>> getHttpMap() {
+        return httpMap;
+    }
+
+    public void setHost(String host) {
+        this.host = host;
+    }
+
+    public void setDataMap(Map<String, List<Object>> dataMap) {
+        for (Map.Entry<String, List<Object>> entry : dataMap.entrySet()) {
+            List<String> values = new ArrayList<>();
+            for (Object value : entry.getValue()) {
+                try {
+                    values.add(new String((byte[]) value, StandardCharsets.UTF_8));
+                } catch (Exception e) {
+                    values.add(value.toString());
+                }
+            }
+            this.dataMap.put(entry.getKey(), values);
+        }
+    }
+
+    public void setHttpMap(Map<String, Map<String, Object>> httpMap) {
+        for (Map.Entry<String, Map<String, Object>> entry : httpMap.entrySet()) {
+            Map<String, String> newValues = new HashMap<>();
+            Map<String, Object> values = entry.getValue();
+            for (String key : values.keySet()) {
+                try {
+                    newValues.put(key, new String((byte[]) values.get(key), StandardCharsets.UTF_8));
+                } catch (Exception e) {
+                    newValues.put(key, values.get(key).toString());
+                }
+            }
+            this.httpMap.put(entry.getKey(), newValues);
+        }
+    }
+}