import { Slot as RUISlot } from '@radix-ui/react-slot'
import {
  type Component,
  type ForwardRefExoticComponent,
  forwardRef,
  type RefAttributes,
  useMemo,
} from 'react'
import { StyleSheet, type ViewProps } from 'react-native'

/**
 * RadixUI has special logic to handle the merging of `style` and `className` props.
 * On the web styles are not allowed so Radix does not handle this scenario.
 * This could be fixed upstream (PR open), but it may not as RN is not their target
 * platform.
 *
 * This shim calls `StyleSheet.flatten` on the styles before we render the <Slot />
 *
 * @see https://github.com/expo/expo/issues/31352
 * @see https://github.com/radix-ui/primitives/issues/3107
 * @param Component
 * @returns
 */
function ShimSlotForReactNative(Component: typeof RUISlot): typeof RUISlot {
  return forwardRef(function RNSlotHOC({ style, ...props }, ref) {
    if (process.env.NODE_ENV !== 'production') {
      if (Array.isArray(style)) {
        throw new Error(
          'Slot does not support array styles. Use StyleSheet.flatten() or an object style instead.'
        )
      }
    }
    style = useMemo(() => StyleSheet.flatten(style), [style])
    return <Component ref={ref} {...props} style={style} />
  })
}

export interface Slot<
  Props = ViewProps,
  Ref = Component<ViewProps>,
> extends ForwardRefExoticComponent<Props & RefAttributes<Ref>> {}

export const Slot: Slot = ShimSlotForReactNative(RUISlot) as Slot