Merge pull request #3 from Chave0v0/dev

完成基本功能开发

Author: Chave0v0

.idea/API-Highlighter.iml

@@ -56,11 +56,5 @@
             <version>1.18.30</version>
         </dependency>
 
-        <dependency>
-            <groupId>com.jgoodies</groupId>
-            <artifactId>forms</artifactId>
-            <version>1.2.1</version>
-        </dependency>
-
     </dependencies>
 </project>

.idea/compiler.xml

@@ -21,6 +21,6 @@ public void initialize(MontoyaApi montoyaApi) {
         // 初始化ui
         MainUI ui = new MainUI(montoyaApi);
         API.userInterface().registerSuiteTab("API Highlighter", ui.getMainPanel());
-        API.http().registerHttpHandler(new APIHighLighterHandler(montoyaApi));
+        API.http().registerHttpHandler(new APIHighLighterHandler());
     }
 }

.idea/libraries/Maven__com_jgoodies_forms_1_2_1.xml

@@ -1,18 +1,21 @@
 package com.chave.config;
 
 import com.chave.pojo.APIItem;
-
 import java.util.ArrayList;
 import java.util.HashMap;
 
 public class APIConfig {
     public static ArrayList<APIItem> TARGET_API = new ArrayList<>();
-    public static ArrayList<String> USER_INPUT_API = new ArrayList<>();
 
-    // 初始化一些测试数据
+    public static final HashMap ITEM_FIELD = new HashMap();
+
     static {
-        TARGET_API.add(new APIItem("GET", "/test"));
-        TARGET_API.add(new APIItem("GET", "/api/v1/{id}/info"));
-        TARGET_API.add(new APIItem("GET", "/api/v1/{id}/product/{name}"));
+        ITEM_FIELD.put(0, "method");
+        ITEM_FIELD.put(1, "path");
+        ITEM_FIELD.put(2, "result");
+        ITEM_FIELD.put(3, "state");
+        ITEM_FIELD.put(4, "note");
+        ITEM_FIELD.put(5, "domain");
     }
+
 }

.idea/misc.xml

@@ -0,0 +1,14 @@
+package com.chave.config;
+
+public class Color {
+    public static String NONE = "None";
+    public static String RED = "Red";
+    public static String ORANGE = "Orange";
+    public static String YELLOW = "Yellow";
+    public static String GREEN = "Green";
+    public static String CYAN ="Cyan";
+    public static String BLUE = "Blue";
+    public static String PINK = "Pink";
+    public static String MAGENTA = "Magenta";
+    public static String GRAY = "Gray";
+}

pom.xml

@@ -4,6 +4,8 @@
 
 public class UserConfig {
     public static MatchMod MATCH_MOD = MatchMod.exactMatch;
-    public static Boolean IS_REGULAR_REPLACE = Boolean.FALSE;
     public static Boolean IS_URL_ENCODE = Boolean.FALSE;
+    public static Boolean IS_CHECK_ENTIRE_REQUEST = Boolean.FALSE;
+    public static Boolean IS_CHECK_HTTP_METHOD = Boolean.FALSE;
+    public static Boolean IS_ANALYZE_PATHVARIABLE = Boolean.FALSE;
 }

src/main/java/com/chave/Main.java

@@ -1,32 +1,40 @@
 package com.chave.handler;
 
-import burp.api.montoya.MontoyaApi;
 import burp.api.montoya.http.handler.*;
+import burp.api.montoya.http.message.requests.HttpRequest;
 import burp.api.montoya.logging.Logging;
+import com.chave.Main;
 import com.chave.config.UserConfig;
 import com.chave.service.APIMatchService;
-
+import com.chave.utils.Util;
 import java.lang.reflect.Method;
 
 public class APIHighLighterHandler implements HttpHandler {
-    private MontoyaApi api;
     private Logging log;
     private APIMatchService apiMatchService;
 
-    public APIHighLighterHandler(MontoyaApi api) {
-        this.api = api;
-        this.log = api.logging();
+    public APIHighLighterHandler() {
+        this.log = Main.API.logging();
         this.apiMatchService = new APIMatchService();
     }
 
     @Override
     public RequestToBeSentAction handleHttpRequestToBeSent(HttpRequestToBeSent requestToBeSent) {
         try {
-            Method matchMethod = APIMatchService.class.getMethod(UserConfig.MATCH_MOD.name(), HttpRequestToBeSent.class);
+            Method matchMethod = APIMatchService.class.getMethod(UserConfig.MATCH_MOD.name(), HttpRequest.class);
             if ((Boolean) matchMethod.invoke(apiMatchService, requestToBeSent)) {
                 // 处理匹配到的逻辑
                 log.logToOutput("匹配到接口：" + requestToBeSent.path());
+
+                // 匹配到进行高亮处理
+                Util.setHighlightColor(requestToBeSent, com.chave.config.Color.YELLOW);
+
+
+                // 后续这里可以添加HaE匹配规则
+
             }
+
+
         } catch (Exception e) {
             log.logToError(e);
         }