@kiwicom/smart-faq/src/SmartFAQ/common/PageVariant.js

// @flow

import * as React from 'react';

import PageVariantContext from '../context/PageVariant';

const ExclusiveToVariant = expectedVariant => {
  const ExclusiveComponent = ({ children }: { children: React.Node }) => (
    <PageVariantContext.Consumer>
      {value => {
        if (value === undefined) {
          return children;
        }

        const { variant } = value;
        if (variant === expectedVariant) {
          return children;
        }
        return null;
      }}
    </PageVariantContext.Consumer>
  );

  return ExclusiveComponent;
};

export const SidebarVersion = ExclusiveToVariant('sidebar');
export const FullPageVersion = ExclusiveToVariant('fullPage');