#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());
    nestedArtboard->file(file());
    if (m_referencedArtboard == nullptr)
    {
        return nestedArtboard;
    }
    auto ni = m_referencedArtboard->instance();
    nestedArtboard->referencedArtboard(ni.release());
    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;
    }
}