#include "./toonzqt/updatechecker.h"

#include <assert.h></assert.h>

#include <qnetworkreply></qnetworkreply>

#include <qeventloop></qeventloop>

//=============================================================================

// UpdateChecker

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

UpdateChecker::UpdateChecker(const QString &requestToServer)

#if QT_VERSION >= 0x050000

	:

#else

	: QHttp(),

#endif

	  m_webPageUrl()

{

	int i = 0;

	QUrl url(requestToServer);

#if QT_VERSION >= 0x050000

	QString urlTemp = requestToServer;

	QStringList urlList = urlTemp.split('?');

	if (urlList.count() <= 1) {

		abort();

		return;

	}

	QStringList paramList = urlList.at(1).split('&');

	//if (!url.userName().isEmpty())

	//	setUser(url.userName(), url.password());

	m_httpRequestAborted = false;

	QString param;

	param = QString("?");

	for (i = 0; i < paramList.size(); i++) {

		param += paramList.at(i);

		if (i < paramList.size() - 1)

			param += QString("&");

	}

	QNetworkAccessManager manager;

	QNetworkReply *reply = manager.get(QNetworkRequest(url.path() + param));

	QEventLoop loop;

	connect(reply, SIGNAL(requestStarted(int)), &loop, SLOT(httpRequestStarted(int)));

	connect(reply, SIGNAL(authenticationRequired(const QString &, quint16, QAuthenticator *)), &loop, SLOT(slotAuthenticationRequired(const QString &, quint16, QAuthenticator *)));

	connect(reply, SIGNAL(stateChanged(int)), &loop, SLOT(httpStateChanged(int)));

	connect(&manager, SIGNAL(finished(QNetworkReply *)), this, SLOT(httpRequestFinished(QNetworkReply *)));

	loop.exec();

#else

	//connect(this, SIGNAL(readyRead(const QHttpResponseHeader &)), this,

	//	SLOT(readyReadexec(const QHttpResponseHeader &)));

	connect(this, SIGNAL(requestFinished(int, bool)), this,

			SLOT(httpRequestFinished(int, bool)));

	connect(this, SIGNAL(requestStarted(int)), this,

			SLOT(httpRequestStarted(int)));

	connect(this, SIGNAL(responseHeaderReceived(const QHttpResponseHeader &)), this,

			SLOT(readResponseHeader(const QHttpResponseHeader &)));

	connect(this, SIGNAL(authenticationRequired(const QString &, quint16, QAuthenticator *)), this,

			SLOT(slotAuthenticationRequired(const QString &, quint16, QAuthenticator *)));

	connect(this, SIGNAL(stateChanged(int)), this,

			SLOT(httpStateChanged(int)));

	QHttp::ConnectionMode mode = url.scheme().toLower() == "https" ? QHttp::ConnectionModeHttps : QHttp::ConnectionModeHttp;

	setHost(url.host(), mode, url.port() == -1 ? 0 : url.port());

	QString urlTemp = requestToServer;

	QStringList urlList = urlTemp.split('?');

	if (urlList.count() <= 1) {

		abort();

		return;

	}

	QStringList paramList = urlList.at(1).split('&');

	if (!url.userName().isEmpty())

		setUser(url.userName(), url.password());

	m_httpRequestAborted = false;

	QString param;

	param = QString("?");

	for (i = 0; i < paramList.size(); i++) {

		param += paramList.at(i);

		if (i < paramList.size() - 1)

			param += QString("&");

	}

	/*param = QString("?")+paramList.at(0)+QString("&")

		+paramList.at(1) + QString("&") + paramList.at(2);*/

	m_httpGetId = get(url.path() + param); //, file);

#endif

}

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

#if QT_VERSION < 0x050000

void UpdateChecker::readResponseHeader(const QHttpResponseHeader &responseHeader)

{

	int err = responseHeader.statusCode();

	if (err != 200 && err != 502) {

		m_httpRequestAborted = true;

		abort();

	}

}

#endif

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

void UpdateChecker::httpStateChanged(int status)

{

	std::string stateStr;

	switch (status) {

	case 1:

		stateStr = "A host name lookup is in progress...";

		break;

	case 2:

		stateStr = "Connecting...";

		break;

	case 3:

		stateStr = "Sending informations...";

		break;

	case 4:

		stateStr = "Reading informations...";

		break;

	case 5:

		stateStr = "Connected.";

		break;

	case 6:

		stateStr = "The connection is closing down, but is not yet closed. (The state will be Unconnected when the connection is closed.)";

		break;

	default:

		stateStr = "There is no connection to the host.";

	}

#if QT_VERSION < 0x050000

	status = state();

	qDebug("Status: %d : %s", status, stateStr.c_str());

#endif

}

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

#if QT_VERSION >= 0x050000

void UpdateChecker::httpRequestFinished(QNetworkReply *reply)

#else

void UpdateChecker::httpRequestFinished(int requestId, bool error)

#endif

{

	QByteArray arr;

	std::string webPageString;

	std::string dateString;

	QString qstr;

#if QT_VERSION >= 0x050000

	if (reply->error() != QNetworkReply::NoError)

		return;

	arr = reply->readAll();

#else

	if (requestId != m_httpGetId) {

		return;

	}

	if (error || m_httpRequestAborted) {

		abort();

		return;

	}

	arr = readAll();

#endif

	qstr = QString(arr);

	int startIndex = qstr.indexOf("Startdate");

	int endIndex = qstr.indexOf("Enddate");

	if ((endIndex - startIndex - 9) <= 0) {

		abort();

		return;

	}

	dateString = qstr.toStdString();

	dateString = dateString.substr(startIndex + 9, endIndex - startIndex - 9);

	QString qDateString = QString::fromStdString(dateString);

	qDateString.remove("\"");

	// Make sure that the format is exactly MM/dd/yyyy

	if (qDateString.size() != 10) {

		QStringList fields = qDateString.split("/");

		if (fields.size() == 3) {

			QString month = fields.at(0);

			QString day = fields.at(1);

			QString year = fields.at(2);

			if (day.size() == 1)

				day.prepend("0");

			if (month.size() == 1)

				month.prepend("0");

			qDateString = month + "/" + day + "/" + year;

		}

	}

	m_updateDate = QDate::fromString(qDateString, "MM/dd/yyyy");

	startIndex = qstr.indexOf("Starturl");

	endIndex = qstr.indexOf("Endurl");

	webPageString = qstr.toStdString();

	webPageString = webPageString.substr(startIndex + 9, endIndex - startIndex - 9);

	QString qWebPageString = QString::fromStdString(webPageString);

	qWebPageString.remove("\"");

	QUrl webPageUrl(qWebPageString);

	if (webPageUrl.isValid())

		m_webPageUrl = webPageUrl;

}

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

void UpdateChecker::slotAuthenticationRequired(const QString &hostName, quint16, QAuthenticator *authenticator)

{

	assert(false);

}