Skip to content

Commit ed5f0c8

Browse files
rlirli
committed
Merge remote-tracking branch 'origin/main' into rli/lsp-proxy
2 parents 548fa25 + 36b6e11 commit ed5f0c8

File tree

22 files changed

+160
-226
lines changed

22 files changed

+160
-226
lines changed

‎.changes/next-release/bugfix-2301aeeb-f36e-4776-8e9b-4238c55e43f4.json‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
{
2+
"type" : "bugfix",
3+
"description" : "/dev: Fix prompt to enable devfile build not triggering when devfile is present."
4+
}

‎.changes/next-release/bugfix-271e3abd-91e9-40a4-9a3d-7c7457ff6f22.json‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
{
2+
"type" : "bugfix",
3+
"description" : "/review disable auto scan by default"
4+
}

‎.changes/next-release/feature-ae7d15f4-ae07-4faf-963a-6d83a85852e4.json‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
{
2+
"type" : "feature",
3+
"description" : "Amazon Q: Support selecting customizations across all Q profiles with automatic profile switching for enterprise users"
4+
}

‎.changes/next-release/feature-f49ee314-0987-45cd-84fe-d852e67c8723.json‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
{
2+
"type" : "feature",
3+
"description" : "Amazon Q /test: Remove unsupported message for non-java python languages"
4+
}

‎plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/controller/CodeTestChatController.kt‎

Lines changed: 14 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -244,26 +244,22 @@ class CodeTestChatController(
244244
})
245245
}
246246
.build()
247+
if (!fileInfo.fileInWorkspace) {
248+
val messageContent =
249+
"<span style=\"color: #EE9D28;\">&#9888;<b> I can't generate tests for ${fileInfo.fileName}" +
250+
" because it's outside the project directory.</b><br></span> " +
251+
"I can still provide examples, instructions and code suggestions."
247252

248-
val messageContent = if (fileInfo.fileInWorkspace) {
249-
"<span style=\"color: #EE9D28;\">&#9888;<b> ${fileInfo.fileLanguage.languageId} is not a " +
250-
"language I support specialized unit test generation for at the moment.</b><br></span>The languages " +
251-
"I support now are Python and Java. I can still provide examples, instructions and code suggestions."
252-
} else {
253-
"<span style=\"color: #EE9D28;\">&#9888;<b> I can't generate tests for ${fileInfo.fileName}" +
254-
" because it's outside the project directory.</b><br></span> " +
255-
"I can still provide examples, instructions and code suggestions."
253+
codeTestChatHelper.addNewMessage(
254+
CodeTestChatMessageContent(
255+
message = messageContent,
256+
type = ChatMessageType.Answer,
257+
canBeVoted = false
258+
),
259+
message.tabId,
260+
false
261+
)
256262
}
257-
258-
codeTestChatHelper.addNewMessage(
259-
CodeTestChatMessageContent(
260-
message = messageContent,
261-
type = ChatMessageType.Answer,
262-
canBeVoted = false
263-
),
264-
message.tabId,
265-
false
266-
)
267263
testResponseMessageId = codeTestChatHelper.addAnswer(
268264
CodeTestChatMessageContent(
269265
message = "",

‎plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/controller/FeatureDevController.kt‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -479,7 +479,7 @@ class FeatureDevController(
479479
),
480480
)
481481

482-
if (!session.context.checkForDevFile()) {
482+
if (!session.context.hasDevFile()) {
483483
followUps.add(
484484
FollowUp(
485485
pillText = message("amazonqFeatureDev.follow_up.generate_dev_file"),
@@ -741,7 +741,7 @@ class FeatureDevController(
741741
}
742742

743743
val codeWhispererSettings = CodeWhispererSettings.getInstance().getAutoBuildSetting()
744-
val hasDevFile = session.context.checkForDevFile()
744+
val hasDevFile = session.context.hasDevFile()
745745
val isPromptedForAutoBuildFeature = codeWhispererSettings.containsKey(session.context.workspaceRoot.path)
746746

747747
if (hasDevFile && !isPromptedForAutoBuildFeature) {

‎plugins/amazonq/codewhisperer/jetbrains-community/src/migration/software/aws/toolkits/jetbrains/services/codewhisperer/explorer/CodeWhispererExplorerActionManager.kt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ class CodeWhispererExplorerActionManager : PersistentStateComponent<CodeWhispere
4848
actionState.value[CodeWhispererExploreStateType.IsAutoEnabled] = isAutoEnabled
4949
}
5050

51-
fun isAutoEnabledForCodeScan(): Boolean = actionState.value.getOrDefault(CodeWhispererExploreStateType.IsAutoCodeScanEnabled, true)
51+
fun isAutoEnabledForCodeScan(): Boolean = actionState.value.getOrDefault(CodeWhispererExploreStateType.IsAutoCodeScanEnabled, false)
5252

5353
fun setAutoEnabledForCodeScan(isAutoEnabledForCodeScan: Boolean) {
5454
actionState.value[CodeWhispererExploreStateType.IsAutoCodeScanEnabled] = isAutoEnabledForCodeScan

‎plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/credentials/CodeWhispererClientAdaptor.kt‎

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ import software.amazon.awssdk.services.codewhispererruntime.model.UserIntent
3838
import software.aws.toolkits.core.utils.debug
3939
import software.aws.toolkits.core.utils.getLogger
4040
import software.aws.toolkits.jetbrains.services.amazonq.codeWhispererUserContext
41+
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
4142
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileManager
4243
import software.aws.toolkits.jetbrains.services.codewhisperer.customization.CodeWhispererCustomization
4344
import software.aws.toolkits.jetbrains.services.codewhisperer.language.CodeWhispererProgrammingLanguage
@@ -78,7 +79,7 @@ interface CodeWhispererClientAdaptor {
7879

7980
fun getCodeFixJob(request: GetCodeFixJobRequest): GetCodeFixJobResponse
8081

81-
fun listAvailableCustomizations(): List<CodeWhispererCustomization>
82+
fun listAvailableCustomizations(profile: QRegionProfile): List<CodeWhispererCustomization>
8283

8384
fun startTestGeneration(uploadId: String, targetCode: List<TargetCode>, userInput: String): StartTestGenerationResponse
8485

@@ -282,9 +283,9 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
282283
override fun getCodeFixJob(request: GetCodeFixJobRequest): GetCodeFixJobResponse = bearerClient().getCodeFixJob(request)
283284

284285
// DO NOT directly use this method to fetch customizations, use wrapper [CodeWhispererModelConfigurator.listCustomization()] instead
285-
override fun listAvailableCustomizations(): List<CodeWhispererCustomization> =
286-
bearerClient().listAvailableCustomizationsPaginator(
287-
ListAvailableCustomizationsRequest.builder().profileArn(QRegionProfileManager.getInstance().activeProfile(project)?.arn).build()
286+
override fun listAvailableCustomizations(profile: QRegionProfile): List<CodeWhispererCustomization> =
287+
QRegionProfileManager.getInstance().getQClient<CodeWhispererRuntimeClient>(project, profile).listAvailableCustomizationsPaginator(
288+
ListAvailableCustomizationsRequest.builder().profileArn(profile.arn).build()
288289
)
289290
.stream()
290291
.toList()
@@ -298,7 +299,8 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
298299
CodeWhispererCustomization(
299300
arn = it.arn(),
300301
name = it.name(),
301-
description = it.description()
302+
description = it.description(),
303+
profile = profile
302304
)
303305
}
304306
}

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

Lines changed: 25 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ import software.amazon.awssdk.arns.Arn
2626
import software.aws.toolkits.core.utils.debug
2727
import software.aws.toolkits.core.utils.getLogger
2828
import software.aws.toolkits.core.utils.tryOrNull
29+
import software.aws.toolkits.jetbrains.services.amazonq.profile.QProfileSwitchIntent
30+
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
31+
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileManager
2932
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants.Q_CUSTOM_LEARN_MORE_URI
3033
import software.aws.toolkits.jetbrains.ui.AsyncComboBox
3134
import software.aws.toolkits.jetbrains.utils.notifyInfo
@@ -34,7 +37,7 @@ import javax.swing.JComponent
3437
import javax.swing.JList
3538

3639
private val NoDataToDisplay = CustomizationUiItem(
37-
CodeWhispererCustomization("", message("codewhisperer.custom.dialog.option.no_data"), ""),
40+
CodeWhispererCustomization("", message("codewhisperer.custom.dialog.option.no_data"), "", QRegionProfile("", "")),
3841
false,
3942
false
4043
)
@@ -106,6 +109,15 @@ class CodeWhispererCustomizationDialog(
106109
RadioButtonOption.Customization -> run {
107110
CodeWhispererModelConfigurator.getInstance().switchCustomization(project, modal.selectedCustomization?.customization)
108111
notifyCustomizationIsSelected(project, modal.selectedCustomization)
112+
// Switch profile if it doesn't match the customization's profile.
113+
// Customizations are profile-scoped and must be used under the correct context.
114+
if (modal.selectedCustomization?.customization?.profile?.arn != QRegionProfileManager.getInstance().activeProfile(project)?.arn) {
115+
QRegionProfileManager.getInstance().switchProfile(
116+
project,
117+
modal.selectedCustomization?.customization?.profile,
118+
QProfileSwitchIntent.Customization
119+
)
120+
}
109121
}
110122
}
111123

@@ -179,12 +191,14 @@ class CodeWhispererCustomizationDialog(
179191
proposeModelUpdate { model ->
180192
val activeCustomization = CodeWhispererModelConfigurator.getInstance().activeCustomization(project)
181193
val unsorted = myCustomizations ?: CodeWhispererModelConfigurator.getInstance().listCustomizations(project).orEmpty()
182-
183-
val sorted = activeCustomization?.let {
184-
unsorted.putPickedUpFront(setOf(it))
185-
} ?: run {
186-
unsorted.sortedBy { it.customization.name }
187-
}
194+
val activeProfile = QRegionProfileManager.getInstance().activeProfile(project)
195+
// Group customizations by profile name (active profile first, then alphabetical), with the active customization on top
196+
val sorted = unsorted.sortedWith(
197+
compareBy<CustomizationUiItem> { it.customization.profile?.profileName != activeProfile?.profileName }
198+
.thenBy { it.customization.profile?.profileName.orEmpty() }
199+
.thenBy { it.customization.name }
200+
)
201+
.let { list -> activeCustomization?.let { list.putPickedUpFront(setOf(it)) } ?: list }
188202

189203
if (
190204
sorted.isNotEmpty() &&
@@ -259,6 +273,10 @@ private object CustomizationRenderer : ColoredListCellRenderer<CustomizationUiIt
259273
}
260274
}
261275

276+
if (it.customization.profile?.profileName?.isNotEmpty() == true) {
277+
append(" [${it.customization.profile?.profileName}]", SimpleTextAttributes.REGULAR_ATTRIBUTES)
278+
}
279+
262280
if (it.isNew) {
263281
append(" New", SimpleTextAttributes.GRAYED_SMALL_ATTRIBUTES)
264282
}

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

Lines changed: 30 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,9 @@ import software.aws.toolkits.core.utils.getLogger
2323
import software.aws.toolkits.jetbrains.services.amazonq.CodeWhispererFeatureConfigService
2424
import software.aws.toolkits.jetbrains.services.amazonq.calculateIfIamIdentityCenterConnection
2525
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
26+
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileManager
2627
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
2728
import software.aws.toolkits.jetbrains.services.codewhisperer.credentials.CodeWhispererClientAdaptor
28-
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants
2929
import software.aws.toolkits.jetbrains.utils.notifyInfo
3030
import software.aws.toolkits.jetbrains.utils.notifyWarn
3131
import software.aws.toolkits.jetbrains.utils.pluginAwareExecuteOnPooledThread
@@ -108,25 +108,24 @@ class DefaultCodeWhispererModelConfigurator : CodeWhispererModelConfigurator, Pe
108108
@RequiresBackgroundThread
109109
override fun listCustomizations(project: Project, passive: Boolean): List<CustomizationUiItem>? =
110110
calculateIfIamIdentityCenterConnection(project) {
111-
// 1. invoke API and get result
112-
val listAvailableCustomizationsResult = try {
113-
CodeWhispererClientAdaptor.getInstance(project).listAvailableCustomizations()
114-
} catch (e: Exception) {
115-
val requestId = (e as? CodeWhispererRuntimeException)?.requestId()
116-
val logMessage = if (CodeWhispererConstants.Customization.noAccessToCustomizationExceptionPredicate(e)) {
117-
// TODO: not required for non GP users
118-
"ListAvailableCustomizations: connection ${it.id} is not allowlisted, requestId: ${requestId.orEmpty()}"
119-
} else {
120-
"ListAvailableCustomizations: failed due to unknown error ${e.message}, requestId: ${requestId.orEmpty()}"
121-
}
122-
123-
LOG.debug { logMessage }
124-
null
111+
// 1. fetch all profiles, invoke fetch customizations API and get result for each profile and aggregate all the results
112+
val listAvailableProfilesResult = QRegionProfileManager.getInstance().listRegionProfiles(project)
113+
?: error("Attempted to fetch profiles while there does not exist")
114+
115+
val aggregatedCustomizations = listAvailableProfilesResult.flatMap { profile ->
116+
runCatching {
117+
CodeWhispererClientAdaptor.getInstance(project).listAvailableCustomizations(profile)
118+
}.onFailure { e ->
119+
val requestId = (e as? CodeWhispererRuntimeException)?.requestId()
120+
val logMessage = "ListAvailableCustomizations: failed due to unknown error ${e.message}, " +
121+
"requestId: ${requestId.orEmpty()}, profileName: ${profile.profileName}"
122+
LOG.debug { logMessage }
123+
}.getOrDefault(emptyList())
125124
}
126125

127126
// 2. get diff
128127
val previousCustomizationsShapshot = connectionToCustomizationsShownLastTime.getOrElse(it.id) { emptyList() }
129-
val diff = listAvailableCustomizationsResult?.filterNot { customization -> previousCustomizationsShapshot.contains(customization.arn) }?.toSet()
128+
val diff = aggregatedCustomizations.filterNot { customization -> previousCustomizationsShapshot.contains(customization.arn) }.toSet()
130129

131130
// 3 if passive,
132131
// (1) update allowlisting
@@ -135,42 +134,36 @@ class DefaultCodeWhispererModelConfigurator : CodeWhispererModelConfigurator, Pe
135134
// if not passive,
136135
// (1) update the customization list snapshot (seen by users last time) if it will be displayed
137136
if (passive) {
138-
connectionIdToIsAllowlisted[it.id] = listAvailableCustomizationsResult != null
139-
if (diff?.isNotEmpty() == true && !hasShownNewCustomizationNotification.getAndSet(true)) {
137+
connectionIdToIsAllowlisted[it.id] = aggregatedCustomizations.isNotEmpty()
138+
if (diff.isNotEmpty() && !hasShownNewCustomizationNotification.getAndSet(true)) {
140139
notifyNewCustomization(project)
141140
}
142141
} else {
143-
listAvailableCustomizationsResult?.let { customizations ->
144-
connectionToCustomizationsShownLastTime[it.id] = customizations.map { customization -> customization.arn }.toMutableList()
145-
}
142+
connectionToCustomizationsShownLastTime[it.id] = aggregatedCustomizations.map { customization -> customization.arn }.toMutableList()
146143
}
147144

148145
// 4. invalidate selected customization if
149146
// (1) the API call failed
150147
// (2) the selected customization is not in the resultset of API call
148+
// (3) the existing q region profile associated with the selected customization does not match the currently active profile
151149
activeCustomization(project)?.let { activeCustom ->
152-
if (listAvailableCustomizationsResult == null) {
150+
if (aggregatedCustomizations.isEmpty()) {
153151
invalidateSelectedAndNotify(project)
154-
} else if (!listAvailableCustomizationsResult.any { latestCustom -> latestCustom.arn == activeCustom.arn }) {
152+
} else if (!aggregatedCustomizations.any { latestCustom -> latestCustom.arn == activeCustom.arn } ||
153+
(activeCustom.profile != null && activeCustom.profile != QRegionProfileManager.getInstance().activeProfile(project))
154+
) {
155155
invalidateSelectedAndNotify(project)
156156
}
157157
}
158158

159159
// 5. transform result to UI items and return
160-
val customizationUiItems = if (diff != null) {
161-
listAvailableCustomizationsResult.let { customizations ->
162-
val nameToCount = customizations.groupingBy { customization -> customization.name }.eachCount()
163-
164-
customizations.map { customization ->
165-
CustomizationUiItem(
166-
customization,
167-
isNew = diff.contains(customization),
168-
shouldPrefixAccountId = (nameToCount[customization.name] ?: 0) > 1
169-
)
170-
}
171-
}
172-
} else {
173-
null
160+
val nameToCount = aggregatedCustomizations.groupingBy { customization -> customization.name }.eachCount()
161+
val customizationUiItems = aggregatedCustomizations.map { customization ->
162+
CustomizationUiItem(
163+
customization,
164+
isNew = diff.contains(customization),
165+
shouldPrefixAccountId = (nameToCount[customization.name] ?: 0) > 1
166+
)
174167
}
175168
connectionToCustomizationUiItems[it.id] = customizationUiItems
176169

0 commit comments

Comments
 (0)