From b8f24b952b4e8d1fc4d93768bca9440b795b2b5f Mon Sep 17 00:00:00 2001
From: Chi Vinh Le <vinh@wikiwi.io>
Date: Thu, 3 Aug 2017 21:41:01 +0700
Subject: [PATCH] Automatically close stream when passed closedAt date

---
 .../src/components/Stream.js                  |  2 +
 .../src/containers/AutomaticAssetClosure.js   | 77 +++++++++++++++++++
 .../src/containers/Stream.js                  |  1 +
 3 files changed, 80 insertions(+)
 create mode 100644 client/coral-embed-stream/src/containers/AutomaticAssetClosure.js

diff --git a/client/coral-embed-stream/src/components/Stream.js b/client/coral-embed-stream/src/components/Stream.js
index 265dcfe8b..1e0ff7ea9 100644
--- a/client/coral-embed-stream/src/components/Stream.js
+++ b/client/coral-embed-stream/src/components/Stream.js
@@ -19,6 +19,7 @@ import cn from 'classnames';
 
 import {getTopLevelParent, attachCommentToParent} from '../graphql/utils';
 import AllCommentsPane from './AllCommentsPane';
+import AutomaticAssetClosure from '../containers/AutomaticAssetClosure';
 
 import styles from './Stream.css';
 
@@ -141,6 +142,7 @@ class Stream extends React.Component {
 
     return (
       <div id="stream" className={styles.root}>
+        <AutomaticAssetClosure assetId={asset.id} closedAt={asset.closedAt}/>
         {comment &&
           <Button
             cStyle="darkGrey"
diff --git a/client/coral-embed-stream/src/containers/AutomaticAssetClosure.js b/client/coral-embed-stream/src/containers/AutomaticAssetClosure.js
new file mode 100644
index 000000000..ad27ef3a9
--- /dev/null
+++ b/client/coral-embed-stream/src/containers/AutomaticAssetClosure.js
@@ -0,0 +1,77 @@
+import React from 'react';
+import PropTypes from 'prop-types';
+import {gql} from 'react-apollo';
+
+const FRAGMENT = gql`
+  fragment CoralEmbedStream_AutomaticAssetClosure_Fragment on Asset {
+    id
+    isClosed
+  }
+`;
+
+function getFragmentId(assetId) {
+  return `Asset_${assetId}`;
+}
+
+class AutomaticAssetClosure extends React.Component {
+  static contextTypes = {
+    client: PropTypes.object.isRequired,
+  };
+
+  timer = null;
+
+  componentWillMount() {
+    this.setupTimer(this.props.assetId, this.props.closedAt);
+  }
+
+  componentWillReceiveProps(next) {
+    if (
+      this.props.assetId !== next.assetId ||
+      this.props.closedAt !== next.closedAt
+    ) {
+      this.setupTimer(next.assetId, next.closedAt);
+    }
+  }
+
+  closeAsset(assetId) {
+    this.context.client.writeFragment({
+      fragment: FRAGMENT,
+      id: getFragmentId(assetId),
+      data: {
+        isClosed: true,
+      },
+    });
+  }
+
+  setupTimer(assetId, closedAt) {
+    clearTimeout(this.timer);
+    this.timer = null;
+
+    if (assetId && closedAt) {
+      const asset = this.context.client.readFragment({
+        fragment: FRAGMENT,
+        id: getFragmentId(assetId),
+      });
+
+      if (!asset.isClosed && closedAt) {
+        const diff = (new Date(closedAt) - new Date());
+        if (diff >= 0) {
+          this.timer = setTimeout(() => this.closeAsset(assetId), diff);
+        } else {
+          this.closeAsset(assetId);
+        }
+      }
+    }
+  }
+
+  render() {
+    return null;
+  }
+}
+
+AutomaticAssetClosure.PropTypes = {
+  assetId: PropTypes.string,
+  closedAt: PropTypes.string,
+};
+
+export default AutomaticAssetClosure;
diff --git a/client/coral-embed-stream/src/containers/Stream.js b/client/coral-embed-stream/src/containers/Stream.js
index de726f373..fcedbcb0b 100644
--- a/client/coral-embed-stream/src/containers/Stream.js
+++ b/client/coral-embed-stream/src/containers/Stream.js
@@ -250,6 +250,7 @@ const fragments = {
         id
         title
         url
+        closedAt
         isClosed
         created_at
         settings {