expo/src/dom/dom-hooks.ts

The Expo SDK

import { useCallback, useEffect, useImperativeHandle, type DependencyList, type Ref } from 'react';

import type { DOMImperativeFactory } from './dom.types';
import { REGISTER_DOM_IMPERATIVE_HANDLE_PROPS } from './injection';

declare namespace globalThis {
  let _domRefProxy: undefined | unknown;
}

/**
 * A React `useImperativeHandle` like hook for DOM components.
 *
 */
export function useDOMImperativeHandle<T extends DOMImperativeFactory>(
  ref: Ref<T>,
  init: () => T,
  deps?: DependencyList
) {
  const isTargetWeb =
    // @ts-expect-error: Added via react-native-webview
    typeof window.ReactNativeWebView === 'undefined' &&
    // @ts-expect-error: Added via expo/dom
    typeof window.$$EXPO_INITIAL_PROPS === 'undefined';

  const stubHandlerFactory = useCallback(() => ({}) as T, deps ?? []);

  // This standard useImperativeHandle hook is serving for web
  useImperativeHandle(ref, isTargetWeb ? init : stubHandlerFactory, deps);

  // This `globalThis._domRefProxy` is serving for native
  useEffect(() => {
    if (!isTargetWeb) {
      globalThis._domRefProxy = init();
      // TODO(@kitten): Type `ReactNativeWebView` and the message data
      // @ts-expect-error: Added via react-native-webview
      window.ReactNativeWebView.postMessage(
        JSON.stringify({
          type: REGISTER_DOM_IMPERATIVE_HANDLE_PROPS,
          data: Object.keys(globalThis._domRefProxy as any),
        })
      );
    }
    return () => {
      if (!isTargetWeb) {
        globalThis._domRefProxy = undefined;
      }
    };
  }, deps);
}