Add ability for container components to register their own mapStateToProps function.

This commit is contained in:
Owen Conti
2017-08-11 16:01:01 -06:00
parent c237f25638
commit d9b0632d16

View File

@@ -20,8 +20,11 @@ const RootWrapper = (reduxStore, ComponentToWrap) => class extends Component {
} }
const makeContainer = (getSystem, component, reduxStore) => { const makeContainer = (getSystem, component, reduxStore) => {
const mapStateToProps = component.prototype.constructor.mapStateToProps || function(state) {
return {state}
}
let wrappedWithSystem = SystemWrapper(getSystem, component, reduxStore) let wrappedWithSystem = SystemWrapper(getSystem, component, reduxStore)
let connected = connect(state => ({state}))(wrappedWithSystem) let connected = connect( mapStateToProps )(wrappedWithSystem)
if(reduxStore) if(reduxStore)
return RootWrapper(reduxStore, connected) return RootWrapper(reduxStore, connected)
return connected return connected