move logic about replacement and pasting comment to store instead of root component

web/app/common/store.js

@@ -11,16 +11,91 @@ class Store {
     return _instance;
   }
 
-  set(obj) {
-    this.data = {
-      ...this.data,
-      ...obj,
-    };
+  set(key, obj) {
+    this.data[key] = obj;
   }
 
   get(key) {
     return this.data[key];
   }
+
+  addComment({ text, id, pid }) {
+    const comments = this.data.comments;
+    const newComment = {
+      comment: {
+        id,
+        text,
+        user: this.get('user'),
+        time: new Date(),
+        ...(pid ? { pid } : {}),
+      },
+    };
+
+    if (pid) {
+      this.pasteReply(newComment);
+    } else {
+      this.pasteComment(newComment);
+    }
+  }
+
+  pasteReply(newReply) {
+    let again = true;
+
+    const concatReply = (root, reply) => {
+      root.replies = root.replies || [];
+      root.replies = [reply].concat(root.replies);
+
+      again = false;
+
+      return root;
+    }
+
+    const paste = (root, commentObj) => {
+      if (!again) return root;
+
+      if (root.comment.id === commentObj.comment.pid) {
+        return concatReply(root, commentObj);
+      }
+
+      if (root.replies) {
+        root.replies = root.replies.map(reply => {
+          if (reply.comment.id === commentObj.comment.pid) {
+            return concatReply(reply, commentObj);
+          } else {
+            return paste(reply, commentObj);
+          }
+        })
+      }
+
+      return root;
+    };
+
+    this.data.comments = this.data.comments.map(thread => paste(thread, newReply));
+  }
+
+  pasteComment(newComment) {
+    this.data.comments = [newComment].concat(this.data.comments);
+  }
+
+  replaceComment(newComment) {
+    let again = true;
+
+    const replace = (thread, comment) => {
+      if (!again) return thread;
+
+      if (thread.comment.id === comment.id) {
+        thread.comment = comment;
+        again = false;
+        return thread;
+      }
+
+      thread.replies = thread.replies.map(reply => replace(reply, comment));
+
+      return thread;
+    };
+
+    this.data.comments = this.data.comments.map(thread => replace(thread, newComment));
+  }
 }
 
 export default new Store();