auth_wrapper.jsx

import React from 'react';
import { connect } from 'react-redux';

/**
 * A higher order function that returns a function whose argument is the component
 * to be wrapped.
 *
 * usage example:
 * const EnsureLoggedIn = AuthWrapper();
 * <EnsureLoggedIn(Profile) />
 */
export default function authWrapper(options = {}) {

  return (WrappedComponent) => {

    class AuthComponent extends React.Component {
      render() {
        const { user } = this.props;

        if (options.guest) {
          return (
            user ? <div>You are already logged in!</div> : <WrappedComponent />
          );
        } else {
          return (
            user ? <WrappedComponent /> : <div>Please sign in</div>
          );
        }
      }
    }

    if (WrappedComponent.reduxAsyncConnect) {
      AuthComponent.reduxAsyncConnect = WrappedComponent.reduxAsyncConnect;
    }

    return connect(
      (state) => ({
        user: state.auth.user
      })
    )(AuthComponent);
  };
}