Skip to content

Commit 77d066f

Browse files
authored
fix(amazonq): chat enablement isn't updated correctly on profile changed
1 parent 9c257ed commit 77d066f

File tree

6 files changed

+16
-4
lines changed

6 files changed

+16
-4
lines changed

packages/amazonq/src/app/amazonqScan/app.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,9 @@ export function init(appContext: AmazonQAppInitContext) {
7070
AuthUtil.instance.secondaryAuth.onDidChangeActiveConnection(() => {
7171
return debouncedEvent()
7272
})
73+
AuthUtil.instance.regionProfileManager.onDidChangeRegionProfile(() => {
74+
return debouncedEvent()
75+
})
7376

7477
Commands.register('aws.amazonq.security.scan-statusbar', async () => {
7578
if (AuthUtil.instance.isConnectionExpired()) {

packages/core/src/amazonq/webview/generators/webViewContent.ts

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,10 +94,7 @@ export class WebViewContentGenerator {
9494
}
9595

9696
const regionProfileString: string = JSON.stringify(regionProfile)
97-
98-
// AuthUtil.instance.getChatAuthState is throttled version which possibly return an old snapshot of auth state however webview initialization here requires the latest accurate
99-
// otherwise features will be disabled as auth still says it's not connected & profile selected
100-
const authState = (await AuthUtil.instance._getChatAuthState()).amazonQ
97+
const authState = (await AuthUtil.instance.getChatAuthState()).amazonQ
10198

10299
return `
103100
<script type="text/javascript" src="${javascriptEntrypoint.toString()}" defer onload="init()"></script>

packages/core/src/amazonqDoc/app.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,4 +100,7 @@ export function init(appContext: AmazonQAppInitContext) {
100100
AuthUtil.instance.secondaryAuth.onDidChangeActiveConnection(() => {
101101
return debouncedEvent()
102102
})
103+
AuthUtil.instance.regionProfileManager.onDidChangeRegionProfile(() => {
104+
return debouncedEvent()
105+
})
103106
}

packages/core/src/amazonqFeatureDev/app.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,4 +106,7 @@ export function init(appContext: AmazonQAppInitContext) {
106106
AuthUtil.instance.secondaryAuth.onDidChangeActiveConnection(() => {
107107
return debouncedEvent()
108108
})
109+
AuthUtil.instance.regionProfileManager.onDidChangeRegionProfile(() => {
110+
return debouncedEvent()
111+
})
109112
}

packages/core/src/amazonqGumby/app.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,9 @@ export function init(appContext: AmazonQAppInitContext) {
6767
AuthUtil.instance.secondaryAuth.onDidChangeActiveConnection(() => {
6868
return debouncedEvent()
6969
})
70+
AuthUtil.instance.regionProfileManager.onDidChangeRegionProfile(() => {
71+
return debouncedEvent()
72+
})
7073

7174
showTransformationHub.register()
7275

packages/core/src/amazonqTest/app.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,9 @@ export function init(appContext: AmazonQAppInitContext) {
6868
AuthUtil.instance.secondaryAuth.onDidChangeActiveConnection(() => {
6969
return debouncedEvent()
7070
})
71+
AuthUtil.instance.regionProfileManager.onDidChangeRegionProfile(() => {
72+
return debouncedEvent()
73+
})
7174
testGenState.setChatControllers(testChatControllerEventEmitters)
7275
// TODO: Add testGen provider for creating new files after test generation if they does not exist
7376
}

0 commit comments

Comments
 (0)