From 84bc1ebef7d7a6be30b53fa0cd6fe134cd704a22 Mon Sep 17 00:00:00 2001
From: Abhas Bhattacharya <abhasbhattacharya2@gmail.com>
Date: Mon, 9 Apr 2018 11:05:33 +0530
Subject: [PATCH] Add state saving to Container

---
 .../CreateEditorApp/CreateEditorApp.tsx       |  2 +-
 .../DragContainer/DragContainer.tsx           | 26 +++++++++++++++++++
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/src/Components/CreateEditorApp/CreateEditorApp.tsx b/src/Components/CreateEditorApp/CreateEditorApp.tsx
index 2b02cfb..fd5218e 100644
--- a/src/Components/CreateEditorApp/CreateEditorApp.tsx
+++ b/src/Components/CreateEditorApp/CreateEditorApp.tsx
@@ -17,7 +17,7 @@ interface CreateEditorAppState {
 }
 
 export class CreateEditorApp extends React.Component<{}, CreateEditorAppState> {
-  localStorageKey: 'appData';
+  localStorageKey: string = 'appData';
 
   constructor(props: {}) {
     super(props);
diff --git a/src/Components/DragContainer/DragContainer.tsx b/src/Components/DragContainer/DragContainer.tsx
index f861762..a069682 100644
--- a/src/Components/DragContainer/DragContainer.tsx
+++ b/src/Components/DragContainer/DragContainer.tsx
@@ -22,6 +22,8 @@ interface DragContainerState {
 }
 
 export class DragContainer extends React.Component<DragContainerProps, DragContainerState> {
+    localStorageKey: string = 'containerData';
+
     constructor(props: DragContainerProps) {
         super(props);
 
@@ -42,6 +44,30 @@ export class DragContainer extends React.Component<DragContainerProps, DragConta
         this.handleDragStop = this.handleDragStop.bind(this);
     }
 
+    componentDidMount() {
+        this.loadFromLocalStorage();
+    }
+
+    componentDidUpdate() {
+        this.saveToLocalStorage();
+    }
+
+    saveToLocalStorage() {
+        const pastOffsetJSON = JSON.stringify(this.state.pastCumulativeOffsetMap);
+        localStorage.setItem(this.localStorageKey, pastOffsetJSON);
+    }
+
+    loadFromLocalStorage() {
+        const pastOffsetJSON = localStorage.getItem(this.localStorageKey);
+
+        if (pastOffsetJSON) {
+            const pastOffsetObject = JSON.parse(pastOffsetJSON);
+            this.setState({
+                pastCumulativeOffsetMap: pastOffsetObject
+            });
+        }
+    }
+
     handleDragStartOnDraggableNode(draggableId: string, ev: React.MouseEvent<HTMLDivElement>): void {
         this.setState({
             ongoing: {