patch

Author: Will-ShaoHua

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/project/ProjectContextController.kt

@@ -26,6 +26,12 @@ class ProjectContextController(private val project: Project, private val cs: Cor
         }
     }
 
+    enum class IndexUpdateMode(val value: String) {
+        UPDATE("update"),
+        REMOVE("remove"),
+        ADD("add")
+    }
+
     fun getProjectContextIndexComplete() = projectContextProvider.isIndexComplete.get()
 
     fun query(prompt: String): List<RelevantDocument> {
@@ -46,9 +52,9 @@ class ProjectContextController(private val project: Project, private val cs: Cor
         }
     }
 
-    fun updateIndex(filePath: String) {
+    fun updateIndex(filePaths: List<String>, mode: String) {
         try {
-            return projectContextProvider.updateIndex(filePath)
+            return projectContextProvider.updateIndex(filePaths, mode)
         } catch (e: Exception) {
             logger.warn { "error while updating index for project context $e.message" }
         }

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/project/ProjectContextProvider.kt

@@ -86,8 +86,8 @@ class ProjectContextProvider(val project: Project, private val encoderServer: En
     )
 
     data class UpdateIndexRequestV2(
-        val filePath: String,
-        val content: String,
+        val filePaths: List<String>,
+        val updateMode: String,
     )
 
     data class Usage(
@@ -292,19 +292,19 @@ class ProjectContextProvider(val project: Project, private val encoderServer: En
         }
     }
 
-    fun updateIndex(filePath: String) {
+    fun updateIndex(filePaths: List<String>, mode: String) {
         if (!isIndexComplete.get()) return
-        logger.info { "project context: updating index for $filePath on port ${encoderServer.port}" }
-        val url = URL("http://localhost:${encoderServer.port}/updateIndex")
-        val payload = UpdateIndexRequestPayload(filePath)
-        val payloadJson = mapper.writeValueAsString(payload)
-        val encrypted = encoderServer.encrypt(payloadJson)
+
+        val url = URL("http://localhost:${encoderServer.port}/updateIndexV2")
+        val payload = UpdateIndexRequestV2(filePaths, mode)
+        val encrypted = encoderServer.encrypt(mapper.writeValueAsString(payload))
+
         with(url.openConnection() as HttpURLConnection) {
             setConnectionProperties(this)
             setConnectionTimeout(this)
             setConnectionRequest(this, encrypted)
             val responseCode = responseCode
-            logger.debug { "project context update index response code: $responseCode for $filePath" }
+            logger.debug { "project context update index response code: $responseCode" }
             return
         }
     }