Handle show document message from server

Author: manodnyab

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

@@ -4,15 +4,21 @@
 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.jetbrains.core.credentials.AwsBearerTokenConnection
 import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnectionManager
 import software.aws.toolkits.jetbrains.core.credentials.pinning.QConnection
@@ -56,6 +62,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)
@@ -145,4 +175,8 @@ class AmazonQLanguageClientImpl(private val project: Project) : AmazonQLanguageC
             error("Cannot handle partial chat")
         }
     }
+
+    companion object {
+        private val LOG = getLogger<AmazonQLanguageClientImpl>()
+    }
 }