From da761468b66720478c3bd7f9b8385dd9d590ad22 Mon Sep 17 00:00:00 2001
From: David Jay <davidgljay@gmail.com>
Date: Mon, 23 Jan 2017 16:18:51 -0500
Subject: [PATCH] Enabling like buttons.

---
 client/coral-embed-stream/src/Embed.js  |   2 +-
 client/coral-plugin-likes/LikeButton.js | 101 ++++++++++++++++--------
 2 files changed, 67 insertions(+), 36 deletions(-)

diff --git a/client/coral-embed-stream/src/Embed.js b/client/coral-embed-stream/src/Embed.js
index 4d44a10b6..cd55c13cd 100644
--- a/client/coral-embed-stream/src/Embed.js
+++ b/client/coral-embed-stream/src/Embed.js
@@ -151,7 +151,7 @@ class Embed extends Component {
                   currentUser={user}
                   postAction={this.props.postAction}
                   deleteAction={this.props.deleteAction}
-                  showSignInDialog={showSignInDialog}
+                  showSignInDialog={this.props.showSignInDialog}
                   comments={asset.comments} />
                 <Notification
                   notifLength={4500}
diff --git a/client/coral-plugin-likes/LikeButton.js b/client/coral-plugin-likes/LikeButton.js
index 796ad159c..cc2c6d83e 100644
--- a/client/coral-plugin-likes/LikeButton.js
+++ b/client/coral-plugin-likes/LikeButton.js
@@ -1,45 +1,76 @@
-import React from 'react';
+import React, {Component, PropTypes} from 'react';
 import {I18n} from '../coral-framework';
 import translations from './translations.json';
 
 const name = 'coral-plugin-likes';
 
-const LikeButton = ({like, id, postAction, deleteAction, showSignInDialog, currentUser}) => {
-  const liked = like && like.current;
-  const onLikeClick = () => {
-    if (!currentUser) {
-      const offset = document.getElementById(`c_${id}`).getBoundingClientRect().top - 75;
-      showSignInDialog(offset);
-      return;
-    }
-    if (currentUser.banned) {
-      return;
-    }
-    if (!liked) {
-      postAction({
-        item_id: id,
-        item_type: 'COMMENTS',
-        action_type: 'LIKE'
-      });
-      // TODO: frontend update from mutation
-    } else {
-      deleteAction(liked.id);
-    }
-  };
+class LikeButton extends Component {
 
-  return <div className={`${name}-container`}>
-    <button onClick={onLikeClick} className={`${name}-button ${liked && 'likedButton'}`}>
-      {
-        liked
-        ? <span className={`${name}-button-text`}>{lang.t('liked')}</span>
-      : <span className={`${name}-button-text`}>{lang.t('like')}</span>
+  static propTypes = {
+    like: PropTypes.shape({
+      current: PropTypes.obect,
+      count: PropTypes.number
+    }),
+    id: PropTypes.string,
+    postAction: PropTypes.func.isRequired,
+    deleteAction: PropTypes.func.isRequired,
+    showSignInDialog: PropTypes.func.isRequired,
+    currentUser: PropTypes.shape({
+      banned: PropTypes.boolean
+    }),
+  }
+
+  state = {
+    localPost: null, // Set to the ID of an action if one is posted
+    localDelete: false // Set to true is the user deletes an action, unless localPost is already set.
+  }
+
+  render() {
+    const {like, id, postAction, deleteAction, showSignInDialog, currentUser} = this.props;
+    const {localPost, localDelete} = this.state;
+    const liked = (like && like.current && !localDelete) || localPost;
+    let count = like ? like.count : 0;
+    if (localPost) {count += 1;}
+    if (localDelete) {count -= 1;}
+
+    const onLikeClick = () => {
+      if (!currentUser) {
+        const offset = document.getElementById(`c_${id}`).getBoundingClientRect().top - 75;
+        showSignInDialog(offset);
+        return;
       }
-      <i className={`${name}-icon material-icons`}
-        aria-hidden={true}>thumb_up</i>
-      <span className={`${name}-like-count`}>{like && like.count > 0 && like.count}</span>
-    </button>
-  </div>;
-};
+      if (currentUser.banned) {
+        return;
+      }
+      if (!liked) {
+        this.setState({localPost: 'temp', localDelete: false});
+        postAction({
+          item_id: id,
+          item_type: 'COMMENTS',
+          action_type: 'LIKE'
+        }).then(({data}) => {
+          this.setState({localPost: data.createAction.id});
+        });
+      } else {
+        this.setState((prev) => prev.localPost ? {...prev, localPost: null} : {...prev, localDelete: true});
+        deleteAction(localPost || like.current.id);
+      }
+    };
+
+    return <div className={`${name}-container`}>
+      <button onClick={onLikeClick} className={`${name}-button ${liked && 'likedButton'}`}>
+        {
+          liked
+          ? <span className={`${name}-button-text`}>{lang.t('liked')}</span>
+        : <span className={`${name}-button-text`}>{lang.t('like')}</span>
+        }
+        <i className={`${name}-icon material-icons`}
+          aria-hidden={true}>thumb_up</i>
+        <span className={`${name}-like-count`}>{count > 0 && count}</span>
+      </button>
+    </div>;
+  }
+}
 
 export default LikeButton;