sapphire/src/servers/sapphire_zone/Session.cpp

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