#include "rive/layout/grid_item_placement.hpp"
#include "rive/container_component.hpp"
#include "rive/layout/grid_track.hpp"
#include "rive/layout/layout_node_provider.hpp"

using namespace rive;

GridItemPlacement* GridItemPlacement::from(const ContainerComponent* owner)
{
  if (owner == nullptr)
  {
    return nullptr;
  }
  for (auto* child : owner->children())
  {
    if (child->is<GridItemPlacement>())
    {
      return child->as<GridItemPlacement>();
    }
  }
  return nullptr;
}

StatusCode GridItemPlacement::onAddedClean(CoreContext* context)
{
  StatusCode code = Super::onAddedClean(context);
  if (code != StatusCode::Ok)
  {
    return code;
  }
#ifdef WITH_RIVE_LAYOUT
  // from() already resolves both shapes: a LayoutComponent provides for
  // itself, while Text/Image/Shape provide via their LayoutParticipant child.
  // onAddedClean runs after every object has resolved its parent, so the
  // sibling provider is findable now.
  if (auto* provider = LayoutNodeProvider::from(parent()))
  {
    provider->addLayoutStyleApplier(this);
  }
#endif
  return StatusCode::Ok;
}

void GridItemPlacement::buildDependencies()
{
  Super::buildDependencies();
  if (parent() != nullptr)
  {
    parent()->addDependent(this);
  }
}

#ifdef WITH_RIVE_LAYOUT
void GridItemPlacement::applyPlacementStyle(
    YGStyle& style,
    const LayoutSyncContext& context)
{
  // A stack collapses every child into its single cell, so explicit placement
  // has nothing to say. Note this is a no-op rather than an absent applier:
  // reparenting between a grid and a stack must not create or destroy the
  // object, or the authored placement would be lost on the round trip.
  if (context.parentIsStack || !context.parentIsGrid)
  {
    return;
  }
  GridTrack::syncItemLines(
      style,
      gridColumn(),
      gridRow(),
      gridColumnSpan(),
      gridRowSpan());
}
#endif

void GridItemPlacement::markOwnerDirty()
{
  if (auto* provider = LayoutNodeProvider::from(parent()))
  {
    provider->markLayoutNodeDirty();
  }
}

void GridItemPlacement::gridColumnChanged() { markOwnerDirty(); }
void GridItemPlacement::gridRowChanged() { markOwnerDirty(); }
void GridItemPlacement::gridColumnSpanChanged() { markOwnerDirty(); }
void GridItemPlacement::gridRowSpanChanged() { markOwnerDirty(); }