var _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; };

import React from 'react';

export default function controlWrapper(Component) {

    class CrudControl extends React.Component {

        componentWillMount() {
            // register the handler
            const self = this;
            const handler = this.props.controller.on((evt, data) => {
                self.refs.ctrl.eventHandler(evt, data);
            });
            this.setState({ handler: handler });
        }

        componentWillUnmount() {
            // remove registration
            this.props.controller.removeListener(this.state.handler);
        }

        render() {
            return React.createElement(Component, _extends({ ref: "ctrl" }, this.props));
        }

    }

    CrudControl.propTypes = {
        controller: React.PropTypes.object
    };

    return CrudControl;
}