withPropsStyles.ts

type StyleRulesCallback<ClassKey extends string = string> = (
  props: any,
  theme: Theme
) => StyleRules<ClassKey>;

const withPropsStyles = (style: StyleRulesCallback) => (
  component: React.ComponentType<any>
) => {
  return forwardRef((props, ref) => {
    const proxy = (theme: Theme) => style(props, theme);
    const hoc = withStyles(proxy)(component);
    return props.children
      ? createElement(hoc, { ...props, ref }, props.children)
      : createElement(hoc, { ...props, ref });
  });
};