[skip ci] Fabric: Export shadow node family to javascript (facebook#51146)

Summary:

InteresectionObserver caches shadow nodes in javascript.

This is hazardous: caching shadow nodes could lead to memory leaks.

Therefore, this diff exports shadow node families to javascript via fabric ui manager.

This way, we don't have to cache the shadow nodes in js.

Changelog: [

Reviewed By: yungsters

Differential Revision: D74262805

Author: RSNara

packages/react-native/Libraries/ReactNative/FabricUIManager.js

@@ -27,6 +27,7 @@ import defineLazyObjectProperty from '../Utilities/defineLazyObjectProperty';
 
 export type NodeSet = Array<Node>;
 export type NodeProps = {...};
+export opaque type ShadowNodeFamilyReference = mixed;
 export interface Spec {
   +createNode: (
     reactTag: number,
@@ -40,6 +41,7 @@ export interface Spec {
   +cloneNodeWithNewProps: (node: Node, newProps: NodeProps) => Node;
   +cloneNodeWithNewChildrenAndProps: (node: Node, newProps: NodeProps) => Node;
   +createChildSet: (rootTag: RootTag) => NodeSet;
+  +createShadowNodeFamilyReference: (node: Node) => ShadowNodeFamilyReference;
   +appendChild: (parentNode: Node, child: Node) => Node;
   +appendChildToSet: (childSet: NodeSet, child: Node) => void;
   +completeRoot: (rootTag: RootTag, childSet: NodeSet) => void;
@@ -111,6 +113,7 @@ const CACHED_PROPERTIES = [
   'cloneNodeWithNewProps',
   'cloneNodeWithNewChildrenAndProps',
   'createChildSet',
+  'createShadowNodeFamilyReference',
   'appendChild',
   'appendChildToSet',
   'completeRoot',

packages/react-native/ReactCommon/react/renderer/core/ShadowNodeFamily.h

@@ -39,7 +39,7 @@ struct ShadowNodeFamilyFragment {
  * Represents all things that shadow nodes from the same family have in common.
  * To be used inside `ShadowNode` class *only*.
  */
-class ShadowNodeFamily final {
+class ShadowNodeFamily final : public jsi::NativeState {
  public:
   using Shared = std::shared_ptr<const ShadowNodeFamily>;
   using Weak = std::weak_ptr<const ShadowNodeFamily>;

packages/react-native/ReactCommon/react/renderer/uimanager/UIManagerBinding.cpp

@@ -235,6 +235,24 @@ jsi::Value UIManagerBinding::get(
         });
   }
 
+  if (methodName == "createShadowNodeFamilyReference") {
+    auto paramCount = 1;
+    return jsi::Function::createFromHostFunction(
+        runtime,
+        name,
+        paramCount,
+        [methodName, paramCount](
+            jsi::Runtime& runtime,
+            const jsi::Value& /*thisValue*/,
+            const jsi::Value* arguments,
+            size_t count) {
+          validateArgumentCount(runtime, methodName, paramCount, count);
+
+          auto node = shadowNodeFromValue(runtime, arguments[0]);
+          return valueFromShadowNodeFamily(runtime, node->getFamilyShared());
+        });
+  }
+
   if (methodName == "setIsJSResponder") {
     auto paramCount = 3;
     return jsi::Function::createFromHostFunction(

packages/react-native/ReactCommon/react/renderer/uimanager/primitives.h

@@ -42,6 +42,16 @@ inline static ShadowNode::Shared shadowNodeFromValue(
       ->shadowNode;
 }
 
+inline static ShadowNodeFamily::Shared shadowNodeFamilyFromValue(
+    jsi::Runtime& runtime,
+    const jsi::Value& value) {
+  if (value.isNull()) {
+    return nullptr;
+  }
+
+  return value.getObject(runtime).getNativeState<ShadowNodeFamily>(runtime);
+}
+
 inline static jsi::Value valueFromShadowNode(
     jsi::Runtime& runtime,
     ShadowNode::Shared shadowNode,
@@ -60,6 +70,17 @@ inline static jsi::Value valueFromShadowNode(
   return obj;
 }
 
+inline static jsi::Value valueFromShadowNodeFamily(
+    jsi::Runtime& runtime,
+    ShadowNodeFamily::Shared shadowNodeFamily) {
+  jsi::Object obj(runtime);
+  // Need to const_cast since JSI only allows non-const pointees
+  obj.setNativeState(
+      runtime,
+      std::const_pointer_cast<ShadowNodeFamily>(std::move(shadowNodeFamily)));
+  return obj;
+}
+
 // TODO: once we no longer need to mutate the return value (appendChildToSet)
 // make this a SharedListOfShared
 inline static ShadowNode::UnsharedListOfShared shadowNodeListFromValue(