import React, { Component, PropTypes } from 'react';

export const ScrollProvider = ({ context }) => (Wrapped) => {
  return class Scroll extends Component {

    constructor(props) {
      super(props);
      this.state = {
        scrollTop: 0
      };
    }

    static childContextTypes = {
      [context]: PropTypes.number
    }

    getChildContext() {
      return { [context]: this.state.scrollTop };
    }

    handleScroll = (e) => {
      this.setState({
        scrollTop: e.target.scrollTop
      });
    }

    render() {
      return (
        <div onScroll={this.handleScroll}>
          <Wrapped {...this.props} />
        </div>
      );
    }

  };
};