import { compose, gql } from 'react-apollo';
import { withFragments, excludeIf } from 'plugin-api/beta/client/hocs';
import ToxicLabel from '../components/ToxicLabel';
import { isToxic } from '../utils';

const enhance = compose(
  withFragments({
    comment: gql`
      fragment TalkToxicComments_ToxicLabel_Comment on Comment {
        actions {
          __typename
          ... on FlagAction {
            reason
          }
        }
      }
    `,
  }),
  excludeIf(({ comment: { actions } }) => !isToxic(actions))
);

export default enhance(ToxicLabel);