#include "EffectBuilder.h"
#include "EffectResult.h"

#include <Actor/Player.h>

#include <Network/PacketWrappers/EffectPacket.h>

#include <Territory/Territory.h>

#include <Util/Util.h>
#include <Util/UtilMath.h>

#include <Logging/Logger.h>

using namespace Sapphire;
using namespace Sapphire::World::Action;
using namespace Sapphire::Network::Packets;

EffectBuilder::EffectBuilder( Entity::CharaPtr source, uint32_t actionId, uint16_t sequence ) :
  m_sourceChara( std::move( source ) ),
  m_actionId( actionId ),
  m_sequence( sequence )
{

}

uint64_t EffectBuilder::getResultDelayMs()
{
  // todo: actually figure this retarded shit out

  return Common::Util::getTimeMs() + 850;
}

std::shared_ptr< std::vector< EffectResultPtr > > EffectBuilder::getResultList( Entity::CharaPtr& chara )
{
  auto it = m_resolvedEffects.find( chara->getId() );
  if( it == m_resolvedEffects.end() )
  {
    // create a new one and return it
    auto resultList = std::make_shared< std::vector< EffectResultPtr > >();

    m_resolvedEffects[ chara->getId() ] = resultList;

    return resultList;
  }

  return it->second;
}

void EffectBuilder::heal( Entity::CharaPtr& effectTarget, Entity::CharaPtr& healingTarget, uint32_t amount, Common::ActionHitSeverityType severity, Common::ActionEffectResultFlag flag )
{
  auto resultList = getResultList( effectTarget );
  assert( resultList );

  EffectResultPtr nextResult = make_EffectResult( healingTarget, getResultDelayMs() );
  nextResult->heal( amount, severity, flag );
  resultList->push_back( std::move( nextResult ) );
}

void EffectBuilder::restoreMP( Entity::CharaPtr& target, Entity::CharaPtr& restoringTarget, uint32_t amount, Common::ActionEffectResultFlag flag )
{
  auto resultList = getResultList( target );
  assert( resultList );

  EffectResultPtr nextResult = make_EffectResult( restoringTarget, getResultDelayMs() ); // restore mp source actor
  nextResult->restoreMP( amount, flag );
  resultList->push_back( std::move( nextResult ) );
}

void EffectBuilder::damage( Entity::CharaPtr& effectTarget, Entity::CharaPtr& damagingTarget, uint32_t amount, Common::ActionHitSeverityType severity, Common::ActionEffectResultFlag flag )
{
  auto resultList = getResultList( effectTarget );
  assert( resultList );

  EffectResultPtr nextResult = make_EffectResult( damagingTarget, getResultDelayMs() );
  nextResult->damage( amount, severity, flag );
  resultList->push_back( std::move( nextResult ) );
}

void EffectBuilder::startCombo( Entity::CharaPtr& target, uint16_t actionId )
{
  auto resultList = getResultList( target );
  assert( resultList  );

  EffectResultPtr nextResult = make_EffectResult( target, 0 );
  nextResult->startCombo( actionId );
  resultList->push_back( std::move( nextResult ) );
}

void EffectBuilder::comboSucceed( Entity::CharaPtr& target )
{
  auto resultList = getResultList( target );
  assert( resultList  );

  EffectResultPtr nextResult = make_EffectResult( target, 0 );
  nextResult->comboSucceed();
  resultList->push_back( std::move( nextResult ) );
}

void EffectBuilder::buildAndSendPackets()
{
  auto targetCount = m_resolvedEffects.size();
  Logger::debug( "EffectBuilder result: " );
  Logger::debug( "Targets afflicted: {}", targetCount );

  auto globalSequence = m_sourceChara->getCurrentTerritory()->getNextEffectSequence();

  while( m_resolvedEffects.size() > 0 )
  {
    auto packet = buildNextEffectPacket( globalSequence );
    m_sourceChara->sendToInRangeSet( packet, true );
  }
}

std::shared_ptr< FFXIVPacketBase > EffectBuilder::buildNextEffectPacket( uint32_t globalSequence )
{
  auto remainingTargetCount = m_resolvedEffects.size();

  if( remainingTargetCount > 1 ) // use AoeEffect packets
  {
    int packetSize = remainingTargetCount <= 8 ? 8 :
                   ( remainingTargetCount <= 16 ? 16 :
                   ( remainingTargetCount <= 24 ? 24 : 32 ) );

    using EffectHeader = Server::FFXIVIpcAoeEffect< 8 >; // dummy type to access header part of the packet

    FFXIVPacketBasePtr effectPacket = nullptr;
    EffectHeader* pHeader;
    Common::EffectEntry* pEntry;
    uint64_t* pEffectTargetId;
    uint16_t* pFlag;
    switch( packetSize )
    {
      case 8:
      {
        auto p = makeZonePacket< Server::FFXIVIpcAoeEffect8 >( m_sourceChara->getId() );
        pHeader = reinterpret_cast< EffectHeader* >( &p->data() );
        pEntry = reinterpret_cast< Common::EffectEntry* >( &p->data().effects );
        pEffectTargetId = reinterpret_cast< uint64_t* >( &p->data().effectTargetId );
        pFlag = reinterpret_cast< uint16_t* >( &p->data().unkFlag );
        effectPacket = std::move( p );
        break;
      }
      case 16:
      {
        auto p = makeZonePacket< Server::FFXIVIpcAoeEffect16 >( m_sourceChara->getId() );
        pHeader = reinterpret_cast< EffectHeader* >( &p->data() );
        pEntry = reinterpret_cast< Common::EffectEntry* >( &p->data().effects );
        pEffectTargetId = reinterpret_cast< uint64_t* >( &p->data().effectTargetId );
        pFlag = reinterpret_cast< uint16_t* >( &p->data().unkFlag );
        effectPacket = std::move( p );
        break;
      }
      case 24:
      {
        auto p = makeZonePacket< Server::FFXIVIpcAoeEffect24 >( m_sourceChara->getId() );
        pHeader = reinterpret_cast< EffectHeader* >( &p->data() );
        pEntry = reinterpret_cast< Common::EffectEntry* >( &p->data().effects );
        pEffectTargetId = reinterpret_cast< uint64_t* >( &p->data().effectTargetId );
        pFlag = reinterpret_cast< uint16_t* >( &p->data().unkFlag );
        effectPacket = std::move( p );
        break;
      }
      case 32:
      {
        auto p = makeZonePacket< Server::FFXIVIpcAoeEffect32 >( m_sourceChara->getId() );
        pHeader = reinterpret_cast< EffectHeader* >( &p->data() );
        pEntry = reinterpret_cast< Common::EffectEntry* >( &p->data().effects );
        pEffectTargetId = reinterpret_cast< uint64_t* >( &p->data().effectTargetId );
        pFlag = reinterpret_cast< uint16_t* >( &p->data().unkFlag );
        effectPacket = std::move( p );
        break;
      }
    }
    assert( effectPacket != nullptr );

    pHeader->actionId = m_actionId;
    pHeader->actionAnimationId = static_cast< uint16_t >( m_actionId );
    pHeader->animationTargetId = m_sourceChara->getId();
    pHeader->someTargetId = 0xE0000000;
    pHeader->rotation = Common::Util::floatToUInt16Rot( m_sourceChara->getRot() );
    pHeader->effectDisplayType = Common::ActionEffectDisplayType::ShowActionName;
    pHeader->effectCount = static_cast< uint8_t >( remainingTargetCount > packetSize ? packetSize : remainingTargetCount );
    pHeader->sourceSequence = m_sequence;
    pHeader->globalSequence = globalSequence;

    uint8_t targetIndex = 0;
    for( auto it = m_resolvedEffects.begin(); it != m_resolvedEffects.end(); )
    {
      auto resultList = it->second;
      assert( resultList->size() > 0 );
      auto firstResult = resultList->data()[ 0 ];
      pEffectTargetId[ targetIndex ] = firstResult->getTarget()->getId();
      Logger::debug( " - id: {}", pEffectTargetId[ targetIndex ] );

      for( auto i = 0; i < resultList->size(); i++ )
      {
        auto result = resultList->data()[ i ];
        pEntry[ targetIndex * 8 + i ] = result->buildEffectEntry();
        m_sourceChara->getCurrentTerritory()->addEffectResult( std::move( result ) );
      }
      resultList->clear();

      it = m_resolvedEffects.erase( it );

      targetIndex++;

      if( targetIndex == packetSize )
        break;
    }

    pFlag[0] = 0x7FFF;
    pFlag[1] = 0x7FFF;
    pFlag[2] = 0x7FFF;

    return effectPacket;
  }
  else
  {
    auto resultList = m_resolvedEffects.begin()->second;
    assert( !resultList->empty() );
    auto firstResult = resultList->data()[ 0 ];
    Logger::debug( " - id: {}", firstResult->getTarget()->getId() );

    auto seq = m_sourceChara->getCurrentTerritory()->getNextEffectSequence();

    auto effectPacket = std::make_shared< Server::EffectPacket >( m_sourceChara->getId(), firstResult->getTarget()->getId(), m_actionId );
    effectPacket->setRotation( Common::Util::floatToUInt16Rot( m_sourceChara->getRot() ) );
    effectPacket->setSequence( seq, m_sequence );

    for( int i = 0; i < resultList->size(); i++ )
    {
      auto result = resultList->data()[ i ];
      effectPacket->addEffect( result->buildEffectEntry() );
      m_sourceChara->getCurrentTerritory()->addEffectResult( std::move( result ) );
    }

    resultList->clear();

    m_resolvedEffects.clear();

    return effectPacket;
  }
}