Chore: rename old addContext/request to addContext/notify

Signed-off-by: Daniel Hu <[email protected]>
devchat-ai · Nov 15, 2023 · affc7e1 · affc7e1
1 parent 3863d21
commit affc7e1
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 36 deletions.
diff --git a/src/main/java/ai/devchat/devchat/DevChatActions.java b/src/main/java/ai/devchat/devchat/DevChatActions.java
@@ -7,6 +7,7 @@ public class DevChatActions {
     public static final String SET_OR_UPDATE_KEY_RESPONSE = "setOrUpdateKey/response";
     public static final String ADD_CONTEXT_REQUEST = "addContext/request";
     public static final String ADD_CONTEXT_RESPONSE = "addContext/response";
+    public static final String ADD_CONTEXT_NOTIFY = "addContext/notify";
     public static final String LIST_COMMANDS_REQUEST = "listCommands/request";
     public static final String LIST_COMMANDS_RESPONSE = "listCommands/response";
     public static final String LIST_CONVERSATIONS_REQUEST = "listConversations/request";

diff --git a/src/main/java/ai/devchat/devchat/handler/AddContextNotifyHandler.java b/src/main/java/ai/devchat/devchat/handler/AddContextNotifyHandler.java
@@ -0,0 +1,40 @@
+package ai.devchat.devchat.handler;
+
+import ai.devchat.devchat.ActionHandler;
+import ai.devchat.devchat.DevChatActionHandler;
+import ai.devchat.devchat.DevChatActions;
+
+import com.alibaba.fastjson.JSONObject;
+
+public class AddContextNotifyHandler implements ActionHandler {
+    private JSONObject metadata;
+    private JSONObject payload;
+    private final DevChatActionHandler devChatActionHandler;
+
+    public final String RESPONSE_FUNC = "AddContextFromEditor";
+
+    public AddContextNotifyHandler(DevChatActionHandler devChatActionHandler) {
+        this.devChatActionHandler = devChatActionHandler;
+    }
+
+    @Override
+    public void executeAction() {
+        devChatActionHandler.sendResponse(DevChatActions.ADD_CONTEXT_NOTIFY, RESPONSE_FUNC, (metadata, payload) -> {
+            metadata.put("status", "success");
+            metadata.put("error", "");
+
+            payload.put("path", this.payload.getString("path"));
+            payload.put("content", this.payload.getString("content"));
+            payload.put("languageId", this.payload.getString("languageId"));
+            payload.put("startLine", this.payload.getInteger("startLine"));
+        });
+    }
+
+    public void setMetadata(JSONObject metadata) {
+        this.metadata = metadata;
+    }
+
+    public void setPayload(JSONObject payload) {
+        this.payload = payload;
+    }
+}
diff --git a/src/main/java/ai/devchat/devchat/handler/AddContextRequestHandler.java b/src/main/java/ai/devchat/devchat/handler/AddContextRequestHandler.java
diff --git a/src/main/java/ai/devchat/idea/action/AddToDevChatAction.java b/src/main/java/ai/devchat/idea/action/AddToDevChatAction.java
@@ -2,7 +2,7 @@
 
 import ai.devchat.common.Log;
 import ai.devchat.devchat.DevChatActionHandler;
-import ai.devchat.devchat.handler.AddContextRequestHandler;
+import ai.devchat.devchat.handler.AddContextNotifyHandler;
 import com.alibaba.fastjson.JSONObject;
 
 public class AddToDevChatAction {
@@ -24,8 +24,8 @@ public void execute(String filePath, String fileContent, String language, int st
         payload.put("languageId", language);
         payload.put("startLine", startLine);
 
-        AddContextRequestHandler addContextRequestHandler = new AddContextRequestHandler(actionHandler);
-        addContextRequestHandler.setPayload(payload);
-        addContextRequestHandler.executeAction();
+        AddContextNotifyHandler addContextNotifyHandler = new AddContextNotifyHandler(actionHandler);
+        addContextNotifyHandler.setPayload(payload);
+        addContextNotifyHandler.executeAction();
     }
 }