#include "tfarmserver.h"

#include "tfarmexecutor.h"

#include "tsystem.h"

#include "tconvert.h"

#include <vector></vector>

using namespace std;

//------------------------------------------------------------------------------

class TFarmServerStub::Imp : public TFarmExecutor

{

public:

	Imp(TFarmServer *server, int port)

		: TFarmExecutor(port), m_server(server)

	{

		assert(m_server);

	}

	// TFarmExecutor overrides

	QString execute(const std::vector<qstring> &argv);</qstring>

	TFarmServer *m_server;

};

//------------------------------------------------------------------------------

QString TFarmServerStub::Imp::execute(const std::vector<qstring> &argv)</qstring>

{

#ifdef DEBUG

	std::cout << "executing " << argv[0].c_str() << endl;

#endif

	if (argv.size() > 0) {

#ifdef DEBUG

		for (int i = 0; i < (int)argv.size(); ++i)

			std::cout << argv[i] << " ";

#endif

		if (argv[0] == "addTask" && argv.size() == 3) {

			int ret = m_server->addTask(argv[1], argv[2]);

			return QString::number(ret);

		} else if (argv[0] == "terminateTask" && argv.size() > 1) {

			int ret = m_server->terminateTask(argv[1]);

			return QString::number(ret);

		} else if (argv[0] == "getTasks") {

			std::vector<qstring> tasks;</qstring>

			int ret = m_server->getTasks(tasks);

			QString reply(QString::number(ret));

			reply += ",";

			std::vector<qstring>::iterator it = tasks.begin();</qstring>

			for (; it != tasks.end(); ++it) {

				reply += *it;

				reply += ",";

			}

			if (!reply.isEmpty())

				reply.remove(reply.length() - 1, 1);

			return reply;

		} else if (argv[0] == "queryHwInfo") {

			TFarmServer::HwInfo hwInfo;

			m_server->queryHwInfo(hwInfo);

			QString ret;

			ret += QString::number((int)hwInfo.m_cpuCount);

			ret += ",";

			ret += QString::number((int)(hwInfo.m_totPhysMem / 1024));

			ret += ",";

			ret += QString::number((int)(hwInfo.m_availPhysMem / 1024));

			ret += ",";

			ret += QString::number((int)(hwInfo.m_totVirtMem / 1024));

			ret += ",";

			ret += QString::number((int)(hwInfo.m_availVirtMem / 1024));

			ret += ",";

			ret += QString::number(hwInfo.m_type);

			return ret;

		} else if (argv[0] == "attachController" && argv.size() > 3) {

			int port = argv[3].toInt();

			m_server->attachController(argv[1], argv[2], port);

			return "";

		} else if (argv[0] == "detachController" && argv.size() > 3) {

			int port = argv[3].toInt();

			m_server->detachController(argv[1], argv[2], port);

			return "";

		}

	}

	return "";

}

//------------------------------------------------------------------------------

//------------------------------------------------------------------------------

TFarmServerStub::TFarmServerStub(TFarmServer *farmServer, int port)

	: m_imp(new Imp(farmServer, port))

{

}

//------------------------------------------------------------------------------

TFarmServerStub::~TFarmServerStub()

{

}

//------------------------------------------------------------------------------

int TFarmServerStub::run()

{

	m_imp->run();

	return m_imp->getExitCode();

}

//------------------------------------------------------------------------------

int TFarmServerStub::shutdown()

{

	TTcpIpClient client;

	int socketId;

	int ret = client.connect(TSystem::getHostName(), "", m_imp->getPort(), socketId);

	if (ret == OK) {

		ret = client.send(socketId, "shutdown");

	}

	return ret;

}

//------------------------------------------------------------------------------

int TFarmServerStub::getPort() const

{

	return m_imp->getPort();

}