feat(amazonq): implement aws/credentials/getConnectionMetadata for lsp (#5472)

* implement getConnectionMetadata

* tests

* in case of failure, default as builderID start URL

Author: samgst-amazon

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

@@ -4,12 +4,16 @@
 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.jetbrains.core.credentials.AwsBearerTokenConnection
+import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnectionManager
+import software.aws.toolkits.jetbrains.core.credentials.pinning.QConnection
 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 +22,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 +50,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 return default builderID start url
+                    ConnectionMetadata(
+                        SsoProfileData(AmazonQLspConstants.AWS_BUILDER_ID_URL)
+                    )
+                }
+            }
+        }
 
     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/AmazonQLspConstants.kt

@@ -6,4 +6,5 @@ package software.aws.toolkits.jetbrains.services.amazonq.lsp
 object AmazonQLspConstants {
     const val LSP_CW_CONFIGURATION_KEY = "aws.codeWhisperer"
     const val LSP_CW_OPT_OUT_KEY = "shareCodeWhispererContentWithAWS"
+    const val AWS_BUILDER_ID_URL = "https://view.awsapps.com/start"
 }

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:

plugins/amazonq/shared/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonq/lsp/AmazonQLanguageClientImplTest.kt

@@ -5,17 +5,54 @@
 package software.aws.toolkits.jetbrains.services.amazonq.lsp
 
 import com.google.gson.Gson
+import com.intellij.openapi.components.service
+import com.intellij.openapi.project.Project
 import com.intellij.testFramework.ApplicationExtension
+import io.mockk.every
+import io.mockk.mockk
 import org.assertj.core.api.Assertions.assertThat
 import org.eclipse.lsp4j.ConfigurationItem
 import org.eclipse.lsp4j.ConfigurationParams
 import org.junit.jupiter.api.Test
 import org.junit.jupiter.api.extension.ExtendWith
+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
+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
 
 @ExtendWith(ApplicationExtension::class)
 class AmazonQLanguageClientImplTest {
-    private val sut = AmazonQLanguageClientImpl()
+    private val project: Project = mockk(relaxed = true)
+    private val sut = AmazonQLanguageClientImpl(project)
+
+    @Test
+    fun `getConnectionMetadata returns connection metadata with start URL for bearer token connection`() {
+        val mockConnectionManager = mockk<ToolkitConnectionManager>()
+        every { project.service<ToolkitConnectionManager>() } returns mockConnectionManager
+
+        val expectedStartUrl = "https://test.aws.com"
+        val mockConnection = mockk<AwsBearerTokenConnection> {
+            every { startUrl } returns expectedStartUrl
+        }
+
+        every { mockConnectionManager.activeConnectionForFeature(QConnection.getInstance()) } returns mockConnection
+
+        assertThat(sut.getConnectionMetadata().get())
+            .isEqualTo(ConnectionMetadata(SsoProfileData(expectedStartUrl)))
+    }
+
+    @Test
+    fun `getConnectionMetadata returns empty start URL when no active connection`() {
+        val mockConnectionManager = mockk<ToolkitConnectionManager>()
+        every { project.service<ToolkitConnectionManager>() } returns mockConnectionManager
+
+        every { mockConnectionManager.activeConnectionForFeature(QConnection.getInstance()) } returns null
+
+        assertThat(sut.getConnectionMetadata().get())
+            .isEqualTo(ConnectionMetadata(SsoProfileData(AmazonQLspConstants.AWS_BUILDER_ID_URL)))
+    }
 
     @Test
     fun `configuration null if no attributes requested`() {