react-native-tvos/Libraries/Components/TV/tagForComponentOrHandle.js

A framework for building native apps using React

// Replacement for findNodeHandle, since we need this for converting
// components to tags when rendering nextFocus props
// @flow

// TODO: make this work for Fabric

// import {findNodeHandle} from '../../Renderer/shims/ReactNative';

type TagForComponentOrHandleType = (
    component: ?(
      | {_nativeTag: ?number, canonical: ?{_nativeTag: ?number}}
      | number
    ),
  ) => ?number;

const tagForComponentOrHandle: TagForComponentOrHandleType = (
  component: ?(
    | {_nativeTag: ?number, canonical: ?{_nativeTag: ?number}}
    | number
  ),
): ?number => {
  if (typeof component === 'number') {
    return component;
  }
  if (typeof component?._nativeTag === 'number') {
    return component?._nativeTag;
  }
  if (typeof component?.canonical?._nativeTag === 'number') {
    return component?.canonical?._nativeTag;
  }
  return undefined;
};

export default tagForComponentOrHandle;