📦 retyui / react-native-stylex

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36/* eslint-disable @typescript-eslint/no-explicit-any */
import React, {
  ComponentProps,
  ComponentType,
  forwardRef,
  Ref,
  ElementRef
} from "react";

interface InjectedStyledProps<Styles> {
  styles: Styles;
}

export type InferInjectedStyledProps<
  Fn extends (...args: any) => any
> = InjectedStyledProps<ReturnType<Fn>>;

export function withStyles<Styles>(useStyles: () => Styles) {
  function WithStyles<TComponent extends ComponentType<any>>(
    Component: TComponent
  ) {
    const renderComponent = (
      props: Omit<ComponentProps<TComponent>, "styles">,
      ref: Ref<ElementRef<TComponent>>
    ) => {
      const styles = useStyles();
      // @ts-expect-error: 'ref' as never
      return <Component {...props} ref={ref} styles={styles} />;
    };

    return forwardRef(renderComponent);
  }

  return WithStyles;
}