import * as React from 'react';
import withQueryView from 'client/ui/blocks/QueryView';
import { IQueryView } from 'modules/interfaces';
import { clearToken } from 'client/shared/storage/persistence/token';
import { WithUserQuery } from 'client/__generated__';

export default (Component, queryOptions = {}) => {
  return withQueryView(
    (props: IQueryView) => {
      const { data, ...rest } = props;

      let user = null;
      let profile = null;
      let features = null;

      if (data && data.userManagement) {
        user = data.userManagement.currentUser;
        profile = data.userManagement.currentProfile;
        features = data.features;
      }

      const defaultCurrencyCode = extractDefaultCurrencyCode(data);

      return (
        <Component
          user={user}
          profile={profile}
          features={features}
          defaultCurrencyCode={defaultCurrencyCode}
          {...rest}
        />
      );
    },
    {
      query: Component.fragment || WithUserQuery.Document,
      resolveError: () => {
        console.log(`Had error on setting token to empty`);
        clearToken();
      },
      ...queryOptions,
    },
  );
};