From dca5b9f56c519088d2c8c47192aa9ecf8ad10dd8 Mon Sep 17 00:00:00 2001
From: Bogdan Fazakas <bogdan.fazakas@gmail.com>
Date: Wed, 24 Feb 2021 13:32:15 +0200
Subject: [PATCH] fix persistance for both publish forms

---
 src/components/pages/Publish/index.tsx | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/components/pages/Publish/index.tsx b/src/components/pages/Publish/index.tsx
index 36664181e..fd9ddd030 100644
--- a/src/components/pages/Publish/index.tsx
+++ b/src/components/pages/Publish/index.tsx
@@ -73,6 +73,12 @@ export default function PublishPage({
   const [error, setError] = useState<string>()
   const [title, setTitle] = useState<string>()
   const [did, setDid] = useState<string>()
+  const [algoInitialValues, setAlgoInitialValues] = useState<
+    Partial<MetadataPublishFormAlgorithm>
+  >(initialValuesAlgorithm)
+  const [datasetInitialValues, setdatasetInitialValues] = useState<
+    Partial<MetadataPublishFormDataset>
+  >(initialValues)
   const [publishType, setPublishType] = useState<MetadataMain['type']>(
     'dataset'
   )
@@ -164,7 +170,7 @@ export default function PublishPage({
   return isInPurgatory && purgatoryData ? null : (
     <Formik
       initialValues={
-        publishType === 'dataset' ? initialValues : initialValuesAlgorithm
+        publishType === 'dataset' ? datasetInitialValues : algoInitialValues
       }
       initialStatus="empty"
       validationSchema={
@@ -178,6 +184,7 @@ export default function PublishPage({
           ? await handleSubmit(values, resetForm)
           : await handleAlgorithmSubmit(values, resetForm)
       }}
+      enableReinitialize
     >
       {({ values }) => {
         const tabs = [
@@ -225,9 +232,12 @@ export default function PublishPage({
                 <Tabs
                   className={styles.tabs}
                   items={tabs}
-                  handleTabChange={(title) =>
+                  handleTabChange={(title) => {
                     setPublishType(title.toLowerCase().replace(' ', '') as any)
-                  }
+                    title === 'Algorithm'
+                      ? setdatasetInitialValues(values)
+                      : setAlgoInitialValues(values)
+                  }}
                 />
               </>
             )}