/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2015 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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 */

#include <QApplication>

#include <QString>

#include <QCheckBox>

#include <QByteArray>

#include <QUuid>

#include <QColor>

#include <QStringListModel>

#include <QTextStream>

#include <utility>

#include "hwform.h"

#include "ui/page/pageoptions.h"

#include "ui/page/pagetraining.h"

#include "ui/page/pagecampaign.h"

#include "game.h"

#include "hwconsts.h"

#include "gameuiconfig.h"

#include "gamecfgwidget.h"

#include "teamselect.h"

#include "proto.h"

#include "binds.h"

#include "campaign.h"

#include <QTextStream>

#include "ThemeModel.h"

// last game info

QList<QVariant> lastGameStartArgs = QList<QVariant>();

GameType lastGameType = gtNone;

GameCFGWidget * lastGameCfg = NULL;

QString lastGameAmmo = NULL;

TeamSelWidget * lastGameTeamSel = NULL;

QString trainingName, trainingScript, trainingTeam, campaign, campaignScript, campaignTeam; // TODO: Cleaner solution?

HWGame::HWGame(GameUIConfig * config, GameCFGWidget * gamecfg, QString ammo, TeamSelWidget* pTeamSelWidget) :

    TCPBase(true, !config->language().isEmpty(), 0),

    ammostr(ammo),

    m_pTeamSelWidget(pTeamSelWidget)

{

    this->config = config;

    this->gamecfg = gamecfg;

    netSuspend = false;

    lastGameCfg = gamecfg;

    lastGameAmmo = ammo;

    lastGameTeamSel = pTeamSelWidget;

    gameState = gsNotStarted;

    gameType = gtNone;

}

HWGame::~HWGame()

{

    SetGameState(gsDestroyed);

}

void HWGame::onClientDisconnect()

{

    switch (gameType)

    {

        case gtDemo:

            // for video recording we need demo anyway

            emit HaveRecord(rtNeither, demo);

            break;

        case gtNet:

            emit HaveRecord(rtDemo, demo);

            break;

        default:

            if (gameState == gsInterrupted || gameState == gsHalted)

                emit HaveRecord(rtSave, demo);

            else if (gameState == gsFinished)

                emit HaveRecord(rtDemo, demo);

            else

                emit HaveRecord(rtNeither, demo);

    }

    SetGameState(gsStopped);

}

void HWGame::commonConfig()

{

    QByteArray buf;

    QString gt;

    switch (gameType)

    {

        case gtDemo:

            gt = "TD";

            break;

        case gtNet:

            gt = "TN";

            break;

        default:

            gt = "TL";

    }

    HWProto::addStringToBuffer(buf, gt);

    buf += gamecfg->getFullConfig();

    if (m_pTeamSelWidget)

    {

        foreach(HWTeam team, m_pTeamSelWidget->getPlayingTeams())

        {

            HWProto::addStringToBuffer(buf, QString("eammloadt %1").arg(ammostr.mid(0, cAmmoNumber)));

            HWProto::addStringToBuffer(buf, QString("eammprob %1").arg(ammostr.mid(cAmmoNumber, cAmmoNumber)));

            HWProto::addStringToBuffer(buf, QString("eammdelay %1").arg(ammostr.mid(2 * cAmmoNumber, cAmmoNumber)));

            HWProto::addStringToBuffer(buf, QString("eammreinf %1").arg(ammostr.mid(3 * cAmmoNumber, cAmmoNumber)));

            if(gamecfg->schemeData(15).toBool() || !gamecfg->schemeData(21).toBool()) HWProto::addStringToBuffer(buf, QString("eammstore"));

            HWProto::addStringListToBuffer(buf,

                                           team.teamGameConfig(gamecfg->getInitHealth()));

            ;

        }

    }

    RawSendIPC(buf);

}

void HWGame::SendConfig()

{

    commonConfig();

}

void HWGame::SendQuickConfig()

{

    QByteArray teamscfg;

    QAbstractItemModel * themeModel = DataManager::instance().themeModel()->withoutHidden();

    HWProto::addStringToBuffer(teamscfg, "TL");

    // Random seed

    HWProto::addStringToBuffer(teamscfg, "eseed " + QUuid::createUuid().toString());

    int r, minhogs, maxhogs;

    r = rand() % 10000;

    if(r < 3000) { // 30%

        r = 0;

    } else if(r < 5250) { // 22.5%

    } else if(r < 7500) { // 22.5%

    } else if(r < 8750) { // 12.5%

        r = 3;

    } else { // 12.5%

        r = 4;

    }

    switch(r)

    {

        default:

        case 0: {

            r = rand() % 3;

            if(r == 0)

            {

                HWProto::addStringToBuffer(teamscfg, "e$template_filter 1");

                minhogs = 3;

                maxhogs = 4;

            }

            {

                HWProto::addStringToBuffer(teamscfg, "e$template_filter 2");

                minhogs = 4;

                maxhogs = 5;

            }

            else

            {

                HWProto::addStringToBuffer(teamscfg, "e$template_filter 4");

                minhogs = 4;

                maxhogs = 6;

            }

            HWProto::addStringToBuffer(teamscfg, "e$feature_size "+QString::number(rand()%18+4));

            break;

        }

        // Maze

        case 1: {

            minhogs = 4;

            maxhogs = 6;

            HWProto::addStringToBuffer(teamscfg, "e$mapgen 1");

            HWProto::addStringToBuffer(teamscfg, "e$template_filter "+QString::number(rand()%6));

            HWProto::addStringToBuffer(teamscfg, "e$feature_size "+QString::number(rand()%16+6));

            break;

        }

        // Perlin

        case 2: {

            minhogs = 4;

            maxhogs = 6;

            HWProto::addStringToBuffer(teamscfg, "e$mapgen 2");

            HWProto::addStringToBuffer(teamscfg, "e$template_filter "+QString::number(rand()%6));

            HWProto::addStringToBuffer(teamscfg, "e$feature_size "+QString::number(rand()%18+4));

            break;

        }

        // Image map

        case 3: {

            minhogs = 4;

            maxhogs = 6;

            HWProto::addStringToBuffer(teamscfg, "e$mapgen 3");

            // Select map from hardcoded list.

            // TODO: find a more dynamic solution.

            r = rand() % cQuickGameMaps.count();

            HWProto::addStringToBuffer(teamscfg, "e$map " + cQuickGameMaps[r]);

            break;

        }

        // Forts

        case 4: {

            minhogs = 4;

            maxhogs = 6;

            HWProto::addStringToBuffer(teamscfg, "e$mapgen 4");

            HWProto::addStringToBuffer(teamscfg, "e$feature_size "+QString::number(rand()%20+1));

            break;

        }

    }

    // Theme

    HWProto::addStringToBuffer(teamscfg, QString("etheme %1")

        .arg((themeModel->rowCount() > 0) ? themeModel->index(rand() % themeModel->rowCount(), 0).data(ThemeModel::ActualNameRole).toString() : "Nature"));

    int hogs = minhogs + rand() % (maxhogs-minhogs+1);

    // Teams

    // Player team

    HWTeam team1;

    team1.setDifficulty(0);

    team1.setColor(0);

    team1.setNumHedgehogs(hogs);

    HWNamegen::teamRandomEverything(team1);

    team1.setVoicepack("Default_qau");

    // Computer team

    HWTeam team2;

    team2.setDifficulty(r);

    team2.setColor(1);

    team2.setNumHedgehogs(hogs);

    // Make sure the team names are not equal

    do

        HWNamegen::teamRandomEverything(team2);

    while(!team2.name().compare(team1.name()) || !team2.hedgehog(0).Hat.compare(team1.hedgehog(0).Hat));

    team2.setVoicepack("Default_qau");

    r = rand() % 2;

    {

        HWProto::addStringListToBuffer(teamscfg, team1.teamGameConfig(100));

        HWProto::addStringListToBuffer(teamscfg, team2.teamGameConfig(100));

    }

    {

        HWProto::addStringListToBuffer(teamscfg, team2.teamGameConfig(100));

        HWProto::addStringListToBuffer(teamscfg, team1.teamGameConfig(100));

    }

    // Ammo scheme "Default"

    // TODO: Random schemes

    HWProto::addStringToBuffer(teamscfg, QString("eammloadt %1").arg(cDefaultAmmoStore->mid(0, cAmmoNumber)));

    HWProto::addStringToBuffer(teamscfg, QString("eammprob %1").arg(cDefaultAmmoStore->mid(cAmmoNumber, cAmmoNumber)));

    HWProto::addStringToBuffer(teamscfg, QString("eammdelay %1").arg(cDefaultAmmoStore->mid(2 * cAmmoNumber, cAmmoNumber)));

    HWProto::addStringToBuffer(teamscfg, QString("eammreinf %1").arg(cDefaultAmmoStore->mid(3 * cAmmoNumber, cAmmoNumber)));

    HWProto::addStringToBuffer(teamscfg, QString("eammstore"));

    HWProto::addStringToBuffer(teamscfg, QString("eammstore"));

    RawSendIPC(teamscfg);

}

void HWGame::SendTrainingConfig()

{

    QByteArray traincfg;

    HWProto::addStringToBuffer(traincfg, "TL");

    HWTeam missionTeam = HWTeam();

    missionTeam.setName(config->Form->ui.pageTraining->CBTeam->currentText());

    missionTeam.loadFromFile();

    missionTeam.setNumHedgehogs(HEDGEHOGS_PER_TEAM);

    missionTeam.setMissionTeam(true);

    HWProto::addStringListToBuffer(traincfg, missionTeam.teamGameConfig(100));

    HWProto::addStringToBuffer(traincfg, "eseed " + QUuid::createUuid().toString());

    HWProto::addStringToBuffer(traincfg, "escript " + trainingScript);

    RawSendIPC(traincfg);

}

void HWGame::SendCampaignConfig()

{

    QByteArray campaigncfg;

    HWProto::addStringToBuffer(campaigncfg, "TL");

    HWTeam missionTeam = HWTeam();

    missionTeam.setName(config->Form->ui.pageCampaign->CBTeam->currentText());

    missionTeam.loadFromFile();

    missionTeam.setNumHedgehogs(HEDGEHOGS_PER_TEAM);

    missionTeam.setMissionTeam(true);

    HWProto::addStringListToBuffer(campaigncfg, missionTeam.teamGameConfig(100));

    HWProto::addStringToBuffer(campaigncfg, "eseed " + QUuid::createUuid().toString());

    HWProto::addStringToBuffer(campaigncfg, "escript " + campaignScript);

    RawSendIPC(campaigncfg);

}

void HWGame::SendNetConfig()

{

    commonConfig();

}

void HWGame::ParseMessage(const QByteArray & msg)

{

    switch(msg.at(1))

    {

        case '?':

        {

            SendIPC("!");

            break;

        }

        case 'C':

        {

            switch (gameType)

            {

                case gtLocal:

                {

                    SendConfig();

                    break;

                }

                case gtQLocal:

                {

                    SendQuickConfig();

                    break;

                }

                case gtNone:

                case gtSave:

                case gtDemo:

                    break;

                case gtNet:

                {

                    SendNetConfig();

                    break;

                }

                case gtTraining:

                {

                    SendTrainingConfig();

                    break;

                }

                case gtCampaign:

                {

                    SendCampaignConfig();

                    break;

                }

            }

            break;

        }

        case 'E':

        {

            int size = msg.size();

            emit ErrorMessage(

                tr("A fatal ERROR occured! The game engine had to stop.\n\n"

                "We are very sorry for the inconvenience. :-(\n\n"

                "If this keeps happening, please click the 'Feedback' button in the main menu!\n\n"

                "Last engine message:\n%1")

                .arg(QString::fromUtf8(msg.mid(2).left(size - 4))));

            return;

        }

        case 'i':

        {

            emit GameStats(msg.at(2), QString::fromUtf8(msg.mid(3)));

            break;

        }

        case 'Q':

        {

            SetGameState(gsInterrupted);

            break;

        }

        case 'q':

        {

            SetGameState(gsFinished);

            break;

        }

        case 'H':

        {

            SetGameState(gsHalted);

            break;

        }

        case 's':

        {

            int size = msg.size();

            QString msgbody = QString::fromUtf8(msg.mid(2).left(size - 4));

            emit SendChat(msgbody);

            QByteArray buf;

            HWProto::addStringToBuffer(buf, "s" + HWProto::formatChatMsg(config->netNick(), msgbody) + "\x20\x20");

            demo.append(buf);

            break;

        }

        case 'b':

        {

            int size = msg.size();

            QString msgbody = QString::fromUtf8(msg.mid(2).left(size - 4));

            emit SendTeamMessage(msgbody);

            break;

        }

        case 'V':

        {

            if (msg.at(2) == '?')

                sendCampaignVar(msg.right(msg.size() - 3));

            else if (msg.at(2) == '!')

                writeCampaignVar(msg.right(msg.size() - 3));

            break;

        }

        case 'v':

        {

            if (msg.at(2) == '?')

                sendMissionVar(msg.right(msg.size() - 3));

            else if (msg.at(2) == '!')

                writeMissionVar(msg.right(msg.size() - 3));

            break;

        }

        case 'W':

        {

            // fetch new window resolution via IPC and save it in the settings

            int size = msg.size();