#include "session.hpp"
#include "game.hpp"
#include "state/commands.hpp"

#include <iostream>

using namespace std;

bool Session::parse(string s)
{
    s.erase(std::find_if(s.rbegin(), s.rend(),
            std::not1(std::ptr_fun<int, int>(std::isspace))).base(), s.end());

    if(s.size()==1&&s[0]=='q'){
        //quit
        m_socket.close();
        m_state->quitPlayer(m_pid);
        return false;
    }else if(s.size()==1&&s[0]=='c'){
        //clear
        m_state->clear(m_pid);
    }else if(s.size()>=3&&s[0]=='n'&&s[1]==' '){
        //set name
        m_state->setName(m_pid, s.substr(2));
    }else if(s.size()>=3&&s[0]=='v'&&s[1]==' '){
        //set speed
        double speed = atof(s.substr(2).c_str());
        m_state->setSpeed(m_pid, speed);
    }else{
        //shoot
        double angle = atof(s.c_str());
        m_state->commandForPlayer(m_pid, new game::ShootCommand(angle));
    }
    return true;
}

void Session::start()
{
    if(m_started) return;
    m_started=true;
    m_pid = m_state->addPlayer();
    char hello[] = "\nUse \"n name\" to change name, \"v velocity\" to change velocity, \"c\" to clear past shots or \"q\" to close the connection.\nEverything else is interpreted as a shooting angle.\n\n> ";
    do_write(hello, strlen(hello));
}

void Session::do_read()
{
    auto self(shared_from_this());
    m_socket.async_read_some(asio::buffer(m_rcv_data, max_length),
        [this, self](std::error_code ec, std::size_t length)
        {
            if (!ec)
            {
                std::string s(m_rcv_data,length);
                if(parse(s)){
                    char c[]="> ";
                    do_write(c,strlen(c));
                }
            }
        });
}

void Session::do_write(char m_snd_data[], std::size_t length)
{
    auto self(shared_from_this());
    asio::async_write(m_socket, asio::buffer(m_snd_data, length),
        [this, self](std::error_code ec, std::size_t /*length*/)
        {
            if (!ec)
            {
                do_read();
            }
        });
}