#include "teststatemanager.h"
#include "MyEventTypes.h"
#include "MyStateTypes.h"

#include <vector>
#include <iostream>

#include <dtUtil/stringutils.h>
#include <dtUtil/log.h>

void LOG(const std::string& msg)
{
   dtUtil::Log::GetInstance().LogMessage(dtUtil::Log::LOG_ALWAYS,"",msg.c_str());
}

////////////////////////////////////////////////////////////////////////////////
StateWalker::StateWalker(dtABC::StateManager* gm)
   : BaseClass()
   , mStateManager(gm)
   , mStartState(0)
{
   if (mStateManager.valid())
   {
      mStateManager->AddSender(this);
   }

   AddSender(&dtCore::System::GetInstance());

   mStateManager->RegisterEvent<Alt>(&MyEventType::ALT);
   mStateManager->RegisterEvent<Start>(&MyEventType::START);

   mStateManager->RegisterState<Shell>(&MyStateType::SHELL);
   mStateManager->RegisterState<Options>(&MyStateType::OPTIONS);
   mStateManager->RegisterState<Game>(&MyStateType::GAME);

   if (dtABC::State* current = mStateManager->GetCurrentState())
   {
      mStartState = current;
   }
}

////////////////////////////////////////////////////////////////////////////////
StateWalker::~StateWalker()
{
   mStateManager->RemoveSender(this);
   RemoveSender(&dtCore::System::GetInstance());
}

////////////////////////////////////////////////////////////////////////////////
void StateWalker::OnMessage(dtCore::Base::MessageData* msg)
{
   if (msg->message == dtCore::System::MESSAGE_PRE_FRAME)
   {
      DisplayEventChoicesAndWaitForInput();
   }
}

////////////////////////////////////////////////////////////////////////////////
void StateWalker::DisplayEventChoicesAndWaitForInput()
{
   if (!mStateManager->GetCurrentState())
   {
      LOG_ERROR("No valid current State within StateManager, stopping the System.");
      dtCore::System::GetInstance().Stop();
      return;
   }

   LOG("Current State: "+mStateManager->GetCurrentState()->GetName());
   LOG("The Event choices are:");

   std::vector<const dtABC::Event::Type*> eventvec(mStateManager->GetNumOfEvents( mStateManager->GetCurrentState()));
   mStateManager->GetEvents(mStateManager->GetCurrentState(), eventvec);
   unsigned int esize = eventvec.size();

   // display the Event choices for those transitions
   unsigned int i(0);
   for (; i < esize; i++)
   {
      LOG(dtUtil::ToString(i) + ": " + eventvec[i]->GetName());
   }

   // print "extra" event choices
   DisplayExtraEventChoices(i);

   // --------------------------- handle input --------------------------- //
   LOG("Please choose an Event:");

   unsigned int eventchoice;
   std::cin >> eventchoice;
   LOG("Your choice was: " + dtUtil::ToString(eventchoice));

   if (eventchoice < esize)
   {
      const dtABC::Event::Type* eventtype = eventvec[eventchoice];
      dtCore::RefPtr<dtABC::StateManager::EventFactory> ef = mStateManager->GetEventFactory();
      if (ef->IsTypeSupported(eventtype))
      {
         dtCore::RefPtr<dtABC::Event> event = ef->CreateObject(eventtype);
         if (event.valid())
         {
            LOG("Sending Event of type: " + eventtype->GetName());
            SendMessage("event", event.get());
         }
      }
      else
      {  // some logging feedback
         LOG("Can not create Event of type: " + eventtype->GetName());
      }
   }

   else
   {  // honor the "extra" event choices
      HandleExtraEventChoices(esize, eventchoice);
   }
}

////////////////////////////////////////////////////////////////////////////////
// --- "extra" choices --- //
void StateWalker::DisplayExtraEventChoices(unsigned int index)
{
   LOG(dtUtil::ToString(index) + ": " + "Quit");
}

////////////////////////////////////////////////////////////////////////////////
void StateWalker::HandleExtraEventChoices(unsigned int eventvecsize, unsigned int choice)
{
   if (choice == eventvecsize)
   {
      LOG("StateWalker: Quit was chosen.");
      dtCore::System::GetInstance().Stop();
   }
}