#include "rive/nested_artboard_leaf.hpp"
#include "rive/renderer.hpp"
#include "rive/layout_component.hpp"
#include "rive/artboard.hpp"

using namespace rive;

Core* NestedArtboardLeaf::clone() const
{
  NestedArtboardLeaf* nestedArtboard = static_cast<NestedArtboardLeaf*>(
      NestedArtboardLeafBase::clone());
  cloneReferencesInto(nestedArtboard);
  return nestedArtboard;
}

void NestedArtboardLeaf::update(ComponentDirt value)
{
  Super::update(value);
  auto artboard = artboardInstance();
  if (hasDirt(value, ComponentDirt::WorldTransform) && artboard != nullptr)
  {
    // The layout we size against. fitToLayoutParent is absent from every
    // file written before it existed, so legacy files default to false and
    // take the direct-parent reach they were authored against: a Solo or
    // Bone between us and a layout stops the sizing, and we frame
    // ourselves. Opted in, sizing reaches through those transparent
    // containers to the layout that actually owns the space.
    auto* p = parent();
    auto* sizingLayout = fitToLayoutParent()
                             ? contentSizingLayout(p)
                             : (p != nullptr && p->is<LayoutComponent>()
                                    ? p->as<LayoutComponent>()
                                    : nullptr);

    AABB bounds = sizingLayout != nullptr ? sizingLayout->localBounds()
                                          : artboard->bounds();

    auto artboardFit = (Fit)fit();
    if (artboardFit == Fit::layout)
    {
      // Layout fit owns the artboard's size: match the leaf's bounds
      // and let the artboard's own layout reflow.
      bool resized = false;
      if (artboard->width() != bounds.width())
      {
        artboard->width(bounds.width());
        resized = true;
      }
      if (artboard->height() != bounds.height())
      {
        artboard->height(bounds.height());
        resized = true;
      }
      if (resized)
      {
        // The instance already advanced; reflow now or this frame
        // draws the old layout.
        artboard->updatePass(false);
      }
    }

    auto viewTransform = computeAlignment(
        artboardFit,
        Alignment(alignmentX(), alignmentY()),
        bounds,
        artboard->bounds());

    m_WorldTransform *= viewTransform;
  }
}