#include "rive/draw_rules.hpp"
#include "rive/artboard.hpp"
#include "rive/core_context.hpp"
#include "rive/draw_target.hpp"

using namespace rive;

StatusCode DrawRules::onAddedDirty(CoreContext* context)
{
    StatusCode result = Super::onAddedDirty(context);
    if (result != StatusCode::Ok)
    {
        return result;
    }
    auto coreObject = context->resolve(drawTargetId());
    if (coreObject != nullptr && coreObject->is<DrawTarget>())
    {
#ifdef WITH_RIVE_EDITOR
        setActiveTargetForEditor(static_cast<DrawTarget*>(coreObject));
#else
        m_ActiveTarget = static_cast<DrawTarget*>(coreObject);
#endif
    }

    return StatusCode::Ok;
}

StatusCode DrawRules::onAddedClean(CoreContext* context)
{
    return StatusCode::Ok;
}

void DrawRules::drawTargetIdChanged()
{
    auto coreObject = artboard()->resolve(drawTargetId());
    DrawTarget* t = (coreObject != nullptr && coreObject->is<DrawTarget>())
                        ? static_cast<DrawTarget*>(coreObject)
                        : nullptr;
#ifdef WITH_RIVE_EDITOR
    setActiveTargetForEditor(t);
#else
    m_ActiveTarget = t;
#endif
    artboard()->addDirt(ComponentDirt::DrawOrder);
}