Progress

app/javascript/gabsocial/features/compose/components/poll_button.js

@@ -1,31 +1,31 @@
-import { defineMessages, injectIntl } from 'react-intl';
-import { addPoll, removePoll } from '../../../actions/compose';
+import { defineMessages, injectIntl } from 'react-intl'
+import { addPoll, removePoll } from '../../../actions/compose'
 import ComposeExtraButton from './compose_extra_button'
 
 const messages = defineMessages({
   add_poll: { id: 'poll_button.add_poll', defaultMessage: 'Add poll' },
   title: { id: 'poll_button.title', defaultMessage: 'Poll' },
   remove_poll: { id: 'poll_button.remove_poll', defaultMessage: 'Remove poll' },
-});
+})
 
 const mapStateToProps = state => ({
   unavailable: state.getIn(['compose', 'is_uploading']) || (state.getIn(['compose', 'media_attachments']).size > 0),
   active: state.getIn(['compose', 'poll']) !== null,
-});
+})
 
 const mapDispatchToProps = dispatch => ({
 
   onClick() {
     dispatch((_, getState) => {
       if (getState().getIn(['compose', 'poll'])) {
-        dispatch(removePoll());
+        dispatch(removePoll())
       } else {
-        dispatch(addPoll());
+        dispatch(addPoll())
       }
-    });
+    })
   },
 
-});
+})
 
 export default
 @connect(mapStateToProps, mapDispatchToProps)
@@ -39,16 +39,16 @@ class PollButton extends PureComponent {
     onClick: PropTypes.func.isRequired,
     intl: PropTypes.object.isRequired,
     small: PropTypes.bool,
-  };
+  }
 
   handleClick = () => {
-    this.props.onClick();
+    this.props.onClick()
   }
 
   render() {
-    const { intl, active, unavailable, disabled, small } = this.props;
+    const { intl, active, unavailable, disabled, small } = this.props
 
-    if (unavailable) return null;
+    if (unavailable) return null
 
     return (
       <ComposeExtraButton
@@ -57,6 +57,7 @@ class PollButton extends PureComponent {
         onClick={this.handleClick}
         icon='poll'
         small={small}
+        active={active}
       />
     )
   }