From 8c1a89403a44672376425b764570a8fead171862 Mon Sep 17 00:00:00 2001
From: Laurent Nguyen <laurent.nguyen@microsoft.com>
Date: Wed, 20 Mar 2024 11:15:25 +0100
Subject: [PATCH] Add react editor and pass column headers

---
 .../Tabs/DocumentsTabV2/DocumentsTabV2.tsx    | 28 +++++++++++++++++--
 .../DocumentsTableComponent.tsx               |  4 +++
 2 files changed, 29 insertions(+), 3 deletions(-)

diff --git a/src/Explorer/Tabs/DocumentsTabV2/DocumentsTabV2.tsx b/src/Explorer/Tabs/DocumentsTabV2/DocumentsTabV2.tsx
index e3309ee86..29cc224f5 100644
--- a/src/Explorer/Tabs/DocumentsTabV2/DocumentsTabV2.tsx
+++ b/src/Explorer/Tabs/DocumentsTabV2/DocumentsTabV2.tsx
@@ -6,6 +6,7 @@ import { getErrorMessage, getErrorStack } from 'Common/ErrorHandlingUtils';
 import { queryDocuments } from 'Common/dataAccess/queryDocuments';
 import { readDocument } from 'Common/dataAccess/readDocument';
 import { useDialog } from 'Explorer/Controls/Dialog';
+import { EditorReact } from 'Explorer/Controls/Editor/EditorReact';
 import { querySampleDocuments, readSampleDocument } from 'Explorer/QueryCopilot/QueryCopilotUtilities';
 import DocumentsTab from 'Explorer/Tabs/DocumentsTab';
 import { dataExplorerLightTheme } from 'Explorer/Theme/ThemeUtil';
@@ -456,6 +457,11 @@ const DocumentsTabComponent: React.FunctionComponent<{
     return () => resizeObserver.disconnect(); // clean up
   }, []);
 
+  const columnHeaders = {
+    idHeader: isPreferredApiMongoDB ? "_id" : "id",
+    pkeyHeaders: partitionKeyPropertyHeaders
+  };
+
   return <FluentProvider theme={dataExplorerLightTheme} style={{ height: "100%" }}>
     <div
       className="tab-pane active"
@@ -569,13 +575,29 @@ const DocumentsTabComponent: React.FunctionComponent<{
         <Split>
           <div style={{ minWidth: 480, width: "20%" }}
             ref={tableContainerRef}>
-            <DocumentsTableComponent style={{ width: 200 }} items={tableItems} onSelectedItem={onSelectedDocument} size={tableContainerSizePx} />
+            <DocumentsTableComponent
+              style={{ width: 200 }}
+              items={tableItems}
+              onSelectedItem={onSelectedDocument}
+              size={tableContainerSizePx}
+              columnHeaders={columnHeaders}
+            />
             <a className="loadMore" role="button" onClick={() => loadNextPage(false)}>Load more</a>
           </div>
-          <div style={{ minWidth: "20%" }}><pre>{JSON.stringify(currentDocument, undefined, " ")}</pre></div>
+          <div style={{ minWidth: "20%", width: "100%" }}>
+            <EditorReact
+              language={"json"}
+              content={JSON.stringify(currentDocument, undefined, "  ")}
+              isReadOnly={false}
+              ariaLabel={"Stored procedure body"}
+              lineNumbers={"on"}
+              theme={"_theme"}
+              onContentChanged={(newContent: string) => { }}
+            />
+          </div>
         </Split>
       </div>
     </div >
-  </FluentProvider>;
+  </FluentProvider >;
 }
 
diff --git a/src/Explorer/Tabs/DocumentsTabV2/DocumentsTableComponent.tsx b/src/Explorer/Tabs/DocumentsTabV2/DocumentsTableComponent.tsx
index ab1db5c13..013af9de6 100644
--- a/src/Explorer/Tabs/DocumentsTabV2/DocumentsTableComponent.tsx
+++ b/src/Explorer/Tabs/DocumentsTabV2/DocumentsTableComponent.tsx
@@ -11,6 +11,10 @@ export interface IDocumentsTableComponentProps {
   items: DocumentsTableComponentItem[];
   onSelectedItem: (index: number) => void;
   size: { height: number; width: number };
+  columnHeaders: {
+    idHeader: string;
+    partitionKeyHeader: string;
+  };
   style?: React.CSSProperties;
 }