import React from 'react';

import TreeContext from '../contexts/TreeContext';

function withTree(Component) {
  return function TreeComponent(props) {
    return (
      <TreeContext.Consumer>
        {tree => <Component {...props} tree={tree} />}
      </TreeContext.Consumer>
    );
  };
}

export default withTree;