port: [#4465][#6560] Allow TokenCredential authentication in CosmosDbPartitionedStorage (#4473)

* add tokenCredential authentication to cosmosDB

* fix ts issues with new version

* fix pipelines issues

* add EOF

---------

Co-authored-by: JhontSouth <[email protected]>

Author: ceciliaavila

libraries/botbuilder-azure-blobs/src/ignoreError.ts

@@ -19,7 +19,7 @@ export type IgnoreError = (err: Error) => boolean;
 export async function ignoreError<T>(promise: Promise<T>, ignore: IgnoreError): Promise<T | null> {
     try {
         return await promise;
-    } catch (err) {
+    } catch (err: any) {
         if (!ignore(err)) {
             throw err;
         } else {

libraries/botbuilder-azure/etc/botbuilder-azure.api.md

@@ -9,6 +9,7 @@ import { CosmosClientOptions } from '@azure/cosmos';
 import { PagedResult } from 'botbuilder';
 import { Storage as Storage_2 } from 'botbuilder';
 import { StoreItems } from 'botbuilder';
+import { TokenCredential } from '@azure/core-auth';
 import { TranscriptInfo } from 'botbuilder';
 import { TranscriptStore } from 'botbuilder';
 
@@ -64,6 +65,7 @@ export interface CosmosDbPartitionedStorageOptions {
     cosmosDbEndpoint?: string;
     databaseId: string;
     keySuffix?: string;
+    tokenCredential?: TokenCredential;
 }
 
 // @public (undocumented)

libraries/botbuilder-azure/package.json

@@ -28,7 +28,8 @@
     }
   },
   "dependencies": {
-    "@azure/cosmos": "^3.3.1",
+    "@azure/cosmos": "3.10.0",
+    "@azure/core-auth": "^1.1.3",
     "azure-storage": "2.10.7",
     "botbuilder": "4.1.6",
     "lodash": "^4.17.20",

libraries/botbuilder-azure/src/cosmosDbPartitionedStorage.ts

@@ -11,6 +11,7 @@ import { Container, CosmosClient, CosmosClientOptions } from '@azure/cosmos';
 import { CosmosDbKeyEscape } from './cosmosDbKeyEscape';
 import { DoOnce } from './doOnce';
 import { Storage, StoreItems } from 'botbuilder';
+import { TokenCredential } from '@azure/core-auth';
 
 // eslint-disable-next-line @typescript-eslint/no-var-requires
 const pjson: Record<'name' | 'version', string> = require('../package.json');
@@ -67,6 +68,10 @@ export interface CosmosDbPartitionedStorageOptions {
      * compatibilityMode cannot be true if keySuffix is used.
      */
     compatibilityMode?: boolean;
+    /**
+     * The authentication tokenCredential for Cosmos DB.
+     */
+    tokenCredential?: TokenCredential;
 }
 
 //Internal data structure for storing items in a CosmosDB Collection.
@@ -143,8 +148,12 @@ export class CosmosDbPartitionedStorage implements Storage {
             throw new ReferenceError('cosmosDbEndpoint for CosmosDB is required.');
         }
         cosmosDbStorageOptions.authKey ??= cosmosClientOptions?.key;
-        if (!cosmosDbStorageOptions.authKey && !cosmosClientOptions?.tokenProvider) {
-            throw new ReferenceError('authKey for CosmosDB is required.');
+        if (
+            !cosmosDbStorageOptions.authKey &&
+            !cosmosClientOptions?.tokenProvider &&
+            !cosmosDbStorageOptions.tokenCredential
+        ) {
+            throw new ReferenceError('authKey or tokenCredential for CosmosDB is required.');
         }
         if (!cosmosDbStorageOptions.databaseId) {
             throw new ReferenceError('databaseId is for CosmosDB required.');
@@ -322,12 +331,21 @@ export class CosmosDbPartitionedStorage implements Storage {
     async initialize(): Promise<void> {
         if (!this.container) {
             if (!this.client) {
-                this.client = new CosmosClient({
-                    endpoint: this.cosmosDbStorageOptions.cosmosDbEndpoint,
-                    key: this.cosmosDbStorageOptions.authKey,
-                    userAgentSuffix: `${pjson.name} ${pjson.version}`,
-                    ...this.cosmosDbStorageOptions.cosmosClientOptions,
-                });
+                if (this.cosmosDbStorageOptions.tokenCredential) {
+                    this.client = new CosmosClient({
+                        endpoint: this.cosmosDbStorageOptions.cosmosDbEndpoint,
+                        aadCredentials: this.cosmosDbStorageOptions.tokenCredential,
+                        userAgentSuffix: `${pjson.name} ${pjson.version}`,
+                        ...this.cosmosDbStorageOptions.cosmosClientOptions,
+                    });
+                } else {
+                    this.client = new CosmosClient({
+                        endpoint: this.cosmosDbStorageOptions.cosmosDbEndpoint,
+                        key: this.cosmosDbStorageOptions.authKey,
+                        userAgentSuffix: `${pjson.name} ${pjson.version}`,
+                        ...this.cosmosDbStorageOptions.cosmosClientOptions,
+                    });
+                }
             }
             const dbAndContainerKey = `${this.cosmosDbStorageOptions.databaseId}-${this.cosmosDbStorageOptions.containerId}`;
             this.container = await _doOnce.waitFor(

libraries/botbuilder-azure/tests/cosmosDbPartitionedStorage.test.js

@@ -146,12 +146,12 @@ describe('CosmosDbPartitionedStorage - Constructor Tests', function () {
         );
     });
 
-    it('throws when no authKey provided', function () {
+    it('throws when no authKey or tokenCredential provided', function () {
         const noAuthKey = getSettings();
         noAuthKey.authKey = null;
         assert.throws(
             () => new CosmosDbPartitionedStorage(noAuthKey),
-            ReferenceError('authKey for CosmosDB is required.')
+            ReferenceError('authKey or tokenCredential for CosmosDB is required.')
         );
     });
 

libraries/botbuilder-dialogs-adaptive-runtime-core/test/serviceCollection.test.ts

@@ -57,7 +57,7 @@ describe('ServiceCollection', function () {
                 services.addFactory<number[]>('values', (values = [1]) => values.concat(2));
                 services.composeFactory<number[]>('values', (values) => values.concat(3));
 
-                const { values } = services.makeInstances();
+                const { values } = services.makeInstances<{ values: number[] }>();
                 assert.deepStrictEqual(values, [1, 2, 3]);
             });
 
@@ -69,7 +69,7 @@ describe('ServiceCollection', function () {
                 services.composeFactory<number[]>('values', (values) => values.concat(2));
                 services.composeFactory<number[]>('values', (values) => values.concat(3));
 
-                const { values } = services.makeInstances();
+                const { values } = services.makeInstances<{ values: number[] }>();
                 assert.deepStrictEqual(values, [1, 2, 3]);
             });
         });
@@ -95,8 +95,7 @@ describe('ServiceCollection', function () {
 
         it('optionally fully reconstructs dependencies', function () {
             const services = makeServiceCollection();
-
-            const { foo, bar, baz } = services.makeInstances();
+            const { foo, bar, baz } = services.makeInstances<{ foo: Foo; bar: Bar; baz: Baz }>();
             assert.ok(foo);
             assert.ok(bar);
             assert.ok(baz);

libraries/botbuilder-dialogs-adaptive-runtime-integration-azure-functions/src/index.ts

@@ -209,7 +209,7 @@ export function makeTriggers(
                 res.status(200);
                 res.set('Content-Type', contentType);
                 res.end(contents);
-            } catch (err) {
+            } catch (err: any) {
                 if (err.message.includes('ENOENT')) {
                     return res.status(404).end();
                 }

libraries/botbuilder-dialogs-adaptive-runtime-integration-express/src/index.ts

@@ -144,7 +144,7 @@ export async function makeApp(
             await adapter.process(req, res, async (turnContext) => {
                 await bot.run(turnContext);
             });
-        } catch (err) {
+        } catch (err: any) {
             return errorHandler(err, res);
         }
     });
@@ -173,7 +173,7 @@ export async function makeApp(
                         await adapter.process(req, res, async (turnContext) => {
                             await bot.run(turnContext);
                         });
-                    } catch (err) {
+                    } catch (err: any) {
                         return errorHandler(err, res);
                     }
                 });
@@ -200,7 +200,7 @@ export async function makeApp(
                     await adapter.process(req, socket, head, async (context) => {
                         await bot.run(context);
                     });
-                } catch (err) {
+                } catch (err: any) {
                     return errorHandler(err);
                 }
             });

libraries/botbuilder-dialogs-adaptive-runtime-integration-restify/src/index.ts

@@ -140,7 +140,7 @@ export async function makeServer(
             await adapter.process(req, res, async (turnContext) => {
                 await bot.run(turnContext);
             });
-        } catch (err) {
+        } catch (err: any) {
             return errorHandler(err, res);
         }
     });
@@ -169,7 +169,7 @@ export async function makeServer(
                         await adapter.process(req, res, async (turnContext) => {
                             await bot.run(turnContext);
                         });
-                    } catch (err) {
+                    } catch (err: any) {
                         return errorHandler(err, res);
                     }
                 });
@@ -197,7 +197,7 @@ export async function makeServer(
             await adapter.process(req, socket, head, async (context) => {
                 await bot.run(context);
             });
-        } catch (err) {
+        } catch (err: any) {
             return errorHandler(err);
         }
     });

libraries/botbuilder-dialogs-adaptive-runtime/src/index.ts

@@ -442,7 +442,7 @@ async function addSettingsBotComponents(services: ServiceCollection, configurati
             const botComponent = await loadBotComponent(name);
 
             botComponent.configureServices(services, configuration.bind([settingsPrefix ?? name]));
-        } catch (error) {
+        } catch (error: any) {
             loadErrors.push({ error, name });
         }
     }

libraries/botbuilder-stdlib/src/retry.ts

@@ -25,7 +25,7 @@ export async function retry<T>(
         try {
             // Note: return await intentional so we can catch errors
             return await promise(n);
-        } catch (err) {
+        } catch (err: any) {
             maybeError = err;
 
             await new Promise((resolve) => setTimeout(resolve, delay));

package.json

@@ -91,11 +91,11 @@
     "source-map-support": "^0.5.19",
     "sponge": "^0.1.0",
     "tinyify": "^3.0.0",
-    "ts-node": "^9.0.0",
+    "ts-node": "^10.0.0",
     "typedoc": "^0.19.2",
     "typedoc-plugin-external-module-name": "^4.0.3",
     "typedoc-plugin-markdown": "^3.0.11",
-    "typescript": "^4.0.5",
+    "typescript": "^4.1.0",
     "wsrun": "^5.2.4"
   },
   "nyc": {