Version: 2.1 Update

README.md

@@ -36,6 +36,7 @@ https://gh0st.cn/HaE/
 4. 配置文件采用YAML格式存储，更加便于阅读和修改
 5. 内置简单缓存，在“多正则、大数据”的场景下减少卡顿现象
 6. **支持标签分页**，点击`...`即可添加新的标签页，对着标签页右键即可删除
+7. 高亮信息添加的同时添加Comment，便于查找请求
 
 ![-w477](images/16000720732851.jpg)
 

src/main/java/burp/BurpExtender.java

@@ -5,7 +5,6 @@ import burp.ui.MainUI;
 
 import javax.swing.*;
 import java.awt.*;
-import java.io.UnsupportedEncodingException;
 import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
 import java.util.List;
@@ -33,12 +32,12 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito
         this.callbacks = callbacks;
         BurpExtender.helpers = callbacks.getHelpers();
 
-        String version = "2.0.7";
+        String version = "2.1";
         callbacks.setExtensionName(String.format("HaE (%s) - Highlighter and Extractor", version));
         // 定义输出
         stdout = new PrintWriter(callbacks.getStdout(), true);
-        stdout.println("@UI Author: 0chencc");
         stdout.println("@Core Author: EvilChen");
+        stdout.println("@UI Author: 0chencc");
         stdout.println("@Github: https://github.com/gh0stkey/HaE");
         // UI
         SwingUtilities.invokeLater(this::initialize);
@@ -102,11 +101,19 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito
                 obj = ec.matchRegex(byteResponse, responseHeaders, responseBody, "response");
             }
 
-            List<String> colorList = da.highlightList(obj);
+            List<List<String>> resultList = da.highlightAndComment(obj);
+            List<String> colorList = resultList.get(0);
+            stdout.println(colorList);
+            List<String> commentList = resultList.get(1);
             if (colorList.size() != 0) {
                 String color = uc.getEndColor(gck.getColorKeys(colorList));
                 messageInfo.setHighlight(color);
             }
+
+            if (commentList.size() != 0) {
+                String originalComment = messageInfo.getComment();
+                messageInfo.setComment(String.join(", ", commentList));
+            }
         }
 
     }

src/main/java/burp/action/DoAction.java

@@ -21,13 +21,18 @@ public class DoAction {
         return result[0];
     }
 
-    public List<String> highlightList(Map<String, Map<String, Object>> obj) {
+    public List<List<String>> highlightAndComment(Map<String, Map<String, Object>> obj) {
         List<String> colorList = new ArrayList<>();
+        List<String> commentList = new ArrayList<>();
+        List<List<String>> result = new ArrayList<>();
         obj.keySet().forEach(i->{
             Map<String, Object> tmpMap = obj.get(i);
             String color = tmpMap.get("color").toString();
             colorList.add(color);
+            commentList.add(i);
         });
-        return colorList;
+        result.add(colorList);
+        result.add(commentList);
+        return result;
     }
 }