import { withHandlers } from 'recompose';

export const withRef = (onRef, getRefElement) => withHandlers(
  () => {
    let element = null;
    return {
      [onRef]: () => ref => {
        element = ref;
      },
      [getRefElement]: () => () => element,
    };
  }
);