deps(amazonq): Handle show document message from server (#5638)

* Show context options in prompt

* Handle show document message from server

* fix regression

* Fix nesting for project context

* detekt

* detekt

Author: manodnyab

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/AmazonQLanguageClientImpl.kt

@@ -4,15 +4,22 @@
 package software.aws.toolkits.jetbrains.services.amazonq.lsp
 
 import com.intellij.notification.NotificationType
+import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.fileEditor.FileEditorManager
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFileManager
 import migration.software.aws.toolkits.jetbrains.settings.AwsSettings
 import org.eclipse.lsp4j.ConfigurationParams
 import org.eclipse.lsp4j.MessageActionItem
 import org.eclipse.lsp4j.MessageParams
 import org.eclipse.lsp4j.MessageType
 import org.eclipse.lsp4j.ProgressParams
 import org.eclipse.lsp4j.PublishDiagnosticsParams
+import org.eclipse.lsp4j.ShowDocumentParams
+import org.eclipse.lsp4j.ShowDocumentResult
 import org.eclipse.lsp4j.ShowMessageRequestParams
+import software.aws.toolkits.core.utils.getLogger
+import software.aws.toolkits.core.utils.warn
 import software.aws.toolkits.jetbrains.core.credentials.AwsBearerTokenConnection
 import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnectionManager
 import software.aws.toolkits.jetbrains.core.credentials.pinning.QConnection
@@ -56,6 +63,30 @@ class AmazonQLanguageClientImpl(private val project: Project) : AmazonQLanguageC
         showMessage(message)
     }
 
+    override fun showDocument(params: ShowDocumentParams?): CompletableFuture<ShowDocumentResult> {
+        try {
+            if (params == null || params.uri.isNullOrEmpty()) {
+                return CompletableFuture.completedFuture(ShowDocumentResult(false))
+            }
+
+            ApplicationManager.getApplication().invokeLater {
+                try {
+                    val virtualFile = VirtualFileManager.getInstance().findFileByUrl(params.uri)
+                        ?: throw IllegalArgumentException("Cannot find file: ${params.uri}")
+
+                    FileEditorManager.getInstance(project).openFile(virtualFile, true)
+                } catch (e: Exception) {
+                    LOG.warn { "Failed to show document: ${params.uri}" }
+                }
+            }
+
+            return CompletableFuture.completedFuture(ShowDocumentResult(true))
+        } catch (e: Exception) {
+            LOG.warn { "Error showing document" }
+            return CompletableFuture.completedFuture(ShowDocumentResult(false))
+        }
+    }
+
     override fun getConnectionMetadata(): CompletableFuture<ConnectionMetadata> =
         CompletableFuture.supplyAsync {
             val connection = ToolkitConnectionManager.getInstance(project)
@@ -132,4 +163,8 @@ class AmazonQLanguageClientImpl(private val project: Project) : AmazonQLanguageC
             error("Cannot handle partial chat")
         }
     }
+
+    companion object {
+        private val LOG = getLogger<AmazonQLanguageClientImpl>()
+    }
 }

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/model/aws/chat/ChatPrompt.kt

@@ -3,6 +3,9 @@
 
 package software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat
 
+import com.fasterxml.jackson.annotation.JsonProperty
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.Command
+
 data class ChatPrompt(
     val prompt: String,
     val escapedPrompt: String,
@@ -17,10 +20,17 @@ data class SendChatPromptRequest(
 data class MidChatPrompt(
     val prompt: InnerChatPrompt,
     val tabId: String,
+    val context: List<Command>?,
 )
 
 data class InnerChatPrompt(
     val prompt: String,
     val escapedPrompt: String,
-    val context: List<String>? = null,
+    val context: List<Command>? = null,
+    val options: InnerChatOptions?,
+)
+
+data class InnerChatOptions(
+    @JsonProperty("pair-programmer-mode")
+    val pairProgrammingMode: String,
 )