/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2005-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 <QFile>

#include <QTextStream>

#include <QMessageBox>

#include <QPushButton>

#include <QListWidget>

#include <QStackedLayout>

#include <QLineEdit>

#include <QLabel>

#include <QRadioButton>

#include <QSpinBox>

#include <QCloseEvent>

#include <QCheckBox>

#include <QTextBrowser>

#include <QAction>

#include <QTimer>

#include <QScrollBar>

#include <QDataWidgetMapper>

#include <QTableView>

#include <QCryptographicHash>

#include "hwform.h"

#include "game.h"

#include "team.h"

#include "namegen.h"

#include "teamselect.h"

#include "selectWeapon.h"

#include "gameuiconfig.h"

#include "pages.h"

#include "statsPage.h"

#include "hwconsts.h"

#include "newnetclient.h"

#include "gamecfgwidget.h"

#include "netserverslist.h"

#include "netudpserver.h"

#include "chatwidget.h"

#include "playrecordpage.h"

#include "input_ip.h"

#include "ammoSchemeModel.h"

#include "bgwidget.h"

#include "xfire.h"

#ifdef __APPLE__

#include "CocoaInitializer.h"

#include "M3Panel.h"

#ifdef SPARKLE_ENABLED

#define SPARKLE_APPCAST_URL "http://www.hedgewars.org/download/appcast.xml"

#include "SparkleAutoUpdater.h"

#endif

#endif

// I started handing this down to each place it touches, but it was getting ridiculous

// and this one flag does not warrant a static class

bool frontendEffects = true;

QString playerHash;

HWForm::HWForm(QWidget *parent)

  : QMainWindow(parent), pnetserver(0), pRegisterServer(0), editedTeam(0), hwnet(0)

{

#ifdef USE_XFIRE

    xfire_init();

#endif

    gameSettings = new QSettings(cfgdir->absolutePath() + "/hedgewars.ini", QSettings::IniFormat);

    frontendEffects = gameSettings->value("frontend/effects", true).toBool();

    playerHash = QString(QCryptographicHash::hash(gameSettings->value("net/nick","").toString().toLatin1(), QCryptographicHash::Md5).toHex());

    ui.setupUi(this);

    setMinimumSize(800, 600);

    CustomizePalettes();

    ui.pageOptions->CBResolution->addItems(sdli.getResolutions());

    config = new GameUIConfig(this, cfgdir->absolutePath() + "/hedgewars.ini");

    namegen = new HWNamegen();

#ifdef __APPLE__

        panel = new M3Panel;

#ifdef SPARKLE_ENABLED

        AutoUpdater* updater;

        CocoaInitializer initializer;

        updater = new SparkleAutoUpdater(SPARKLE_APPCAST_URL);

    if(updater && config->isAutoUpdateEnabled())

            updater->checkForUpdates();

#endif        

#endif

    UpdateTeamsLists();

    UpdateWeapons();

    connect(config, SIGNAL(frontendFullscreen(bool)), this, SLOT(onFrontendFullscreen(bool)));

    onFrontendFullscreen(config->isFrontendFullscreen());

    connect(ui.pageMain->BtnSinglePlayer, SIGNAL(clicked()), this, SLOT(GoToSinglePlayer()));

    connect(ui.pageMain->BtnSetup, SIGNAL(clicked()), this, SLOT(GoToSetup()));

    connect(ui.pageMain->BtnNet, SIGNAL(clicked()), this, SLOT(GoToNetType()));

    connect(ui.pageMain->BtnInfo, SIGNAL(clicked()), this, SLOT(GoToInfo()));

    connect(ui.pageMain->BtnExit, SIGNAL(pressed()), this, SLOT(btnExitPressed()));

    connect(ui.pageMain->BtnExit, SIGNAL(clicked()), this, SLOT(btnExitClicked()));

    connect(ui.pageEditTeam->BtnTeamSave, SIGNAL(clicked()), this, SLOT(TeamSave()));

    connect(ui.pageEditTeam->BtnTeamDiscard, SIGNAL(clicked()), this, SLOT(TeamDiscard()));

    connect(ui.pageEditTeam->signalMapper, SIGNAL(mapped(const int &)), this, SLOT(RandomName(const int &)));

    connect(ui.pageEditTeam->randTeamButton, SIGNAL(clicked()), this, SLOT(RandomNames()));

    connect(ui.pageMultiplayer->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageMultiplayer->BtnStartMPGame, SIGNAL(clicked()), this, SLOT(StartMPGame()));

    connect(ui.pageMultiplayer->teamsSelect, SIGNAL(setEnabledGameStart(bool)),

        ui.pageMultiplayer->BtnStartMPGame, SLOT(setEnabled(bool)));

    connect(ui.pageMultiplayer->teamsSelect, SIGNAL(SetupClicked()), this, SLOT(IntermediateSetup()));

    connect(ui.pageMultiplayer->gameCFG, SIGNAL(goToSchemes()), this, SLOT(GoToSchemes()));

    connect(ui.pageMultiplayer->gameCFG, SIGNAL(goToWeapons(const QString &)), this, SLOT(GoToSelectWeaponSet(const QString &)));

    connect(ui.pagePlayDemo->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pagePlayDemo->BtnPlayDemo, SIGNAL(clicked()), this, SLOT(PlayDemo()));

    connect(ui.pagePlayDemo->DemosList, SIGNAL(doubleClicked (const QModelIndex &)), this, SLOT(PlayDemo()));

    connect(ui.pageOptions->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageOptions->BtnNewTeam, SIGNAL(clicked()), this, SLOT(NewTeam()));

    connect(ui.pageOptions->BtnEditTeam, SIGNAL(clicked()), this, SLOT(EditTeam()));

    connect(ui.pageOptions->BtnDeleteTeam, SIGNAL(clicked()), this, SLOT(DeleteTeam()));

    connect(ui.pageOptions->BtnSaveOptions, SIGNAL(clicked()), config, SLOT(SaveOptions()));

    connect(ui.pageOptions->BtnSaveOptions, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageOptions->WeaponEdit, SIGNAL(clicked()), this, SLOT(GoToSelectWeapon()));

    connect(ui.pageOptions->WeaponsButt, SIGNAL(clicked()), this, SLOT(GoToSelectNewWeapon()));

    connect(ui.pageSelectWeapon->pWeapons, SIGNAL(weaponsChanged()), this, SLOT(UpdateWeapons()));

    connect(ui.pageNet->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageNet->BtnSpecifyServer, SIGNAL(clicked()), this, SLOT(NetConnect()));

    connect(ui.pageNet->BtnNetSvrStart, SIGNAL(clicked()), this, SLOT(GoToNetServer()));

    connect(ui.pageNet, SIGNAL(connectClicked(const QString &, quint16)), this, SLOT(NetConnectServer(const QString &, quint16)));

    connect(ui.pageNetServer->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageNetServer->BtnStart, SIGNAL(clicked()), this, SLOT(NetStartServer()));

    connect(ui.pageNetGame->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageNetGame->pNetTeamsWidget, SIGNAL(setEnabledGameStart(bool)),

        ui.pageNetGame->BtnGo, SLOT(setEnabled(bool)));

    connect(ui.pageNetGame->pNetTeamsWidget, SIGNAL(setEnabledGameStart(bool)),

        ui.pageNetGame->BtnStart, SLOT(setEnabled(bool)));

    connect(ui.pageNetGame->pNetTeamsWidget, SIGNAL(SetupClicked()), this, SLOT(IntermediateSetup()));

    connect(ui.pageNetGame->pGameCFG, SIGNAL(goToSchemes()), this, SLOT(GoToSchemes()));

    connect(ui.pageNetGame->pGameCFG, SIGNAL(goToWeapons(const QString &)), this, SLOT(GoToSelectWeaponSet(const QString &)));

    connect(ui.pageRoomsList->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageRoomsList->BtnAdmin, SIGNAL(clicked()), this, SLOT(GoToAdmin()));

    connect(ui.pageInfo->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageGameStats->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageSinglePlayer->BtnSimpleGamePage, SIGNAL(clicked()), this, SLOT(SimpleGame()));

    connect(ui.pageSinglePlayer->BtnTrainPage, SIGNAL(clicked()), this, SLOT(GoToTraining()));

    connect(ui.pageSinglePlayer->BtnMultiplayer, SIGNAL(clicked()), this, SLOT(GoToMultiplayer()));

    connect(ui.pageSinglePlayer->BtnLoad, SIGNAL(clicked()), this, SLOT(GoToSaves()));

    connect(ui.pageSinglePlayer->BtnDemos, SIGNAL(clicked()), this, SLOT(GoToDemos()));

    connect(ui.pageSinglePlayer->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageTraining->BtnStartTrain, SIGNAL(clicked()), this, SLOT(StartTraining()));

    connect(ui.pageTraining->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageSelectWeapon->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageSelectWeapon->BtnDelete, SIGNAL(clicked()),

        ui.pageSelectWeapon->pWeapons, SLOT(deleteWeaponsName())); // executed first

    connect(ui.pageSelectWeapon->pWeapons, SIGNAL(weaponsDeleted()),

        this, SLOT(UpdateWeapons())); // executed second

    connect(ui.pageSelectWeapon->pWeapons, SIGNAL(weaponsDeleted()),

        this, SLOT(GoBack())); // executed third

    connect(ui.pageScheme->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageAdmin->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageNetType->BtnBack, SIGNAL(clicked()), this, SLOT(GoBack()));

    connect(ui.pageNetType->BtnLAN, SIGNAL(clicked()), this, SLOT(GoToNet()));

    connect(ui.pageNetType->BtnOfficialServer, SIGNAL(clicked()), this, SLOT(NetConnectOfficialServer()));

    ammoSchemeModel = new AmmoSchemeModel(this, cfgdir->absolutePath() + "/schemes.ini");

    ui.pageScheme->setModel(ammoSchemeModel);

    ui.pageMultiplayer->gameCFG->GameSchemes->setModel(ammoSchemeModel);

    wBackground = NULL;

    if (config->isFrontendEffects()) {

       wBackground = new BGWidget(this);

       wBackground->setFixedSize(this->width(), this->height());

       wBackground->lower();

       wBackground->init();

       wBackground->startAnimation();

    }

    PagesStack.push(ID_PAGE_MAIN);

    GoBack();

}

#ifdef USE_XFIRE

void HWForm::updateXfire(void)

{

    if(hwnet)

    {

        xfire_setvalue(XFIRE_SERVER, !hwnet->getHost().compare("netserver.hedgewars.org:46631") ? "Official server" : hwnet->getHost().toAscii());

        switch(hwnet->getClientState())

        {

            case 1: // Connecting

            xfire_setvalue(XFIRE_STATUS, "Connecting");

            xfire_setvalue(XFIRE_NICKNAME, "-");

            xfire_setvalue(XFIRE_ROOM, "-");

            case 2: // In lobby

            xfire_setvalue(XFIRE_STATUS, "Online");

            xfire_setvalue(XFIRE_NICKNAME, hwnet->getNick().toAscii());

            xfire_setvalue(XFIRE_ROOM, "In game lobby");

            break;

            case 3: // In room

            xfire_setvalue(XFIRE_STATUS, "Online");

            xfire_setvalue(XFIRE_NICKNAME, hwnet->getNick().toAscii());

            xfire_setvalue(XFIRE_ROOM, (hwnet->getRoom() + " (waiting for players)").toAscii());

            break;

            case 5: // In game

            xfire_setvalue(XFIRE_STATUS, "Online");

            xfire_setvalue(XFIRE_NICKNAME, hwnet->getNick().toAscii());

            xfire_setvalue(XFIRE_ROOM, (hwnet->getRoom() + " (playing or spectating)").toAscii());

            break;

            default:

            break;

        }

    }

    else

    {

        xfire_setvalue(XFIRE_STATUS, "Offline");

        xfire_setvalue(XFIRE_NICKNAME, "-");

        xfire_setvalue(XFIRE_ROOM, "-");

        xfire_setvalue(XFIRE_SERVER, "-");

    }

    xfire_update();

}

#endif

void HWForm::onFrontendFullscreen(bool value)

{

  if (value)

    setWindowState(windowState() | Qt::WindowFullScreen);

  else {

    setWindowState(windowState() & !Qt::WindowFullScreen);

  }

}

void HWForm::CustomizePalettes()

{

    QList<QScrollBar *> allSBars = findChildren<QScrollBar *>();

    QPalette pal = palette();

    pal.setColor(QPalette::WindowText, QColor(0xff, 0xcc, 0x00));

    pal.setColor(QPalette::Button, QColor(0x00, 0x35, 0x1d));

    pal.setColor(QPalette::Base, QColor(0x00, 0x35, 0x1d));

    pal.setColor(QPalette::Window, QColor(0x00, 0x00, 0x00));

    for (int i = 0; i < allSBars.size(); ++i)

        allSBars.at(i)->setPalette(pal);

}

void HWForm::UpdateWeapons()

{

    QVector<QComboBox*> combos;

    combos.push_back(ui.pageOptions->WeaponsName);

    combos.push_back(ui.pageMultiplayer->gameCFG->WeaponsName);

    combos.push_back(ui.pageNetGame->pGameCFG->WeaponsName);

    QStringList names = ui.pageSelectWeapon->pWeapons->getWeaponNames();

    for(QVector<QComboBox*>::iterator it = combos.begin(); it != combos.end(); ++it) {

        (*it)->clear();

        for(int i = 0; i < names.size(); ++i)

            (*it)->addItem(names[i], ui.pageSelectWeapon->pWeapons->getWeaponsString(names[i]));

        int pos = (*it)->findText("Default");

        if (pos != -1) {

            (*it)->setCurrentIndex(pos);

        }

    }

}

void HWForm::UpdateTeamsLists(const QStringList* editable_teams)

{

    QStringList teamslist;

    if(editable_teams) {

      teamslist =* editable_teams;

    } else {

      teamslist = config->GetTeamsList();

    }

    if(teamslist.empty()) {

        HWTeam defaultTeam("DefaultTeam");

        defaultTeam.SaveToFile();

        teamslist.push_back("DefaultTeam");

    }

    ui.pageOptions->CBTeamName->clear();

    ui.pageOptions->CBTeamName->addItems(teamslist);

}

void HWForm::GoToMain()

{

    GoToPage(ID_PAGE_MAIN);

}

void HWForm::GoToSinglePlayer()

{

    GoToPage(ID_PAGE_SINGLEPLAYER);

}

void HWForm::GoToTraining()

{

    GoToPage(ID_PAGE_TRAINING);

}

void HWForm::GoToSetup()

{

    GoToPage(ID_PAGE_SETUP);

}

void HWForm::GoToSelectNewWeapon()

{

    ui.pageSelectWeapon->pWeapons->setWeaponsName(tr("new"));

    GoToPage(ID_PAGE_SELECTWEAPON);

}

void HWForm::GoToSelectWeapon()

{

    ui.pageSelectWeapon->pWeapons->setWeaponsName(ui.pageOptions->WeaponsName->currentText());

    GoToPage(ID_PAGE_SELECTWEAPON);

}

void HWForm::GoToSelectWeaponSet(const QString & name)

{

    ui.pageSelectWeapon->pWeapons->setWeaponsName(name);

    GoToPage(ID_PAGE_SELECTWEAPON);

}

void HWForm::GoToInfo()

{

    GoToPage(ID_PAGE_INFO);

}

void HWForm::GoToMultiplayer()

{

    GoToPage(ID_PAGE_MULTIPLAYER);

}

void HWForm::GoToSaves()

{

    ui.pagePlayDemo->FillFromDir(PagePlayDemo::RT_Save);

    GoToPage(ID_PAGE_DEMOS);

}

void HWForm::GoToDemos()

{

    ui.pagePlayDemo->FillFromDir(PagePlayDemo::RT_Demo);

    GoToPage(ID_PAGE_DEMOS);

}

void HWForm::GoToNet()

{

    ui.pageNet->updateServersList();

    GoToPage(ID_PAGE_NET);

}

void HWForm::GoToNetType()

{

    GoToPage(ID_PAGE_NETTYPE);

}

void HWForm::GoToNetServer()

{

    GoToPage(ID_PAGE_NETSERVER);

}

void HWForm::GoToSchemes()

{

    GoToPage(ID_PAGE_SCHEME);

}

void HWForm::GoToAdmin()

{

    GoToPage(ID_PAGE_ADMIN);

}

void HWForm::OnPageShown(quint8 id, quint8 lastid)

{

#ifdef USE_XFIRE

    updateXfire();

#endif

    if (id == ID_PAGE_MULTIPLAYER || id == ID_PAGE_NETGAME) {

        QStringList tmNames = config->GetTeamsList();

        TeamSelWidget* curTeamSelWidget;

        ui.pageOptions->BtnNewTeam->setEnabled(false);

        ui.pageOptions->BtnEditTeam->setEnabled(false);

        ui.pageOptions->BtnDeleteTeam->setEnabled(false);

        if(id == ID_PAGE_MULTIPLAYER) {

          curTeamSelWidget = ui.pageMultiplayer->teamsSelect;

        } else {

          curTeamSelWidget = ui.pageNetGame->pNetTeamsWidget;

        }

        QList<HWTeam> teamsList;

        for(QStringList::iterator it = tmNames.begin(); it != tmNames.end(); it++) {

          HWTeam team(*it);

          team.LoadFromFile();

          teamsList.push_back(team);

        }

        if(lastid == ID_PAGE_SETUP) { // _TEAM

          if (editedTeam) {

            curTeamSelWidget->addTeam(*editedTeam);

          }

        } else if(lastid != ID_PAGE_GAMESTATS

                && lastid != ID_PAGE_INGAME

                && lastid != ID_PAGE_SCHEME

                && lastid != ID_PAGE_SELECTWEAPON) {

            curTeamSelWidget->resetPlayingTeams(teamsList);

        }

    } else

    if (id == ID_PAGE_GAMESTATS)

    {

        ui.pageGameStats->renderStats();

    }

    if(id == ID_PAGE_MAIN)

    {

        ui.pageOptions->BtnNewTeam->setEnabled(true);

        ui.pageOptions->BtnEditTeam->setEnabled(true);

        ui.pageOptions->BtnDeleteTeam->setEnabled(true);

    }

    // load and save ignore/friends lists

    if(lastid == ID_PAGE_NETGAME) // leaving a room

        ui.pageNetGame->pChatWidget->saveLists(ui.pageOptions->editNetNick->text());

    else if(lastid == ID_PAGE_ROOMSLIST) // leaving the lobby

        ui.pageRoomsList->chatWidget->saveLists(ui.pageOptions->editNetNick->text());

    if(id == ID_PAGE_NETGAME) // joining a room

        ui.pageNetGame->pChatWidget->loadLists(ui.pageOptions->editNetNick->text());

    else if(id == ID_PAGE_ROOMSLIST) // joining the lobby

        ui.pageRoomsList->chatWidget->loadLists(ui.pageOptions->editNetNick->text());

}

void HWForm::GoToPage(quint8 id)

{

    quint8 lastid = ui.Pages->currentIndex();

    PagesStack.push(ui.Pages->currentIndex());

    OnPageShown(id, lastid);

    ui.Pages->setCurrentIndex(id);

}

void HWForm::GoBack()

{

    quint8 id = PagesStack.isEmpty() ? ID_PAGE_MAIN : PagesStack.pop();