paseo/packages/app/src/utils/web-focus.ts

interface FocusWithRetriesOptions {
  focus: () => void;
  isFocused: () => boolean;
  deferInitialAttempt?: boolean;
  timeoutMs?: number;
  onSuccess?: () => void;
  onTimeout?: () => void;
}

export function focusWithRetries({
  focus,
  isFocused,
  deferInitialAttempt = false,
  timeoutMs = 1500,
  onSuccess,
  onTimeout,
}: FocusWithRetriesOptions): () => void {
  let cancelled = false;
  const deadlineMs = Date.now() + timeoutMs;

  const tick = () => {
    if (cancelled) return;

    try {
      focus();
    } catch {
      // ignore
    }

    if (isFocused()) {
      onSuccess?.();
      return;
    }

    if (Date.now() >= deadlineMs) {
      onTimeout?.();
      return;
    }

    requestAnimationFrame(() => {
      requestAnimationFrame(tick);
    });
  };

  if (deferInitialAttempt) {
    requestAnimationFrame(tick);
  } else {
    tick();
  }

  return () => {
    cancelled = true;
  };
}