import { useMemo, type PropsWithChildren } from 'react';
import {
  useSharedValue,
  type WithSpringConfig,
  type WithTimingConfig,
} from 'react-native-reanimated';

import {
  DefaultAnimationConfigs,
  DefaultPressableConfig,
  type DefaultPressableProps,
  type PressableConfig,
} from './constants';
import {
  PressablesContext,
  PressablesGroupContext,
  type AnimatedPressableOptions,
  type AnimationType,
  type PressableContextType,
} from './context';

export type PressablesConfigProps<
  T extends AnimationType,
  TMetadata = unknown,
> = {
  children?: React.ReactNode;
  animationType?: T;
  animationConfig?: T extends 'timing' ? WithTimingConfig : WithSpringConfig;
  globalHandlers?: {
    onPressIn?: (options: AnimatedPressableOptions<TMetadata>) => void;
    onPressOut?: (options: AnimatedPressableOptions<TMetadata>) => void;
    onPress?: (options: AnimatedPressableOptions<TMetadata>) => void;
  };
  metadata?: TMetadata;
  /**
   * Activates the pressable animation on hover (web only)
   * @platform web
   */
  activateOnHover?: boolean;
  /**
   * Pressable configuration values (opacity, scale, etc.)
   */
  config?: Partial<PressableConfig>;
  /**
   * Default props applied to all pressables.
   * Individual pressable props will override these values.
   *
   * @example
   * // Disable ripple effect globally on Android
   * <PressablesConfig defaultProps={{ rippleColor: 'transparent' }}>
   *   ...
   * </PressablesConfig>
   */
  defaultProps?: DefaultPressableProps;
};

export const PressablesGroup = ({ children }: PropsWithChildren) => {
  const lastTouchedPressable = useSharedValue<string | null>(null);

  const groupValue = useMemo(() => {
    return {
      lastTouchedPressable: lastTouchedPressable,
    };
  }, [lastTouchedPressable]);

  return (
    <PressablesGroupContext.Provider value={groupValue}>
      {children}
    </PressablesGroupContext.Provider>
  );
};

export const PressablesConfig = <T extends AnimationType, TMetadata = unknown>({
  children,
  animationType = 'timing' as T,
  animationConfig,
  globalHandlers,
  metadata,
  activateOnHover,
  config,
  defaultProps,
}: PressablesConfigProps<T, TMetadata>) => {
  const value = useMemo(() => {
    return {
      animationType,
      animationConfig: animationConfig ?? DefaultAnimationConfigs[animationType],
      globalHandlers,
      metadata,
      activateOnHover,
      config: { ...DefaultPressableConfig, ...config },
      defaultProps,
    };
  }, [animationType, animationConfig, globalHandlers, metadata, activateOnHover, config, defaultProps]);

  return (
    // metadata-typed handlers are erased to `unknown` at the context boundary;
    // BasePressable re-applies TMetadata via createAnimatedPressable<TMetadata>.
    <PressablesContext.Provider
      value={value as PressableContextType<AnimationType>}
    >
      <PressablesGroup>{children}</PressablesGroup>
    </PressablesContext.Provider>
  );
};

export * from './hooks';
export type { DefaultPressableProps, PressableConfig } from './constants';