Restore stored control states on init

ScottLogic · Aug 2, 2024 · 72f6622 · 72f6622
1 parent dbd8d29
commit 72f6622
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/app/carbon-estimator-form/carbon-estimator-form.component.ts b/src/app/carbon-estimator-form/carbon-estimator-form.component.ts
@@ -180,6 +180,20 @@ export class CarbonEstimatorFormComponent implements OnInit, OnDestroy {
     if (storedFormData) {
       this.estimatorForm.setValue(storedFormData);
     }
+
+    const storedControlStates = this.getSavedControlStates();
+
+    if (storedControlStates) {
+      for (const [controlKey, controlState] of Object.entries(storedControlStates)) {
+        const control = this.estimatorForm.get(controlKey)!;
+        if (controlState.dirty) {
+          control.markAsDirty();
+        }
+        if (controlState.touched) {
+          control.markAsTouched();
+        }
+      }
+    }
   }
 
   ngOnDestroy(): void {
@@ -255,6 +269,11 @@ export class CarbonEstimatorFormComponent implements OnInit, OnDestroy {
     this.storageService.set('controlStates', JSON.stringify(controlStates));
   }
 
+  private getSavedControlStates() {
+    const storedControlStates = this.storageService.get('controlStates');
+    return storedControlStates ? (JSON.parse(storedControlStates) as Record<string, ControlState>) : null;
+  }
+
   private saveFormState() {
     this.saveFormData();
     this.saveControlStates();