From 098b6caf25153b8723f33885be1caee46ad45d4b Mon Sep 17 00:00:00 2001
From: Agnieszka Jarosik <agnieszka.jarosik-loj@pagepro.co>
Date: Thu, 23 Feb 2023 21:32:51 +0100
Subject: [PATCH] fixed types

---
 .../ExpenditurePage/Stages/FormStages.tsx     |  2 +-
 .../ExpenditurePage/Stages/Stages.tsx         |  5 +-
 .../IncorporationPage/IncorporationPage.tsx   | 67 ++++++++-----------
 3 files changed, 33 insertions(+), 41 deletions(-)

diff --git a/src/modules/dashboard/components/ExpenditurePage/Stages/FormStages.tsx b/src/modules/dashboard/components/ExpenditurePage/Stages/FormStages.tsx
index ef3c45ac52..f6573bb4ce 100644
--- a/src/modules/dashboard/components/ExpenditurePage/Stages/FormStages.tsx
+++ b/src/modules/dashboard/components/ExpenditurePage/Stages/FormStages.tsx
@@ -139,7 +139,7 @@ const FormStages = ({
 
     if (!hasErrors) {
       handleSubmit(values as any);
-      activeStage?.buttonAction();
+      activeStage?.buttonAction?.();
     }
   }, [activeStage, handleSubmit, setTouched, validateForm, values]);
 
diff --git a/src/modules/dashboard/components/ExpenditurePage/Stages/Stages.tsx b/src/modules/dashboard/components/ExpenditurePage/Stages/Stages.tsx
index 4782b99681..00d6b8c77f 100644
--- a/src/modules/dashboard/components/ExpenditurePage/Stages/Stages.tsx
+++ b/src/modules/dashboard/components/ExpenditurePage/Stages/Stages.tsx
@@ -76,9 +76,10 @@ export interface Appearance {
   size?: 'small' | 'medium';
 }
 
-export interface StageType extends Omit<StageObject, 'buttonText'> {
+export interface StageType
+  extends Omit<StageObject, 'buttonText' | 'buttonAction'> {
   buttonText?: MessageDescriptor | string;
-  buttonAction: VoidFunction;
+  buttonAction?: VoidFunction;
   description?: MessageDescriptor;
 }
 
diff --git a/src/modules/pages/components/IncorporationPage/IncorporationPage.tsx b/src/modules/pages/components/IncorporationPage/IncorporationPage.tsx
index a8d3e7c059..a0514cefdd 100644
--- a/src/modules/pages/components/IncorporationPage/IncorporationPage.tsx
+++ b/src/modules/pages/components/IncorporationPage/IncorporationPage.tsx
@@ -7,6 +7,7 @@ import { getMainClasses } from '~utils/css';
 import { SpinnerLoader } from '~core/Preloaders';
 import IncorporationForm from '~dashboard/Incorporation/IncorporationForm';
 import Stages, { FormStages } from '~dashboard/ExpenditurePage/Stages';
+import LockedIncorporationForm from '~dashboard/Incorporation/IncorporationForm/LockedIncorporationForm';
 
 import {
   initialValues,
@@ -16,7 +17,6 @@ import {
 } from './constants';
 import { ValuesType } from './types';
 import styles from './IncorporationPage.css';
-import LockedIncorporationForm from '~dashboard/DAOIncorporation/IncorporationForm/LockedIncorporationForm';
 
 const displayName = 'pages.IncorporationPage';
 
@@ -102,38 +102,19 @@ const IncorporationPage = () => {
             <main className={styles.mainContent}>
               <div />
               {colonyData && (
-                <>
-                  {activeStageId === StagesEnum.Draft ? (
-                    <FormStages
-                      activeStageId={activeStageId}
-                      stages={stages.map((stage) => ({
-                        ...stage,
-                        id: stage.id.toString(),
-                        label: stage.title,
-                        buttonAction,
-                      }))}
-                      setActiveStageId={setActiveStageId}
-                      colony={colonyData.processedColony}
-                      setFormValues={setFormValues}
-                      handleCancelExpenditure={() => {}}
-                    />
-                  ) : (
-                    <Stages
-                      activeStageId={activeStageId}
-                      stages={stages.map((stage) => ({
-                        ...stage,
-                        id: stage.id.toString(),
-                        label: stage.title,
-                        buttonAction,
-                      }))}
-                      appearance={{ size: 'medium' }}
-                      handleButtonClick={buttonAction}
-                      handleSaveDraft={handleSubmit}
-                      colony={colonyData?.processedColony}
-                      viewFor="incorporation"
-                    />
-                  )}
-                </>
+                <FormStages
+                  activeStageId={activeStageId}
+                  stages={stages.map((stage) => ({
+                    ...stage,
+                    id: stage.id.toString(),
+                    label: stage.title,
+                    buttonAction,
+                  }))}
+                  setActiveStageId={setActiveStageId}
+                  colony={colonyData.processedColony}
+                  setFormValues={setFormValues}
+                  handleCancelExpenditure={() => {}}
+                />
               )}
             </main>
           </div>
@@ -160,11 +141,21 @@ const IncorporationPage = () => {
       <div className={styles.mainContainer}>
         <main className={styles.mainContent}>
           <div />
-          <Stages
-            activeStageId={activeStageId}
-            stages={stages}
-            buttonAction={buttonAction}
-          />
+          {colonyData && (
+            <Stages
+              activeStageId={activeStageId}
+              stages={stages.map((stage) => ({
+                ...stage,
+                id: stage.id.toString(),
+                label: stage.title,
+                buttonAction,
+              }))}
+              appearance={{ size: 'medium' }}
+              handleButtonClick={buttonAction || (() => {})}
+              colony={colonyData?.processedColony}
+              viewFor="incorporation"
+            />
+          )}
         </main>
       </div>
     </div>