From 693879647eccba05bf106ba06e075ed6f5fff473 Mon Sep 17 00:00:00 2001
From: Chi Vinh Le <vinh@wikiwi.io>
Date: Wed, 5 Jul 2017 21:55:25 +0700
Subject: [PATCH] Cleanup

---
 .../src/components/Comment.js                 | 34 +++++++++----------
 .../src/containers/Stream.js                  | 13 +------
 .../coral-embed-stream/src/graphql/utils.js   | 32 +++++++++++++++--
 3 files changed, 48 insertions(+), 31 deletions(-)

diff --git a/client/coral-embed-stream/src/components/Comment.js b/client/coral-embed-stream/src/components/Comment.js
index 0e6201caf..7cc8b1480 100644
--- a/client/coral-embed-stream/src/components/Comment.js
+++ b/client/coral-embed-stream/src/components/Comment.js
@@ -406,7 +406,7 @@ export default class Comment extends React.Component {
         return false;
       });
 
-    const rootClassNames = [
+    const rootClassName = cn(
       'talk-stream-comment-wrapper',
       `talk-stream-comment-wrapper-level-${depth}`,
       styles.root,
@@ -415,28 +415,28 @@ export default class Comment extends React.Component {
         ...conditionalClassNames,
         [styles.enter]: this.state.animateEnter,
       },
-    ];
+    );
+
+    const commentClassName = cn(
+      'talk-stream-comment',
+      `talk-stream-comment-level-${depth}`,
+      styles.comment,
+      styles[`commentLevel${depth}`],
+      {
+        [styles.pendingComment]: isPending,
+        [styles.highlightedComment]: isHighlighted,
+        'talk-stream-pending-comment': isPending,
+        'talk-stream-highlighted-comment': isHighlighted,
+      }
+    );
 
     return (
       <div
-        className={cn(...rootClassNames)}
+        className={rootClassName}
         id={`c_${comment.id}`}
       >
         {!isReply && <hr aria-hidden={true} />}
-        <div
-          className={cn(
-            'talk-stream-comment',
-            `talk-stream-comment-level-${depth}`,
-            styles.comment,
-            styles[`commentLevel${depth}`],
-            {
-              [styles.pendingComment]: isPending,
-              [styles.highlightedComment]: isHighlighted,
-              'talk-stream-pending-comment': isPending,
-              'talk-stream-highlighted-comment': isHighlighted,
-            }
-          )}
-        >
+        <div className={commentClassName}>
           <AuthorName author={comment.user} className={'talk-stream-comment-user-name'} />
           {isStaff(comment.tags) ? <TagLabel>Staff</TagLabel> : null}
 
diff --git a/client/coral-embed-stream/src/containers/Stream.js b/client/coral-embed-stream/src/containers/Stream.js
index 28947b77e..e52aa000f 100644
--- a/client/coral-embed-stream/src/containers/Stream.js
+++ b/client/coral-embed-stream/src/containers/Stream.js
@@ -21,6 +21,7 @@ import {
   insertCommentIntoEmbedQuery,
   removeCommentFromEmbedQuery,
   insertFetchedCommentsIntoEmbedQuery,
+  nest,
 } from '../graphql/utils';
 
 const {showSignInDialog} = authActions;
@@ -151,18 +152,6 @@ class StreamContainer extends React.Component {
   }
 }
 
-const nest = (def, level) => {
-  let result = '';
-  for (let x = 0; x < level; x++) {
-    if (x === 0) {
-      result += def;
-      continue;
-    }
-    result = result.replace('...nest', def);
-  }
-  return result.replace('...nest', '');
-};
-
 const commentFragment = gql`
   fragment CoralEmbedStream_Stream_comment on Comment {
     id
diff --git a/client/coral-embed-stream/src/graphql/utils.js b/client/coral-embed-stream/src/graphql/utils.js
index 518f4b62f..03d9fe02a 100644
--- a/client/coral-embed-stream/src/graphql/utils.js
+++ b/client/coral-embed-stream/src/graphql/utils.js
@@ -4,13 +4,11 @@ import {THREADING_LEVEL} from '../constants/stream';
 function applyToCommentsOrigin(root, callback) {
   if (root.comment) {
     let comment = root.comment;
-    console.log(comment);
     for (let depth = 0; depth <= THREADING_LEVEL; depth++) {
       let changes = {$apply: (node) => node ? callback(node) : node};
       for (let i = 0; i < depth; i++) {
         changes = {parent: changes};
       }
-      console.log(changes);
       comment = update(comment, changes);
     }
 
@@ -183,3 +181,33 @@ function findAndInsertFetchedComments(parent, comments, parent_id) {
 export function insertFetchedCommentsIntoEmbedQuery(root, comments, parent_id) {
   return applyToCommentsOrigin(root, (origin) => findAndInsertFetchedComments(origin, comments, parent_id));
 }
+
+/**
+ * Nest a string in itself repeatly until `level` has been reached.
+ *
+ * Example:
+ * nest(`
+ *   a
+ *   ...nest
+ *   b
+ * `, 2)
+ *
+ * Output:
+ * `
+ *   a
+ *   a
+ *   b
+ *   b
+ * `
+ */
+export function nest(document, level) {
+  let result = '';
+  for (let x = 0; x < level; x++) {
+    if (x === 0) {
+      result += document;
+      continue;
+    }
+    result = result.replace('...nest', document);
+  }
+  return result.replace('...nest', '');
+}