From 04ae1b11f37216b75defafafe14418ca0fcd3c73 Mon Sep 17 00:00:00 2001
From: Gage Krumbach <gkrumbach@gmail.com>
Date: Thu, 7 Mar 2024 10:31:45 -0600
Subject: [PATCH] Add useIsAreaAvailable hook to check if experiments are
 available

---
 .../pipelines/content/createRun/RunPage.tsx       | 15 ++++++++++-----
 .../concepts/pipelines/content/createRun/const.ts |  4 ++--
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/frontend/src/concepts/pipelines/content/createRun/RunPage.tsx b/frontend/src/concepts/pipelines/content/createRun/RunPage.tsx
index 97e3f11d85..308522b900 100644
--- a/frontend/src/concepts/pipelines/content/createRun/RunPage.tsx
+++ b/frontend/src/concepts/pipelines/content/createRun/RunPage.tsx
@@ -25,6 +25,7 @@ import { useGetSearchParamValues } from '~/utilities/useGetSearchParamValues';
 import { PipelineRunSearchParam } from '~/concepts/pipelines/content/types';
 import { PipelineRunType } from '~/pages/pipelines/global/runs';
 import useExperimentById from '~/concepts/pipelines/apiHooks/useExperimentById';
+import { useIsAreaAvailable, SupportedArea } from '~/concepts/areas';
 
 type RunPageProps = {
   cloneRun?: PipelineRunKFv2 | PipelineRunJobKFv2 | null;
@@ -48,6 +49,14 @@ const RunPage: React.FC<RunPageProps> = ({ cloneRun, contextPath, testId }) => {
   const [cloneRunPipeline] = usePipelineById(cloneRunPipelineId);
   const [cloneRunExperiment] = useExperimentById(cloneRunExperimentId);
 
+  const isExperimentsAvailable = useIsAreaAvailable(SupportedArea.PIPELINE_EXPERIMENTS).status;
+
+  const sections = isExperimentsAvailable
+    ? Object.values(CreateRunPageSections)
+    : Object.values(CreateRunPageSections).filter(
+        (section) => section !== CreateRunPageSections.EXPERIMENT,
+      );
+
   const [formData, setFormDataValue] = useRunFormData(cloneRun, {
     runType: {
       ...(runType === PipelineRunType.Scheduled
@@ -73,11 +82,7 @@ const RunPage: React.FC<RunPageProps> = ({ cloneRun, contextPath, testId }) => {
   return (
     <div data-testid={testId}>
       <PageSection isFilled variant="light">
-        <GenericSidebar
-          sections={Object.values(CreateRunPageSections)}
-          titles={runPageSectionTitles}
-          maxWidth={175}
-        >
+        <GenericSidebar sections={sections} titles={runPageSectionTitles} maxWidth={175}>
           <RunForm
             data={formData}
             runType={runType as PipelineRunType}
diff --git a/frontend/src/concepts/pipelines/content/createRun/const.ts b/frontend/src/concepts/pipelines/content/createRun/const.ts
index cac0c4d0a9..b3f72014bd 100644
--- a/frontend/src/concepts/pipelines/content/createRun/const.ts
+++ b/frontend/src/concepts/pipelines/content/createRun/const.ts
@@ -19,16 +19,16 @@ export const RUN_OPTION_LABEL_SIZE = 100;
 
 export enum CreateRunPageSections {
   NAME_DESC = 'run-section-name-desc',
+  EXPERIMENT = 'run-section-experiment',
   PIPELINE = 'run-section-pipeline',
   PIPELINE_VERSION = 'run-section-pipeline-version',
-  EXPERIMENT = 'run-section-experiment',
   PARAMS = 'run-section-params',
 }
 
 export const runPageSectionTitles: Record<CreateRunPageSections, string> = {
   [CreateRunPageSections.NAME_DESC]: 'Name and description',
+  [CreateRunPageSections.EXPERIMENT]: 'Experiment',
   [CreateRunPageSections.PIPELINE]: 'Pipeline',
   [CreateRunPageSections.PIPELINE_VERSION]: 'Pipeline version',
-  [CreateRunPageSections.EXPERIMENT]: 'Experiment',
   [CreateRunPageSections.PARAMS]: 'Pipeline input parameters',
 };