Function withWrapper

packages/styletron-react/src/index.tsx:261–287

(component, wrapper)

export const withStyle = withStyleDeep;

export const withWrapper: WithWrapperFn = (component, wrapper) => {
  const styletron = component.__STYLETRON__;

  if (__DEV__) {
    if (!styletron) {
      /* eslint-disable no-console */
      console.warn(
        "The first parameter to `withWrapper` must be a styled component (without extra wrappers).",
      );
      /* eslint-enable no-console */
    }
  }

  const composed = {
    getInitialStyle: styletron.getInitialStyle,
    base: styletron.base,
    driver: styletron.driver,
    wrapper: wrapper,
    reducers: styletron.reducers,
  };

  if (__BROWSER__ && __DEV__) {
    addDebugMetadata(composed, 2);
  }

  return createStyledElementComponent(composed);
};

export function autoComposeShallow<Props>(
  styletron: Styletron,