auth-mutation.js

export function withAuthContext<Props: Object>(
  Component: ComponentType<Props>
): ComponentType<$Diff<Props, State>> {
  return function AuthContextConsumerWrapper(props: Props) {
    return (
      <AuthContextConsumer>{context => <Component {...context} {...props} />}</AuthContextConsumer>
    );
  };
}

export function withAuth<Props: Object>(
  Component: ComponentType<Props>
): ComponentType<$Diff<Props, State>> {
  return compose(graphql(LOGIN_MUTATION, { name: "loginUser" }))(withAuthContext(Component));
}