/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2005-2007, 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 <QApplication>

#include <QStringList>

#include <QLineEdit>

#include "team.h"

#include "hwform.h"

#include "predefteams.h"

#include "pages.h"

#include "hwconsts.h"

#include "hats.h"

HWTeam::HWTeam(const QString & teamname) :

	difficulty(0),

	numHedgehogs(4),

	m_isNetTeam(false)

{

	TypesLoad();

	TeamName = teamname;

	OldTeamName = TeamName;

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

	{

		HHName[i].sprintf("hedgehog %d", i);

		HHHat[i] = "NoHat";

	}

	Grave = "Statue";

	Fort = "Plane";

	Voicepack = "Default";

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

	{

		binds[i].action = cbinds[i].action;

		binds[i].strbind = cbinds[i].strbind;

	}

}

HWTeam::HWTeam(const QStringList& strLst) :

  numHedgehogs(4),

  m_isNetTeam(true)

{

	TypesLoad();

	// net teams are configured from QStringList

	if(strLst.size() != 22) throw HWTeamConstructException();

	TeamName = strLst[0];

	Grave = strLst[1];

	Fort = strLst[2];

	Voicepack = strLst[3];

	Owner = strLst[4];

	difficulty = strLst[5].toUInt();

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

	{

		HHName[i]=strLst[i * 2 + 6];

		HHHat[i]=strLst[i * 2 + 7];

	}

}

HWTeam::HWTeam(quint8 num) :

  difficulty(0),

  numHedgehogs(4),

  m_isNetTeam(false)

{

	TypesLoad();

	num %= PREDEFTEAMS_COUNT;

/*

	//This is now generated by TeamRandomNames(TRUE);

	TeamName = QApplication::translate("teams", pteams[num].TeamName);

	HHName[0] = QApplication::translate("teams", pteams[num].hh0name);

	HHName[1] = QApplication::translate("teams", pteams[num].hh1name);

	HHName[2] = QApplication::translate("teams", pteams[num].hh2name);

	HHName[3] = QApplication::translate("teams", pteams[num].hh3name);

	HHName[4] = QApplication::translate("teams", pteams[num].hh4name);

	HHName[5] = QApplication::translate("teams", pteams[num].hh5name);

	HHName[6] = QApplication::translate("teams", pteams[num].hh6name);

	HHName[7] = QApplication::translate("teams", pteams[num].hh7name);

	HHHat[0] = pteams[num].hh0hat;

	HHHat[1] = pteams[num].hh1hat;

	HHHat[2] = pteams[num].hh2hat;

	HHHat[3] = pteams[num].hh3hat;

	HHHat[4] = pteams[num].hh4hat;

	HHHat[5] = pteams[num].hh5hat;

	HHHat[6] = pteams[num].hh6hat;

	HHHat[7] = pteams[num].hh7hat;

*/	

	TeamName = QString("Team");

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

	{

		HHName[i].sprintf("hedgehog %d", i);

		HHHat[i] = "NoHat";

	}

	Grave = pteams[num].Grave;

	Fort = pteams[num].Fort;

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

	{

		binds[i].action = cbinds[i].action;

		binds[i].strbind = cbinds[i].strbind;

	}

	TeamRandomNames(TRUE);

}

bool HWTeam::LoadFromFile()

{

	numHedgehogs=4;

	QFile cfgfile(cfgdir->absolutePath() + "/" + TeamName + ".cfg");

	if (!cfgfile.open(QIODevice::ReadOnly)) return false;

	QTextStream stream(&cfgfile);

	stream.setCodec("UTF-8");

	QString str;

	QString action;

	while (!stream.atEnd())

	{

		str = stream.readLine();

		if (str.startsWith(";")) continue;

		/*if (str.startsWith("name team "))

		{

			str.remove(0, 10);

			TeamName = str;

		} else*/

		if (str.startsWith("name hh"))

		{

			str.remove(0, 7);

			long i = str.left(1).toLong();

			if ((i < 0) || (i > 7)) continue;

			str.remove(0, 2);

			HHName[i] = str;

		} else

		if (str.startsWith("hat"))

		{

			str.remove(0, 3);

			long i = str.left(1).toLong();

			if ((i < 0) || (i > 7)) continue;

			str.remove(0, 2);

			HHHat[i] = str;

		} else

		if (str.startsWith("grave "))

		{

			str.remove(0, 6);

			Grave = str;

		} else

		if (str.startsWith("fort "))

		{

			str.remove(0, 5);

			Fort = str;

		} else

		if (str.startsWith("voicepack "))

		{

			str.remove(0, 10);

			Voicepack = str;

		} else

		if (str.startsWith("bind "))

		{

			str.remove(0, 5);

			action = str.section(' ', 1);

			str = str.section(' ', 0, 0);

			str.truncate(15);

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

				if (action == binds[i].action)

				{

					binds[i].strbind = str;

					break;

				}

		} else

		if (str.startsWith("difficulty "))

		{

		  str.remove(0, 11);

		  difficulty=str.toUInt();

		  if (difficulty>5) difficulty=0; // this shouldn't normally happen

		}

	}

	cfgfile.close();

	return true;

}

bool HWTeam::SaveToFile()

{

	if (OldTeamName != TeamName)

	{

		QFile cfgfile(cfgdir->absolutePath() + "/" + OldTeamName + ".cfg");

		cfgfile.remove();

		OldTeamName = TeamName;

	}

	QFile cfgfile(cfgdir->absolutePath() + "/" + TeamName + ".cfg");

	if (!cfgfile.open(QIODevice::WriteOnly)) return false;

	QTextStream stream(&cfgfile);

	stream.setCodec("UTF-8");

	stream << "; Generated by Hedgewars, do not modify" << endl;

	stream << "name team " << TeamName << endl;

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

	{

		stream << "name hh" << i << " " << HHName[i] << endl;

		stream << "hat" << i << " " << HHHat[i] << endl;

	}

	stream << "grave " << Grave << endl;

	stream << "fort " << Fort << endl;

	stream << "voicepack " << Voicepack << endl;

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

	{

		stream << "bind " << binds[i].strbind << " " << binds[i].action << endl;

	}

	stream << "difficulty " << difficulty << endl;

	cfgfile.close();

	return true;

}

void HWTeam::SetToPage(HWForm * hwform)

{

	hwform->ui.pageEditTeam->TeamNameEdit->setText(TeamName);

	hwform->ui.pageEditTeam->CBTeamLvl->setCurrentIndex(difficulty);

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

	{

		hwform->ui.pageEditTeam->HHNameEdit[i]->setText(HHName[i]);

		hwform->ui.pageEditTeam->HHHats[i]->setCurrentIndex(hwform->ui.pageEditTeam->HHHats[i]->findData(HHHat[i], Qt::DisplayRole));

	}

	hwform->ui.pageEditTeam->CBGrave->setCurrentIndex(hwform->ui.pageEditTeam->CBGrave->findText(Grave));

	hwform->ui.pageEditTeam->CBFort->setCurrentIndex(hwform->ui.pageEditTeam->CBFort->findText(Fort));

	hwform->ui.pageEditTeam->CBVoicepack->setCurrentIndex(hwform->ui.pageEditTeam->CBVoicepack->findText(Voicepack));

	//hwform->ui.pageEditTeam->CBFort_activated(Fort);

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

	{

		hwform->ui.pageEditTeam->CBBind[i]->setCurrentIndex(hwform->ui.pageEditTeam->CBBind[i]->findText(binds[i].strbind));

	}

}

void HWTeam::GetFromPage(HWForm * hwform)

{

	TeamName  = hwform->ui.pageEditTeam->TeamNameEdit->text();

	difficulty = hwform->ui.pageEditTeam->CBTeamLvl->currentIndex();

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

	{

		HHName[i] = hwform->ui.pageEditTeam->HHNameEdit[i]->text();

		HHHat[i] = hwform->ui.pageEditTeam->HHHats[i]->currentText();

	}

	Grave = hwform->ui.pageEditTeam->CBGrave->currentText();

	Fort = hwform->ui.pageEditTeam->CBFort->currentText();

	Voicepack = hwform->ui.pageEditTeam->CBVoicepack->currentText();

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

	{

		binds[i].strbind = hwform->ui.pageEditTeam->CBBind[i]->currentText();

	}

}

QStringList HWTeam::TeamGameConfig(quint32 InitHealth) const

{

	QStringList sl;

	sl.push_back(QString("eaddteam %1 %2").arg(teamColor.rgb() & 0xffffff).arg(TeamName));

	if (m_isNetTeam)

		sl.push_back("erdriven");

	sl.push_back(QString("egrave " + Grave));

	sl.push_back(QString("efort " + Fort));

	sl.push_back(QString("evoicepack " + Voicepack));

	if (!m_isNetTeam)

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

			sl.push_back(QString("ebind " + binds[i].strbind + " " + binds[i].action));

	for (int t = 0; t < numHedgehogs; t++)

	{

	  sl.push_back(QString("eaddhh %1 %2 %3")

		       .arg(QString::number(difficulty),

			    QString::number(InitHealth),

			    HHName[t]));

	  sl.push_back(QString("ehat %1")

		       .arg(HHHat[t]));

	}

	return sl;

}

void HWTeam::RandomNameByHat(const int &i)

{

	QStringList Dictionaries;

	HatCfgLoad(HHHat[i],Dictionaries);

/*

"Dismissed",

"Dragon",

"Mindblower",

*/

	QStringList Dictionary;

	DictLoad(Dictionaries[rand()%(Dictionaries.size())],Dictionary);

	HHName[i] = Dictionary[rand()%(Dictionary.size())];

}

void HWTeam::TeamRandomName(const int &i)

{

	RandomNameByHat(i);

}

void HWTeam::TeamRandomNames(bool changeteamname)

{

	if ((TypesHatnames.size() > 0) && TypesAvliable){

		int kind = (rand()%(TypesHatnames.size()));

		if (changeteamname){

			if (TypesTeamnames[kind].size() > 0){

				TeamName = TypesTeamnames[kind][rand()%(TypesTeamnames[kind].size())];

			}

		}

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

		{

			if ((TypesHatnames[kind].size()) > 0){

				HHHat[i] = TypesHatnames[kind][rand()%(TypesHatnames[kind].size())];

			}

			RandomNameByHat(i);

		}

	}

}

bool HWTeam::isNetTeam() const

{

  return m_isNetTeam;

}

bool HWTeam::operator==(const HWTeam& t1) const {

  return TeamName==t1.TeamName;

}

bool HWTeam::operator<(const HWTeam& t1) const {

  return TeamName<t1.TeamName; // if names are equal - test if it is net team

}

void HWTeam::TypesLoad()

{

     QFile file(QString("%1/Names/types.ini").arg(datadir->absolutePath()));

     if (!file.open(QIODevice::ReadOnly | QIODevice::Text))

         TypesAvliable = FALSE;

     int counter = 0; //counter starts with 0 (teamnames mode)

     TypesTeamnames.append(QStringList());

     TypesHatnames.append(QStringList());

     QTextStream in(&file);

     while (!in.atEnd()) {

         QString line = in.readLine();

         if (line == QString("#####")){

             counter++; //toggle mode (teamnames || hats)

             if ((counter%2) == 0){

                 TypesTeamnames.append(QStringList());

                 TypesHatnames.append(QStringList());

             }

         } else if ((line == QString("*****")) || (line == QString("*END*"))){

             TypesAvliable = TRUE; return; // bye bye

         } else {

             if ((counter%2) == 0){ // even => teamnames mode

                 TypesTeamnames[(counter/2)].append(line);

             } else { // odd => hats mode

                 TypesHatnames[((counter-1)/2)].append(line);

             }

         }

//         Types.append(line);

     }

         TypesAvliable = TRUE;

     return;

}

void HWTeam::DictLoad(const QString filename, QStringList &list)

{

     list.clear();

     QFile file(QString("%1/Names/%2.txt").arg(datadir->absolutePath()).arg(filename));

     if (file.open(QIODevice::ReadOnly | QIODevice::Text))

     {

         QTextStream in(&file);

         while (!in.atEnd()) {

             QString line = in.readLine();

             if(line != QString(""))

                 {list.append(line);}

         }

     }

     if (list.size()==0)

         list.append(filename);

}

void HWTeam::HatCfgLoad(const QString hatname, QStringList &list)

{

     list.clear();

     QFile file(QString("%1/Names/%2.cfg").arg(datadir->absolutePath()).arg(hatname));

     if (file.open(QIODevice::ReadOnly | QIODevice::Text))

     {

         QTextStream in(&file);

         while (!in.atEnd()) {

             QString line = in.readLine();

             if(line != QString(""))

                 {list.append(line);}

         }

     }

     if (list.size()==0)

         list.append(QString("generic"));

}