import type { ReactElementFiber } from './types';
import { getRoute } from '../reactNavigation/navigationStack';

export function readRoute(node: ReactElementFiber): any {
  const descriptor = node.memoizedProps?.descriptor;
  if (descriptor && descriptor.route) {
    return descriptor.route;
  }

  let route = node.memoizedProps?.route;
  if (route) {
    return route;
  }

  const state = node.memoizedProps?.state;
  route = getRoute(state);
  if (route) {
    return route;
  }

  return undefined;
}

function findRoute(node: ReactElementFiber): any {
  if (!node) {
    return undefined;
  }

  const route = readRoute(node);
  if (route) {
    return route;
  }

  return findRoute(node.return);
}

export default findRoute;