/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2012 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 <QDir>

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

#include <QInputDialog>

#include <QPropertyAnimation>

#include <QSettings>

#if (QT_VERSION >= 0x040600)

#include <QGraphicsEffect>

#include <QParallelAnimationGroup>

#endif

#include "hwform.h"

#include "game.h"

#include "team.h"

#include "campaign.h"

#include "teamselect.h"

#include "selectWeapon.h"

#include "gameuiconfig.h"

#include "pageinfo.h"

#include "pagetraining.h"

#include "pagesingleplayer.h"

#include "pageselectweapon.h"

#include "pageadmin.h"

#include "pagecampaign.h"

#include "pagescheme.h"

#include "pagenetgame.h"

#include "pageroomslist.h"

#include "pageconnecting.h"

#include "pageoptions.h"

#include "pageeditteam.h"

#include "pagemultiplayer.h"

#include "pagenet.h"

#include "pagemain.h"

#include "pagefeedback.h"

#include "pagenetserver.h"

#include "pagedrawmap.h"

#include "pagenettype.h"

#include "pagegamestats.h"

#include "pageplayrecord.h"

#include "pagedata.h"

#include "pagevideos.h"

#include "hwconsts.h"

#include "newnetclient.h"

#include "gamecfgwidget.h"

#include "netserverslist.h"

#include "netudpserver.h"

#include "chatwidget.h"

#include "input_ip.h"

#include "input_password.h"

#include "ammoSchemeModel.h"

#include "bgwidget.h"

#include "xfire.h"

#include "drawmapwidget.h"

#include "mouseoverfilter.h"

#include "roomslistmodel.h"

#include "recorder.h"

#include "playerslistmodel.h"

#include "DataManager.h"

#ifdef __APPLE__

#include "M3Panel.h"

#ifdef SPARKLE_ENABLED

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

GameUIConfig* HWForm::config = NULL;

QSettings* HWForm::gameSettings = NULL;

HWForm::HWForm(QWidget *parent, QString styleSheet)

    : QMainWindow(parent)

    , game(0)

    , pnetserver(0)

    , pRegisterServer(0)

    , editedTeam(0)

    , hwnet(0)

{

    // set music track

    SDLInteraction::instance().setMusicTrack("/Music/main_theme.ogg");

#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().toUtf8(), QCryptographicHash::Md5).toHex());

    this->setStyleSheet(styleSheet);

    ui.setupUi(this);

    setMinimumSize(760, 580);

    //setFocusPolicy(Qt::StrongFocus);

    CustomizePalettes();

    ui.pageOptions->CBResolution->addItems(SDLInteraction::instance().getResolutions());

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

    ui.pageVideos->init(config);

#ifdef __APPLE__

    panel = new M3Panel;

#ifdef SPARKLE_ENABLED

    AutoUpdater* updater;

    updater = new SparkleAutoUpdater();

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

        updater->checkForUpdates();

#endif

    QShortcut *hideFrontend = new QShortcut(QKeySequence("Ctrl+M"), this);

    connect (hideFrontend, SIGNAL(activated()), this, SLOT(showMinimized()));

#else

    // ctrl+q closes frontend for consistency

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

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

    //QShortcut * updateData = new QShortcut(QKeySequence("F5"), this);

    //connect (updateData, SIGNAL(activated()), &DataManager::instance(), SLOT(reload()));

#endif

    UpdateTeamsLists();

    InitCampaignPage();

    UpdateCampaignPage(0);

    UpdateWeapons();

    // connect all goBack signals

    int nPages = ui.Pages->count();

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

        connect(ui.Pages->widget(i), SIGNAL(goBack()), this, SLOT(GoBack()));

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

    pageSwitchMapper->setMapping(ui.pageMain->BtnFeedback, ID_PAGE_FEEDBACK);

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

    pageSwitchMapper->setMapping(ui.pageMain->BtnDataDownload, ID_PAGE_DATADOWNLOAD);

#ifdef VIDEOREC

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

    pageSwitchMapper->setMapping(ui.pageMain->BtnVideos, ID_PAGE_VIDEOS);

#endif

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

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

    connect(ui.pageFeedback->BtnSend, SIGNAL(clicked()), this, SLOT(SendFeedback()));

    connect(ui.pageEditTeam, SIGNAL(goBack()), this, SLOT(AfterTeamEdit()));

    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->BtnPlayDemo, SIGNAL(clicked()), this, SLOT(PlayDemo()));

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

    connect(ui.pageOptions, SIGNAL(newTeamRequested()), this, SLOT(NewTeam()));

    connect(ui.pageOptions, SIGNAL(editTeamRequested(const QString&)), this, SLOT(EditTeam(const QString&)));

    connect(ui.pageOptions, SIGNAL(deleteTeamRequested(const QString&)), this, SLOT(DeleteTeam(const QString&)));

    connect(ui.pageOptions, SIGNAL(goBack()), config, SLOT(SaveOptions()));

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

    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.pageOptions->CBFrontendEffects, SIGNAL(toggled(bool)), this, SLOT(onFrontendEffects(bool)) );

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

    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->BtnStart, SIGNAL(clicked()), this, SLOT(NetStartServer()));

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

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

    connect(ui.pageInfo->BtnSnapshots, SIGNAL(clicked()), this, SLOT(OpenSnapshotFolder()));

    connect(ui.pageGameStats, SIGNAL(saveDemoRequested()), this, SLOT(saveDemoWithCustomName()));

    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.pageTraining, SIGNAL(startMission(const QString&)), this, SLOT(startTraining(const QString&)));

    connect(ui.pageCampaign->BtnStartCampaign, SIGNAL(clicked()), this, SLOT(StartCampaign()));

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

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

    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.pageNetType->BtnLAN, SIGNAL(clicked()), this, SLOT(GoToNet()));

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

    connect(ui.pageConnecting, SIGNAL(cancelConnection()), this, SLOT(GoBack()));

    connect(ui.pageVideos, SIGNAL(goBack()), config, SLOT(SaveVideosOptions()));

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

    ui.pageScheme->setModel(ammoSchemeModel);

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

    ui.pageOptions->SchemesName->setModel(ammoSchemeModel);

    wBackground = new BGWidget(this);

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

    wBackground->lower();

    wBackground->init();

    wBackground->enabled = config->isFrontendEffects();

    wBackground->startAnimation();

    //Install all eventFilters :

    MouseOverFilter *filter = new MouseOverFilter();

    filter->setUi(&ui);

    QList<QWidget *> widgets;

    for (int i=0; i < ui.Pages->count(); i++)

    {

        widgets = ui.Pages->widget(i)->findChildren<QWidget *>();

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

        {

            widgets.at(i)->installEventFilter(filter);

        }

    }

    PagesStack.push(ID_PAGE_MAIN);

    GoBack();

}

#ifdef USE_XFIRE

void HWForm::updateXfire(void)

{

    if(hwnet && (hwnet->clientState() != HWNewNet::Disconnected))

    {

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

        switch(hwnet->clientState())

        {

            case HWNewNet::Connecting: // Connecting

            case HWNewNet::Connected:

                xfire_setvalue(XFIRE_STATUS, "Connecting");

                xfire_setvalue(XFIRE_NICKNAME, "-");

                xfire_setvalue(XFIRE_ROOM, "-");

            case HWNewNet::InLobby: // In lobby

                xfire_setvalue(XFIRE_STATUS, "Online");

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

                xfire_setvalue(XFIRE_ROOM, "In game lobby");

                break;

            case HWNewNet::InRoom: // 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 HWNewNet::InGame: // 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() & static_cast<int>(!Qt::WindowFullScreen));

    }

}

void HWForm::onFrontendEffects(bool value)

{

    wBackground->enabled = value;

    if (value)

        wBackground->startAnimation();

    else

        wBackground->stopAnimation();

}

/*

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

}