#include <boost/filesystem/operations.hpp>
#include <time.h>

#include <Util/Util.h>
#include <Network/PacketContainer.h>
#include <Logging/Logger.h>

#include "Network/GameConnection.h"
#include "Actor/Player.h"

#include "Session.h"
#include "Framework.h"


extern Core::Framework g_fw;

Core::Session::Session( uint32_t sessionId ) :
   m_sessionId( sessionId ),
   m_lastDataTime( Util::getTimeSeconds() ),
   m_lastSqlTime( Util::getTimeSeconds() ),
   m_isValid( false )
{
}

Core::Session::~Session()
{
}

void Core::Session::setZoneConnection( Network::GameConnectionPtr pZoneCon )
{
   pZoneCon->m_conType = Network::ConnectionType::Zone;
   m_pZoneConnection = pZoneCon;
}

void Core::Session::setChatConnection( Network::GameConnectionPtr pChatCon )
{
    pChatCon->m_conType = Network::ConnectionType::Chat;
    m_pChatConnection = pChatCon;
}

Core::Network::GameConnectionPtr Core::Session::getZoneConnection() const
{
   return m_pZoneConnection;
}

Core::Network::GameConnectionPtr Core::Session::getChatConnection() const
{
    return m_pChatConnection;
}


bool Core::Session::loadPlayer()
{

   m_pPlayer = Entity::make_Player();

   if( !m_pPlayer->load( m_sessionId, shared_from_this() ) )
   {
      m_isValid = false;
      return false;
   }
   
   m_isValid = true;

   return true;

}

void Core::Session::close()
{
   if( m_pZoneConnection )
      m_pZoneConnection->Disconnect();

   if( m_pChatConnection )
      m_pChatConnection->Disconnect();

   // remove the session from the player
   if( m_pPlayer )
      // reset the zone, so the zone handler knows to remove the actor
      m_pPlayer->setCurrentZone( nullptr );
}

uint32_t Core::Session::getId() const
{
   return m_sessionId;
}

int64_t Core::Session::getLastDataTime() const
{
   return m_lastDataTime;
}

int64_t Core::Session::getLastSqlTime() const
{
   return m_lastSqlTime;
}

bool Core::Session::isValid() const
{
   return m_isValid;
}

void Core::Session::updateLastDataTime()
{
   m_lastDataTime = Util::getTimeSeconds();
}

void Core::Session::updateLastSqlTime()
{
   m_lastSqlTime = Util::getTimeSeconds();
}

void Core::Session::startReplay( const std::string& path )
{
   auto pLog = g_fw.get< Logger >();
   if( !boost::filesystem::exists( path ) )
   {
      getPlayer()->sendDebug( "Couldn't find folder." );
      return;
   }

   m_replayCache.clear();

   std::vector< std::tuple< uint64_t, std::string > > loadedSets;

   for( auto it = boost::filesystem::directory_iterator( boost::filesystem::path( path ) );
        it != boost::filesystem::directory_iterator(); ++it )
   {
      // Get the filename of the current element
      auto fileName = it->path().filename().string();
      auto unixTime = std::stoull( fileName.substr( 0, 14 ).c_str() );

      if( unixTime > 1000000000 )
      {
         loadedSets.push_back( std::tuple< uint64_t, std::string >( unixTime, it->path().string() ) );
      }
   }

   sort( loadedSets.begin(), loadedSets.end(),
         []( const std::tuple< uint64_t, std::string >& left, const std::tuple< uint64_t, std::string >& right)
   {
      return std::get< 0 >( left ) < std::get< 0 >( right );
   } );

   uint64_t startTime = std::get< 0 >( loadedSets.at( 0 ) );

   for( auto set : loadedSets )
   {
      m_replayCache.push_back( std::tuple< uint64_t, std::string >(
         Util::getTimeMs() + ( std::get< 0 >( set ) - startTime ), std::get< 1 >( set ) ) );

      pLog->info( "Registering " + std::get< 1 >( set ) + " for " + std::to_string( std::get< 0 >( set ) - startTime ) );
   }

   getPlayer()->sendDebug( "Registered " + std::to_string( m_replayCache.size() ) + " sets for replay" );
   m_isReplaying = true;
}

void Core::Session::stopReplay()
{
   m_isReplaying = false;
   m_replayCache.clear();
}

void Core::Session::processReplay()
{
   int at = 0;
   for( const auto& set : m_replayCache )
   {
      if( std::get< 0 >( set ) <= Util::getTimeMs() )
      {
         m_pZoneConnection->injectPacket( std::get< 1 >( set ), *getPlayer().get() );
         m_replayCache.erase( m_replayCache.begin() + at );
         //g_framework.getLogger().info( "Sent for " + std::to_string( std::get< 0 >( set ) ) + ", left: " + std::to_string( m_replayCache.size() ) );
      }
      at++;
   }

   if( m_replayCache.size() == 0 )
      m_isReplaying = false;
}

void Core::Session::sendReplayInfo()
{
   std::string message = std::to_string( m_replayCache.size() ) + " Sets left in cache, ";

   if( m_isReplaying )
      message += " is active";
   else
      message += " is idle";

   getPlayer()->sendDebug( message );
}

void Core::Session::update()
{
   if( m_isReplaying )
      processReplay();

   if( m_pZoneConnection )
   {
      m_pZoneConnection->processInQueue();
      
      // SESSION LOGIC
      m_pPlayer->update( Util::getTimeMs() );

      if( ( static_cast< uint32_t >( Util::getTimeSeconds() ) - static_cast< uint32_t >( getLastSqlTime() ) ) > 10  )
      {
         updateLastSqlTime();
         m_pPlayer->updateSql();
      }

      m_pZoneConnection->processOutQueue();
   }

   if( m_pChatConnection )
   {
       m_pChatConnection->processInQueue();
       m_pChatConnection->processOutQueue();
   }

}

Core::Entity::PlayerPtr Core::Session::getPlayer() const
{
   return m_pPlayer;
}