Move node cloning to YGConfig

Summary:
@public

Encapsulates node cloning within `YGConfig`.
This is necessary for allowing for context-aware cloning functions, which will ultimately allow for removal of weak global JNI references.

Reviewed By: shergin

Differential Revision: D14132608

fbshipit-source-id: 0dec114c8e172b1e34a4b7fd146c43f13c151ade

yoga/YGConfig.cpp

@@ -6,7 +6,7 @@
  */
 #include "YGConfig.h"
 
-YGConfig::YGConfig(YGLogger logger) {
+YGConfig::YGConfig(YGLogger logger) : cloneNodeCallback_{nullptr} {
   logger_.noContext = logger;
   loggerUsesContext_ = false;
 }
@@ -24,3 +24,14 @@ void YGConfig::log(
     logger_.noContext(config, node, logLevel, format, args);
   }
 }
+
+YGNodeRef YGConfig::cloneNode(YGNodeRef node, YGNodeRef owner, int childIndex) {
+  YGNodeRef clone = nullptr;
+  if (cloneNodeCallback_ != nullptr) {
+    clone = cloneNodeCallback_(node, owner, childIndex);
+  }
+  if (clone == nullptr) {
+    clone = YGNodeClone(node);
+  }
+  return clone;
+}

yoga/YGConfig.h

@@ -19,6 +19,7 @@ struct YGConfig {
       va_list args);
 
 private:
+  YGCloneNodeFunc cloneNodeCallback_ = nullptr;
   union {
     LogWithContextFn withContext;
     YGLogger noContext;
@@ -31,7 +32,6 @@ public:
   bool shouldDiffLayoutWithoutLegacyStretchBehaviour = false;
   bool printTree = false;
   float pointScaleFactor = 1.0f;
-  YGCloneNodeFunc cloneNodeCallback = nullptr;
   std::array<bool, facebook::yoga::enums::count<YGExperimentalFeature>()>
       experimentalFeatures = {};
   void* context = nullptr;
@@ -50,4 +50,9 @@ public:
   void setLogger(std::nullptr_t) {
     setLogger(YGLogger{nullptr});
   }
+
+  YGNodeRef cloneNode(YGNodeRef node, YGNodeRef owner, int childIndex);
+  void setCloneNodeCallback(YGCloneNodeFunc cloneNode) {
+    cloneNodeCallback_ = cloneNode;
+  }
 };

yoga/YGNode.cpp

@@ -404,16 +404,9 @@ void YGNode::cloneChildrenIfNeeded() {
     return;
   }
 
-  const YGCloneNodeFunc cloneNodeCallback = config_->cloneNodeCallback;
   for (uint32_t i = 0; i < childCount; ++i) {
     const YGNodeRef oldChild = children_[i];
-    YGNodeRef newChild = nullptr;
-    if (cloneNodeCallback) {
-      newChild = cloneNodeCallback(oldChild, this, i);
-    }
-    if (newChild == nullptr) {
-      newChild = YGNodeClone(oldChild);
-    }
+    YGNodeRef newChild = config_->cloneNode(oldChild, this, i);
     replaceChild(newChild, i);
     newChild->setOwner(this);
   }

yoga/Yoga.cpp

@@ -425,8 +425,6 @@ void YGNodeRemoveChild(const YGNodeRef owner, const YGNodeRef excludedChild) {
   // Otherwise we have to clone the node list except for the child we're trying
   // to delete. We don't want to simply clone all children, because then the
   // host will need to free the clone of the child that was just deleted.
-  const YGCloneNodeFunc cloneNodeCallback =
-      owner->getConfig()->cloneNodeCallback;
   uint32_t nextInsertIndex = 0;
   for (uint32_t i = 0; i < childCount; i++) {
     const YGNodeRef oldChild = owner->getChild(i);
@@ -437,13 +435,8 @@ void YGNodeRemoveChild(const YGNodeRef owner, const YGNodeRef excludedChild) {
       owner->markDirtyAndPropogate();
       continue;
     }
-    YGNodeRef newChild = nullptr;
-    if (cloneNodeCallback) {
-      newChild = cloneNodeCallback(oldChild, owner, nextInsertIndex);
-    }
-    if (newChild == nullptr) {
-      newChild = YGNodeClone(oldChild);
-    }
+    YGNodeRef newChild =
+        owner->getConfig()->cloneNode(oldChild, owner, nextInsertIndex);
     owner->replaceChild(newChild, nextInsertIndex);
     newChild->setOwner(owner);
 
@@ -4293,7 +4286,7 @@ void* YGConfigGetContext(const YGConfigRef config) {
 void YGConfigSetCloneNodeFunc(
     const YGConfigRef config,
     const YGCloneNodeFunc callback) {
-  config->cloneNodeCallback = callback;
+  config->setCloneNodeCallback(callback);
 }
 
 static void YGTraverseChildrenPreOrder(