fix(auth): add inflightPromise in token orchestrator

aws-amplify · Mar 21, 2024 · 7698d53 · 7698d53
1 parent 1a6d358
commit 7698d53
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/packages/auth/src/providers/cognito/tokenProvider/TokenOrchestrator.ts b/packages/auth/src/providers/cognito/tokenProvider/TokenOrchestrator.ts
@@ -23,13 +23,23 @@ import {
 	DeviceMetadata,
 	TokenRefresher,
 } from './types';
+import { oAuthStore } from '../utils/oauth';
+import { addInflightPromise } from '../utils/oauth/inflightPromise';
 
 export class TokenOrchestrator implements AuthTokenOrchestrator {
 	private authConfig?: AuthConfig;
 	tokenStore?: AuthTokenStore;
 	tokenRefresher?: TokenRefresher;
 	waitForInflightOAuth: () => Promise<void> = async () => {
 		// no-op
+		if (!(await oAuthStore.loadOAuthInFlight())) {
+			return;
+		}
+
+		return new Promise<void>((resolve, _reject) => {
+			addInflightPromise(resolve);
+		});
+
 	};
 
 	setAuthConfig(authConfig: AuthConfig) {
@@ -83,6 +93,7 @@ export class TokenOrchestrator implements AuthTokenOrchestrator {
 			// Token provider not configured
 			return null;
 		}
+
 		await this.waitForInflightOAuth();
 		tokens = await this.getTokenStore().loadTokens();
 		const username = await this.getTokenStore().getLastAuthUser();