import React, { createContext } from 'react';
import { ConnectProps } from '../types';

export const ConnectPropsContext = createContext<ConnectProps>({} as ConnectProps);
export const ConnectPropsProvider = ConnectPropsContext.Provider;
export const ConnectPropsConsumer = ConnectPropsContext.Consumer;

export const withConnectProps = (InnerComponent: any) => {
  return (props: any) => {
    return (
      <ConnectPropsConsumer>
        {connectProps => {
          return <InnerComponent {...props} {...connectProps} />;
        }}
      </ConnectPropsConsumer>
    );
  };
};