@aws-amplify/auth 5.5.4-api-v6.29 → 5.5.4-api-v6-models.35

package/src/OAuth/OAuth.ts

@@ -117,7 +117,7 @@ export default class OAuth {
 		this._urlOpener(URL, redirectSignIn);
 	}
 
-	private async _handleCodeFlow(currentUrl: string) {
+	private async _handleCodeFlow(currentUrl: string, userAgentValue?: string) {
 		/* Convert URL into an object with parameters as keys
     { redirect_uri: 'http://localhost:3000/', response_type: 'code', ...} */
 		const { code } = (parse(currentUrl).query || '')
@@ -169,17 +169,12 @@ export default class OAuth {
 			.map(([k, v]) => `${encodeURIComponent(k)}=${encodeURIComponent(v)}`)
 			.join('&');
 
-		const customUserAgentDetails: CustomUserAgentDetails = {
-			category: Category.Auth,
-			action: AuthAction.FederatedSignIn,
-		};
-
 		const { access_token, refresh_token, id_token, error } = await (
 			(await fetch(oAuthTokenEndpoint, {
 				method: 'POST',
 				headers: {
 					'Content-Type': 'application/x-www-form-urlencoded',
-					[USER_AGENT_HEADER]: getAmplifyUserAgent(customUserAgentDetails),
+					[USER_AGENT_HEADER]: userAgentValue,
 				},
 				body,
 			})) as any
@@ -217,7 +212,10 @@ export default class OAuth {
 		};
 	}
 
-	public async handleAuthResponse(currentUrl?: string) {
+	public async handleAuthResponse(
+		currentUrl?: string,
+		userAgentValue?: string
+	) {
 		try {
 			const urlParams = currentUrl
 				? ({
@@ -244,7 +242,10 @@ export default class OAuth {
 				`Starting ${this._config.responseType} flow with ${currentUrl}`
 			);
 			if (this._config.responseType === 'code') {
-				return { ...(await this._handleCodeFlow(currentUrl)), state };
+				return {
+					...(await this._handleCodeFlow(currentUrl, userAgentValue)),
+					state,
+				};
 			} else {
 				return { ...(await this._handleImplicitFlow(currentUrl)), state };
 			}