/*
 * Copyright 2024 Rive
 */

#include "rive/input/focus_node.hpp"
#include "rive/input/focus_manager.hpp"
#include <algorithm>
#include <cstddef>

namespace rive
{

FocusNode::~FocusNode()
{
    // A child held elsewhere via rcp (a NestedArtboard's persistent focus
    // scope, an ArtboardComponentList row) can outlive this node, so clear its
    // back-pointer to keep m_parent from dangling.
    for (const auto& child : m_children)
    {
        child->m_parent = nullptr;
    }
}

void FocusNode::addChild(rcp<FocusNode> child)
{
    insertChild(m_children.size(), std::move(child));
}

void FocusNode::insertChild(size_t index, rcp<FocusNode> child)
{
    if (!child)
    {
        return;
    }
    child->removeFromParent();
    child->m_parent = this;
    if (index > m_children.size())
    {
        index = m_children.size();
    }
    m_children.insert(m_children.begin() + static_cast<ptrdiff_t>(index),
                      std::move(child));
    invalidateFocusableContent();
}

void FocusNode::removeChild(rcp<FocusNode> child)
{
    if (!child || child->m_parent != this)
    {
        return;
    }

    child->m_parent = nullptr;
    auto it = std::find(m_children.begin(), m_children.end(), child);
    if (it != m_children.end())
    {
        m_children.erase(it);
    }
    invalidateFocusableContent();
}

void FocusNode::invalidateFocusableContent()
{
    if (m_manager != nullptr)
    {
        m_manager->markFocusableContentDirty();
    }
}

void FocusNode::removeFromParent()
{
    if (m_parent)
    {
        m_parent->removeChild(ref_rcp(this));
    }
}

} // namespace rive