implement getConnectionMetadata

Author: samgst-amazon

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

@@ -4,12 +4,17 @@
 package software.aws.toolkits.jetbrains.services.amazonq.lsp
 
 import com.intellij.notification.NotificationType
+import com.intellij.openapi.project.Project
 import org.eclipse.lsp4j.ConfigurationParams
 import org.eclipse.lsp4j.MessageActionItem
 import org.eclipse.lsp4j.MessageParams
 import org.eclipse.lsp4j.MessageType
 import org.eclipse.lsp4j.PublishDiagnosticsParams
 import org.eclipse.lsp4j.ShowMessageRequestParams
+import software.aws.toolkits.core.TokenConnectionSettings
+import software.aws.toolkits.jetbrains.core.credentials.AwsBearerTokenConnection
+import software.aws.toolkits.jetbrains.core.credentials.pinning.QConnection
+import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnectionManager
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.ConnectionMetadata
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.SsoProfileData
 import software.aws.toolkits.jetbrains.settings.CodeWhispererSettings
@@ -18,7 +23,7 @@ import java.util.concurrent.CompletableFuture
 /**
  * Concrete implementation of [AmazonQLanguageClient] to handle messages sent from server
  */
-class AmazonQLanguageClientImpl : AmazonQLanguageClient {
+class AmazonQLanguageClientImpl(private val project: Project) : AmazonQLanguageClient {
     override fun telemetryEvent(`object`: Any) {
         println(`object`)
     }
@@ -46,11 +51,25 @@ class AmazonQLanguageClientImpl : AmazonQLanguageClient {
         showMessage(message)
     }
 
-    override fun getConnectionMetadata() = CompletableFuture.completedFuture(
-        ConnectionMetadata(
-            SsoProfileData("TODO")
-        )
-    )
+    override fun getConnectionMetadata(): CompletableFuture<ConnectionMetadata> =
+        CompletableFuture.supplyAsync {
+            val connection = ToolkitConnectionManager.getInstance(project)
+                .activeConnectionForFeature(QConnection.getInstance())
+
+            when (connection) {
+                is AwsBearerTokenConnection -> {
+                    ConnectionMetadata(
+                        SsoProfileData(connection.startUrl)
+                    )
+                }
+                else -> {
+                    // If no connection or not a bearer token connection
+                    ConnectionMetadata(
+                        SsoProfileData("")
+                    )
+                }
+            }
+        }
 
     override fun configuration(params: ConfigurationParams): CompletableFuture<List<Any>> {
         if (params.items.isEmpty()) {

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

@@ -264,7 +264,7 @@ private class AmazonQServerInstance(private val project: Project, private val cs
         launcherHandler.startNotify()
 
         launcher = LSPLauncher.Builder<AmazonQLanguageServer>()
-            .setLocalService(AmazonQLanguageClientImpl())
+            .setLocalService(AmazonQLanguageClientImpl(project))
             .setRemoteInterface(AmazonQLanguageServer::class.java)
             .configureGson {
                 // TODO: maybe need adapter for initialize: