talk/plugins/talk-plugin-featured-comments/client/containers/ModSubscription.js

import React from 'react';
import { gql } from 'react-apollo';
import { connect } from 'react-redux';
import Comment from 'coral-admin/src/routes/Moderation/containers/Comment';
import { getDefinitionName } from 'coral-framework/utils';
import truncate from 'lodash/truncate';
import t from 'coral-framework/services/i18n';
import { subscriptionFields } from 'coral-admin/src/routes/Moderation/graphql';

function prepareNotificationText(text) {
  return truncate(text, { length: 50 }).replace('\n', ' ');
}

class ModSubscription extends React.Component {
  subscriptions = null;

  componentWillMount() {
    const configs = [
      {
        document: COMMENT_FEATURED_SUBSCRIPTION,
        variables: {
          assetId: this.props.data.variables.asset_id,
        },
        updateQuery: (
          prev,
          { subscriptionData: { data: { commentFeatured: { user, comment } } } }
        ) => {
          const notifyText =
            this.props.user.id === user.id
              ? ''
              : t(
                  'talk-plugin-featured-comments.notify_featured',
                  user.username,
                  prepareNotificationText(comment.body)
                );
          return this.props.handleCommentChange(prev, comment, notifyText);
        },
      },
      {
        document: COMMENT_UNFEATURED_SUBSCRIPTION,
        variables: {
          assetId: this.props.data.variables.asset_id,
        },
        updateQuery: (
          prev,
          {
            subscriptionData: {
              data: { commentUnfeatured: { user, comment } },
            },
          }
        ) => {
          const notify =
            this.props.user.id === user.id
              ? ''
              : t(
                  'talk-plugin-featured-comments.notify_unfeatured',
                  user.username,
                  prepareNotificationText(comment.body)
                );
          return this.props.handleCommentChange(prev, comment, notify);
        },
      },
    ];
    this.subscriptions = configs.map(config =>
      this.props.data.subscribeToMore(config)
    );
  }

  componentWillUnmount() {
    this.subscriptions.forEach(unsubscribe => unsubscribe());
  }

  render() {
    return null;
  }
}

const COMMENT_FEATURED_SUBSCRIPTION = gql`
  subscription CommentFeatured($assetId: ID){
    commentFeatured(asset_id: $assetId) {
      comment {
        ...${getDefinitionName(Comment.fragments.comment)}
        ${subscriptionFields}
      }
      user {
        id
        username
      }
    }
  }
  ${Comment.fragments.comment}
`;

const COMMENT_UNFEATURED_SUBSCRIPTION = gql`
  subscription CommentUnfeatured($assetId: ID){
    commentUnfeatured(asset_id: $assetId){
      comment {
        ...${getDefinitionName(Comment.fragments.comment)}
        ${subscriptionFields}
      }
      user {
        id
        username
      }
    }
  }
  ${Comment.fragments.comment}
`;

const mapStateToProps = state => ({
  user: state.auth.user,
});

export default connect(mapStateToProps, null)(ModSubscription);