From bc951b4e13d4f26089fe83b8891cfb5a1a9097a9 Mon Sep 17 00:00:00 2001
From: Darko Radivojevic <v-darkora@microsoft.com>
Date: Mon, 19 Jun 2023 13:31:49 +0200
Subject: [PATCH] Add button to route to copilot tab

---
 images/CopilotQueryTablesTab.svg  |  3 +++
 src/Explorer/Tabs/DocumentsTab.ts | 30 ++++++++++++++++++++++++++++++
 2 files changed, 33 insertions(+)
 create mode 100644 images/CopilotQueryTablesTab.svg

diff --git a/images/CopilotQueryTablesTab.svg b/images/CopilotQueryTablesTab.svg
new file mode 100644
index 000000000..16cee2a53
--- /dev/null
+++ b/images/CopilotQueryTablesTab.svg
@@ -0,0 +1,3 @@
+<svg width="20" height="22" viewBox="0 0 20 29" fill="none" xmlns="http://www.w3.org/2000/svg">
+<path d="M12.5849 1.28251C11.0125 0.262601 8.98746 0.2626 7.41509 1.28251L2.16509 4.68792C0.8149 5.56372 0 7.06362 0 8.67298V13.3266C0 14.9359 0.814898 16.4358 2.16509 17.3116L7.41509 20.717C8.98746 21.737 11.0125 21.737 12.5849 20.717L17.8349 17.3116C19.1851 16.4358 20 14.9359 20 13.3266V8.67298C20 7.06362 19.1851 5.56372 17.8349 4.68792L12.5849 1.28251ZM9.50097 2.05606C10.2759 1.93579 11.0848 2.09742 11.7686 2.54095L17.0186 5.94636C17.9424 6.54559 18.5 7.57184 18.5 8.67298V11.3953C18.287 11.1924 18.0465 11.0115 17.7802 10.8583L11.8647 7.45689C10.7104 6.79321 9.29086 6.79131 8.13486 7.45187L7.5 7.81465V4.41921C7.5 3.54159 8.01028 2.74407 8.80712 2.3763L9.50097 2.05606ZM17.8644 15.2255L17.7932 15.3501C17.5776 15.6212 17.3172 15.8595 17.0186 16.0532L11.7686 19.4586C10.6928 20.1564 9.30721 20.1564 8.23138 19.4586L5.86807 17.9256C6.11557 17.8358 6.35595 17.719 6.58487 17.5752L12.2457 14.0169C13.3374 13.3307 14 12.1316 14 10.8421V10.415L17.0324 12.1587C18.1077 12.7769 18.4798 14.1486 17.8644 15.2255ZM12.5 9.55251V10.8421C12.5 11.6158 12.1025 12.3352 11.4474 12.747L10.0303 13.6377L8.57078 12.7396C7.90535 12.3301 7.5 11.6047 7.5 10.8233V9.54228L8.87907 8.75424C9.57267 8.3579 10.4244 8.35904 11.1169 8.75725L12.5 9.55251ZM8.61445 14.5277L5.78662 16.3052C5.02045 16.7868 4.04031 16.7625 3.29894 16.2435L2.5521 15.7207C1.88767 15.1109 1.5 14.2448 1.5 13.3266V8.67298C1.5 7.57184 2.05756 6.54559 2.98138 5.94636L6.0268 3.97095C6.00907 4.11852 6 4.26815 6 4.41921V10.8233C6 12.1256 6.67558 13.3345 7.78463 14.017L8.61445 14.5277Z" fill="#0078D4"/>
+</svg>
diff --git a/src/Explorer/Tabs/DocumentsTab.ts b/src/Explorer/Tabs/DocumentsTab.ts
index f75a99fac..a95ed7b2a 100644
--- a/src/Explorer/Tabs/DocumentsTab.ts
+++ b/src/Explorer/Tabs/DocumentsTab.ts
@@ -1,6 +1,8 @@
 import { extractPartitionKey, ItemDefinition, PartitionKeyDefinition, QueryIterator, Resource } from "@azure/cosmos";
+import { ReactTabKind, useTabs } from "hooks/useTabs";
 import * as ko from "knockout";
 import Q from "q";
+import CopilotQueryTablesTab from "../../../images/CopilotQueryTablesTab.svg";
 import DeleteDocumentIcon from "../../../images/DeleteDocument.svg";
 import DiscardIcon from "../../../images/discard.svg";
 import NewDocumentIcon from "../../../images/NewDocument.svg";
@@ -54,6 +56,7 @@ export default class DocumentsTab extends TabsBase {
   public discardNewDocumentChangesButton: ViewModels.Button;
   public discardExisitingDocumentChangesButton: ViewModels.Button;
   public deleteExisitingDocumentButton: ViewModels.Button;
+  public copliotEntityButton: ViewModels.Button;
   public displayedError: ko.Observable<string>;
   public accessibleDocumentList: AccessibleVerticalList;
   public dataContentsGridScrollHeight: ko.Observable<string>;
@@ -304,6 +307,16 @@ export default class DocumentsTab extends TabsBase {
         return true;
       }),
     };
+
+    this.copliotEntityButton = {
+      enabled: ko.computed<boolean>(() => {
+        return userContext.features.enableCopilot;
+      }),
+      visible: ko.computed<boolean>(() => {
+        return userContext.features.enableCopilot;
+      })
+    };
+
     this.buildCommandBarOptions();
     this.shouldShowEditor = ko.computed<boolean>(() => {
       const documentHasContent: boolean =
@@ -768,6 +781,10 @@ export default class DocumentsTab extends TabsBase {
     }
   };
 
+  public onCopilotEntityClick = (): void => {
+    useTabs.getState().openAndActivateReactTab(ReactTabKind.QueryCopilot)
+  }
+
   protected _loadNextPageInternal(): Q.Promise<DataModels.DocumentId[]> {
     return Q(this._documentsIterator.fetchNext().then((response) => response.resources));
   }
@@ -884,6 +901,19 @@ export default class DocumentsTab extends TabsBase {
       buttons.push(DocumentsTab._createUploadButton(this.collection.container));
     }
 
+    if (this.copliotEntityButton.visible()) {
+      const label = "Query Copliot";
+      buttons.push({
+        iconSrc: CopilotQueryTablesTab,
+        iconAlt: label,
+        onCommandClick: this.onCopilotEntityClick,
+        commandButtonLabel: label,
+        ariaLabel: label,
+        hasPopup: true,
+        disabled: !this.copliotEntityButton.enabled(),
+      });
+    }
+
     return buttons;
   }