// TypeScript Version: 3.0

import * as React from 'react';

export type Omit<T, U> = Pick<T, Exclude<keyof T, U>>;

export interface InferableComponentEnhancerWithProps<
  TInjectedProps,
  TNeedsProps
> {
  <P extends TInjectedProps>(
    component: React.ComponentType<P>,
  ): React.ComponentType<Omit<P, keyof TInjectedProps> & TNeedsProps>;
}

export function mapProps<TInner, TOuter>(
  mapper: (input: TOuter) => TInner,
): InferableComponentEnhancerWithProps<TInner, TOuter>;

export function withProps<TInner, TOuter>(
  objOrMapper: Partial<TInner> | ((input: TOuter) => Partial<TInner>),
): InferableComponentEnhancerWithProps<TInner & TOuter, TOuter>;