import * as React from 'react';
import { StyleSheet, View, type ViewStyle } from 'react-native';
import useLatestCallback from 'use-latest-callback';

import type { DrawerProps } from '../types';
import { DrawerProgressContext } from '../utils/DrawerProgressContext';
import { getDrawerWidthWeb } from '../utils/getDrawerWidth';
import { useFakeSharedValue } from '../utils/useFakeSharedValue';
import { Overlay } from './Overlay';

export function Drawer({
  direction = 'ltr',
  drawerPosition = direction === 'rtl' ? 'right' : 'left',
  drawerStyle,
  drawerType = 'front',
  onClose,
  onTransitionStart,
  onTransitionEnd,
  open,
  overlayStyle,
  overlayAccessibilityLabel,
  renderDrawerContent,
  children,
  style,
}: DrawerProps) {
  const drawerWidth = getDrawerWidthWeb({
    drawerStyle,
  });

  const progress = useFakeSharedValue(open ? 1 : 0);

  React.useEffect(() => {
    progress.value = open ? 1 : 0;
  }, [open, progress]);

  const drawerRef = React.useRef<View>(null);

  const onTransitionStartLatest = useLatestCallback(() => {
    onTransitionStart?.(open === false);
  });

  const onTransitionEndLatest = useLatestCallback(() => {
    onTransitionEnd?.(open === false);
  });

  React.useEffect(() => {
    const element = drawerRef.current as HTMLDivElement | null;

    element?.addEventListener('transitionstart', onTransitionStartLatest);
    element?.addEventListener('transitionend', onTransitionEndLatest);

    return () => {
      element?.removeEventListener('transitionstart', onTransitionStartLatest);
      element?.removeEventListener('transitionend', onTransitionEndLatest);
    };
  }, [onTransitionEndLatest, onTransitionStartLatest]);

  const isOpen = drawerType === 'permanent' ? true : open;
  const isRight = drawerPosition === 'right';

  const drawerTranslateX =
    // The drawer stays in place at open position when `drawerType` is `back`
    open || drawerType === 'back'
      ? drawerPosition === 'left'
        ? '100%'
        : '-100%'
      : 0;

  const drawerAnimatedStyle =
    drawerType !== 'permanent'
      ? {
          transition: 'transform 0.3s',
          transform: `translateX(${drawerTranslateX})`,
        }
      : null;

  const contentTranslateX = open
    ? // The screen content stays in place when `drawerType` is `front`
      drawerType === 'front'
      ? 0
      : `calc(${drawerWidth} * ${drawerPosition === 'left' ? 1 : -1})`
    : 0;

  const contentAnimatedStyle =
    drawerType !== 'permanent'
      ? {
          transition: 'transform 0.3s',
          transform: `translateX(${contentTranslateX})`,
        }
      : null;

  const drawerElement = (
    <View
      key="drawer"
      ref={drawerRef}
      style={[
        styles.drawer,
        {
          position: drawerType === 'permanent' ? 'relative' : 'absolute',
          zIndex: drawerType === 'back' ? -1 : 1,
        },
        // FIXME: width contains `px` on web
        { width: drawerWidth } as ViewStyle,
        // @ts-expect-error offset contains `calc` for web
        drawerType !== 'permanent'
          ? // Position drawer off-screen by default in closed state
            // And add a translation only when drawer is open
            // So changing position in closed state won't trigger a visible transition
            drawerPosition === 'right'
            ? { right: `calc(${drawerWidth} * -1)` }
            : { left: `calc(${drawerWidth} * -1)` }
          : null,
        drawerAnimatedStyle,
        drawerStyle,
      ]}
    >
      {renderDrawerContent()}
    </View>
  );

  const mainContent = (
    <View key="content" style={[styles.content, contentAnimatedStyle]}>
      <View
        aria-hidden={isOpen && drawerType !== 'permanent'}
        style={styles.content}
      >
        {children}
      </View>
      {drawerType !== 'permanent' ? (
        <Overlay
          open={open}
          progress={progress}
          onPress={() => onClose()}
          style={overlayStyle}
          accessibilityLabel={overlayAccessibilityLabel}
        />
      ) : null}
    </View>
  );

  return (
    <DrawerProgressContext.Provider value={progress}>
      <View style={[styles.container, style]}>
        {!isRight && drawerElement}
        {mainContent}
        {isRight && drawerElement}
      </View>
    </DrawerProgressContext.Provider>
  );
}

const styles = StyleSheet.create({
  container: {
    flex: 1,
    flexDirection: 'row',
  },
  drawer: {
    top: 0,
    bottom: 0,
    maxWidth: '100%',
    backgroundColor: 'white',
  },
  content: {
    flex: 1,
  },
});