From bf4e2b17270acdddf845f7232aac0077b792926c Mon Sep 17 00:00:00 2001
From: Tyler Ohlsen <ohltyler@amazon.com>
Date: Fri, 14 Jun 2024 10:48:10 -0700
Subject: [PATCH] Add stepped component in input form

Signed-off-by: Tyler Ohlsen <ohltyler@amazon.com>
---
 .../workflow_inputs/workflow_inputs.tsx       | 39 ++++++++++++-------
 1 file changed, 25 insertions(+), 14 deletions(-)

diff --git a/public/pages/workflow_detail/workflow_inputs/workflow_inputs.tsx b/public/pages/workflow_detail/workflow_inputs/workflow_inputs.tsx
index 208dd1dc..058c0a90 100644
--- a/public/pages/workflow_detail/workflow_inputs/workflow_inputs.tsx
+++ b/public/pages/workflow_detail/workflow_inputs/workflow_inputs.tsx
@@ -14,7 +14,7 @@ import {
   EuiHorizontalRule,
   EuiLoadingSpinner,
   EuiPanel,
-  EuiTitle,
+  EuiStepsHorizontal,
 } from '@elastic/eui';
 import {
   Workflow,
@@ -46,9 +46,9 @@ interface WorkflowInputsProps {
   setIngestResponse: (ingestResponse: string) => void;
 }
 
-export enum CREATE_STEP {
-  INGEST = 'Step 1: Define ingestion pipeline',
-  SEARCH = 'Step 2: Define search pipeline',
+export enum STEP {
+  INGEST = 'Ingestion pipeline',
+  SEARCH = 'Search pipeline',
 }
 
 /**
@@ -63,9 +63,7 @@ export function WorkflowInputs(props: WorkflowInputsProps) {
   const dispatch = useAppDispatch();
 
   // selected step state
-  const [selectedStep, setSelectedStep] = useState<CREATE_STEP>(
-    CREATE_STEP.INGEST
-  );
+  const [selectedStep, setSelectedStep] = useState<STEP>(STEP.INGEST);
 
   // ingest state
   const [ingestDocs, setIngestDocs] = useState<{}[]>([]);
@@ -201,9 +199,22 @@ export function WorkflowInputs(props: WorkflowInputsProps) {
           }}
         >
           <EuiFlexItem grow={false}>
-            <EuiTitle size="s">
-              <h4>{selectedStep}</h4>
-            </EuiTitle>
+            <EuiStepsHorizontal
+              steps={[
+                {
+                  title: STEP.INGEST,
+                  isComplete: selectedStep === STEP.SEARCH,
+                  isSelected: selectedStep === STEP.INGEST,
+                  onClick: () => {},
+                },
+                {
+                  title: STEP.SEARCH,
+                  isComplete: false,
+                  isSelected: selectedStep === STEP.SEARCH,
+                  onClick: () => {},
+                },
+              ]}
+            ></EuiStepsHorizontal>
           </EuiFlexItem>
           <EuiFlexItem
             grow={true}
@@ -212,7 +223,7 @@ export function WorkflowInputs(props: WorkflowInputsProps) {
               overflowX: 'hidden',
             }}
           >
-            {selectedStep === CREATE_STEP.INGEST ? (
+            {selectedStep === STEP.INGEST ? (
               <IngestInputs
                 onFormChange={props.onFormChange}
                 ingestDocs={ingestDocs}
@@ -231,11 +242,11 @@ export function WorkflowInputs(props: WorkflowInputsProps) {
               </EuiFlexItem>
               <EuiFlexItem>
                 <EuiFlexGroup direction="row" justifyContent="flexEnd">
-                  {selectedStep === CREATE_STEP.INGEST ? (
+                  {selectedStep === STEP.INGEST ? (
                     <>
                       <EuiFlexItem grow={false}>
                         <EuiButtonEmpty
-                          onClick={() => setSelectedStep(CREATE_STEP.SEARCH)}
+                          onClick={() => setSelectedStep(STEP.SEARCH)}
                         >
                           Skip
                         </EuiButtonEmpty>
@@ -255,7 +266,7 @@ export function WorkflowInputs(props: WorkflowInputsProps) {
                     <>
                       <EuiFlexItem grow={false}>
                         <EuiButtonEmpty
-                          onClick={() => setSelectedStep(CREATE_STEP.INGEST)}
+                          onClick={() => setSelectedStep(STEP.INGEST)}
                         >
                           Back
                         </EuiButtonEmpty>