#include "setupHelpers.h"
#include "signalHelpers.h"
#include <cassert>
#include <Eris/PollDefault.h>
#include <Eris/Entity.h>
#include <sigc++/functors/mem_fun.h>
#include <iostream>
#include <wfmath/timestamp.h>

using WFMath::TimeStamp;
using WFMath::TimeDiff;

using std::cout;
using std::endl;

AutoConnection stdConnect()
{
    AutoConnection con(new Eris::Connection("eris-test", "localhost", 7450, false));
    SignalCounter0 connectCount;
    con->Connected.connect(sigc::mem_fun(connectCount, &SignalCounter0::fired));
    
    SignalCounter1<const std::string&> fail;
    con->Failure.connect(sigc::mem_fun(fail, &SignalCounter1<const std::string&>::fired));
    
    con->connect();
    
    while (!connectCount.fireCount() && !fail.fireCount())
    {
        Eris::PollDefault::poll();
    }
    
    assert(con->isConnected());
    return con;
}

AutoAccount stdLogin(const std::string& uname, const std::string& pwd, Eris::Connection* con)
{
    AutoAccount player(new Eris::Account(con));
    
    SignalCounter0 loginCount;
    player->LoginSuccess.connect(sigc::mem_fun(loginCount, &SignalCounter0::fired));
   
    SignalCounter1<const std::string&> loginErrorCounter;
    player->LoginFailure.connect(sigc::mem_fun(loginErrorCounter, &SignalCounter1<const std::string&>::fired));
    
    player->login(uname, pwd);

    while (!loginCount.fireCount() && !loginErrorCounter.fireCount())
    {
        Eris::PollDefault::poll();
    }
    
    assert(loginErrorCounter.fireCount() == 0);
    return player;
}

AvatarGetter::AvatarGetter(Eris::Account* acc) : 
    m_acc(acc),
    m_expectFail(false),
    m_failed(false),
    m_earlyReturn(false)
{
    m_acc->AvatarSuccess.connect(sigc::mem_fun(*this, &AvatarGetter::success));
    m_acc->AvatarFailure.connect(sigc::mem_fun(*this, &AvatarGetter::failure));
}
    
AutoAvatar AvatarGetter::take(const std::string& charId)
{        
    m_waiting = true;
    m_failed = false;
    
    m_acc->takeCharacter(charId);
    
    while (m_waiting) Eris::PollDefault::poll();
    
    // we wanted to fail, bail out now
    if (m_failed && m_expectFail) return AutoAvatar();
    
    assert(m_av->getEntity() == NULL); // shouldn't have the entity yet
    assert(m_av->getId() == charId); // but should have it's ID

    if (m_earlyReturn) return m_av;

    SignalCounter1<Eris::Entity*> gotChar;
    m_av->GotCharacterEntity.connect(sigc::mem_fun(gotChar, &SignalCounter1<Eris::Entity*>::fired));

    TimeStamp end = TimeStamp::now() + TimeDiff(2 * 1000);
    
    while ((gotChar.fireCount() == 0) && (TimeStamp::now() < end)) {
        Eris::PollDefault::poll();
    }
    
    if (gotChar.fireCount() == 0) cout << "timed-out waiting to go in-game" << endl;
    
    assert(m_av->getEntity());
    assert(m_av->getEntity()->getId() == charId);

    return m_av;
}

AutoAvatar AvatarGetter::create(const Atlas::Objects::Entity::RootEntity& charEnt)
{        
    m_waiting = true;
    m_failed = false;
    
    m_acc->createCharacter(charEnt);
    
    while (m_waiting) Eris::PollDefault::poll();
    
    // we wanted to fail, bail out now
    if (m_failed && m_expectFail) return AutoAvatar();
    
    assert(m_av->getEntity() == NULL); // shouldn't have the entity yet

    if (m_earlyReturn) return m_av;

    SignalCounter1<Eris::Entity*> gotChar;
    m_av->GotCharacterEntity.connect(sigc::mem_fun(gotChar, &SignalCounter1<Eris::Entity*>::fired));

    TimeStamp end = TimeStamp::now() + TimeDiff(2 * 1000);
    
    while ((gotChar.fireCount() == 0) && (TimeStamp::now() < end)) {
        Eris::PollDefault::poll();
    }
    
    if (gotChar.fireCount() == 0) cout << "timed-out waiting to go in-game" << endl;
    
    assert(m_av->getEntity());

    return m_av;
}
    
void AvatarGetter::success(Eris::Avatar* av)
{
    m_av.reset(av);
    m_waiting = false;
}

void AvatarGetter::failure(const std::string& msg)
{
    if (msg != "deliberate") {
        std::cerr << "failure getting an avatar: " << msg << std::endl;
    }
    m_waiting = false;
    m_failed = true;
}
