onion/js/utils/react_utils.js

import React from 'react';
import { currentUserShape } from '../components/prop_types';


/**
 * Taken from react-router (https://github.com/reactjs/react-router/blob/master/modules/withRouter.js)
 * FIXME: should be put into react-component's utils
 */
export function getDisplayName(WrappedComponent) {
    return WrappedComponent.displayName || WrappedComponent.name || 'Component';
}

/**
 * Similar to react-router's `withRouter`, this injects the `currentUser` from the Component's
 * context into the Component as a prop. It also injects whether the user's logged in or not as
 * `isLoggedIn`.
 *
 * @param  {Component} Component Component to inject `context.currentUser` and `isLoggedIn` into
 * @return {Component}           Wrapped component
 */
export function withCurrentUser(Component) {
    const contextTypes = {
        currentUser: currentUserShape.isRequired
    };

    const WithCurrentUser = (props, { currentUser }) => (
        <Component {...props} currentUser={currentUser} isLoggedIn={!!currentUser.email} />
    );

    WithCurrentUser.displayName = `WithCurrentUser(${getDisplayName(Component)})`;
    WithCurrentUser.contextTypes = contextTypes;

    return WithCurrentUser;
}