Skip to content

Commit fa412ce

Browse files
committed
Add cacheChangedEvent enum
1 parent 96d7d42 commit fa412ce

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

packages/core/src/auth/auth2.ts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,8 @@ interface BaseLogin {
6767
readonly loginType: LoginType
6868
}
6969

70+
export type cacheChangedEvent = 'delete' | 'create'
71+
7072
export type Login = SsoLogin // TODO: add IamLogin type when supported
7173

7274
export type TokenSource = IamIdentityCenterSsoTokenSource | AwsBuilderIdSsoTokenSource
@@ -168,7 +170,7 @@ export class LanguageClientAuth {
168170
this.client.onNotification(ssoTokenChangedRequestType.method, ssoTokenChangedHandler)
169171
}
170172

171-
registerCacheWatcher(cacheChangedHandler: (event: string) => any) {
173+
registerCacheWatcher(cacheChangedHandler: (event: cacheChangedEvent) => any) {
172174
this.cacheWatcher.onDidCreate(() => cacheChangedHandler('create'))
173175
this.cacheWatcher.onDidDelete(() => cacheChangedHandler('delete'))
174176
}

packages/core/src/codewhisperer/util/authUtil.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ import { showAmazonQWalkthroughOnce } from '../../amazonq/onboardingPage/walkthr
3131
import { setContext } from '../../shared/vscode/setContext'
3232
import { openUrl } from '../../shared/utilities/vsCodeUtils'
3333
import { telemetry } from '../../shared/telemetry/telemetry'
34-
import { AuthStateEvent, LanguageClientAuth, LoginTypes, SsoLogin } from '../../auth/auth2'
34+
import { AuthStateEvent, cacheChangedEvent, LanguageClientAuth, LoginTypes, SsoLogin } from '../../auth/auth2'
3535
import { builderIdStartUrl, internalStartUrl } from '../../auth/sso/constants'
3636
import { VSCODE_EXTENSION_ID } from '../../shared/extensions'
3737
import { RegionProfileManager } from '../region/regionProfileManager'
@@ -90,7 +90,7 @@ export class AuthUtil implements IAuthProvider {
9090
this.regionProfileManager.onDidChangeRegionProfile(async () => {
9191
await this.setVscodeContextProps()
9292
})
93-
lspAuth.registerCacheWatcher(async (event: string) => await this.cacheChangedHandler(event))
93+
lspAuth.registerCacheWatcher(async (event: cacheChangedEvent) => await this.cacheChangedHandler(event))
9494
}
9595

9696
// Do NOT use this in production code, only used for testing
@@ -277,7 +277,7 @@ export class AuthUtil implements IAuthProvider {
277277
})
278278
}
279279

280-
private async cacheChangedHandler(event: 'create' | 'delete') {
280+
private async cacheChangedHandler(event: cacheChangedEvent) {
281281
if (event === 'delete') {
282282
await this.logout()
283283
} else if (event === 'create') {

0 commit comments

Comments
 (0)