Add Sample Data to Resource Tree (#1499)

* Add Sample Data to Resource Tree

* Format

* Fix strict build

* Fix lint

* Fixed implementation to show Sample data container

* Udated logic based on TokenCollection

* Re-configure copilot flag

---------

Co-authored-by: Predrag Klepic <v-prklepic@microsoft.com>

src/Common/SampleDataClient.ts

@@ -0,0 +1,26 @@
+import * as Cosmos from "@azure/cosmos";
+import { userContext } from "UserContext";
+
+let _sampleDataclient: Cosmos.CosmosClient;
+
+export function sampleDataClient(): Cosmos.CosmosClient {
+  if (_sampleDataclient) {
+    return _sampleDataclient;
+  }
+
+  const sampleDataConnectionInfo = userContext.sampleDataConnectionInfo;
+  const options: Cosmos.CosmosClientOptions = {
+    endpoint: sampleDataConnectionInfo.accountEndpoint,
+    tokenProvider: async () => {
+      const sampleDataConnectionInfo = userContext.sampleDataConnectionInfo;
+      return Promise.resolve(sampleDataConnectionInfo.resourceToken);
+    },
+    connectionPolicy: {
+      enableEndpointDiscovery: false,
+    },
+    userAgentSuffix: "Azure Portal",
+  };
+
+  _sampleDataclient = new Cosmos.CosmosClient(options);
+  return _sampleDataclient;
+}

src/Common/dataAccess/readCollection.ts

@@ -1,13 +1,39 @@
+import { CosmosClient } from "@azure/cosmos";
+import { sampleDataClient } from "Common/SampleDataClient";
+import { userContext } from "UserContext";
 import * as DataModels from "../../Contracts/DataModels";
+import { logConsoleProgress } from "../../Utils/NotificationConsoleUtils";
 import { client } from "../CosmosClient";
 import { handleError } from "../ErrorHandlingUtils";
-import { logConsoleProgress } from "../../Utils/NotificationConsoleUtils";
 
 export async function readCollection(databaseId: string, collectionId: string): Promise<DataModels.Collection> {
+  const cosmosClient = client();
+  return await readCollectionInternal(cosmosClient, databaseId, collectionId);
+}
+
+export async function readSampleCollection(): Promise<DataModels.Collection> {
+  const cosmosClient = sampleDataClient();
+
+  const sampleDataConnectionInfo = userContext.sampleDataConnectionInfo;
+  const databaseId = sampleDataConnectionInfo?.databaseId;
+  const collectionId = sampleDataConnectionInfo?.collectionId;
+
+  if (!databaseId || !collectionId) {
+    return undefined;
+  }
+
+  return await readCollectionInternal(cosmosClient, databaseId, collectionId);
+}
+
+export async function readCollectionInternal(
+  cosmosClient: CosmosClient,
+  databaseId: string,
+  collectionId: string
+): Promise<DataModels.Collection> {
   let collection: DataModels.Collection;
   const clearMessage = logConsoleProgress(`Querying container ${collectionId}`);
   try {
-    const response = await client().database(databaseId).container(collectionId).read();
+    const response = await cosmosClient.database(databaseId).container(collectionId).read();
     collection = response.resource as DataModels.Collection;
   } catch (error) {
     handleError(error, "ReadCollection", `Error while querying container ${collectionId}`);