erxes/erxes

import * as React from "react";
import { AppConsumer } from "../AppContext";
import Header from "../../components/common/Header";
import { IBookingData, ICategoryTree } from "../../types";
import { ChildProps } from "react-apollo";

type Props = {
  items: ICategoryTree[];
  parentId?: string;
  goToCategory?: (categoryId: string) => void;
  goToProduct?: (productId: string) => void;
  goToIntro?: () => void;
  booking?: IBookingData;
  selectedItem?: string;
};

function HeaderContainer(props: ChildProps<Props>) {
  const { goToCategory, goToIntro, goToProduct } = props;

  const changeRoute = (treeItem: ICategoryTree) => {
    if (treeItem.type === "category" && goToCategory) {
      return goToCategory(treeItem._id);
    }

    if (treeItem.type === "product" && goToProduct) {
      return goToProduct(treeItem._id);
    }
  };
  return <Header {...props} changeRoute={changeRoute} goToIntro={goToIntro} />;
}

const WithContext = (props: Props) => {
  return (
    <AppConsumer>
      {({ goToCategory, goToProduct, goToIntro, getBooking, selectedItem }) => {
        const booking = getBooking();
        return (
          <HeaderContainer
            {...props}
            goToCategory={goToCategory}
            goToProduct={goToProduct}
            goToIntro={goToIntro}
            booking={booking}
            selectedItem={selectedItem}
          />
        );
      }}
    </AppConsumer>
  );
};

export default WithContext;