#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;
    }
  }

  // Removed m_Exit and m_Any from this validation in order to be able to
  // support states with no exit or any state in the future once this runtime
  // has been around for some time
  if (m_Entry == nullptr)
  {
    // The layer is corrupt, we must have an entry state.
    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);
}