[Extension Service Provider] add option to override access token (#3665)

* add option to override access token to support dual tokens Co-authored-by: Arun George <aruniverse@users.noreply.github.com>
iTwin · May 24, 2022 · 8b52c5b · 8b52c5b
1 parent d8f86ad
commit 8b52c5b
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 1 deletion.
diff --git a/common/api/core-frontend.api.md b/common/api/core-frontend.api.md
@@ -9580,6 +9580,8 @@ export class ServiceExtensionProvider implements ExtensionProvider {
 
 // @alpha
 export interface ServiceExtensionProviderProps {
+    // @internal (undocumented)
+    getAccessToken?: () => Promise<AccessToken>;
     iTwinId: string;
     name: string;
     version: string;

diff --git a/common/changes/@itwin/core-frontend/ExtServiceProviderAccessTokenCB_2022-05-23-16-50.json b/common/changes/@itwin/core-frontend/ExtServiceProviderAccessTokenCB_2022-05-23-16-50.json
@@ -0,0 +1,10 @@
+{
+  "changes": [
+    {
+      "packageName": "@itwin/core-frontend",
+      "comment": "",
+      "type": "none"
+    }
+  ],
+  "packageName": "@itwin/core-frontend"
+}
diff --git a/core/frontend/src/extension/providers/ServiceExtensionProvider.ts b/core/frontend/src/extension/providers/ServiceExtensionProvider.ts
@@ -11,6 +11,7 @@ import { ExtensionClient, ExtensionMetadata } from "./ExtensionServiceClient";
 import type {
   ExtensionManifest, ExtensionProvider,
 } from "../Extension";
+import type { AccessToken } from "@itwin/core-bentley";
 
 /**
  * Required props for an extension uploaded to Bentley's Extension Service
@@ -23,6 +24,8 @@ export interface ServiceExtensionProviderProps {
   version: string;
   /** iTwin Id */
   iTwinId: string;
+  /** @internal */
+  getAccessToken?: () => Promise<AccessToken>;
 }
 
 /**
@@ -83,7 +86,7 @@ export class ServiceExtensionProvider implements ExtensionProvider {
   private async _getExtensionFiles(props: ServiceExtensionProviderProps) {
     const extensionClient = new ExtensionClient();
 
-    const accessToken = await IModelApp.authorizationClient?.getAccessToken();
+    const accessToken = await (props.getAccessToken?.() ?? IModelApp.authorizationClient?.getAccessToken());
     if (!accessToken)
       return undefined;