/*

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

#include <QFile>

#include <QTextStream>

#include <QMessageBox>

#include <QPushButton>

#include <QSpinBox>

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

#include <QApplication>

#include <QInputDialog>

#include <QPropertyAnimation>

#include <QSettings>

#include <QSortFilterProxyModel>

#include <QIcon>

#include <QImage>

#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 "pagenetserver.h"

#include "pagedrawmap.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 "gameSchemeModel.h"

#include "bgwidget.h"

#include "drawmapwidget.h"

#include "mouseoverfilter.h"

#include "roomslistmodel.h"

#include "recorder.h"

#include "playerslistmodel.h"

#include "feedbackdialog.h"

#include "MessageDialog.h"

#include "DataManager.h"

#include "AutoUpdater.h"

#ifdef Q_OS_WIN

#ifndef WINVER

#define WINVER 0x0500

#endif

#include <windows.h>

#else

#include <unistd.h>

#include <sys/types.h>

#endif

#ifdef Q_OS_MAC

#include <sys/sysctl.h>

#endif

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

QIcon finishedIcon;

QIcon notFinishedIcon;

GameUIConfig* HWForm::config = 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");

    this->setStyleSheet(styleSheet);

    QIcon * hwIcon = new QIcon();

    hwIcon->addFile(":/res/hh_small.png");

    //hwIcon->addFile(":/res/hh25x25.png");

    // crop-workaround for the fact that hh25x25.png is actually 25x35

    QPixmap pm(":/res/hh25x25.png");

    hwIcon->addPixmap(pm.copy(0,(pm.height()-25)/2,25,25));

    hwIcon->addFile(":/res/teamicon.png");

    hwIcon->addFile(":/res/teamicon2.png");

    this->setWindowIcon(*hwIcon);

    ui.setupUi(this);

    setMinimumSize(760, 580);

    //setFocusPolicy(Qt::StrongFocus);

    CustomizePalettes();

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

    config = new GameUIConfig(this, DataManager::instance().settingsFileName());

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

    playerHash = QString(QCryptographicHash::hash(config->value("net/nick",tr("Guest")+QString("%1").arg(rand())).toString().toUtf8(), QCryptographicHash::Md5).toHex());

    // Icons for finished missions

    finishedIcon.addFile(":/res/missionFinished.png", QSize(), QIcon::Normal, QIcon::On);

    finishedIcon.addFile(":/res/missionFinishedSelected.png", QSize(), QIcon::Selected, QIcon::On);

    // A transparent icon, used to nicely align the unfinished missions with the finished ones

    QPixmap emptySpace = QPixmap(15, 15);

    emptySpace.fill(QColor(0, 0, 0, 0));

    notFinishedIcon = QIcon(emptySpace);

    ui.pageRoomsList->setSettings(config);

    ui.pageNetGame->setSettings(config);

    ui.pageNetGame->chatWidget->setSettings(config);

    ui.pageRoomsList->chatWidget->setSettings(config);

    ui.pageOptions->setConfig(config);

#ifdef VIDEOREC

    ui.pageVideos->init(config);

#endif

#if defined(__APPLE__) && defined(SPARKLE_ENABLED)

    if (config->isAutoUpdateEnabled())

    {

        AutoUpdater* updater = NULL;

        updater = new SparkleAutoUpdater();

        if (updater)

        {

            updater->checkForUpdates();

            delete updater;

        }

    }

#endif

#ifdef __APPLE__

    panel = new M3Panel;

    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

    previousCampaignName = "";

    previousTeamName = "";

    UpdateTeamsLists();

    InitCampaignPage();

    UpdateCampaignPage(0);

    UpdateCampaignPageTeam(0);

    UpdateCampaignPageMission(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()), this, SLOT(showFeedbackDialog()));

    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.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(GoToWeapons(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(GoToEditWeapons()));

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

    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.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(GoToWeapons(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.pageGameStats, SIGNAL(restartGameRequested()), this, SLOT(restartGame()));

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

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

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

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

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

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

    connect(ui.pageCampaign->CBMission, SIGNAL(currentIndexChanged(int)), this, SLOT(UpdateCampaignPageMission(int)));

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

             this, SLOT(DeleteWeapons(QString)));

    connect(ui.pageSelectWeapon->pWeapons, SIGNAL(weaponsAdded(QString, QString)),

             this, SLOT(AddWeapons(QString, QString)));

    connect(ui.pageSelectWeapon->pWeapons, SIGNAL(weaponsEdited(QString, QString, QString)),

             this, SLOT(EditWeapons(QString, QString, QString)));

    connect(ui.pageSelectWeapon->pWeapons, SIGNAL(weaponsEdited(QString, QString, QString)),

             ui.pageNetGame->pGameCFG, SLOT(resendAmmoData()));

    connect(ui.pageMain->BtnNetLocal, SIGNAL(clicked()), this, SLOT(GoToNet()));

    connect(ui.pageMain->BtnNetOfficial, SIGNAL(clicked()), this, SLOT(NetConnectOfficialServer()));

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

    gameSchemeModel = new GameSchemeModel(this, cfgdir->absolutePath() + "/Schemes/Game");

    ui.pageScheme->setModel(gameSchemeModel);

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

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

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

        }

    }

    ui.Pages->setCurrentIndex(ID_PAGE_INFO);

    PagesStack.push(ID_PAGE_MAIN);

    ((AbstractPage*)ui.Pages->widget(ID_PAGE_MAIN))->triggerPageEnter();

    GoBack();

}

void HWForm::onFrontendFullscreen(bool value)

{

    if (value)

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

    else

    {

        setWindowState(windowState() & ~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()

{

    // Scroll bar widget palette

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

    // Set default hyperlink color

    QPalette appPal = qApp->palette();

    appPal.setColor(QPalette::Link, QColor(0xff, 0xff, 0x6e));

    qApp->setPalette(appPal);

}

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::AddWeapons(QString weaponsName, QString ammo)

{

    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)->addItem(weaponsName, QVariant(ammo));

    }

    ui.pageSelectWeapon->selectWeaponSet->setCurrentIndex(ui.pageSelectWeapon->selectWeaponSet->count()-1);

}

void HWForm::DeleteWeapons(QString weaponsName)

{

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