/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2006-2008 Ulyanov Igor <iulyanov@gmail.com>

 * Copyright (c) 2010 Andrey Korotaev <unC0Rr@gmail.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; version 2 of the License

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA

 */

#include <QDebug>

#include <QInputDialog>

#include <QCryptographicHash>

#include "hwconsts.h"

#include "newnetclient.h"

#include "proto.h"

#include "gameuiconfig.h"

#include "game.h"

#include "gamecfgwidget.h"

#include "teamselect.h"

#include "misc.h"

char delimeter='\n';

HWNewNet::HWNewNet(GameUIConfig * config, GameCFGWidget* pGameCFGWidget, TeamSelWidget* pTeamSelWidget) :

  config(config),

  m_pGameCFGWidget(pGameCFGWidget),

  m_pTeamSelWidget(pTeamSelWidget),

  isChief(false),

  m_game_connected(false),

  loginStep(0),

  netClientState(0)

{

// socket stuff

    connect(&NetSocket, SIGNAL(readyRead()), this, SLOT(ClientRead()));

    connect(&NetSocket, SIGNAL(connected()), this, SLOT(OnConnect()));

    connect(&NetSocket, SIGNAL(disconnected()), this, SLOT(OnDisconnect()));

    connect(&NetSocket, SIGNAL(error(QAbstractSocket::SocketError)), this,

            SLOT(displayError(QAbstractSocket::SocketError)));

// config stuff

    connect(this, SIGNAL(paramChanged(const QString &, const QStringList &)), pGameCFGWidget, SLOT(setParam(const QString &, const QStringList &)));

    connect(pGameCFGWidget, SIGNAL(paramChanged(const QString &, const QStringList &)), this, SLOT(onParamChanged(const QString &, const QStringList &)));

    connect(this, SIGNAL(configAsked()), pGameCFGWidget, SLOT(fullNetConfig()));

}

HWNewNet::~HWNewNet()

{

    if (m_game_connected)

    {

        RawSendNet(QString("QUIT%1%2").arg(delimeter).arg("User quit"));

        emit Disconnected();

    }

    NetSocket.flush();

}

void HWNewNet::Connect(const QString & hostName, quint16 port, const QString & nick)

{

    mynick = nick.isEmpty() ? QLineEdit::tr("unnamed") : nick;

    myhost = hostName + QString(":%1").arg(port);

    NetSocket.connectToHost(hostName, port);

}

void HWNewNet::Disconnect()

{

    if (m_game_connected)

        RawSendNet(QString("QUIT%1%2").arg(delimeter).arg("User quit"));

    m_game_connected = false;

    NetSocket.disconnectFromHost();

}

void HWNewNet::CreateRoom(const QString & room)

{

    if(netClientState != 2)

    {

        qWarning("Illegal try to create room!");

        return;

    }

    myroom = room;

    RawSendNet(QString("CREATE_ROOM%1%2").arg(delimeter).arg(room));

    isChief = true;

}

void HWNewNet::JoinRoom(const QString & room)

{

    if(netClientState != 2)

    {

        qWarning("Illegal try to join room!");

        return;

    }

    myroom = room;

    RawSendNet(QString("JOIN_ROOM%1%2").arg(delimeter).arg(room));

    isChief = false;

}

void HWNewNet::AddTeam(const HWTeam & team)

{

    QString cmd = QString("ADD_TEAM") + delimeter +

         team.TeamName + delimeter +

         team.teamColor.name() + delimeter +

         team.Grave + delimeter +

         team.Fort + delimeter +

         team.Voicepack + delimeter +

         team.Flag + delimeter +

         QString::number(team.difficulty);

    for(int i = 0; i < 8; ++i)

    {

        cmd.append(delimeter);

        cmd.append(team.HHName[i]);

        cmd.append(delimeter);

        cmd.append(team.HHHat[i]);

    }

    RawSendNet(cmd);

}

void HWNewNet::RemoveTeam(const HWTeam & team)

{

    RawSendNet(QString("REMOVE_TEAM") + delimeter + team.TeamName);

}

void HWNewNet::ToggleReady()

{

  RawSendNet(QString("TOGGLE_READY"));

}

void HWNewNet::SendNet(const QByteArray & buf)

{

  QString msg = QString(buf.toBase64());

  RawSendNet(QString("EM%1%2").arg(delimeter).arg(msg));

}

void HWNewNet::RawSendNet(const QString & str)

{

  RawSendNet(str.toUtf8());

}

void HWNewNet::RawSendNet(const QByteArray & buf)

{

//  qDebug() << "Client: " << QString(buf).split("\n");

    NetSocket.write(buf);

    NetSocket.write("\n\n", 2);

}

void HWNewNet::ClientRead()

{

    while (NetSocket.canReadLine()) {

        QString s = QString::fromUtf8(NetSocket.readLine());

        if (s.endsWith('\n')) s.chop(1);

        if (s.size() == 0) {

            ParseCmd(cmdbuf);

            cmdbuf.clear();

        } else

            cmdbuf << s;

    }

}

void HWNewNet::OnConnect()

{

}

void HWNewNet::OnDisconnect()

{

    if(m_game_connected) emit Disconnected();

    m_game_connected = false;

}

void HWNewNet::displayError(QAbstractSocket::SocketError socketError)

{

    emit Disconnected();

    switch (socketError) {

        case QAbstractSocket::RemoteHostClosedError:

            break;

        case QAbstractSocket::HostNotFoundError:

            emit showMessage(tr("The host was not found. Please check the host name and port settings."));

            break;

        case QAbstractSocket::ConnectionRefusedError:

            emit showMessage(tr("Connection refused"));

            break;

        default:

            emit showMessage(NetSocket.errorString());

        }

}

void HWNewNet::ParseCmd(const QStringList & lst)

{

//  qDebug() << "Server: " << lst;

    if(!lst.size())

    {

        qWarning("Net client: Bad message");

        return;

    }

    if (lst[0] == "NICK")

    {

        mynick = lst[1];

        return ;

    }

    if (lst[0] == "PROTO")

        return ;

    if (lst[0] == "ERROR") {

        if (lst.size() == 2)

            emit showMessage("Error: " + lst[1]);

        else

            emit showMessage("Unknown error");

        return;

    }

    if (lst[0] == "WARNING") {

        if (lst.size() == 2)

            emit showMessage("Warning: " + lst[1]);

        else

            emit showMessage("Unknown warning");

        return;

    }

    if (lst[0] == "CONNECTED") {

        RawSendNet(QString("NICK%1%2").arg(delimeter).arg(mynick));

        RawSendNet(QString("PROTO%1%2").arg(delimeter).arg(*cProtoVer));

        netClientState = 1;

        m_game_connected = true;

        emit adminAccess(false);

        return;

    }

    if (lst[0] == "PING") {

        if (lst.size() > 1)

            RawSendNet(QString("PONG%1%2").arg(delimeter).arg(lst[1]));

        else

            RawSendNet(QString("PONG"));

        return;

    }

    if (lst[0] == "ROOMS") {

        QStringList tmp = lst;

        tmp.removeFirst();

        emit roomsList(tmp);

        return;

    }

    if (lst[0] == "SERVER_MESSAGE") {

        if(lst.size() < 2)

        {

            qWarning("Net: Empty SERVERMESSAGE message");

            return;

        }

        emit serverMessage(lst[1]);

        return;

    }

    if (lst[0] == "CHAT") {

        if(lst.size() < 3)

        {

            qWarning("Net: Empty CHAT message");

            return;

        }

        if (netClientState == 2)

            emit chatStringLobby(HWProto::formatChatMsg(lst[1], lst[2]));

        else

            emit chatStringFromNet(HWProto::formatChatMsg(lst[1], lst[2]));

        return;

    }

    if (lst[0] == "INFO") {

        if(lst.size() < 5)

        {

            qWarning("Net: Malformed INFO message");

            return;

        }

        QStringList tmp = lst;

        tmp.removeFirst();

        if (netClientState == 2)

            emit chatStringLobby(tmp.join("\n").prepend('\x01'));

        else

            emit chatStringFromNet(tmp.join("\n").prepend('\x01'));

        return;

    }

    if (lst[0] == "READY") {

        if(lst.size() < 2)

        {

            qWarning("Net: Malformed READY message");

            return;

        }

        for(int i = 1; i < lst.size(); ++i)

        {

            if (lst[i] == mynick)

                emit setMyReadyStatus(true);

            emit setReadyStatus(lst[i], true);

        }

        return;

    }

    if (lst[0] == "NOT_READY") {

        if(lst.size() < 2)

        {

            qWarning("Net: Malformed NOT_READY message");

            return;

        }

        for(int i = 1; i < lst.size(); ++i)

        {

            if (lst[i] == mynick)

                emit setMyReadyStatus(false);

            emit setReadyStatus(lst[i], false);

        }

        return;

    }

    if (lst[0] == "ADD_TEAM") {

        if(lst.size() != 24)

        {

            qWarning("Net: Bad ADDTEAM message");

            return;

        }

        QStringList tmp = lst;

        tmp.removeFirst();

        emit AddNetTeam(tmp);

        return;

    }

    if (lst[0] == "REMOVE_TEAM") {

        if(lst.size() != 2)

        {

            qWarning("Net: Bad REMOVETEAM message");

            return;

        }

        m_pTeamSelWidget->removeNetTeam(HWTeam(lst[1]));

        return;

    }

    if(lst[0] == "ROOMABANDONED") {

        netClientState = 2;

        emit showMessage(HWNewNet::tr("Room destroyed"));

        emit LeftRoom();

        return;

    }

    if(lst[0] == "KICKED") {

        netClientState = 2;

        emit showMessage(HWNewNet::tr("You got kicked"));

        emit LeftRoom();

        return;

    }

    if(lst[0] == "JOINED") {

        if(lst.size() < 2)

        {

            qWarning("Net: Bad JOINED message");

            return;

        }

        for(int i = 1; i < lst.size(); ++i)

        {

            if (lst[i] == mynick)

            {

                netClientState = 3;

                emit EnteredGame();

                emit roomMaster(isChief);

                if (isChief)

                    emit configAsked();

            }

            emit nickAdded(lst[i], isChief && (lst[i] != mynick));

            emit chatStringFromNet(tr("%1 *** %2 has joined the room").arg('\x03').arg(lst[i]));

        }

        return;

    }

    if(lst[0] == "LOBBY:JOINED") {

        if(lst.size() < 2)

        {

            qWarning("Net: Bad JOINED message");

            return;

        }

        for(int i = 1; i < lst.size(); ++i)

        {

            if (lst[i] == mynick)

            {

                netClientState = 2;

                RawSendNet(QString("LIST"));

                emit Connected();

            }

            emit nickAddedLobby(lst[i], false);

            emit chatStringLobby(tr("%1 *** %2 has joined").arg('\x03').arg(lst[i]));

        }

        return;

    }

    if(lst[0] == "LEFT") {

        if(lst.size() < 2)

        {

            qWarning("Net: Bad LEFT message");

            return;

        }

        emit nickRemoved(lst[1]);

        if (lst.size() < 3)

            emit chatStringFromNet(tr("%1 *** %2 has left").arg('\x03').arg(lst[1]));

        else

            emit chatStringFromNet(tr("%1 *** %2 has left (%3)").arg('\x03').arg(lst[1], lst[2]));

        return;

    }

    if(lst[0] == "ROOM") {

        if(lst.size() < 2)

        {

            qWarning("Net: Bad ROOM message");

            return;

        }

        RawSendNet(QString("LIST"));

        return;

    }

    if(lst[0] == "LOBBY:LEFT") {

        if(lst.size() < 2)

        {

            qWarning("Net: Bad LOBBY:LEFT message");

            return;

        }

        emit nickRemovedLobby(lst[1]);

        if (lst.size() < 3)

            emit chatStringLobby(tr("%1 *** %2 has left").arg('\x03').arg(lst[1]));

        else

            emit chatStringLobby(tr("%1 *** %2 has left (%3)").arg('\x03').arg(lst[1], lst[2]));

        return;

    }

    if (lst[0] == "RUN_GAME") {

        netClientState = 5;

        emit AskForRunGame();

        return;

    }

    if (lst[0] == "ASKPASSWORD") {

        bool ok = false;

        int passLength = config->value("net/passwordlength", 0).toInt();

        QString hash = config->value("net/passwordhash", "").toString();

        QString password = QInputDialog::getText(0, tr("Password"), tr("Your nickname %1 is\nregistered on Hedgewars.org\nPlease provide your password\nor pick another nickname:").arg(mynick), QLineEdit::Password, passLength==0?NULL:QString(passLength,'\0'), &ok);

        if (!ok) {

            Disconnect();

            emit Disconnected();

            return;

        }

        if (!passLength || password!=QString(passLength, '\0')) {

            hash = QCryptographicHash::hash(password.toLatin1(), QCryptographicHash::Md5).toHex();

            config->setValue("net/passwordhash", hash);

            config->setValue("net/passwordlength", password.size());

        }

        RawSendNet(QString("PASSWORD%1%2").arg(delimeter).arg(hash));

        return;

    }

    if (lst[0] == "TEAM_ACCEPTED") {

        if (lst.size() != 2)

        {

            qWarning("Net: Bad TEAM_ACCEPTED message");

            return;

        }

        m_pTeamSelWidget->changeTeamStatus(lst[1]);

        return;

    }

    if (lst[0] == "CFG") {

        if(lst.size() < 3)

        {

            qWarning("Net: Bad CFG message");