Function classed

packages/shared/src/lib/classed.ts:127–162

(
  type: ElementType,
  className: string | Record<string, string>,
  ...classes: string[]
)

): ForwardRefExoticComponent<PropsWithoutRef<P> & RefAttributes<Component<P>>>;

function classed<T, P extends Record<string, unknown>>(
  type: ElementType,
  className: string | Record<string, string>,
  ...classes: string[]
): ForwardRefExoticComponent<PropsWithoutRef<P> & RefAttributes<T>> {
  return forwardRef<T, P>(function Classed(props, ref) {
    if (typeof className === 'string') {
      const inputClassName =
        typeof props.className === 'string' ? props.className : undefined;

      return React.createElement(type, {
        ...props,
        className: classNames(inputClassName, className, ...classes),
        ref,
      });
    }

    if (
      !isNullOrUndefined(props.className) &&
      typeof className !== typeof props.className
    ) {
      throw new Error('Incompatible className and parameter');
    }

    return React.createElement(type, {
      ...props,
      className: isNullOrUndefined(props.className)
        ? className
        : combineObjectStrings(
            className,
            props.className as Record<string, string>,
          ),
      ref,
    });
  });
}

export default classed;