/**
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#include "RootShadowNode.h"

#include <react/components/view/conversions.h>
#include <react/debug/SystraceSection.h>

namespace facebook {
namespace react {

const char RootComponentName[] = "RootView";

void RootShadowNode::layout() {
  SystraceSection s("RootShadowNode::layout");
  ensureUnsealed();
  layout(getProps()->layoutContext);

  // This is the rare place where shadow node must layout (set `layoutMetrics`)
  // itself because there is no a parent node which usually should do it.
  setLayoutMetrics(layoutMetricsFromYogaNode(yogaNode_));
}

UnsharedRootShadowNode RootShadowNode::clone(
    const LayoutConstraints &layoutConstraints,
    const LayoutContext &layoutContext) const {
  auto props = std::make_shared<const RootProps>(
      *getProps(), layoutConstraints, layoutContext);
  auto newRootShadowNode = std::make_shared<RootShadowNode>(
      *this, ShadowNodeFragment{.props = props});
  return newRootShadowNode;
}

UnsharedRootShadowNode RootShadowNode::clone(
    const SharedShadowNode &oldShadowNode,
    const SharedShadowNode &newShadowNode) const {
  std::vector<std::reference_wrapper<const ShadowNode>> ancestors;
  oldShadowNode->constructAncestorPath(*this, ancestors);

  if (ancestors.size() == 0) {
    return UnsharedRootShadowNode{nullptr};
  }

  auto oldChild = oldShadowNode;
  auto newChild = newShadowNode;

  SharedShadowNodeUnsharedList sharedChildren;

  for (const auto &ancestor : ancestors) {
    auto children = ancestor.get().getChildren();
    std::replace(children.begin(), children.end(), oldChild, newChild);

    sharedChildren = std::make_shared<SharedShadowNodeList>(children);

    oldChild = ancestor.get().shared_from_this();
    newChild = oldChild->clone(ShadowNodeFragment{.children = sharedChildren});
  }

  return std::make_shared<RootShadowNode>(
      *this, ShadowNodeFragment{.children = sharedChildren});
}

} // namespace react
} // namespace facebook