/*

 * Hedgewars, a worms-like game

 * Copyright (c) 2005 Andrey Korotaev <unC0Rr@gmail.com>

 *

 * Distributed under the terms of the BSD-modified licence:

 *

 * Permission is hereby granted, free of charge, to any person obtaining a copy

 * of this software and associated documentation files (the "Software"), to deal

 * with the Software without restriction, including without limitation the

 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or

 * sell copies of the Software, and to permit persons to whom the Software is

 * furnished to do so, subject to the following conditions:

 *

 * 1. Redistributions of source code must retain the above copyright notice,

 *    this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright notice,

 *    this list of conditions and the following disclaimer in the documentation

 *    and/or other materials provided with the distribution.

 * 3. The name of the author may not be used to endorse or promote products

 *    derived from this software without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED

 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF

 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO

 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;

 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,

 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR

 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF

 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

#include <QMessageBox>

#include <QProcess>

#include <QTimer>

#include <QString>

#include <QByteArray>

#include <QTextStream>

#include <QFile>

#include "game.h"

#include "hwconsts.h"

HWGame::HWGame()

{

	IPCServer = new QTcpServer(this);

	IPCServer->setMaxPendingConnections(1);

	if (!IPCServer->listen(QHostAddress("127.0.0.1"), IPC_PORT))

	{

		QMessageBox::critical(this, tr("Error"),

				tr("Unable to start the server: %1.")

				.arg(IPCServer->errorString()));

	}

	connect(IPCServer, SIGNAL(newConnection()), this, SLOT(NewConnection()));

	IPCSocket = 0;

	TeamCount = 0;

	seed = "seed";

}

void HWGame::NewConnection()

{

	QTcpSocket * client = IPCServer->nextPendingConnection();

	if(!IPCSocket)

	{

		IPCSocket = client;

		connect(client, SIGNAL(disconnected()), this, SLOT(ClientDisconnect()));

		connect(client, SIGNAL(readyRead()), this, SLOT(ClientRead()));

		msgsize = 0;

	} else

	{

		client->disconnectFromHost();

		delete client;

	}

}

void HWGame::ClientDisconnect()

{

	IPCSocket = 0;

	SaveDemo("demo.hwd_1");

	delete this;

}

void HWGame::SendTeamConfig(int index)

{

	QFile teamcfg(teams[index]);

	if (!teamcfg.open(QIODevice::ReadOnly))

	{

		return ;

	}

	QTextStream stream(&teamcfg);

	stream.setCodec("UTF-8");

	QString str;

	while (!stream.atEnd())

	{

		str = stream.readLine();

		if (str.startsWith(";") || (str.length() > 254)) continue;

		str.prepend("e");

		SendIPC(str.toLocal8Bit());

	}

	teamcfg.close();

}

void HWGame::SendConfig()

{

	SENDIPC("TL");

	SENDIPC("e$gmflags 0");

	SENDIPC("eaddteam");

	SendTeamConfig(0);

	SENDIPC("ecolor 65535");

	SENDIPC("eadd hh0 0");

	SENDIPC("eadd hh1 0");

	SENDIPC("eadd hh2 0");

	SENDIPC("eadd hh3 0");

	SENDIPC("eaddteam");

	SendTeamConfig(1);

	SENDIPC("ecolor 16776960");

	SENDIPC("eadd hh0 1");

	SENDIPC("eadd hh1 1");

	SENDIPC("eadd hh2 1");

	SENDIPC("eadd hh3 1");

}

void HWGame::ParseMessage()

{

	switch(msgbuf[0])

	{

		case '?':

		{

			SENDIPC("!");

			break;

		}

		case 'C':

		{

			SendConfig();

			break;

		}

		case '+':

		{

			break;

		}

		default:

		{

			demo->append(msgsize);

			demo->append(QByteArray::fromRawData(msgbuf, msgsize));

		}

	}

}

void HWGame::SendIPC(const char * msg, quint8 len)

{

	IPCSocket->write((char *)&len, 1);

	IPCSocket->write(msg, len);

	if ((len > 5) && !((msg[0] == 'e') && (msg[1] == 'b')))

	{

		demo->append(len);

		demo->append(QByteArray::fromRawData(msg, len));

	}

}

void HWGame::SendIPC(const QByteArray & buf)

{

	if (buf.size() > 255) return;

	quint8 len = buf.size();

	IPCSocket->write((char *)&len, 1);

	IPCSocket->write(buf);

	demo->append(len);

	demo->append(buf);

}

void HWGame::ClientRead()

{

	qint64 readbytes = 1;

	while (readbytes > 0)

	{

		if (msgsize == 0)

		{

			msgbufsize = 0;

			readbytes = IPCSocket->read((char *)&msgsize, 1);

		} else

		{

			msgbufsize +=

			readbytes = IPCSocket->read((char *)&msgbuf[msgbufsize], msgsize - msgbufsize);

			if (msgbufsize = msgsize)

			{

				ParseMessage();

				msgsize = 0;

			}

		}

	}

}

void HWGame::Start(int Resolution, bool Fullscreen)

{

	if (TeamCount < 2) return;

	QProcess * process;

	QStringList arguments;

	seedgen.GenRNDStr(seed, 10);

	process = new QProcess;

	arguments << resolutions[0][Resolution];

	arguments << resolutions[1][Resolution];

	arguments << GetThemeBySeed();

	arguments << "46631";

	arguments << seed;

	arguments << (Fullscreen ? "1" : "0");

	process->start("hw", arguments);

	demo = new QByteArray;

	demo->append(seed.toLocal8Bit());

	demo->append(10);

}

void HWGame::AddTeam(const QString & teamname)

{

	if (TeamCount == 5) return;

	teams[TeamCount] = teamname;

	TeamCount++;

}

QString HWGame::GetThemeBySeed()

{

	QFile themesfile("Data/Themes/themes.cfg");

	QStringList themes;

	if (themesfile.open(QIODevice::ReadOnly))

	{

		QTextStream stream(&themesfile);

		QString str;

		while (!stream.atEnd())

		{

			themes << stream.readLine();

		}

		themesfile.close();

	}

	quint32 len = themes.size();

	if (len == 0)

	{

		QMessageBox::critical(this, "Error", "Cannot access themes.cfg or bad data", "OK");

		return "avematan";

	}

	if (seed.isEmpty())

	{

		QMessageBox::critical(this, "Error", "seed not defined", "OK");

		return "avematan";

	}

	quint32 k = 0;

	for (int i = 0; i < seed.length(); i++)

	{

		k += seed[i].unicode();

	}

	return themes[k % len];

}

void HWGame::SaveDemo(const QString & filename)

{

	QFile demofile(filename);

	if (!demofile.open(QIODevice::WriteOnly))

	{

		QMessageBox::critical(this,

				tr("Error"),

				tr("Cannot save demo to file %s").arg(filename),

				tr("Quit"));

		return ;

	}

	QDataStream stream(&demofile);

	stream.writeRawData(demo->constData(), demo->size());

	demofile.close();

}