#include "rive/animation/state_machine_layer.hpp"
#include "rive/importers/import_stack.hpp"
#include "rive/importers/state_machine_importer.hpp"
#include "rive/generated/animation/state_machine_base.hpp"
#include "rive/animation/any_state.hpp"
#include "rive/animation/entry_state.hpp"
#include "rive/animation/exit_state.hpp"

using namespace rive;

StateMachineLayer::~StateMachineLayer()
{
    for (auto state : m_States)
    {
        delete state;
    }
}

StatusCode StateMachineLayer::onAddedDirty(CoreContext* context)
{
    StatusCode code;
    for (auto state : m_States)
    {
        if ((code = state->onAddedDirty(context)) != StatusCode::Ok)
        {
            return code;
        }
        switch (state->coreType())
        {
            case AnyState::typeKey:
                m_Any = state->as<AnyState>();
                break;
            case EntryState::typeKey:
                m_Entry = state->as<EntryState>();
                break;
            case ExitState::typeKey:
                m_Exit = state->as<ExitState>();
                break;
        }
    }
    if (m_Any == nullptr || m_Entry == nullptr || m_Exit == nullptr)
    {
        // The layer is corrupt, we must have all three of these states.
        return StatusCode::InvalidObject;
    }

    return StatusCode::Ok;
}

StatusCode StateMachineLayer::onAddedClean(CoreContext* context)
{
    StatusCode code;
    for (auto state : m_States)
    {
        if ((code = state->onAddedClean(context)) != StatusCode::Ok)
        {
            return code;
        }
    }

    return StatusCode::Ok;
}

void StateMachineLayer::addState(LayerState* state) { m_States.push_back(state); }

StatusCode StateMachineLayer::import(ImportStack& importStack)
{
    auto stateMachineImporter = importStack.latest<StateMachineImporter>(StateMachineBase::typeKey);
    if (stateMachineImporter == nullptr)
    {
        return StatusCode::MissingObject;
    }
    // WOW -- we're handing off ownership of this!
    stateMachineImporter->addLayer(std::unique_ptr<StateMachineLayer>(this));
    return Super::import(importStack);
}