Implement load conversation

Author: pplam

src/main/kotlin/ai/devchat/devchat/DevChatActionHandler.kt

@@ -23,18 +23,26 @@ class DevChatActionHandler private constructor() {
     fun handle(
         action: String,
         jsCallback: String,
-        callback: (JSONObject, JSONObject) -> Unit,
+        callback: (JSONObject) -> Unit,
     ) {
+        val response = JSONObject()
+        response["action"] = action
+        val metadata = JSONObject()
+        val payload = JSONObject()
+        response["metadata"] = metadata
+        response["payload"] = payload
+
         try {
             Log.info("Handling $action request")
-            sendResponse(action, jsCallback, callback)
+            callback(payload)
+            metadata["status"] = "success"
+            metadata["error"] = ""
         } catch (e: Exception) {
             Log.error("Exception occurred while handle action $action: ${e.message}")
-            sendResponse(action, jsCallback) { metadata: JSONObject, _ ->
-                metadata["status"] = "error"
-                metadata["error"] = e.message
-            }
+            metadata["status"] = "error"
+            metadata["error"] = e.message
         }
+        cefBrowser!!.executeJavaScript("$jsCallback($response)", "", 0)
     }
 
     fun sendResponse(action: String, responseFunc: String, callback: (JSONObject, JSONObject) -> Unit) {

src/main/kotlin/ai/devchat/devchat/handler/LoadConversationRequestHandler.kt

@@ -4,33 +4,39 @@ import ai.devchat.common.Log
 import ai.devchat.devchat.ActionHandler
 import ai.devchat.devchat.DevChatActionHandler
 import ai.devchat.devchat.DevChatActions
+import ai.devchat.idea.storage.ActiveConversation
 import com.alibaba.fastjson.JSONObject
 
 class LoadConversationRequestHandler(private val handler: DevChatActionHandler) : ActionHandler {
     private var metadata: JSONObject? = null
     private var payload: JSONObject? = null
 
-    private fun action(resMetadata: JSONObject, resPayload: JSONObject) {
+    private fun action(res: JSONObject) {
         val topicHash = metadata!!.getString("topicHash")
-        val conversations = handler.devChat.logTopic(topicHash, null)
-        // remove request_tokens and response_tokens in the conversations object
-        for (i in conversations.indices) {
-            val conversation = conversations.getJSONObject(i)
-            conversation.remove("request_tokens")
-            conversation.remove("response_tokens")
+        res["reset"] = true
+        when {
+            topicHash.isNullOrEmpty() -> ActiveConversation.reset()
+            topicHash == ActiveConversation.topic -> res["reset"] = false
+            else -> {
+                val arr = handler.devChat.logTopic(topicHash, null)
+                // remove request_tokens and response_tokens in the conversations object
+                val messages = List<JSONObject>(arr.size){i ->
+                    val msg = arr.getJSONObject(i)
+                    msg.remove("request_tokens")
+                    msg.remove("response_tokens")
+                    msg
+                }
+                ActiveConversation.reset(topicHash, messages)
+            }
         }
-        resMetadata["status"] = "success"
-        resMetadata["error"] = ""
-        resPayload["conversations"] = conversations
     }
 
     override fun executeAction() {
         handler.handle(
             DevChatActions.LOAD_CONVERSATIONS_RESPONSE,
             metadata!!.getString("callback"),
-        ) {resMetadata: JSONObject, resPayload: JSONObject ->
-            action(resMetadata, resPayload)
-        }
+            ::action
+        )
     }
 
     override fun setMetadata(metadata: JSONObject) {

src/main/kotlin/ai/devchat/idea/storage/ActiveConversation.kt

@@ -5,11 +5,11 @@ import com.alibaba.fastjson.JSONObject
 
 object ActiveConversation {
     private var messages: MutableList<JSONObject>? = null
-    private var topic: String? = null
+    var topic: String? = null
 
-    fun reset(topic: String, messages: MutableList<JSONObject>) {
+    fun reset(topic: String? = null, messages: List<JSONObject>? = null) {
         this.topic = topic
-        this.messages = messages
+        this.messages = messages?.toMutableList()
     }
 
     fun addMessage(message: JSONObject) {