import { useNavigation } from '@react-native-bedrock/native/@react-navigation/native';
import { NativeStackNavigationProp } from '@react-native-bedrock/native/@react-navigation/native-stack';
import { useCallback, useRef } from 'react';
import { useVisibilityChange, type VisibilityState } from '../visibility';

/**
 * @public
 * @category 화면 제어
 * @name useWaitForReturnNavigator
 * @description
 * 화면 전환을 하고 돌아왔을 때 다음 코드를 동기적으로 실행할 수 있도록 도와주는 Hook 이에요.
 * 화면 이동은 [@react-navigation/native `useNavigation`의 `navigate`](https://reactnavigation.org/docs/6.x/navigation-prop#navigate)를 사용해요.
 *
 * 예를 들어, 사용자가 다른 화면으로 이동했다가 돌아왔다는 로그를 남기고 싶을 때 사용해요.
 *
 * @example
 * ### 화면 이동 후 돌아왔을 때 코드가 실행되는 예제
 *
 * **"이동하기"** 버튼을 누르면 다른 화면으로 이동하고, 돌아왔을 때 로그가 남겨져요.
 *
 * ```tsx
 * import { Button } from 'react-native';
 * import { useWaitForReturnNavigator } from 'react-native-bedrock';
 *
 * export function UseWaitForReturnNavigator() {
 *   const navigate = useWaitForReturnNavigator();
 *
 *   return (
 *     <>
 *       <Button
 *         title="이동하기"
 *         onPress={async () => {
 *           console.log(1);
 *           await navigate('/examples/use-visibility');
 *           // 화면에 돌아오면 이 코드가 실행됩니다.
 *           console.log(2);
 *         }}
 *       />
 *     </>
 *   );
 * }
 * ```
 */

export function useWaitForReturnNavigator<T extends Record<string, object | undefined>>() {
  const callbacks = useRef<Array<() => void>>([]).current;
  const navigation = useNavigation<NativeStackNavigationProp<T>>();

  const startNavigating = useCallback(
    <RouteName extends keyof T>(route: RouteName, params?: T[RouteName]): Promise<void> => {
      return new Promise<void>((resolve) => {
        callbacks.push(resolve);
        navigation.navigate(route as any, params as any);
      });
    },
    [callbacks, navigation]
  );

  const handleVisibilityChange = useCallback(
    (state: VisibilityState) => {
      if (state === 'visible' && callbacks.length > 0) {
        for (const callback of callbacks) {
          callback();
        }

        callbacks.splice(0, callbacks.length);
      }
    },
    [callbacks]
  );

  useVisibilityChange(handleVisibilityChange);

  return startNavigating;
}