fix(amazonq): Invalidate Obsolete Customization on Profile Switch

plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/customization/CodeWhispererModelConfigurator.kt

@@ -22,7 +22,10 @@
 import software.aws.toolkits.core.utils.getLogger
 import software.aws.toolkits.jetbrains.services.amazonq.CodeWhispererFeatureConfigService
 import software.aws.toolkits.jetbrains.services.amazonq.calculateIfIamIdentityCenterConnection
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
 import software.aws.toolkits.jetbrains.services.codewhisperer.credentials.CodeWhispererClientAdaptor
+import software.aws.toolkits.jetbrains.services.codewhisperer.status.CodeWhispererStatusBarWidget.Companion.ID
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants
 import software.aws.toolkits.jetbrains.utils.notifyInfo
 import software.aws.toolkits.jetbrains.utils.notifyWarn
@@ -81,6 +84,18 @@
 
     private var customizationArnOverrideV2: String? = null
 
+    init {
+        ApplicationManager.getApplication().messageBus.connect(this).subscribe(
+            QRegionProfileSelectedListener.TOPIC,
+            object : QRegionProfileSelectedListener {
+                override fun onProfileSelected(project: Project, profile: QRegionProfile?) {
+                    pluginAwareExecuteOnPooledThread {
+                        CodeWhispererModelConfigurator.getInstance().listCustomizations(project, passive = true)
+                    }
+                }
+            }
+        )
+    }
     override fun showConfigDialog(project: Project) {
         runInEdt {
             calculateIfIamIdentityCenterConnection(project) {

plugins/amazonq/codewhisperer/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/codewhisperer/CodeWhispererModelConfiguratorTest.kt

@@ -7,6 +7,7 @@ import com.intellij.openapi.application.ApplicationManager
 import com.intellij.testFramework.ApplicationRule
 import com.intellij.testFramework.DisposableRule
 import com.intellij.testFramework.ProjectRule
+import com.intellij.testFramework.registerServiceInstance
 import com.intellij.testFramework.replaceService
 import com.intellij.util.xmlb.XmlSerializer
 import org.assertj.core.api.Assertions.assertThat
@@ -40,6 +41,8 @@ import software.aws.toolkits.jetbrains.core.credentials.sono.isSono
 import software.aws.toolkits.jetbrains.core.region.MockRegionProviderRule
 import software.aws.toolkits.jetbrains.services.amazonq.CodeWhispererFeatureConfigService
 import software.aws.toolkits.jetbrains.services.amazonq.FeatureContext
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
+import software.aws.toolkits.jetbrains.services.codewhisperer.credentials.CodeWhispererClientAdaptor
 import software.aws.toolkits.jetbrains.services.codewhisperer.customization.CodeWhispererCustomization
 import software.aws.toolkits.jetbrains.services.codewhisperer.customization.CodeWhispererCustomizationState
 import software.aws.toolkits.jetbrains.services.codewhisperer.customization.DefaultCodeWhispererModelConfigurator
@@ -75,6 +78,7 @@ class CodeWhispererModelConfiguratorTest {
     private lateinit var sut: DefaultCodeWhispererModelConfigurator
     private lateinit var mockClient: CodeWhispererRuntimeClient
     private lateinit var abManager: CodeWhispererFeatureConfigService
+    private lateinit var mockClintAdaptor: CodeWhispererClientAdaptor
 
     @Before
     fun setup() {
@@ -110,6 +114,9 @@ class CodeWhispererModelConfiguratorTest {
             abManager,
             disposableRule.disposable
         )
+
+        mockClintAdaptor = mock()
+        projectRule.project.registerServiceInstance(CodeWhispererClientAdaptor::class.java, mockClintAdaptor)
     }
 
     @Test
@@ -550,4 +557,47 @@ class CodeWhispererModelConfiguratorTest {
         assertThat(actual.previousAvailableCustomizations).hasSize(1)
         assertThat(actual.previousAvailableCustomizations["fake-sso-url"]).isEqualTo(listOf("arn_1", "arn_2", "arn_3"))
     }
+
+    @Test
+    fun `profile switch should keep using existing customization if new list still contains that arn`() {
+        val ssoConn = spy(LegacyManagedBearerSsoConnection(region = "us-east-1", startUrl = "url 1", scopes = Q_SCOPES))
+        ToolkitConnectionManager.getInstance(projectRule.project).switchConnection(ssoConn)
+        val oldCustomization = CodeWhispererCustomization("oldArn", "oldName", "oldDescription")
+        sut.switchCustomization(projectRule.project, oldCustomization)
+
+        assertThat(sut.activeCustomization(projectRule.project)).isEqualTo(oldCustomization)
+
+        val fakeCustomizations = listOf(
+            CodeWhispererCustomization("oldArn", "oldName", "oldDescription")
+        )
+
+        mockClintAdaptor.stub { on { listAvailableCustomizations() } doReturn fakeCustomizations}
+
+        ApplicationManager.getApplication().messageBus
+            .syncPublisher(QRegionProfileSelectedListener.TOPIC)
+            .onProfileSelected(projectRule.project, null)
+
+        assertThat(sut.activeCustomization(projectRule.project)).isEqualTo(oldCustomization)
+    }
+
+    @Test
+    fun `profile switch should invalidate obsolete customization if it's not in the new list`() {
+        val ssoConn = spy(LegacyManagedBearerSsoConnection(region = "us-east-1", startUrl = "url 1", scopes = Q_SCOPES))
+        ToolkitConnectionManager.getInstance(projectRule.project).switchConnection(ssoConn)
+        val oldCustomization = CodeWhispererCustomization("oldArn", "oldName", "oldDescription")
+        sut.switchCustomization(projectRule.project, oldCustomization)
+        assertThat(sut.activeCustomization(projectRule.project)).isEqualTo(oldCustomization)
+
+        val fakeCustomizations = listOf(
+            CodeWhispererCustomization("newArn", "newName", "oldDescription")
+        )
+        mockClintAdaptor.stub { on { listAvailableCustomizations() } doReturn fakeCustomizations}
+
+        ApplicationManager.getApplication().messageBus
+            .syncPublisher(QRegionProfileSelectedListener.TOPIC)
+            .onProfileSelected(projectRule.project, null)
+
+        assertThat(sut.activeCustomization(projectRule.project)).isNull()
+    }
+
 }

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/profile/QRegionProfileManager.kt

@@ -144,7 +144,7 @@ class QRegionProfileManager : PersistentStateComponent<QProfileState>, Disposabl
             }
         }
 
-        project.messageBus
+        ApplicationManager.getApplication().messageBus
             .syncPublisher(QRegionProfileSelectedListener.TOPIC)
             .onProfileSelected(project, newProfile)
     }

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/profile/QRegionProfileSelectedListener.kt

@@ -8,7 +8,7 @@ import com.intellij.util.messages.Topic
 
 interface QRegionProfileSelectedListener {
     companion object {
-        @Topic.ProjectLevel
+        @Topic.AppLevel
         val TOPIC = Topic.create("QRegionProfileSelected", QRegionProfileSelectedListener::class.java)
     }