#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;
	}
	stateMachineImporter->addLayer(this);
	return Super::import(importStack);
}