Support custom params for token request

asgardeo · Jul 24, 2024 · 81485cb · 81485cb
1 parent 9769b04
commit 81485cb
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/lib/src/client.ts b/lib/src/client.ts
@@ -283,14 +283,19 @@ export class AsgardeoAuthClient<T> {
         authorizationCode: string,
         sessionState: string,
         state: string,
-        userID?: string
+        userID?: string,
+        tokenRequestConfig?: {
+            params: Record<string, unknown>
+        }
     ): Promise<TokenResponse> {
         if (await this._dataLayer.getTemporaryDataParameter(OP_CONFIG_INITIATED)) {
-            return this._authenticationCore.requestAccessToken(authorizationCode, sessionState, state, userID);
+            return this._authenticationCore.requestAccessToken(authorizationCode, sessionState,
+                state, userID, tokenRequestConfig);
         }
 
         return this._authenticationCore.getOIDCProviderMetaData(false).then(() => {
-            return this._authenticationCore.requestAccessToken(authorizationCode, sessionState, state, userID);
+            return this._authenticationCore.requestAccessToken(authorizationCode, sessionState,
+                state, userID, tokenRequestConfig);
         });
     }
 

diff --git a/lib/src/core/authentication-core.ts b/lib/src/core/authentication-core.ts
@@ -165,9 +165,12 @@ export class AuthenticationCore<T> {
         authorizationCode: string,
         sessionState: string,
         state: string,
-        userID?: string
+        userID?: string,
+        tokenRequestConfig?: {
+            params: Record<string, unknown>
+        }
     ): Promise<TokenResponse> {
-        const tokenEndpoint: string | undefined = (await this._oidcProviderMetaData()).token_endpoint;
+        let tokenEndpoint: string | undefined = (await this._oidcProviderMetaData()).token_endpoint;
         const configData: StrictAuthClientConfig = await this._config();
 
         if (!tokenEndpoint || tokenEndpoint.trim().length === 0) {
@@ -179,6 +182,15 @@ export class AuthenticationCore<T> {
             );
         }
 
+        if (tokenRequestConfig && tokenRequestConfig.params) {
+            const url: URL = new URL(tokenEndpoint);
+
+            Object.entries(tokenRequestConfig.params).forEach(([ key, value ]: [key: string, value: unknown]) => {
+                url.searchParams.append(key, value as string);
+            });
+            tokenEndpoint = url.toString();
+        }
+
         sessionState && (await this._dataLayer.setSessionDataParameter(
             SESSION_STATE as keyof SessionData, sessionState, userID));