From f0e32491d7a740c0f50370abc34a02515eda5c3a Mon Sep 17 00:00:00 2001
From: Asier Isayas <aisayas@microsoft.com>
Date: Wed, 19 Feb 2025 15:16:32 -0500
Subject: [PATCH] Partition Key, Throughput, Unique Keys

---
 .../AddCollectionPanel/AddCollectionPanel.tsx | 40 ++------
 ...ility.ts => AddCollectionPanelUtility.tsx} | 22 +++++
 ...Key.tsx => AddMVPartitionKeyComponent.tsx} |  6 +-
 .../AddMVThroughput.tsx                       | 10 --
 .../AddMVThroughputComponent.tsx              | 91 +++++++++++++++++++
 .../AddMVUniqueKeysComponent.tsx              | 78 ++++++++++++++++
 .../AddMaterializedViewPanel.tsx              | 41 ++++++++-
 7 files changed, 241 insertions(+), 47 deletions(-)
 rename src/Explorer/Panes/AddCollectionPanel/{AddCollectionPanelUtility.ts => AddCollectionPanelUtility.tsx} (67%)
 rename src/Explorer/Panes/AddMaterializedViewPanel/{AddMVPartitionKey.tsx => AddMVPartitionKeyComponent.tsx} (93%)
 delete mode 100644 src/Explorer/Panes/AddMaterializedViewPanel/AddMVThroughput.tsx
 create mode 100644 src/Explorer/Panes/AddMaterializedViewPanel/AddMVThroughputComponent.tsx
 create mode 100644 src/Explorer/Panes/AddMaterializedViewPanel/AddMVUniqueKeysComponent.tsx

diff --git a/src/Explorer/Panes/AddCollectionPanel/AddCollectionPanel.tsx b/src/Explorer/Panes/AddCollectionPanel/AddCollectionPanel.tsx
index 897427d02..541254d6e 100644
--- a/src/Explorer/Panes/AddCollectionPanel/AddCollectionPanel.tsx
+++ b/src/Explorer/Panes/AddCollectionPanel/AddCollectionPanel.tsx
@@ -31,6 +31,8 @@ import {
   getPartitionKeyName,
   getPartitionKeyPlaceHolder,
   getPartitionKeyTooltipText,
+  isFreeTierAccount,
+  UniqueKeysHeader,
 } from "Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility";
 import { useSidePanel } from "hooks/useSidePanel";
 import { useTeachingBubble } from "hooks/useTeachingBubble";
@@ -196,7 +198,7 @@ export class AddCollectionPanel extends React.Component<AddCollectionPanelProps,
           />
         )}
 
-        {!this.state.errorMessage && this.isFreeTierAccount() && (
+        {!this.state.errorMessage && isFreeTierAccount() && (
           <PanelInfoErrorComponent
             message={getUpsellMessage(userContext.portalEnv, true, isFirstResourceCreated, true)}
             messageType="info"
@@ -403,10 +405,10 @@ export class AddCollectionPanel extends React.Component<AddCollectionPanelProps,
 
                 {!isServerlessAccount() && this.state.isSharedThroughputChecked && (
                   <ThroughputInput
-                    showFreeTierExceedThroughputTooltip={this.isFreeTierAccount() && !isFirstResourceCreated}
+                    showFreeTierExceedThroughputTooltip={isFreeTierAccount() && !isFirstResourceCreated}
                     isDatabase={true}
                     isSharded={this.state.isSharded}
-                    isFreeTier={this.isFreeTierAccount()}
+                    isFreeTier={isFreeTierAccount()}
                     isQuickstart={this.props.isQuickstart}
                     setThroughputValue={(throughput: number) => (this.newDatabaseThroughput = throughput)}
                     setIsAutoscale={(isAutoscale: boolean) => (this.isNewDatabaseAutoscale = isAutoscale)}
@@ -734,10 +736,10 @@ export class AddCollectionPanel extends React.Component<AddCollectionPanelProps,
 
           {this.shouldShowCollectionThroughputInput() && (
             <ThroughputInput
-              showFreeTierExceedThroughputTooltip={this.isFreeTierAccount() && !isFirstResourceCreated}
+              showFreeTierExceedThroughputTooltip={isFreeTierAccount() && !isFirstResourceCreated}
               isDatabase={false}
               isSharded={this.state.isSharded}
-              isFreeTier={this.isFreeTierAccount()}
+              isFreeTier={isFreeTierAccount()}
               isQuickstart={this.props.isQuickstart}
               setThroughputValue={(throughput: number) => (this.collectionThroughput = throughput)}
               setIsAutoscale={(isAutoscale: boolean) => (this.isCollectionAutoscale = isAutoscale)}
@@ -752,27 +754,7 @@ export class AddCollectionPanel extends React.Component<AddCollectionPanelProps,
 
           {userContext.apiType === "SQL" && (
             <Stack>
-              <Stack horizontal>
-                <Text className="panelTextBold" variant="small">
-                  Unique keys
-                </Text>
-                <TooltipHost
-                  directionalHint={DirectionalHint.bottomLeftEdge}
-                  content={
-                    "Unique keys provide developers with the ability to add a layer of data integrity to their database. By creating a unique key policy when a container is created, you ensure the uniqueness of one or more values per partition key."
-                  }
-                >
-                  <Icon
-                    iconName="Info"
-                    className="panelInfoIcon"
-                    tabIndex={0}
-                    ariaLabel={
-                      "Unique keys provide developers with the ability to add a layer of data integrity to their database. By creating a unique key policy when a container is created, you ensure the uniqueness of one or more values per partition key."
-                    }
-                  />
-                </TooltipHost>
-              </Stack>
-
+              {UniqueKeysHeader()}
               {this.state.uniqueKeys.map((uniqueKey: string, i: number): JSX.Element => {
                 return (
                   <Stack style={{ marginBottom: 8 }} key={`uniqueKey${i}`} horizontal>
@@ -1143,10 +1125,6 @@ export class AddCollectionPanel extends React.Component<AddCollectionPanelProps,
     return !!selectedDatabase?.offer();
   }
 
-  private isFreeTierAccount(): boolean {
-    return userContext.databaseAccount?.properties?.enableFreeTier;
-  }
-
   private getFreeTierIndexingText(): string {
     return this.state.enableIndexing
       ? "All properties in your documents will be indexed by default for flexible and efficient queries."
@@ -1222,7 +1200,7 @@ export class AddCollectionPanel extends React.Component<AddCollectionPanelProps,
   }
 
   private shouldShowIndexingOptionsForFreeTierAccount(): boolean {
-    if (!this.isFreeTierAccount()) {
+    if (!isFreeTierAccount()) {
       return false;
     }
 
diff --git a/src/Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility.ts b/src/Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility.tsx
similarity index 67%
rename from src/Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility.ts
rename to src/Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility.tsx
index a5b5e5956..c7e150a68 100644
--- a/src/Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility.ts
+++ b/src/Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility.tsx
@@ -1,3 +1,5 @@
+import { DirectionalHint, Icon, Stack, Text, TooltipHost } from "@fluentui/react";
+import React from "react";
 import { userContext } from "UserContext";
 
 export function getPartitionKeyTooltipText(): string {
@@ -56,3 +58,23 @@ export function getPartitionKey(isQuickstart?: boolean): string {
   }
   return "";
 }
+
+export function isFreeTierAccount(): boolean {
+  return userContext.databaseAccount?.properties?.enableFreeTier;
+}
+
+export function UniqueKeysHeader(): JSX.Element {
+  const tooltipContent =
+    "Unique keys provide developers with the ability to add a layer of data integrity to their database. By creating a unique key policy when a container is created, you ensure the uniqueness of one or more values per partition key.";
+
+  return (
+    <Stack horizontal>
+      <Text className="panelTextBold" variant="small">
+        Unique keys
+      </Text>
+      <TooltipHost directionalHint={DirectionalHint.bottomLeftEdge} content={tooltipContent}>
+        <Icon iconName="Info" className="panelInfoIcon" tabIndex={0} ariaLabel={tooltipContent} />
+      </TooltipHost>
+    </Stack>
+  );
+}
diff --git a/src/Explorer/Panes/AddMaterializedViewPanel/AddMVPartitionKey.tsx b/src/Explorer/Panes/AddMaterializedViewPanel/AddMVPartitionKeyComponent.tsx
similarity index 93%
rename from src/Explorer/Panes/AddMaterializedViewPanel/AddMVPartitionKey.tsx
rename to src/Explorer/Panes/AddMaterializedViewPanel/AddMVPartitionKeyComponent.tsx
index 506619de0..bea52d9e9 100644
--- a/src/Explorer/Panes/AddMaterializedViewPanel/AddMVPartitionKey.tsx
+++ b/src/Explorer/Panes/AddMaterializedViewPanel/AddMVPartitionKeyComponent.tsx
@@ -7,7 +7,7 @@ import {
 } from "Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility";
 import React from "react";
 
-export interface AddMVPartitionKeyProps {
+export interface AddMVPartitionKeyComponentProps {
   partitionKey?: string;
   setPartitionKey: React.Dispatch<React.SetStateAction<string>>;
   subPartitionKeys: string[];
@@ -16,7 +16,7 @@ export interface AddMVPartitionKeyProps {
   setUseHashV1: React.Dispatch<React.SetStateAction<boolean>>;
 }
 
-export const AddMVPartitionKey = (props: AddMVPartitionKeyProps): JSX.Element => {
+export const AddMVPartitionKeyComponent = (props: AddMVPartitionKeyComponentProps): JSX.Element => {
   const { partitionKey, setPartitionKey, subPartitionKeys, setSubPartitionKeys, useHashV1, setUseHashV1 } = props;
 
   const partitionKeyValueOnChange = (value: string): void => {
@@ -60,6 +60,7 @@ export const AddMVPartitionKey = (props: AddMVPartitionKeyProps): JSX.Element =>
         aria-label={getPartitionKeyName()}
         pattern=".*"
         value={partitionKey}
+        style={{ marginBottom: 8 }}
         onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
           partitionKeyValueOnChange(event.target.value);
         }}
@@ -88,6 +89,7 @@ export const AddMVPartitionKey = (props: AddMVPartitionKeyProps): JSX.Element =>
               placeholder={getPartitionKeyPlaceHolder(subPartitionKeyIndex)}
               aria-label={getPartitionKeyName()}
               pattern={".*"}
+              title={""}
               value={subPartitionKey}
               onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
                 subPartitionKeysValueOnChange(event.target.value, subPartitionKeyIndex);
diff --git a/src/Explorer/Panes/AddMaterializedViewPanel/AddMVThroughput.tsx b/src/Explorer/Panes/AddMaterializedViewPanel/AddMVThroughput.tsx
deleted file mode 100644
index 036ffdab0..000000000
--- a/src/Explorer/Panes/AddMaterializedViewPanel/AddMVThroughput.tsx
+++ /dev/null
@@ -1,10 +0,0 @@
-import { Stack } from "@fluentui/react";
-import React from "react";
-
-export interface AddMVThroughputProps {
-  // isQuickstart: boolean;
-}
-
-export const AddMVThroughput = (props: AddMVThroughputProps): JSX.Element => {
-  return <Stack></Stack>;
-};
diff --git a/src/Explorer/Panes/AddMaterializedViewPanel/AddMVThroughputComponent.tsx b/src/Explorer/Panes/AddMaterializedViewPanel/AddMVThroughputComponent.tsx
new file mode 100644
index 000000000..0dd40cf24
--- /dev/null
+++ b/src/Explorer/Panes/AddMaterializedViewPanel/AddMVThroughputComponent.tsx
@@ -0,0 +1,91 @@
+import { Checkbox, Stack } from "@fluentui/react";
+import { Collection } from "Contracts/ViewModels";
+import { ThroughputInput } from "Explorer/Controls/ThroughputInput/ThroughputInput";
+import { isFreeTierAccount } from "Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility";
+import { useDatabases } from "Explorer/useDatabases";
+import React from "react";
+import { getCollectionName } from "Utils/APITypeUtils";
+import { isServerlessAccount } from "Utils/CapabilityUtils";
+
+export interface AddMVThroughputComponentProps {
+  selectedSourceContainer: Collection;
+  enableDedicatedThroughput: boolean;
+  setEnabledDedicatedThroughput: React.Dispatch<React.SetStateAction<boolean>>;
+  materializedViewThroughputOnChange: (materializedViewThroughputValue: number) => void;
+  isMaterializedViewAutoscaleOnChange: (isMaterializedViewAutoscaleValue: boolean) => void;
+  setIsThroughputCapExceeded: React.Dispatch<React.SetStateAction<boolean>>;
+  isCostAknowledgedOnChange: (isCostAknowledgedValue: boolean) => void;
+}
+
+export const AddMVThroughputComponent = (props: AddMVThroughputComponentProps): JSX.Element => {
+  const {
+    selectedSourceContainer,
+    enableDedicatedThroughput,
+    setEnabledDedicatedThroughput,
+    materializedViewThroughputOnChange,
+    isMaterializedViewAutoscaleOnChange,
+    setIsThroughputCapExceeded,
+    isCostAknowledgedOnChange,
+  } = props;
+
+  const isSelectedSourceContainerSharedThroughput = (): boolean => {
+    if (!selectedSourceContainer) {
+      return false;
+    }
+
+    return !!selectedSourceContainer.getDatabase().offer();
+  };
+
+  const showCollectionThroughputInput = (): boolean => {
+    if (isServerlessAccount()) {
+      return false;
+    }
+
+    if (enableDedicatedThroughput) {
+      return true;
+    }
+
+    return !!selectedSourceContainer && !isSelectedSourceContainerSharedThroughput();
+  };
+
+  return (
+    <Stack>
+      {!isServerlessAccount() && isSelectedSourceContainerSharedThroughput() && (
+        <Stack horizontal verticalAlign="center">
+          <Checkbox
+            label={`Provision dedicated throughput for this ${getCollectionName().toLocaleLowerCase()}`}
+            checked={enableDedicatedThroughput}
+            styles={{
+              text: { fontSize: 12 },
+              checkbox: { width: 12, height: 12 },
+              label: { padding: 0, alignItems: "center" },
+            }}
+            onChange={(_, isChecked: boolean) => setEnabledDedicatedThroughput(isChecked)}
+          />
+        </Stack>
+      )}
+      {showCollectionThroughputInput() && (
+        <ThroughputInput
+          showFreeTierExceedThroughputTooltip={isFreeTierAccount() && !useDatabases.getState().isFirstResourceCreated()}
+          isDatabase={false}
+          // isSharded={this.state.isSharded}
+          isSharded={false}
+          isFreeTier={isFreeTierAccount()}
+          isQuickstart={false}
+          setThroughputValue={(throughput: number) => {
+            materializedViewThroughputOnChange(throughput);
+          }}
+          setIsAutoscale={(isAutoscale: boolean) => {
+            isMaterializedViewAutoscaleOnChange(isAutoscale);
+          }}
+          setIsThroughputCapExceeded={(isThroughputCapExceeded: boolean) => {
+            setIsThroughputCapExceeded(isThroughputCapExceeded);
+          }}
+          onCostAcknowledgeChange={(isAcknowledged: boolean) => {
+            isCostAknowledgedOnChange(isAcknowledged);
+          }}
+        />
+      )}
+    </Stack>
+  );
+};
diff --git a/src/Explorer/Panes/AddMaterializedViewPanel/AddMVUniqueKeysComponent.tsx b/src/Explorer/Panes/AddMaterializedViewPanel/AddMVUniqueKeysComponent.tsx
new file mode 100644
index 000000000..6c113b5f0
--- /dev/null
+++ b/src/Explorer/Panes/AddMaterializedViewPanel/AddMVUniqueKeysComponent.tsx
@@ -0,0 +1,78 @@
+import { ActionButton, IconButton, Stack } from "@fluentui/react";
+import { UniqueKeysHeader } from "Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility";
+import React from "react";
+import { userContext } from "UserContext";
+
+export interface AddMVUniqueKeysComponentProps {
+  uniqueKeys: string[];
+  setUniqueKeys: React.Dispatch<React.SetStateAction<string[]>>;
+}
+
+export const AddMVUniqueKeysComponent = (props: AddMVUniqueKeysComponentProps): JSX.Element => {
+  const { uniqueKeys, setUniqueKeys } = props;
+
+  const updateUniqueKeysOnChange = (value: string, uniqueKeyToReplaceIndex: number): void => {
+    const updatedUniqueKeys = uniqueKeys.map((uniqueKey: string, uniqueKeyIndex: number) => {
+      if (uniqueKeyToReplaceIndex === uniqueKeyIndex) {
+        return value;
+      }
+      return uniqueKey;
+    });
+    setUniqueKeys(updatedUniqueKeys);
+  };
+
+  const deleteUniqueKeyOnClick = (uniqueKeyToDeleteIndex: number): void => {
+    const updatedUniqueKeys = uniqueKeys.filter((_, uniqueKeyIndex) => uniqueKeyToDeleteIndex !== uniqueKeyIndex);
+    setUniqueKeys(updatedUniqueKeys);
+  };
+
+  const addUniqueKeyOnClick = (): void => {
+    setUniqueKeys([...uniqueKeys, ""]);
+  };
+
+  return (
+    <Stack>
+      {UniqueKeysHeader()}
+
+      {uniqueKeys.map((uniqueKey: string, uniqueKeyIndex: number): JSX.Element => {
+        return (
+          <Stack style={{ marginBottom: 8 }} key={`uniqueKey-${uniqueKeyIndex}`} horizontal>
+            <input
+              type="text"
+              autoComplete="off"
+              placeholder={
+                userContext.apiType === "Mongo"
+                  ? "Comma separated paths e.g. firstName,address.zipCode"
+                  : "Comma separated paths e.g. /firstName,/address/zipCode"
+              }
+              className="panelTextField"
+              autoFocus
+              value={uniqueKey}
+              onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
+                updateUniqueKeysOnChange(event.target.value, uniqueKeyIndex);
+              }}
+            />
+
+            <IconButton
+              iconProps={{ iconName: "Delete" }}
+              style={{ height: 27 }}
+              onClick={() => {
+                deleteUniqueKeyOnClick(uniqueKeyIndex);
+              }}
+            />
+          </Stack>
+        );
+      })}
+
+      <ActionButton
+        iconProps={{ iconName: "Add" }}
+        styles={{ root: { padding: 0 }, label: { fontSize: 12 } }}
+        onClick={() => {
+          addUniqueKeyOnClick();
+        }}
+      >
+        Add unique key
+      </ActionButton>
+    </Stack>
+  );
+};
diff --git a/src/Explorer/Panes/AddMaterializedViewPanel/AddMaterializedViewPanel.tsx b/src/Explorer/Panes/AddMaterializedViewPanel/AddMaterializedViewPanel.tsx
index 6e36e0c16..6a4d5daa5 100644
--- a/src/Explorer/Panes/AddMaterializedViewPanel/AddMaterializedViewPanel.tsx
+++ b/src/Explorer/Panes/AddMaterializedViewPanel/AddMaterializedViewPanel.tsx
@@ -13,8 +13,9 @@ import {
 import { Collection, Database } from "Contracts/ViewModels";
 import Explorer from "Explorer/Explorer";
 import { getPartitionKey } from "Explorer/Panes/AddCollectionPanel/AddCollectionPanelUtility";
-import { AddMVPartitionKey } from "Explorer/Panes/AddMaterializedViewPanel/AddMVPartitionKey";
-import { AddMVThroughput } from "Explorer/Panes/AddMaterializedViewPanel/AddMVThroughput";
+import { AddMVPartitionKeyComponent } from "Explorer/Panes/AddMaterializedViewPanel/AddMVPartitionKeyComponent";
+import { AddMVThroughputComponent } from "Explorer/Panes/AddMaterializedViewPanel/AddMVThroughputComponent";
+import { AddMVUniqueKeysComponent } from "Explorer/Panes/AddMaterializedViewPanel/AddMVUniqueKeysComponent";
 import { useDatabases } from "Explorer/useDatabases";
 import React, { useEffect, useState } from "react";
 
@@ -32,6 +33,9 @@ export const AddMaterializedViewPanel = (props: AddMaterializedViewPanelProps):
   const [partitionKey, setPartitionKey] = useState<string>(getPartitionKey());
   const [subPartitionKeys, setSubPartitionKeys] = useState<string[]>([]);
   const [useHashV1, setUseHashV1] = useState<boolean>();
+  const [enableDedicatedThroughput, setEnabledDedicatedThroughput] = useState<boolean>();
+  const [isThroughputCapExceeded, setIsThroughputCapExceeded] = useState<boolean>();
+  const [uniqueKeys, setUniqueKeys] = useState<string[]>([]);
 
   useEffect(() => {
     const sourceContainerOptions: IDropdownOption[] = [];
@@ -59,6 +63,22 @@ export const AddMaterializedViewPanel = (props: AddMaterializedViewPanelProps):
     setSourceContainerOptions(sourceContainerOptions);
   }, []);
 
+  let materializedViewThroughput: number;
+  let isMaterializedViewAutoscale: boolean;
+  let isCostAcknowledged: boolean;
+
+  const materializedViewThroughputOnChange = (materializedViewThroughputValue: number): void => {
+    materializedViewThroughput = materializedViewThroughputValue;
+  };
+
+  const isMaterializedViewAutoscaleOnChange = (isMaterializedViewAutoscaleValue: boolean): void => {
+    isMaterializedViewAutoscale = isMaterializedViewAutoscaleValue;
+  };
+
+  const isCostAknowledgedOnChange = (isCostAcknowledgedValue: boolean): void => {
+    isCostAcknowledged = isCostAcknowledgedValue;
+  };
+
   return (
     <form className="panelFormWrapper" id="panelMaterializedView">
       <div className="panelMainContent">
@@ -129,8 +149,21 @@ export const AddMaterializedViewPanel = (props: AddMaterializedViewPanelProps):
             value={materializedViewDefinition}
             onChange={(event: React.ChangeEvent<HTMLInputElement>) => setMaterializedViewDefinition(event.target.value)}
           />
-          <AddMVPartitionKey {...{ partitionKey, setPartitionKey, subPartitionKeys, setSubPartitionKeys }} />
-          <AddMVThroughput />
+          <AddMVPartitionKeyComponent
+            {...{ partitionKey, setPartitionKey, subPartitionKeys, setSubPartitionKeys, useHashV1, setUseHashV1 }}
+          />
+          <AddMVThroughputComponent
+            {...{
+              selectedSourceContainer,
+              enableDedicatedThroughput,
+              setEnabledDedicatedThroughput,
+              materializedViewThroughputOnChange,
+              isMaterializedViewAutoscaleOnChange,
+              setIsThroughputCapExceeded,
+              isCostAknowledgedOnChange,
+            }}
+          />
+          <AddMVUniqueKeysComponent {...{ uniqueKeys, setUniqueKeys }} />
         </Stack>
       </div>
     </form>