/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2006-2008 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 <QGridLayout>

#include <QPushButton>

#include <QGroupBox>

#include <QComboBox>

#include <QCheckBox>

#include <QLabel>

#include <QToolBox>

#include <QLineEdit>

#include <QListWidget>

#include <QApplication>

#include <QSpinBox>

#include <QTextEdit>

#include <QRadioButton>

#include <QTableView>

#include <QMessageBox>

#include <QHeaderView>

#include <QTabWidget>

#include "pages.h"

#include "sdlkeys.h"

#include "hwconsts.h"

#include "gamecfgwidget.h"

#include "teamselect.h"

#include "gamecfgwidget.h"

#include "SquareLabel.h"

#include "mapContainer.h"

#include "about.h"

#include "fpsedit.h"

#include "netserverslist.h"

#include "netudpwidget.h"

#include "netwwwwidget.h"

#include "chatwidget.h"

#include "playrecordpage.h"

#include "selectWeapon.h"

#include "igbox.h"

#include "hats.h"

PageMain::PageMain(QWidget* parent) : 

  AbstractPage(parent)

{

	QGridLayout * pageLayout = new QGridLayout(this);

	//pageLayout->setColumnStretch(0, 1);

	//pageLayout->setColumnStretch(1, 2);

	//pageLayout->setColumnStretch(2, 1);

	//QPushButton* btnLogo = addButton(":/res/HedgewarsTitle.png", pageLayout, 0, 0, 1, 4, true);

	//pageLayout->setAlignment(btnLogo, Qt::AlignHCenter);

	pageLayout->setRowStretch(0, 1);

	pageLayout->setRowStretch(1, 1);

	pageLayout->setRowStretch(2, 0);

	pageLayout->setRowStretch(3, 1);

	pageLayout->setRowStretch(4, 1);

	BtnSinglePlayer = addButton(":/res/LocalPlay.png", pageLayout, 2, 0, 1, 2, true);

	pageLayout->setAlignment(BtnSinglePlayer, Qt::AlignHCenter);

	BtnNet = addButton(":/res/NetworkPlay.png", pageLayout, 2, 2, 1, 2, true);

	pageLayout->setAlignment(BtnNet, Qt::AlignHCenter);

	BtnSetup = addButton(":/res/Settings.png", pageLayout, 4, 3, true);

	//BtnInfo = addButton(":/res/About.png", pageLayout, 3, 1, 1, 2, true);

	BtnInfo = addButton(":/res/HedgewarsTitle.png", pageLayout, 0, 0, 1, 4, true);

	BtnInfo->setStyleSheet("border: transparent;background: transparent;");

	pageLayout->setAlignment(BtnInfo, Qt::AlignHCenter);

	//pageLayout->setAlignment(BtnInfo, Qt::AlignHCenter);

	BtnExit = addButton(":/res/Exit.png", pageLayout, 4, 0, 1, 1, true);

}

PageEditTeam::PageEditTeam(QWidget* parent) :

  AbstractPage(parent)

{

	QGridLayout * pageLayout = new QGridLayout(this);

	QTabWidget * tbw = new QTabWidget(this);

	QWidget * page1 = new QWidget();

	QWidget * page2 = new QWidget();

	tbw->addTab(page1, tr("General"));

	tbw->addTab(page2, tr("Advanced"));

	pageLayout->addWidget(tbw, 0, 0, 1, 3);

	BtnTeamDiscard = addButton(":/res/Exit.png", pageLayout, 1, 0, true);

	BtnTeamSave = addButton(":/res/Save.png", pageLayout, 1, 2, true);;

	QHBoxLayout * page1Layout = new QHBoxLayout(page1);

	page1Layout->setAlignment(Qt::AlignTop);

	QGridLayout * page2Layout = new QGridLayout(page2);

// ====== Page 1 ======

	QVBoxLayout * vbox1 = new QVBoxLayout();

	QVBoxLayout * vbox2 = new QVBoxLayout();

	QVBoxLayout * vbox3 = new QVBoxLayout();

	page1Layout->addLayout(vbox1);

	page1Layout->addLayout(vbox2);

	page1Layout->addLayout(vbox3);

	GBoxTeam = new QGroupBox(this);

	GBoxTeam->setTitle(QGroupBox::tr("Team"));

	GBoxTeam->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

	QGridLayout * GBTLayout = new QGridLayout(GBoxTeam);

	TeamNameEdit = new QLineEdit(GBoxTeam);

	TeamNameEdit->setMaxLength(64);

	GBTLayout->addWidget(TeamNameEdit, 0, 0, 1, 0);

	vbox1->addWidget(GBoxTeam);

	GBoxHedgehogs = new QGroupBox(this);

	GBoxHedgehogs->setTitle(QGroupBox::tr("Team Members"));

	GBoxHedgehogs->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

	QGridLayout * GBHLayout = new QGridLayout(GBoxHedgehogs);

	HatsModel * hatsModel = new HatsModel(GBoxHedgehogs);

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

	{

		HHNameEdit[i] = new QLineEdit(GBoxHedgehogs);

		HHNameEdit[i]->setMaxLength(64);

		HHNameEdit[i]->setMinimumWidth(120);

		GBHLayout->addWidget(HHNameEdit[i], i, 0);

		HHHats[i] = new QComboBox(GBoxHedgehogs);

		HHHats[i]->setModel(hatsModel);

		//HHHats[i]->setModelColumn(1);

		//HHHats[i]->setMinimumWidth(132);

		GBHLayout->addWidget(HHHats[i], i, 1);

	}

	vbox1->addWidget(GBoxHedgehogs);

	GBoxGrave = new QGroupBox(this);

	GBoxGrave->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

	GBoxGrave->setTitle(QGroupBox::tr("Grave"));

	QGridLayout * GBGLayout = new QGridLayout(GBoxGrave);

	CBGrave = new QComboBox(GBoxGrave);

	CBGrave->setMaxCount(65535);

	GBGLayout->addWidget(CBGrave, 0, 0, 1, 3);

	GravePreview = new QLabel(GBoxGrave);

	GravePreview->setScaledContents(false);

	GBGLayout->addWidget(GravePreview, 1, 1);

	vbox2->addWidget(GBoxGrave);

	//page1Layout->addWidget(new QWidget(), 3, 1, 1, 1);

	GBoxTeamLvl = new QGroupBox(this);

	GBoxTeamLvl->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

	GBoxTeamLvl->setTitle(QGroupBox::tr("Team level"));

	QGridLayout * GBTLLayout = new QGridLayout(GBoxTeamLvl);

	CBTeamLvl = new QComboBox(GBoxTeamLvl);

	CBTeamLvl->addItem(QComboBox::tr("Human"));

	CBTeamLvl->addItem(QComboBox::tr("Level 5"));

	CBTeamLvl->addItem(QComboBox::tr("Level 4"));

	CBTeamLvl->addItem(QComboBox::tr("Level 3"));

	CBTeamLvl->addItem(QComboBox::tr("Level 2"));

	CBTeamLvl->addItem(QComboBox::tr("Level 1"));

	CBTeamLvl->setMaxCount(6);

	GBTLLayout->addWidget(CBTeamLvl, 0, 0, 1, 3);

	LevelPict = new QLabel(GBoxTeamLvl);

	LevelPict->setScaledContents(false);

	LevelPict->setFixedSize(32, 32);

	GBTLLayout->addWidget(LevelPict, 1, 1);

	vbox2->addWidget(GBoxTeamLvl);

	GBoxFort = new QGroupBox(this);

	GBoxFort->setTitle(QGroupBox::tr("Fort"));

	QGridLayout * GBFLayout = new QGridLayout(GBoxFort);

	CBFort = new QComboBox(GBoxFort);

	CBFort->setMaxCount(65535);

	GBFLayout->addWidget(CBFort, 0, 0);

	FortPreview = new SquareLabel(GBoxFort);

	FortPreview->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

	FortPreview->setPixmap(QPixmap());

	GBFLayout->addWidget(FortPreview, 1, 0);

	vbox3->addWidget(GBoxFort);

	QDir tmpdir;

	tmpdir.cd(datadir->absolutePath());

	tmpdir.cd("Forts");

	tmpdir.setFilter(QDir::Files);

	CBFort->addItems(tmpdir.entryList(QStringList("*L.png")).replaceInStrings(QRegExp("^(.*)L\\.png"), "\\1"));

	tmpdir.cd("../Graphics/Graves");

	QStringList list = tmpdir.entryList(QStringList("*.png"));

	for (QStringList::Iterator it = list.begin(); it != list.end(); ++it )

	{

		CBGrave->addItem((*it).replace(QRegExp("^(.*)\\.png"), "\\1"));

	}

	connect(CBGrave, SIGNAL(activated(const QString &)), this, SLOT(CBGrave_activated(const QString &)));

	connect(CBTeamLvl, SIGNAL(activated(int)), this, SLOT(CBTeamLvl_activated(int)));

	connect(CBFort, SIGNAL(activated(const QString &)), this, SLOT(CBFort_activated(const QString &)));

	vbox1->addStretch();

	vbox2->addStretch();

//	vbox3->addStretch();

// ====== Page 2 ======

	GBoxBinds = new QGroupBox(this);

	GBoxBinds->setTitle(QGroupBox::tr("Key binds"));

	QGridLayout * GBBLayout = new QGridLayout(GBoxBinds);

	BindsBox = new QToolBox(GBoxBinds);

	BindsBox->setLineWidth(0);

	GBBLayout->addWidget(BindsBox);

	page_A = new QWidget();

	BindsBox->addItem(page_A, QToolBox::tr("Actions"));

	page_W = new QWidget();

	BindsBox->addItem(page_W, QToolBox::tr("Weapons"));

	page_WP = new QWidget();

	BindsBox->addItem(page_WP, QToolBox::tr("Weapon properties"));

	page_O = new QWidget();

	BindsBox->addItem(page_O, QToolBox::tr("Other"));

	page2Layout->addWidget(GBoxBinds, 0, 0);

	QStringList binds;

	for(int i = 0; strlen(sdlkeys[i][1]) > 0; i++)

	{

		binds << sdlkeys[i][1];

	}

	quint16 widind = 0, i = 0;

	while (i < BINDS_NUMBER) {

		quint16 num = 0;

		QGridLayout * pagelayout = new QGridLayout(BindsBox->widget(widind));

		do {

			LBind[i] = new QLabel(BindsBox->widget(widind));

			LBind[i]->setText(QApplication::translate("binds", cbinds[i].name));

			LBind[i]->setAlignment(Qt::AlignRight);

			pagelayout->addWidget(LBind[i], num, 0);

			CBBind[i] = new QComboBox(BindsBox->widget(widind));

			CBBind[i]->addItems(binds);

			pagelayout->addWidget(CBBind[i], num, 1);

			num++;

		} while (!cbinds[i++].chwidget);

		pagelayout->addWidget(new QWidget(BindsBox->widget(widind)), num, 0, 1, 2);

		widind++;

	}

}

void PageEditTeam::CBGrave_activated(const QString & gravename)

{

	QPixmap pix(datadir->absolutePath() + "/Graphics/Graves/" + gravename + ".png");

	GravePreview->setPixmap(pix.copy(0, 0, 32, 32));

}

void PageEditTeam::CBFort_activated(const QString & fortname)

{

	QPixmap pix(datadir->absolutePath() + "/Forts/" + fortname + "L.png");

	FortPreview->setPixmap(pix);

}

void PageEditTeam::CBTeamLvl_activated(int id)

{

	QPixmap pix(QString(":/res/botlevels/%1.png").arg(id));

	LevelPict->setPixmap(pix);

}

PageMultiplayer::PageMultiplayer(QWidget* parent) : 

  AbstractPage(parent)

{

	QGridLayout * pageLayout = new QGridLayout(this);

	BtnBack = addButton(":/res/Exit.png", pageLayout, 2, 0, true);

	gameCFG = new GameCFGWidget(this);

	pageLayout->addWidget(gameCFG, 0, 0, 1, 2);

	pageLayout->setRowStretch(1, 1);

	teamsSelect = new TeamSelWidget(this);

	pageLayout->addWidget(teamsSelect, 0, 2, 2, 2);

	BtnStartMPGame = addButton(tr("Start"), pageLayout, 2, 3);

}

PageOptions::PageOptions(QWidget* parent) :

  AbstractPage(parent)

{

	QGridLayout * pageLayout = new QGridLayout(this);

	pageLayout->setColumnStretch(0, 100);

	pageLayout->setColumnStretch(1, 100);

	pageLayout->setColumnStretch(2, 100);

	pageLayout->setRowStretch(0, 0);

	pageLayout->setRowStretch(1, 0);

	pageLayout->setRowStretch(2, 0);

	pageLayout->setRowStretch(3, 100);

	pageLayout->setRowStretch(4, 0);

	QGroupBox * gbTwoBoxes = new QGroupBox(this);

	pageLayout->addWidget(gbTwoBoxes, 0, 0, 1, 3);

	QHBoxLayout * gbTBLayout = new QHBoxLayout(gbTwoBoxes);

	{

		teamsBox = new IconedGroupBox(this);

		teamsBox->setIcon(QIcon(":/res/teamicon.png"));

		teamsBox->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

		teamsBox->setTitle(QGroupBox::tr("Teams"));

		QGridLayout * GBTlayout = new QGridLayout(teamsBox);

		BtnNewTeam = addButton(tr("New team"), GBTlayout, 1, 0);

		CBTeamName = new QComboBox(teamsBox);

		GBTlayout->addWidget(CBTeamName, 0, 0, 1, 2);

		BtnEditTeam = addButton(tr("Edit team"), GBTlayout, 1, 1);

		labelNN = new QLabel(teamsBox);

		labelNN->setText(QLabel::tr("Net nick"));

		GBTlayout->addWidget(labelNN, 2, 0, 1, 2);

		editNetNick = new QLineEdit(teamsBox);

		editNetNick->setMaxLength(20);

		editNetNick->setText(QLineEdit::tr("unnamed"));

		GBTlayout->addWidget(editNetNick, 3, 0, 1, 2);

		gbTBLayout->addWidget(teamsBox);

	}

	{

		IconedGroupBox* groupWeapons = new IconedGroupBox(this);

		groupWeapons->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

		groupWeapons->setIcon(QIcon(":/res/weaponsicon.png"));

		//groupWeapons->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

		groupWeapons->setTitle(QGroupBox::tr("Weapons"));

		QGridLayout * WeaponsLayout = new QGridLayout(groupWeapons);

		WeaponsButt = addButton(tr("Weapons set"), WeaponsLayout, 1, 0);

		WeaponsName = new QComboBox(this);

		WeaponsLayout->addWidget(WeaponsName, 0, 0, 1, 2);

		WeaponEdit = addButton(tr("Edit"), WeaponsLayout, 1, 1);

		gbTBLayout->addWidget(groupWeapons);

	}

	AGGroupBox = new IconedGroupBox(this);

	AGGroupBox->setIcon(QIcon(":/res/graphicsicon.png"));

	AGGroupBox->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

	AGGroupBox->setTitle(QGroupBox::tr("Audio/Graphic options"));

	pageLayout->addWidget(AGGroupBox, 2, 1);

	QVBoxLayout * GBAlayout = new QVBoxLayout(AGGroupBox);

	QHBoxLayout * GBAreslayout = new QHBoxLayout(0);

	QLabel * resolution = new QLabel(AGGroupBox);

	resolution->setText(QLabel::tr("Resolution"));

	GBAreslayout->addWidget(resolution);

	CBResolution = new QComboBox(AGGroupBox);

	GBAreslayout->addWidget(CBResolution);

	GBAlayout->addLayout(GBAreslayout);

	QHBoxLayout * GBAfpslayout = new QHBoxLayout(0);

	QLabel * maxfps = new QLabel(AGGroupBox);

	maxfps->setText(QLabel::tr("FPS limit"));

	GBAfpslayout->addWidget(maxfps);

	GBAlayout->addLayout(GBAfpslayout);

	CBFullscreen = new QCheckBox(AGGroupBox);

	CBFullscreen->setText(QCheckBox::tr("Fullscreen"));

	GBAlayout->addWidget(CBFullscreen);

	CBFrontendFullscreen = new QCheckBox(AGGroupBox);

	CBFrontendFullscreen->setText(QCheckBox::tr("Frontend fullscreen"));

	GBAlayout->addWidget(CBFrontendFullscreen);

	CBEnableSound = new QCheckBox(AGGroupBox);

	CBEnableSound->setText(QCheckBox::tr("Enable sound"));

	GBAlayout->addWidget(CBEnableSound);

	CBEnableMusic = new QCheckBox(AGGroupBox);

	CBEnableMusic->setText(QCheckBox::tr("Enable music"));

	GBAlayout->addWidget(CBEnableMusic);

	CBShowFPS = new QCheckBox(AGGroupBox);

	CBShowFPS->setText(QCheckBox::tr("Show FPS"));

	GBAlayout->addWidget(CBShowFPS);

	CBAltDamage = new QCheckBox(AGGroupBox);

	CBAltDamage->setText(QCheckBox::tr("Alternative damage show"));

	GBAlayout->addWidget(CBAltDamage);

	fpsedit = new FPSEdit(AGGroupBox);

	GBAfpslayout->addWidget(fpsedit);

	BtnSaveOptions = addButton(":/res/Save.png", pageLayout, 4, 2, true);

	BtnBack = addButton(":/res/Exit.png", pageLayout, 4, 0, true);

}

PageNet::PageNet(QWidget* parent) : AbstractPage(parent)

{

	QFont * font14 = new QFont("MS Shell Dlg", 14);

	QGridLayout * pageLayout = new QGridLayout(this);

	pageLayout->setColumnStretch(0, 1);

	pageLayout->setColumnStretch(1, 1);

	pageLayout->setColumnStretch(2, 1);

	BtnNetSvrStart = new QPushButton(this);

	BtnNetSvrStart->setFont(*font14);

	BtnNetSvrStart->setText(QPushButton::tr("Start server"));

	pageLayout->addWidget(BtnNetSvrStart, 3, 2);

	QGroupBox * NetTypeGroupBox = new QGroupBox(this);

	NetTypeGroupBox->setSizePolicy(QSizePolicy::MinimumExpanding, QSizePolicy::Fixed);

	NetTypeGroupBox->setTitle(QGroupBox::tr("Servers list"));

	pageLayout->addWidget(NetTypeGroupBox, 0, 1);

	QVBoxLayout * GBTlayout = new QVBoxLayout(NetTypeGroupBox);

	rbLocalGame = new QRadioButton(NetTypeGroupBox);

	rbLocalGame->setText(tr("Local"));

	rbLocalGame->setChecked(true);

	GBTlayout->addWidget(rbLocalGame);

	rbInternetGame = new QRadioButton(NetTypeGroupBox);

	rbInternetGame->setText(tr("Internet"));

	GBTlayout->addWidget(rbInternetGame);

	ConnGroupBox = new QGroupBox(this);

	ConnGroupBox->setTitle(QGroupBox::tr("Net game"));

	pageLayout->addWidget(ConnGroupBox, 2, 0, 1, 3);

	GBClayout = new QGridLayout(ConnGroupBox);

	GBClayout->setColumnStretch(0, 1);

	GBClayout->setColumnStretch(1, 1);

	GBClayout->setColumnStretch(2, 1);

	BtnNetConnect = new QPushButton(ConnGroupBox);

	BtnNetConnect->setFont(*font14);

	BtnNetConnect->setText(QPushButton::tr("Connect"));

	GBClayout->addWidget(BtnNetConnect, 2, 2);

	tvServersList = new QTableView(ConnGroupBox);

	tvServersList->setSelectionBehavior(QAbstractItemView::SelectRows);

	GBClayout->addWidget(tvServersList, 1, 0, 1, 3);

	BtnUpdateSList = new QPushButton(ConnGroupBox);

	BtnUpdateSList->setFont(*font14);

	BtnUpdateSList->setText(QPushButton::tr("Update"));

	GBClayout->addWidget(BtnUpdateSList, 2, 0);

	BtnSpecifyServer = new QPushButton(ConnGroupBox);

	BtnSpecifyServer->setFont(*font14);

	BtnSpecifyServer->setText(QPushButton::tr("Specify"));

	GBClayout->addWidget(BtnSpecifyServer, 2, 1);

	BtnBack = addButton(":/res/Exit.png", pageLayout, 3, 0, true);

	connect(rbLocalGame, SIGNAL(toggled(bool)), this, SLOT(updateServersList()));

	connect(BtnNetConnect, SIGNAL(clicked()), this, SLOT(slotConnect()));

}

void PageNet::updateServersList()

{

	if (rbLocalGame->isChecked())

		tvServersList->setModel(new HWNetUdpModel(tvServersList));

	else

		tvServersList->setModel(new HWNetWwwModel(tvServersList));

	tvServersList->horizontalHeader()->setResizeMode(0, QHeaderView::Stretch);

	static_cast<HWNetServersModel *>(tvServersList->model())->updateList();

	connect(BtnUpdateSList, SIGNAL(clicked()), static_cast<HWNetServersModel *>(tvServersList->model()), SLOT(updateList()));

	connect(tvServersList, SIGNAL(doubleClicked(const QModelIndex &)), this, SLOT(slotConnect()));

}

void PageNet::slotConnect()

{

	HWNetServersModel * model = static_cast<HWNetServersModel *>(tvServersList->model());

	QModelIndex mi = tvServersList->currentIndex();

	if(!mi.isValid())

	{

		QMessageBox::information(this, tr("Error"), tr("Please, select server from the list above"));