#include <windows.h>
#include <string>
#include <assert.h>
#include "SAMPFUNCS_API.h"
#include <boost/algorithm/string.hpp>

using namespace std;

SAMPFUNCS *SF = new SAMPFUNCS();

void addIDs( string& strMessage )
{
    for( int i = 0; i < SAMP_MAX_PLAYERS; i++ )
    {
  if( SF->getSAMP()->getPlayers()->IsPlayerDefined( i ) )
  {
   string strName = SF->getSAMP()->getPlayers()->GetPlayerName( i );

   boost::algorithm::erase_all( strMessage, "[" + to_string( i ) + "]" );
   boost::algorithm::erase_all( strMessage, "(" + to_string( i ) + ")" );

   boost::algorithm::replace_all( strMessage, strName, strName + " [" + to_string( i ) + "]" );
  };
    };
};

bool CALLBACK incomingRPC( stRakNetHookParams *params )
{
    params->bitStream->ResetReadPointer();

    switch( params->packetId )
    {
  case ScriptRPCEnumeration::RPC_ScrClientMessage:
  {
  DWORD dwColor;
  int iLen;
  char szMsg[256];

  memset( szMsg, 0, sizeof( szMsg ) );

  params->bitStream->ResetReadPointer();
  params->bitStream->Read( dwColor );
  params->bitStream->Read( iLen );

  if( iLen >= sizeof( szMsg ) )
   break;

  params->bitStream->Read( szMsg, iLen );
  params->bitStream->ResetReadPointer();

  szMsg[iLen] = '\0';

  string strNewLine = szMsg;

  addIDs( strNewLine );

  params->bitStream->ResetWritePointer();
  params->bitStream->Write( dwColor );
  params->bitStream->Write( (int)strNewLine.length() );
  params->bitStream->Write( strNewLine.c_str(), strNewLine.length() );

   break;
  };
  case RPCEnumeration::RPC_Chat:
  {
  unsigned short playerId;
  BYTE byteTextLen;
  char szText[256];

  memset( szText, 0, sizeof( szText ) );

  params->bitStream->ResetReadPointer();
  params->bitStream->Read( playerId );
  params->bitStream->Read( byteTextLen );
  params->bitStream->Read( szText, byteTextLen );
  params->bitStream->ResetReadPointer();

  szText[byteTextLen] = '\0';

  string strNewLine = szText;

  addIDs( strNewLine );

  params->bitStream->ResetWritePointer();
  params->bitStream->Write( playerId );
  params->bitStream->Write( (BYTE)strNewLine.length() );
  params->bitStream->Write( strNewLine.c_str(), strNewLine.length() );

   break;
  };
    };

    params->bitStream->ResetReadPointer();

    return true;
};

void CALLBACK mainloop( void )
{
    static bool init = false;
    if( !init )
    {
  if( !SF->getSAMP()->IsInitialized() )
   return;

  SF->getSAMP()->getChat()->AddChatMessage( D3DCOLOR_ARGB( 255, 255, 0, 0 ) , "ChatID by urShadow loaded {00ff00}| BlastHack.Net" );

  SF->getRakNet()->registerRakNetCallback( RakNetScriptHookType::RAKHOOK_TYPE_INCOMING_RPC, incomingRPC );

  init = true;
    };
};

BOOL APIENTRY DllMain( HMODULE hModule, DWORD dwReasonForCall, LPVOID lpReserved )
{
    switch( dwReasonForCall )
    {
    case DLL_PROCESS_ATTACH:
  SF->initPlugin( mainloop, hModule );
   break;
    case DLL_THREAD_ATTACH:
    case DLL_THREAD_DETACH:
    case DLL_PROCESS_DETACH:
  break;
    };
    return TRUE;
};