/*

 * 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 <QSignalMapper>

#include <QShortcut>

#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"

#include "drawmapwidget.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

    game = NULL;

    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(760, 580);

    //setFocusPolicy(Qt::StrongFocus);

    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

#else

    // ctrl+q closes frontend for consistency

    QShortcut *closeFrontend = new QShortcut(QKeySequence("Ctrl+Q"), this);

    connect (closeFrontend, SIGNAL(activated()), this, SLOT(close()));

#endif

    UpdateTeamsLists();

    UpdateCampaignPage(0);

    UpdateWeapons();

    pageSwitchMapper = new QSignalMapper(this);

    connect(pageSwitchMapper, SIGNAL(mapped(int)), this, SLOT(GoToPage(int)));

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

    onFrontendFullscreen(config->isFrontendFullscreen());

    connect(ui.pageMain->BtnSinglePlayer, SIGNAL(clicked()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageMain->BtnSinglePlayer, ID_PAGE_SINGLEPLAYER);

    connect(ui.pageMain->BtnSetup, SIGNAL(clicked()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageMain->BtnSetup, ID_PAGE_SETUP);

    connect(ui.pageMain->BtnNet, SIGNAL(clicked()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageMain->BtnNet, ID_PAGE_NETTYPE);

    connect(ui.pageMain->BtnInfo, SIGNAL(clicked()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageMain->BtnInfo, ID_PAGE_INFO);

    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, SIGNAL(SetupClicked()), this, SLOT(IntermediateSetup()));

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

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

    connect(ui.pageMultiplayer->gameCFG, SIGNAL(goToDrawMap()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageMultiplayer->gameCFG, ID_PAGE_DRAWMAP);

    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()));

#ifndef __APPLE__

    connect(ui.pageOptions->BtnAssociateFiles, SIGNAL(clicked()), this, SLOT(AssociateFiles()));

#endif

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

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

    connect(ui.pageOptions->WeaponDelete, SIGNAL(clicked()), this, SLOT(DeleteWeaponSet()));

    connect(ui.pageOptions->SchemeEdit, SIGNAL(clicked()), this, SLOT(GoToEditScheme()));

    connect(ui.pageOptions->SchemeNew, SIGNAL(clicked()), this, SLOT(GoToNewScheme()));

    connect(ui.pageOptions->SchemeDelete, SIGNAL(clicked()), this, SLOT(DeleteScheme()));

    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()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageNet->BtnNetSvrStart, ID_PAGE_NETSERVER);

    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, SIGNAL(SetupClicked()), this, SLOT(IntermediateSetup()));

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

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

    connect(ui.pageNetGame->pGameCFG, SIGNAL(goToDrawMap()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageNetGame->pGameCFG, ID_PAGE_DRAWMAP);

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

    connect(ui.pageRoomsList->BtnAdmin, SIGNAL(clicked()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageRoomsList->BtnAdmin, ID_PAGE_ADMIN);

    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()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageSinglePlayer->BtnTrainPage, ID_PAGE_TRAINING);

    connect(ui.pageSinglePlayer->BtnCampaignPage, SIGNAL(clicked()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageSinglePlayer->BtnCampaignPage, ID_PAGE_CAMPAIGN);

    connect(ui.pageSinglePlayer->BtnMultiplayer, SIGNAL(clicked()), pageSwitchMapper, SLOT(map()));

    pageSwitchMapper->setMapping(ui.pageSinglePlayer->BtnMultiplayer, ID_PAGE_MULTIPLAYER);

    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.pageCampaign->BtnStartCampaign, SIGNAL(clicked()), this, SLOT(StartCampaign()));

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

    connect(ui.pageCampaign->CBTeam, SIGNAL(currentIndexChanged(int)), this, SLOT(UpdateCampaignPage(int)));

    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()));

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

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

    ui.pageScheme->setModel(ammoSchemeModel);

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

    ui.pageOptions->SchemesName->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::keyReleaseEvent(QKeyEvent *event)

{

  if (event->key() == Qt::Key_Escape) 

    this->GoBack();

}

*/

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);

    combos.push_back(ui.pageSelectWeapon->selectWeaponSet);

    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(tr("DefaultTeam"));

        defaultTeam.SaveToFile();

        teamslist.push_back(tr("DefaultTeam"));

    }

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

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

    ui.pageCampaign->CBTeam->clear();

    ui.pageCampaign->CBTeam->addItems(teamslist);

}

void HWForm::GoToSelectNewWeapon()

{

    ui.pageSelectWeapon->pWeapons->newWeaponsName();

    GoToPage(ID_PAGE_SELECTWEAPON);

}

void HWForm::GoToSelectWeapon()

{

    ui.pageSelectWeapon->selectWeaponSet->setCurrentIndex(ui.pageOptions->WeaponsName->currentIndex());

    GoToPage(ID_PAGE_SELECTWEAPON);

}

void HWForm::GoToSelectWeaponSet(int index)

{

    ui.pageSelectWeapon->selectWeaponSet->setCurrentIndex(index);

    GoToPage(ID_PAGE_SELECTWEAPON);

}

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::GoToScheme(int index)

{

    ui.pageScheme->selectScheme->setCurrentIndex(index);

    GoToPage(ID_PAGE_SCHEME);

}

void HWForm::GoToNewScheme()

{

    ui.pageScheme->newRow();

    GoToPage(ID_PAGE_SCHEME);

}

void HWForm::GoToEditScheme()

{

    ui.pageScheme->selectScheme->setCurrentIndex(ui.pageOptions->SchemesName->currentIndex());

    GoToPage(ID_PAGE_SCHEME);

}

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

{

#ifdef USE_XFIRE

    updateXfire();

#endif

    if(id == ID_PAGE_DRAWMAP)

    {

        DrawMapScene * scene;

        if(lastid == ID_PAGE_MULTIPLAYER)

            scene = ui.pageMultiplayer->gameCFG->pMapContainer->getDrawMapScene();

        else

            scene = ui.pageNetGame->pGameCFG->pMapContainer->getDrawMapScene();

        ui.pageDrawMap->drawMapWidget->setScene(scene);

    }

    if(lastid == ID_PAGE_DRAWMAP)

    {

        if(id == ID_PAGE_MULTIPLAYER)

            ui.pageMultiplayer->gameCFG->pMapContainer->mapDrawingFinished();

        else

            ui.pageNetGame->pGameCFG->pMapContainer->mapDrawingFinished();

    }

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

        QStringList tmNames = config->GetTeamsList();

        TeamSelWidget* curTeamSelWidget;

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

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

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

        ui.pageOptions->CBTeamName->setVisible(false);

        ui.pageOptions->LblNoEditTeam->setVisible(true);

        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();