From dda648fa27fd608555deb57037eafb95a4f18225 Mon Sep 17 00:00:00 2001
From: Chuck Skelton <chskelt@microsoft.com>
Date: Mon, 24 Jul 2023 17:05:23 -0700
Subject: [PATCH] Initial commit

---
 package-lock.json                             | 19 +++++++++++--------
 package.json                                  |  2 +-
 src/Common/IteratorUtilities.ts               |  2 ++
 src/Contracts/ViewModels.ts                   |  1 +
 .../Tabs/QueryTab/QueryTabComponent.tsx       |  2 ++
 5 files changed, 17 insertions(+), 9 deletions(-)

diff --git a/package-lock.json b/package-lock.json
index 0a05d1161..7ddb68245 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -9,7 +9,7 @@
       "version": "1.0.0",
       "dependencies": {
         "@azure/arm-cosmosdb": "9.1.0",
-        "@azure/cosmos": "3.16.2",
+        "@azure/cosmos": "3.17.4-beta.1",
         "@azure/cosmos-language-service": "0.0.5",
         "@azure/identity": "1.2.1",
         "@azure/ms-rest-nodeauth": "3.0.7",
@@ -396,10 +396,12 @@
       "integrity": "sha512-d6xOpEDfsi2CZVlPQzGeux8XMwLT9hssAsaPYExaQMuYskwb+x1x7J371tWlbBdWHroy99KnVB6qIkUbs5X3UQ=="
     },
     "node_modules/@azure/cosmos": {
-      "version": "3.16.2",
-      "resolved": "https://registry.npmjs.org/@azure/cosmos/-/cosmos-3.16.2.tgz",
-      "integrity": "sha512-sceY5LWj0BHGj8PSyaVCfDRQLVZyoCfIY78kyIROJVEw0k+p9XFs8fhpykN8JklkCftL0WlaVY+X25SQwnhZsw==",
+      "version": "3.17.4-beta.1",
+      "resolved": "https://msazure.pkgs.visualstudio.com/_packaging/AzurePortal/npm/registry/@azure/cosmos/-/cosmos-3.17.4-beta.1.tgz",
+      "integrity": "sha1-axE6QafE98LT4IOEeug1NB2hC74=",
+      "license": "MIT",
       "dependencies": {
+        "@azure/abort-controller": "^1.0.0",
         "@azure/core-auth": "^1.3.0",
         "@azure/core-rest-pipeline": "^1.2.0",
         "@azure/core-tracing": "^1.0.0",
@@ -414,7 +416,7 @@
         "uuid": "^8.3.0"
       },
       "engines": {
-        "node": ">=12.0.0"
+        "node": ">=14.0.0"
       }
     },
     "node_modules/@azure/cosmos-language-service": {
@@ -31470,10 +31472,11 @@
       }
     },
     "@azure/cosmos": {
-      "version": "3.16.2",
-      "resolved": "https://registry.npmjs.org/@azure/cosmos/-/cosmos-3.16.2.tgz",
-      "integrity": "sha512-sceY5LWj0BHGj8PSyaVCfDRQLVZyoCfIY78kyIROJVEw0k+p9XFs8fhpykN8JklkCftL0WlaVY+X25SQwnhZsw==",
+      "version": "3.17.4-beta.1",
+      "resolved": "https://msazure.pkgs.visualstudio.com/_packaging/AzurePortal/npm/registry/@azure/cosmos/-/cosmos-3.17.4-beta.1.tgz",
+      "integrity": "sha1-axE6QafE98LT4IOEeug1NB2hC74=",
       "requires": {
+        "@azure/abort-controller": "^1.0.0",
         "@azure/core-auth": "^1.3.0",
         "@azure/core-rest-pipeline": "^1.2.0",
         "@azure/core-tracing": "^1.0.0",
diff --git a/package.json b/package.json
index fd01978f6..4558b65aa 100644
--- a/package.json
+++ b/package.json
@@ -5,7 +5,7 @@
   "main": "index.js",
   "dependencies": {
     "@azure/arm-cosmosdb": "9.1.0",
-    "@azure/cosmos": "3.16.2",
+    "@azure/cosmos": "3.17.4-beta.1",
     "@azure/cosmos-language-service": "0.0.5",
     "@azure/identity": "1.2.1",
     "@azure/ms-rest-nodeauth": "3.0.7",
diff --git a/src/Common/IteratorUtilities.ts b/src/Common/IteratorUtilities.ts
index f85ad7fb2..b0088461c 100644
--- a/src/Common/IteratorUtilities.ts
+++ b/src/Common/IteratorUtilities.ts
@@ -7,6 +7,7 @@ interface QueryResponse {
   hasMoreResults: boolean;
   activityId: string;
   requestCharge: number;
+  indexMetrics: string;
 }
 
 export interface MinimalQueryIterator {
@@ -30,6 +31,7 @@ export function nextPage(documentsIterator: MinimalQueryIterator, firstItemIndex
       headers,
       activityId: response.activityId,
       requestCharge: response.requestCharge,
+      indexMetrics: response.indexMetrics,
     };
   });
 }
diff --git a/src/Contracts/ViewModels.ts b/src/Contracts/ViewModels.ts
index c9bb351bc..c910cbb06 100644
--- a/src/Contracts/ViewModels.ts
+++ b/src/Contracts/ViewModels.ts
@@ -45,6 +45,7 @@ export interface QueryResults extends QueryResultsMetadata {
   roundTrips?: number;
   headers?: any;
   queryMetrics?: QueryMetrics;
+  indexMetrics?: string;
 }
 
 export interface Button {
diff --git a/src/Explorer/Tabs/QueryTab/QueryTabComponent.tsx b/src/Explorer/Tabs/QueryTab/QueryTabComponent.tsx
index 661ec7c60..66d80f553 100644
--- a/src/Explorer/Tabs/QueryTab/QueryTabComponent.tsx
+++ b/src/Explorer/Tabs/QueryTab/QueryTabComponent.tsx
@@ -219,10 +219,12 @@ export default class QueryTabComponent extends React.Component<IQueryTabComponen
     });
 
     try {
+      //CTODO: Add index metrics to the following method?
       const queryResults: ViewModels.QueryResults = await QueryUtils.queryPagesUntilContentPresent(
         firstItemIndex,
         queryDocuments
       );
+      console.log("Index Metrics: " + queryResults.indexMetrics);
       this.setState({ queryResults, error: "" });
     } catch (error) {
       this.props.tabsBaseInstance.isExecutionError(true);