CodeWhisperer release for April 13, 2023

Co-authored-by: Abhilash Kuhikar <[email protected]>
Co-authored-by: Andrew Yu <[email protected]>
Co-authored-by: Lei Gao <[email protected]>
Co-authored-by: Rachel Cross <[email protected]>
Co-authored-by: Tianxing Cheng <[email protected]>
Co-authored-by: Will Lo <[email protected]>
Co-authored-by: Zoe Lin <[email protected]>

Author: 8 people

.changes/next-release/bugfix-ecbed35e-c5a6-49c8-b058-09644c663cdf.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "[CodeWhisperer]: Multiple bug fixes to improve user experience"
+}

.changes/next-release/feature-0f932842-1f40-46d7-9b29-97398fefb834.json

@@ -0,0 +1,4 @@
+{
+  "type" : "feature",
+  "description" : "[CodeWhisperer]: Introducing \"Stop code scan\" feature where users will be able to stop the ongoing code scan and immediately start a new one. "
+}

.changes/next-release/feature-148aeeef-0a57-44c8-8646-adee74cc3a7d.json

@@ -0,0 +1,4 @@
+{
+  "type" : "feature",
+  "description" : "[CodeWhisperer]: Automatic import recommendations"
+}

.changes/next-release/feature-1c3c551e-1a3e-434f-8596-3e6403060f57.json

@@ -0,0 +1,4 @@
+{
+  "type" : "feature",
+  "description" : "[CodeWhisperer]: Now supports cross region calls."
+}

.changes/next-release/feature-dcd24c1c-d99c-4e89-81c6-9416b3de8bf9.json

@@ -0,0 +1,4 @@
+{
+  "type" : "feature",
+  "description" : "[CodeWhisperer]: New supported programming languages: C, C++, Go, Kotlin, Php, Ruby, Rust, Scala, Shell, Sql."
+}

core/src/software/aws/toolkits/core/credentials/ToolkitCredentialsProvider.kt

@@ -117,7 +117,7 @@ interface ToolkitBearerTokenProviderDelegate : SdkTokenProvider, ToolkitAuthenti
 class ToolkitBearerTokenProvider(val delegate: ToolkitBearerTokenProviderDelegate) : SdkTokenProvider by delegate, ToolkitAuthenticationProvider by delegate {
     companion object {
         // TODO: is there a better place for this
-        fun ssoIdentifier(startUrl: String) = "sso;$startUrl"
+        fun ssoIdentifier(startUrl: String, region: String = DEFAULT_SSO_REGION) = "sso;$region;$startUrl"
 
         // TODO: For AWS Builder ID, we only have startUrl for now instead of each users' metadata data i.e. Email address
         fun ssoDisplayName(startUrl: String) = if (startUrl == SONO_URL) {
@@ -133,4 +133,6 @@ class ToolkitBearerTokenProvider(val delegate: ToolkitBearerTokenProviderDelegat
 
 private const val SONO_URL = "https://view.awsapps.com/start"
 
+const val DEFAULT_SSO_REGION = "us-east-1"
+
 private fun extractOrgID(startUrl: String) = startUrl.removePrefix("https://").removeSuffix(".awsapps.com/start")

core/src/software/aws/toolkits/core/credentials/ToolkitCredentialsProviderManager.kt

@@ -10,6 +10,7 @@ interface ToolkitCredentialsChangeListener {
     fun providerAdded(identifier: CredentialIdentifier) {}
     fun providerModified(identifier: CredentialIdentifier) {}
     fun providerRemoved(identifier: CredentialIdentifier) {}
+    fun providerRemoved(providerId: String) {}
 }
 
 class CredentialProviderNotFoundException : RuntimeException {

jetbrains-core/assets/WelcomeToCodeWhisperer.md

@@ -1,7 +1,7 @@
 ## Using Amazon CodeWhisperer
 
 CodeWhisperer uses machine learning to generate code suggestions from the existing code and comments in your IDE.
-Supported languages include: **Java, Python, JavaScript, C# and TypeScript**.
+Supported languages include: **Java, Python, JavaScript, C#, TypeScript, C, C++, Go, Rust, PHP, Scala, Kotlin, Ruby, Shell and SQL**.
 
 ---
 

jetbrains-core/it/software/aws/toolkits/jetbrains/services/codewhisperer/CodeWhispererCompletionIntegrationTest.kt

@@ -53,7 +53,7 @@ class CodeWhispererCompletionIntegrationTest : CodeWhispererIntegrationTestBase(
         assertDoesNotThrow {
             invokeCodeWhispererService()
             verify(popupManager, never()).showPopup(any(), any(), any(), any(), any())
-            verify(clientAdaptor, never()).listRecommendationsPaginator(any(), any())
+            verify(clientAdaptor, never()).generateCompletionsPaginator(any())
             testMessageShown(message("codewhisperer.language.error", file.fileType.name))
         }
     }

jetbrains-core/it/software/aws/toolkits/jetbrains/services/codewhisperer/CodeWhispererIntegrationTestBase.kt

@@ -28,7 +28,7 @@ import org.mockito.kotlin.timeout
 import org.mockito.kotlin.times
 import org.mockito.kotlin.verify
 import org.mockito.kotlin.whenever
-import software.amazon.awssdk.services.codewhisperer.model.ListRecommendationsResponse
+import software.amazon.awssdk.services.codewhispererruntime.model.GenerateCompletionsResponse
 import software.aws.toolkits.core.TokenConnectionSettings
 import software.aws.toolkits.core.credentials.ToolkitBearerTokenProvider
 import software.aws.toolkits.jetbrains.core.MockClientManager
@@ -103,7 +103,6 @@ open class CodeWhispererIntegrationTestBase(val projectRule: CodeInsightTestFixt
         ApplicationManager.getApplication().replaceService(CodeWhispererTelemetryService::class.java, telemetryServiceSpy, disposableRule.disposable)
 
         stateManager = CodeWhispererExplorerActionManager.getInstance()
-        stateManager.setHasAcceptedTermsOfService(true)
         stateManager.setAutoEnabled(false)
 
         popupManager = spy(CodeWhispererPopupManager.getInstance())
@@ -154,8 +153,8 @@ open class CodeWhispererIntegrationTestBase(val projectRule: CodeInsightTestFixt
         }
     }
 
-    fun withCodeWhispererServiceInvokedAndWait(manual: Boolean = true, runnable: (ListRecommendationsResponse) -> Unit) {
-        val responseCaptor = argumentCaptor<ListRecommendationsResponse>()
+    fun withCodeWhispererServiceInvokedAndWait(manual: Boolean = true, runnable: (GenerateCompletionsResponse) -> Unit) {
+        val responseCaptor = argumentCaptor<GenerateCompletionsResponse>()
         val statesCaptor = argumentCaptor<InvocationContext>()
         invokeCodeWhispererService(manual)
         verify(codewhispererService, timeout(5000).atLeastOnce()).validateResponse(responseCaptor.capture())
@@ -204,7 +203,7 @@ open class CodeWhispererIntegrationTestBase(val projectRule: CodeInsightTestFixt
         return runBlocking {
             var issues = emptyList<CodeWhispererCodeScanIssue>()
             if (success) {
-                verify(scanManager, timeout(60000).atLeastOnce()).renderResponseOnUIThread(issuesCaptor.capture())
+                verify(scanManager, timeout(60000).atLeastOnce()).renderResponseOnUIThread(issuesCaptor.capture(), any(), any())
                 issues = issuesCaptor.lastValue
             }
             verify(telemetryServiceSpy, timeout(60000).atLeastOnce()).sendSecurityScanEvent(codeScanEventCaptor.capture(), anyOrNull())
@@ -219,7 +218,7 @@ open class CodeWhispererIntegrationTestBase(val projectRule: CodeInsightTestFixt
         assertThat(response.responseContext.codeScanTotalIssues).isEqualTo(0)
         assertThat(response.responseContext.codeScanJobId).isNull()
         val exceptionCaptor = argumentCaptor<Exception>()
-        verify(scanManager, atLeastOnce()).handleException(exceptionCaptor.capture())
+        verify(scanManager, atLeastOnce()).handleException(any(), exceptionCaptor.capture())
         val e = exceptionCaptor.lastValue
         assertThat(e is CodeWhispererCodeScanException).isTrue
         assertThat(e.message).isEqualTo(message)