sapphire/src/servers/sapphire_lobby/RestConnector.cpp

#include "RestConnector.h"
#include "LobbySession.h"
#include "ServerLobby.h"
#include <Logging/Logger.h>
#include <Crypt/base64.h>
#include <time.h>
#include <iomanip>

#include <nlohmann/json.hpp>

extern Core::Logger g_log;

typedef std::vector< std::tuple< std::string, uint32_t, uint64_t, std::string > > CharList;

Core::Network::RestConnector::RestConnector()
{

}

Core::Network::RestConnector::~RestConnector()
{

}

HttpResponse Core::Network::RestConnector::requestApi( std::string endpoint, std::string data )
{
  HttpClient client( restHost );

  std::string reqstr = "/sapphire-api/lobby/" + endpoint;

  HttpResponse r;
  try
  {
    r = client.request( "POST", reqstr, data );
  }
  catch( std::exception& e )
  {
    g_log.error( endpoint + " failed, REST is not reachable: " + std::string( e.what() ) );
    return nullptr;
  }
  return r;
}

Core::LobbySessionPtr Core::Network::RestConnector::getSession( char* sId )
{
  std::string json_string = "{\"sId\": \"" + std::string( sId ) + "\",\"secret\": \"" + serverSecret + "\"}";

  HttpResponse r = requestApi( "checkSession", json_string );

  if( r == nullptr )
    return nullptr;

  std::string content = std::string( std::istreambuf_iterator< char >( r->content ), {} );
  if( r->status_code.find( "200" ) != std::string::npos )
  {
    nlohmann::json json;  

    try
    {
      json = json.parse( content );
    }
    catch( std::exception& e )
    {
      g_log.debug( "Could not parse REST response: " + std::string( e.what() ) );
      return nullptr;
    }

    if( content.find( "invalid" ) == std::string::npos )
    {
      LobbySessionPtr pSession( new Core::LobbySession() );
      pSession->setAccountID( json["result"].get< uint32_t >() );
      pSession->setSessionId( ( uint8_t* ) sId );
      return pSession;
    }
    else
    {
      return nullptr;
    }
  }
  else
  {
    return nullptr;
  }
}

bool Core::Network::RestConnector::checkNameTaken( std::string name )
{
  std::string json_string = "{\"name\": \"" + name + "\",\"secret\": \"" + serverSecret + "\"}";

  HttpResponse r = requestApi( "checkNameTaken", json_string );

  if( r == nullptr )
    return true;

  std::string content = std::string( std::istreambuf_iterator< char >( r->content ), {} );
  if( r->status_code.find( "200" ) != std::string::npos )
  {
    auto json = nlohmann::json();

    try
    {
      json = json.parse( content );
    }
    catch( std::exception& e )
    {
      g_log.debug( "Could not parse REST response: " + std::string( e.what() ) );
      return true;
    }

    if( json["result"] != "invalid" && json["result"] == "false" )
      return false;
    return true;
  }
  else
  {
    return true;
  }
}

uint32_t Core::Network::RestConnector::getNextCharId()
{
  std::string json_string = "{\"secret\": \"" + serverSecret + "\"}";

  HttpResponse r = requestApi( "getNextCharId", json_string );

  if( r == nullptr )
    return -1;

  std::string content = std::string( std::istreambuf_iterator< char >( r->content ), {} );
  if( r->status_code.find( "200" ) != std::string::npos )
  {
    auto json = nlohmann::json();

    try
    {
      json = json.parse( content );
    }
    catch( std::exception& e )
    {
      g_log.debug( "Could not parse REST response: " + std::string( e.what() ) );
      return -1;
    }

    if( content.find( "invalid" ) == std::string::npos )
    {
      return json["result"].get< uint32_t >();
    }
    else
    {
      return -1;
    }
  }
  else
  {
    return -1;
  }
}

uint64_t Core::Network::RestConnector::getNextContentId()
{
  std::string json_string = "{\"secret\": \"" + serverSecret + "\"}";

  HttpResponse r = requestApi( "getNextContentId", json_string );

  if( r == nullptr )
    return -1;

  std::string content = std::string( std::istreambuf_iterator< char >( r->content ), {} );
  if( r->status_code.find( "200" ) != std::string::npos )
  {
    auto json = nlohmann::json();

    try
    {
      json = json.parse( content );
    }
    catch( std::exception& e )
    {
      g_log.debug( "Could not parse REST response: " + std::string( e.what() ) );
      return -1;
    }

    if( content.find( "invalid" ) == std::string::npos )
    {
      return json["result"].get< uint64_t >();
    }
    else
    {
      return -1;
    }
  }
  else
  {
    return -1;
  }
}

CharList Core::Network::RestConnector::getCharList( char* sId )
{
  std::string json_string = "{\"sId\": \"" + std::string( sId, 56 ) + "\",\"secret\": \"" + serverSecret + "\"}";

  HttpResponse r = requestApi( "getCharacterList", json_string );

  CharList list;
  if( r == nullptr )
    return list;

  std::string content = std::string( std::istreambuf_iterator< char >( r->content ), {} );
  g_log.debug( content );
  if( r->status_code.find( "200" ) != std::string::npos )
  {
    auto json = nlohmann::json();

    try
    {
      json = json.parse( content );
    }
    catch( std::exception& e )
    {
      g_log.debug( "Could not parse REST response: " + std::string( e.what() ) );
      return list;
    }

    if( json["result"].get< std::string >().find( "invalid" ) == std::string::npos )
    {

      g_log.debug( json["result"] );

      for( auto& child : json["charArray"] )
      {
        g_log.debug( child["contentId"] );
        //std::string, uint32_t, uint64_t, std::string
        list.push_back( { child["name"].get< std::string >(),
                          child["charId"].get< uint32_t >(),
                          child["contentId"].get< uint64_t >(),
                          child["infoJson"].get< std::string >() 
                        } );
      }

      return list;

    }
    else
    {
      return list;
    }
  }
  else
  {
    return list;
  }
}

bool Core::Network::RestConnector::deleteCharacter( char* sId, std::string name )
{
  std::string json_string =
    "{\"sId\": \"" + std::string( sId, 56 ) + "\",\"secret\": \"" + serverSecret + "\",\"name\": \"" + name + "\"}";

  HttpResponse r = requestApi( "deleteCharacter", json_string );

  if( r == nullptr )
    return false;

  std::string content = std::string( std::istreambuf_iterator< char >( r->content ), {} );
  if( r->status_code.find( "200" ) != std::string::npos )
  {
    auto json = nlohmann::json();
    try
    {
      json = json.parse( content );
    }
    catch( std::exception& e )
    {
      g_log.debug( "Could not parse REST response: " + std::string( e.what() ) );
      return false;
    }

    if( content.find( "invalid" ) == std::string::npos )
      return true;
    return false;
  }
  else
  {
    return false;
  }
}

int Core::Network::RestConnector::createCharacter( char* sId, std::string name, std::string infoJson )
{
  std::string json_string =
    "{\"sId\": \"" + std::string( sId, 56 ) + "\",\"secret\": \"" + serverSecret + "\",\"name\": \"" + name +
    "\",\"infoJson\": \"" + Core::Util::base64_encode( ( uint8_t* ) infoJson.c_str(), infoJson.length() ) + "\"}";

  HttpResponse r = requestApi( "createCharacter", json_string );

  if( r == nullptr )
    return -1;

  std::string content = std::string( std::istreambuf_iterator< char >( r->content ), {} );
  g_log.debug( content );
  if( r->status_code.find( "200" ) != std::string::npos )
  {
    auto json = nlohmann::json();

    try
    {
      json = json.parse( content );
    }
    catch( std::exception& e )
    {
      g_log.debug( "Could not parse REST response: " + std::string( e.what() ) );
      return -1;
    }

    if( content.find( "invalid" ) == std::string::npos )
      return json["result"].get< int >();
    return -1;
  }
  else
  {
    return -1;
  }
}