patch

Author: Will-ShaoHua

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

@@ -54,11 +54,6 @@ class ProjectContextProvider(val project: Project, private val encoderServer: En
             }
         }
     }
-    data class IndexRequestPayload(
-        val filePaths: List<String>,
-        val projectRoot: String,
-        val refresh: Boolean,
-    )
 
     data class IndexRequestV2(
         val filePaths: List<String>,
@@ -81,10 +76,6 @@ class ProjectContextProvider(val project: Project, private val encoderServer: En
         val filePath: String,
     )
 
-    data class UpdateIndexRequestPayload(
-        val filePath: String,
-    )
-
     data class UpdateIndexRequestV2(
         val filePaths: List<String>,
         val updateMode: String,
@@ -168,7 +159,6 @@ class ProjectContextProvider(val project: Project, private val encoderServer: En
         logger.debug { "project context file collection time: ${duration}ms" }
         logger.debug { "list of files collected: ${filesResult.files.joinToString("\n")}" }
         val projectRoot = project.guessProjectDir()?.path ?: return false
-//        val payload = IndexRequestPayload(filesResult.files, projectRoot, false)
         val payload = IndexRequestV2(filesResult.files, projectRoot, "all", "")
         val payloadJson = mapper.writeValueAsString(payload)
         val encrypted = encoderServer.encrypt(payloadJson)
@@ -227,21 +217,24 @@ class ProjectContextProvider(val project: Project, private val encoderServer: En
         val payloadJson = mapper.writeValueAsString(payload)
         val encrypted = encoderServer.encrypt(payloadJson)
 
-        val connection = url.openConnection() as HttpURLConnection
-        setConnectionProperties(connection)
-        setConnectionTimeout(connection)
-        setConnectionRequest(connection, encrypted)
+        val response = with(url.openConnection() as HttpURLConnection) {
+            setConnectionProperties(this)
+            setConnectionTimeout(this)
+            setConnectionRequest(this, encrypted)
+            val responseCode = this.responseCode
+            logger.info { "project context query response code: $responseCode for $prompt" }
 
-        val responseCode = connection.responseCode
-        logger.info { "project context query response code: $responseCode for $prompt" }
-        val responseBody = if (responseCode == 200) {
-            connection.inputStream.bufferedReader().use { reader -> reader.readText() }
-        } else {
-            ""
+            val responseBody = if (responseCode == 200) {
+                this.inputStream.bufferedReader().use { reader -> reader.readText() }
+            } else {
+                ""
+            }
+
+            responseBody
         }
-        connection.disconnect()
+
         try {
-            return mapper.readValue<List<BM25Chunk>>(responseBody)
+            return mapper.readValue<List<BM25Chunk>>(response)
         } catch (e: Exception) {
             logger.warn { "error parsing query response ${e.message}" }
             throw e