#include "rive/parent_traversal.hpp"
#include "rive/artboard.hpp"
#include "rive/artboard_host.hpp"
#include "rive/component.hpp"
#include "rive/container_component.hpp"

using namespace rive;

ParentTraversal::ParentTraversal(Component* start)
    : m_current(start != nullptr ? start->parent() : nullptr)
    , m_currentArtboard(start != nullptr ? start->artboard() : nullptr)
{}

ContainerComponent* ParentTraversal::next()
{
  // Reset boundary crossing state
  m_didCrossBoundary = false;
  m_crossingHost = nullptr;
  m_sourceArtboard = nullptr;

  if (m_current == nullptr)
  {
    return nullptr;
  }

  ContainerComponent* result = m_current;

  // Advance to next parent
  if (m_current->parent() != nullptr)
  {
    m_current = m_current->parent();
  }
  else if (m_currentArtboard != nullptr && m_currentArtboard->host() != nullptr)
  {
    // No direct parent - try crossing artboard boundary
    auto* host = m_currentArtboard->host();
    auto* hostComponent = host->hostComponent();

    if (hostComponent != nullptr && hostComponent->parent() != nullptr)
    {
      // Record boundary crossing info before updating state
      m_didCrossBoundary = true;
      m_crossingHost = host;
      m_sourceArtboard = m_currentArtboard;

      m_current = hostComponent->parent();
      m_currentArtboard = host->parentArtboard();
    }
    else
    {
      m_current = nullptr;
    }
  }
  else
  {
    m_current = nullptr;
  }

  return result;
}