Merge pull request #19 from jakubvilhan/feature/custom-get-token-fn

Feature add support to provide custom getToken function

Author: WilsonLe

src/callback-endpoint.ts

@@ -42,24 +42,32 @@ export const createCallbackEndpoint = (
       // obtain access token
       // /////////////////////////////////////
 
-      const tokenResponse = await fetch(pluginOptions.tokenEndpoint, {
-        method: "POST",
-        headers: {
-          "Content-Type": "application/x-www-form-urlencoded",
-          Accept: "application/json",
-        },
-        body: new URLSearchParams({
-          code,
-          client_id: pluginOptions.clientId,
-          client_secret: pluginOptions.clientSecret,
-          redirect_uri: redirectUri,
-          grant_type: "authorization_code",
-        }).toString(),
-      });
-      const tokenData = await tokenResponse.json();
-      const access_token = tokenData?.access_token;
+      let access_token: string;
+
+      if (pluginOptions.getToken) {
+        access_token = pluginOptions.getToken(code);
+      } else {
+        const tokenResponse = await fetch(pluginOptions.tokenEndpoint, {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/x-www-form-urlencoded",
+            Accept: "application/json",
+          },
+          body: new URLSearchParams({
+            code,
+            client_id: pluginOptions.clientId,
+            client_secret: pluginOptions.clientSecret,
+            redirect_uri: redirectUri,
+            grant_type: "authorization_code",
+          }).toString(),
+        });
+        const tokenData = await tokenResponse.json();
+
+        access_token = tokenData?.access_token;
+      }
+
       if (typeof access_token !== "string")
-        throw new Error(`No access token: ${JSON.stringify(tokenData)}`);
+        throw new Error(`No access token: ${access_token}`);
 
       // /////////////////////////////////////
       // get user info

src/types.ts

@@ -111,6 +111,12 @@ export interface PluginTypes {
    */
   callbackPath?: string;
 
+  /**
+   * Function to get token from the OAuth providers.
+   * If its not provided default will be used.
+   */
+  getToken?: (code: string) => string;
+
   /**
    * Redirect users after successful login.
    */