///////////////////////////////////////////////////////////////////////////////
// MuldeR's Utilities for Qt
// Copyright (C) 2004-2017 LoRd_MuldeR <MuldeR2@GMX.de>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library 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
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//
// http://www.gnu.org/licenses/lgpl-2.1.txt
//////////////////////////////////////////////////////////////////////////////////

#pragma once

//MUtils
#include <MUtils/Global.h>

//Qt
#include <QThread>
#include <QDate>

///////////////////////////////////////////////////////////////////////////////

namespace MUtils
{
	class MUTILS_API UpdateCheckerInfo
	{
		friend class UpdateChecker;

	public:
		UpdateCheckerInfo(void);
		void resetInfo(void);
		bool isComplete(void);

		const quint32 &getBuildNo(void)          const { return m_buildNo;          }
		const QDate   &getBuildDate(void)        const { return m_buildDate;        }
		const QString &getDownloadSite(void)     const { return m_downloadSite;     }
		const QString &getDownloadAddress(void)  const { return m_downloadAddress;  }
		const QString &getDownloadFilename(void) const { return m_downloadFilename; }
		const QString &getDownloadFilecode(void) const { return m_downloadFilecode; }
		const QString &getDownloadChecksum(void) const { return m_downloadChecksum; }

	private:
		quint32 m_buildNo;
		QDate m_buildDate;
		QString m_downloadSite;
		QString m_downloadAddress;
		QString m_downloadFilename;
		QString m_downloadFilecode;
		QString m_downloadChecksum;
	};

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

	class MUTILS_API UpdateChecker : public QThread
	{
		Q_OBJECT

	public:
		enum
		{
			UpdateStatus_NotStartedYet             = 0,
			UpdateStatus_CheckingConnection        = 1,
			UpdateStatus_FetchingUpdates           = 2,
			UpdateStatus_CompletedUpdateAvailable  = 3,
			UpdateStatus_CompletedNoUpdates        = 4,
			UpdateStatus_CompletedNewVersionOlder  = 5,
			UpdateStatus_ErrorNoConnection         = 6,
			UpdateStatus_ErrorConnectionTestFailed = 7,
			UpdateStatus_ErrorFetchUpdateInfo      = 8,
			UpdateStatus_CancelledByUser           = 9
		}
		update_status_t;

		UpdateChecker(const QString &binWGet, const QString &binMCat, const QString &binGnuPG, const QString &binKeys, const QString &applicationId, const quint32 &installedBuildNo, const bool betaUpdates, const bool testMode = false);
		~UpdateChecker(void);

		const int  getUpdateStatus(void)             const { return m_status; }
		const bool getSuccess(void)                  const { return m_success; };
		const int  getMaximumProgress(void)          const { return m_maxProgress; };
		const int  getCurrentProgress(void)          const { return m_progress; };
		const UpdateCheckerInfo *getUpdateInfo(void) const { return m_updateInfo.data(); }

		bool cancel(void) { return m_cancelled.ref(); }

	public slots:
		void start(Priority = InheritPriority);

	protected:
		void run(void);
		void checkForUpdates(void);
		void testKnownHosts(void);

	signals:
		void statusChanged(const int status);
		void progressChanged(const int progress);
		void messageLogged(const QString &text);

	private:
		const int m_maxProgress;
		QScopedPointer<UpdateCheckerInfo> m_updateInfo;
	
		const bool m_betaUpdates;
		const bool m_testMode;

		const QString m_applicationId;
		const quint32 m_installedBuildNo;

		const QString m_binaryWGet;
		const QString m_binaryMCat;
		const QString m_binaryGnuPG;
		const QString m_binaryKeys;

		QAtomicInt m_success;
		QAtomicInt m_cancelled;

		int m_status;
		int m_progress;

		inline void setStatus(const int status);
		inline void setProgress(const int progress);
		inline void log(const QString &str1, const QString &str2 = QString(), const QString &str3 = QString(), const QString &str4 = QString());

		bool getUpdateInfo(const QString &url, const QString &outFileVers, const QString &outFileSign);
		bool tryContactHost(const QString &hostname, const int &timeoutMsec, quint32 *const ipAddrOut = NULL);
		bool parseVersionInfo(const QString &file, UpdateCheckerInfo *updateInfo);

		bool getFile(const QString &url, const QString &outFile, const unsigned int maxRedir = 5U);
		bool getFile(const QString &url, const bool forceIp4, const QString &outFile, const unsigned int maxRedir = 5U);
		bool checkSignature(const QString &file, const QString &signature);
		bool tryUpdateMirror(UpdateCheckerInfo *updateInfo, const QString &url, const bool &quick);
	};
}