withRouteBlock HOC to prevent transitions

SomeContainer.tsx

export default compose<Props, OwnProps>(
  withHistory,
  withRouteBlock("Already leaving?")
)(SomeContainer);

withRouteBlock.ts

import { compose, lifecycle, withState } from 'recompose';
import { History, UnregisterCallback } from 'history';

type Props = {
  pristine: boolean;
  history: History;
};

type State = {
  historyBlock: UnregisterCallback;
};

/**
 * HOC that blocks history transition if props.pristine is false
 */
export const withRouteBlock = (message: string) => compose(

  lifecycle<Props, State>({

    componentDidMount() {
      const { history, pristine } = this.props;

      if (!history) {
        console.warn('history in undefined');
      }

      if (!pristine) {
        this.setState({ historyBlock: history.block(message) })
      }
    },

    componentWillReceiveProps(nextProps) {
      const { history, pristine } = nextProps;
      const historyBlock = this.state ? this.state.historyBlock : null;

      if (!pristine && !historyBlock) {
        this.setState({ historyBlock: history.block(message) })
      }
      else if (historyBlock && pristine) {
        historyBlock();
      }
    },

    componentWillUnmount() {
      if (this.state && this.state.historyBlock) {
        this.state.historyBlock();
      }
    }
  })

);