/*

 * Hedgewars, a free turn based strategy game

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

#ifdef SPARKLE_ENABLED

#include "CocoaInitializer.h"

#include "SparkleAutoUpdater.h"

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

HWForm::HWForm(QWidget *parent)

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

{

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

    frontendEffects = settings.value("video/frontendeffects", true).toBool();

	ui.setupUi(this);

	CustomizePalettes();

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

	namegen = new HWNamegen();

#ifdef SPARKLE_ENABLED

        AutoUpdater* updater;

        CocoaInitializer initializer;

        updater = new SparkleAutoUpdater("http://files.getdropbox.com/u/24468/appcast.xml"); //this has to change before release!!!

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

            updater->checkForUpdates();

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

}

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)

{

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

		QStringList tmNames = config->GetTeamsList();

		TeamSelWidget* curTeamSelWidget;

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

	}

}

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

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

	ui.Pages->setCurrentIndex(id);

	OnPageShown(id, curid);

	if (id == ID_PAGE_CONNECTING)

		GoBack();

	if (id == ID_PAGE_NETSERVER)

		GoBack();

	if ((!hwnet) && (id == ID_PAGE_ROOMSLIST))

		GoBack();

	if ((!hwnet) || (!hwnet->isInRoom()))

		if (id == ID_PAGE_NETGAME || id == ID_PAGE_NETGAME)

			GoBack();

	if (curid == ID_PAGE_ROOMSLIST) NetDisconnect();

	if (curid == ID_PAGE_SCHEME)

		ammoSchemeModel->Save();

}

void HWForm::btnExitPressed()

{

	eggTimer.start();

}

void HWForm::btnExitClicked()

{

	if (eggTimer.elapsed() < 3000)

		close();

	else

	{

		QPushButton * btn = findChild<QPushButton *>("imageButt");

		if (btn)

		{

			btn->setIcon(QIcon(":/res/bonus.png"));

		}

	}

}

void HWForm::IntermediateSetup()

{

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

	TeamSelWidget* curTeamSelWidget;

	if(id == ID_PAGE_MULTIPLAYER) {

		curTeamSelWidget = ui.pageMultiplayer->teamsSelect;

	} else {

		curTeamSelWidget = ui.pageNetGame->pNetTeamsWidget;

	}

	QList<HWTeam> teams = curTeamSelWidget->getDontPlayingTeams();

	QStringList tmnames;

	for(QList<HWTeam>::iterator it = teams.begin(); it != teams.end(); ++it) {

		tmnames += it->TeamName;

	}

	UpdateTeamsLists(&tmnames); // FIXME: still need more work if teamname is updated while configuring

	GoToPage(ID_PAGE_SETUP);

}

void HWForm::NewTeam()

{

	editedTeam = new HWTeam("unnamed");

	editedTeam->SetToPage(this);

	GoToPage(ID_PAGE_SETUP_TEAM);

}

void HWForm::EditTeam()

{

	editedTeam = new HWTeam(ui.pageOptions->CBTeamName->currentText());

	editedTeam->LoadFromFile();

	editedTeam->SetToPage(this);

	GoToPage(ID_PAGE_SETUP_TEAM);

}

void HWForm::RandomNames()

{

	editedTeam->GetFromPage(this);

	namegen->TeamRandomNames(editedTeam,FALSE);