withRedirect.hoc.ts

import * as React from 'react'
import { Redirect } from 'react-router-dom'
import queryString from 'qs'


export const withRedirect = <P extends Object>(
  reaction: (any) => boolean,
  redirect: string | ((props: P) => string),
) => (WrappedComponent) => {
  const WithRedirectWrapper = (props: P) => {
    const condition = reaction(props)

    if (!condition) {
      return <WrappedComponent {...props} />
    }

    const to: string = (typeof redirect === 'function')
      ? redirect(props)
      : redirect

    return <Redirect to={to} />
  }

  return WithRedirectWrapper
}

export const withRedirectNextProvider = (route: string = '/') => (state) => {
  const query = queryString.stringify({
    next: state.location.pathname,
  })

  return `${route}?${query}`
}

export const withRedirectNextHandler = (defaultNext: string = '/') => (props) => {
  const { '?next': next } = queryString.parse(props.location.search)

  return next || defaultNext
}