add voting context + refresh polls after add

app/components/Voting.js

@@ -6,26 +6,30 @@ import AddPoll from './simple-voting/AddPoll';
 import PollsList from './simple-voting/PollsList';
 import StatusLogo from '../ui/components/StatusLogo';
 import Collapse from '@material-ui/core/Collapse';
+import { VotingContext } from '../context';
 
 class Voting extends PureComponent {
   state = { addPoll: false };
 
   render(){
-    const { toggleAdmin, rawPolls } = this.props;
     const { addPoll } = this.state;
     const togglePoll = () => { this.setState({ addPoll: !addPoll })};
     return (
-      <Fragment>
-        <CssBaseline />
-        <AppBar toggleAdmin={toggleAdmin} togglePoll={togglePoll} />
-        <div style={{ margin: '30px', textAlign: 'center' }}>
-          <img src="images/logo.png" width="200" />
-        </div>
-        <Collapse in={addPoll}>
-          <AddPoll togglePoll={togglePoll} />
-        </Collapse>
-        {rawPolls && <PollsList rawPolls={rawPolls} />}
-      </Fragment>
+      <VotingContext.Consumer>
+        {({ getPolls, rawPolls, toggleAdmin }) =>
+          <Fragment>
+            <CssBaseline />
+            <AppBar toggleAdmin={toggleAdmin} togglePoll={togglePoll} />
+            <div style={{ margin: '30px', textAlign: 'center' }}>
+              <img src="images/logo.png" width="200" />
+            </div>
+            <Collapse in={addPoll}>
+              <AddPoll togglePoll={togglePoll} getPolls={getPolls} />
+            </Collapse>
+            {rawPolls && <PollsList rawPolls={rawPolls} />}
+          </Fragment>
+        }
+      </VotingContext.Consumer>
     )
   }
 }

app/components/simple-voting/AddPoll.js

@@ -106,6 +106,7 @@ const AddPoll = withFormik({
             console.log('sucess:', res);
             setSubmitting(false);
             props.togglePoll();
+            props.getPolls();
           })
           .catch(res => {
             console.log('fail:', res);

app/context.js

@@ -0,0 +1,3 @@
+import React from 'react';
+
+export const VotingContext = React.createContext(null);

app/dapp.js

@@ -5,6 +5,7 @@ import PollManager from 'Embark/contracts/PollManager';
 import AdminView from './components/AdminView';
 import Voting from './components/Voting';
 import SNT from  'Embark/contracts/SNT';
+import { VotingContext } from './context';
 window['SNT'] = SNT;
 
 import './dapp.css';
@@ -36,7 +37,7 @@ class App extends React.Component {
     this.setState({account: _account});
   }
 
-  async _getPolls(){
+  _getPolls = async () => {
     const { nPolls, poll } = PollManager.methods;
     const polls = await nPolls.call();
     const total = await polls.call();
@@ -54,13 +55,17 @@ class App extends React.Component {
 
   render(){
     const { admin, rawPolls } = this.state;
+    const { _getPolls } = this;
     const toggleAdmin = () => this.setState({ admin: true });
+    const votingContext = { getPolls: _getPolls, rawPolls, toggleAdmin };
     return (
-      <Fragment>
-        {admin ?
-         <AdminView setAccount={this.setAccount} /> :
-         <Voting toggleAdmin={toggleAdmin} rawPolls={rawPolls} />}
-      </Fragment>
+      <VotingContext.Provider value={votingContext}>
+        <Fragment>
+          {admin ?
+           <AdminView setAccount={this.setAccount} /> :
+           <Voting />}
+        </Fragment>
+      </VotingContext.Provider>
     );
   }
 }