Redux connect implementation

connect.js

function connect(mapStateToProps, mapDispatchToProps) {
  return function connector(Component) {
    class ContainerComponent extends React.Component {
      componentDidMount() {
        const { store } = this.context;
        this.unsubscribe = store.subscribe(() => this.forceUpdate());
      }

      componentWillUnmount() {
        this.unsubscribe();
      }      
      
      render() {
        const {store} = this.context;
        var props = Object.assign(mapStateToProps(store.getState()), mapDispatchToProps(store.dispatch));
        return <ContainerComponent {...props} />;
      }
    }
    
    ContainerComponent.contextTypes = {
      store: React.PropTypes.object
    };
    
    return ContainerComponent;
  }
}