Merge pull request #170 from workfloworchestrator/2059-restore-values

2059 restore values

Author: DutchBen

frontend/.changeset/mighty-tips-carry.md

@@ -0,0 +1,5 @@
+---
+'pydantic-forms': patch
+---
+
+Adds fieldData storage

frontend/packages/pydantic-forms/src/core/PydanticFormContextProvider.tsx

@@ -162,11 +162,25 @@ function PydanticFormContextProvider({
     });
 
     const awaitReset = useCallback(
-        async (payLoad: FieldValues = {}) => {
-            reactHookForm.reset(payLoad);
-            await new Promise((resolve) => setTimeout(resolve, 0)); // wait one tick
+        async (payLoad?: FieldValues) => {
+            getHashForArray(formInputData).then(async (hash) => {
+                let resetPayload = {};
+
+                if (payLoad) {
+                    resetPayload = { ...payLoad };
+                } else {
+                    const currentStepFromHistory = formInputHistory.get(hash);
+
+                    if (currentStepFromHistory) {
+                        resetPayload = { ...currentStepFromHistory };
+                    }
+                }
+                reactHookForm.reset(resetPayload);
+                await new Promise((resolve) => setTimeout(resolve, 0)); // wait one tick
+            });
         },
-        [reactHookForm],
+
+        [formInputData, formInputHistory, reactHookForm],
     );
 
     const addFormInputData = useCallback(
@@ -242,6 +256,42 @@ function PydanticFormContextProvider({
         setHasNext(false);
     }, [emptyRawSchema]);
 
+    const fieldDataStorageRef = useRef<Map<string, Map<string, unknown>>>(
+        new Map(),
+    );
+
+    const fieldDataStorage = useMemo(
+        () => ({
+            has: (fieldId: string, key: string | number) => {
+                if (
+                    fieldDataStorageRef.current &&
+                    fieldDataStorageRef.current.has(fieldId)
+                ) {
+                    const fieldStorage =
+                        fieldDataStorageRef.current.get(fieldId);
+                    return fieldStorage?.has(key.toString()) ?? false;
+                }
+                return false;
+            },
+            get: (fieldId: string, key: string | number) => {
+                const fieldData = fieldDataStorageRef?.current?.get(fieldId);
+                return fieldData?.get(key.toString());
+            },
+            set: (fieldId: string, key: string | number, value: unknown) => {
+                fieldDataStorageRef.current.set(
+                    fieldId,
+                    new Map([[key.toString(), value]]),
+                );
+            },
+            delete: (fieldId: string) => {
+                if (fieldDataStorageRef.current?.has(fieldId)) {
+                    fieldDataStorageRef.current.delete(fieldId);
+                }
+            },
+        }),
+        [],
+    );
+
     const PydanticFormContextState = {
         // to prevent an issue where the sending state hangs
         // we check both the SWR hook state as our manual state
@@ -272,6 +322,7 @@ function PydanticFormContextProvider({
         formInputData,
         hasNext,
         initialData,
+        fieldDataStorage,
     };
 
     // a useeffect for whenever the error response updates
@@ -316,7 +367,7 @@ function PydanticFormContextProvider({
             // When the formKey changes we need to reset the form input data
             setFormInputData([]);
             setFormInputHistory(new Map<string, object>());
-            awaitReset();
+            awaitReset({});
             formRef.current = formKey;
         }
     }, [awaitReset, formKey]);
@@ -371,15 +422,6 @@ function PydanticFormContextProvider({
         z.config(getLocale());
     }, [locale]);
 
-    useEffect(() => {
-        getHashForArray(formInputData).then((hash) => {
-            const currentStepFromHistory = formInputHistory.get(hash);
-            if (currentStepFromHistory) {
-                awaitReset(currentStepFromHistory);
-            }
-        });
-    }, [awaitReset, formInputData, formInputHistory, reactHookForm]);
-
     return (
         <PydanticFormContext.Provider value={PydanticFormContextState}>
             {children(PydanticFormContextState)}

frontend/packages/pydantic-forms/src/types.ts

@@ -47,6 +47,13 @@ export type PydanticFormControlledElementProps = Omit<
 export type PydanticFormControlledElement =
     React.JSXElementConstructor<PydanticFormControlledElementProps>;
 
+export type PydanticFormFieldDataStorage = {
+    set: (fieldId: string, key: string | number, value: unknown) => void;
+    has: (fieldId: string, key: string | number) => boolean;
+    get: (fieldId: string, key: string) => unknown;
+    delete: (fieldId: string) => void;
+};
+
 export interface PydanticFormContextProps {
     isLoading: boolean;
     isSending: boolean;
@@ -71,6 +78,7 @@ export interface PydanticFormContextProps {
     hasNext: boolean;
     formInputData: object[];
     initialData: FieldValues;
+    fieldDataStorage: PydanticFormFieldDataStorage;
 }
 
 export enum PydanticFormState {