#include <QTest>
#include "TestUtil.h"

#include "minecraft/MojangVersionFormat.h"
#include "minecraft/OneSixVersionFormat.h"
#include "minecraft/Library.h"
#include "net/HttpMetaCache.h"
#include "FileSystem.h"

class LibraryTest : public QObject
{
	Q_OBJECT
private:
	LibraryPtr readMojangJson(const char *file)
	{
		auto path = QFINDTESTDATA(file);
		QFile jsonFile(path);
		jsonFile.open(QIODevice::ReadOnly);
		auto data = jsonFile.readAll();
		jsonFile.close();
		return MojangVersionFormat::libraryFromJson(QJsonDocument::fromJson(data).object(), file);
	}
	// get absolute path to expected storage, assuming default cache prefix
	QStringList getStorage(QString relative)
	{
		return {FS::PathCombine(cache->getBasePath("libraries"), relative)};
	}
private
slots:
	void initTestCase()
	{
		cache.reset(new HttpMetaCache());
		cache->addBase("libraries", QDir("libraries").absolutePath());
		dataDir = QDir("data").absolutePath();
	}
	void test_legacy()
	{
		Library test("test.package:testname:testversion");
		QCOMPARE(test.artifactPrefix(), QString("test.package:testname"));
		QCOMPARE(test.isNative(), false);

		QStringList jar, native, native32, native64;
		test.getApplicableFiles(currentSystem, jar, native, native32, native64, QString());
		QCOMPARE(jar, getStorage("test/package/testname/testversion/testname-testversion.jar"));
		QCOMPARE(native, {});
		QCOMPARE(native32, {});
		QCOMPARE(native64, {});
	}
	void test_legacy_url()
	{
		QStringList failedFiles;
		Library test("test.package:testname:testversion");
		test.setRepositoryURL("file://foo/bar");
		auto downloads = test.getDownloads(currentSystem, cache.get(), failedFiles, QString());
		QCOMPARE(downloads.size(), 1);
		QCOMPARE(failedFiles, {});
		NetActionPtr dl = downloads[0];
		QCOMPARE(dl->m_url, QUrl("file://foo/bar/test/package/testname/testversion/testname-testversion.jar"));
	}
	void test_legacy_url_local_broken()
	{
		Library test("test.package:testname:testversion");
		QCOMPARE(test.isNative(), false);
		QStringList failedFiles;
		test.setHint("local");
		auto downloads = test.getDownloads(currentSystem, cache.get(), failedFiles, QString());
		QCOMPARE(downloads.size(), 0);
		QCOMPARE(failedFiles, getStorage("test/package/testname/testversion/testname-testversion.jar"));
	}
	void test_legacy_url_local_override()
	{
		Library test("com.paulscode:codecwav:20101023");
		QCOMPARE(test.isNative(), false);
		QStringList failedFiles;
		test.setHint("local");
		auto downloads = test.getDownloads(currentSystem, cache.get(), failedFiles, QString("data"));
		QCOMPARE(downloads.size(), 0);
		qDebug() << failedFiles;
		QCOMPARE(failedFiles.size(), 0);

		QStringList jar, native, native32, native64;
		test.getApplicableFiles(currentSystem, jar, native, native32, native64, QString("data"));
		QCOMPARE(jar, {QFileInfo("data/codecwav-20101023.jar").absoluteFilePath()});
		QCOMPARE(native, {});
		QCOMPARE(native32, {});
		QCOMPARE(native64, {});
	}
	void test_legacy_native()
	{
		Library test("test.package:testname:testversion");
		test.m_nativeClassifiers[OpSys::Os_Linux]="linux";
		QCOMPARE(test.isNative(), true);
		test.setRepositoryURL("file://foo/bar");
		{
			QStringList jar, native, native32, native64;
			test.getApplicableFiles(Os_Linux, jar, native, native32, native64, QString());
			QCOMPARE(jar, {});
			QCOMPARE(native, getStorage("test/package/testname/testversion/testname-testversion-linux.jar"));
			QCOMPARE(native32, {});
			QCOMPARE(native64, {});
			QStringList failedFiles;
			auto dls = test.getDownloads(Os_Linux, cache.get(), failedFiles, QString());
			QCOMPARE(dls.size(), 1);
			QCOMPARE(failedFiles, {});
			auto dl = dls[0];
			QCOMPARE(dl->m_url, QUrl("file://foo/bar/test/package/testname/testversion/testname-testversion-linux.jar"));
		}
	}
	void test_legacy_native_arch()
	{
		Library test("test.package:testname:testversion");
		test.m_nativeClassifiers[OpSys::Os_Linux]="linux-${arch}";
		test.m_nativeClassifiers[OpSys::Os_OSX]="osx-${arch}";
		test.m_nativeClassifiers[OpSys::Os_Windows]="windows-${arch}";
		QCOMPARE(test.isNative(), true);
		test.setRepositoryURL("file://foo/bar");
		{
			QStringList jar, native, native32, native64;
			test.getApplicableFiles(Os_Linux, jar, native, native32, native64, QString());
			QCOMPARE(jar, {});
			QCOMPARE(native, {});
			QCOMPARE(native32, getStorage("test/package/testname/testversion/testname-testversion-linux-32.jar"));
			QCOMPARE(native64, getStorage("test/package/testname/testversion/testname-testversion-linux-64.jar"));
			QStringList failedFiles;
			auto dls = test.getDownloads(Os_Linux, cache.get(), failedFiles, QString());
			QCOMPARE(dls.size(), 2);
			QCOMPARE(failedFiles, {});
			QCOMPARE(dls[0]->m_url, QUrl("file://foo/bar/test/package/testname/testversion/testname-testversion-linux-32.jar"));
			QCOMPARE(dls[1]->m_url, QUrl("file://foo/bar/test/package/testname/testversion/testname-testversion-linux-64.jar"));
		}
		{
			QStringList jar, native, native32, native64;
			test.getApplicableFiles(Os_Windows, jar, native, native32, native64, QString());
			QCOMPARE(jar, {});
			QCOMPARE(native, {});
			QCOMPARE(native32, getStorage("test/package/testname/testversion/testname-testversion-windows-32.jar"));
			QCOMPARE(native64, getStorage("test/package/testname/testversion/testname-testversion-windows-64.jar"));
			QStringList failedFiles;
			auto dls = test.getDownloads(Os_Windows, cache.get(), failedFiles, QString());
			QCOMPARE(dls.size(), 2);
			QCOMPARE(failedFiles, {});
			QCOMPARE(dls[0]->m_url, QUrl("file://foo/bar/test/package/testname/testversion/testname-testversion-windows-32.jar"));
			QCOMPARE(dls[1]->m_url, QUrl("file://foo/bar/test/package/testname/testversion/testname-testversion-windows-64.jar"));
		}
		{
			QStringList jar, native, native32, native64;
			test.getApplicableFiles(Os_OSX, jar, native, native32, native64, QString());
			QCOMPARE(jar, {});
			QCOMPARE(native, {});
			QCOMPARE(native32, getStorage("test/package/testname/testversion/testname-testversion-osx-32.jar"));
			QCOMPARE(native64, getStorage("test/package/testname/testversion/testname-testversion-osx-64.jar"));
			QStringList failedFiles;
			auto dls = test.getDownloads(Os_OSX, cache.get(), failedFiles, QString());
			QCOMPARE(dls.size(), 2);
			QCOMPARE(failedFiles, {});
			QCOMPARE(dls[0]->m_url, QUrl("file://foo/bar/test/package/testname/testversion/testname-testversion-osx-32.jar"));
			QCOMPARE(dls[1]->m_url, QUrl("file://foo/bar/test/package/testname/testversion/testname-testversion-osx-64.jar"));
		}
	}
	void test_legacy_native_arch_local_override()
	{
		Library test("test.package:testname:testversion");
		test.m_nativeClassifiers[OpSys::Os_Linux]="linux-${arch}";
		test.setHint("local");
		QCOMPARE(test.isNative(), true);
		test.setRepositoryURL("file://foo/bar");
		{
			QStringList jar, native, native32, native64;
			test.getApplicableFiles(Os_Linux, jar, native, native32, native64, QString("data"));
			QCOMPARE(jar, {});
			QCOMPARE(native, {});
			QCOMPARE(native32, {QFileInfo("data/testname-testversion-linux-32.jar").absoluteFilePath()});
			QCOMPARE(native64, getStorage("test/package/testname/testversion/testname-testversion-linux-64.jar"));
			QStringList failedFiles;
			auto dls = test.getDownloads(Os_Linux, cache.get(), failedFiles, QString("data"));
			QCOMPARE(dls.size(), 0);
			QCOMPARE(failedFiles, {getStorage("test/package/testname/testversion/testname-testversion-linux-64.jar")});
		}
	}
	void test_onenine()
	{
		auto test = readMojangJson("data/lib-simple.json");
		{
			QStringList jar, native, native32, native64;
			test->getApplicableFiles(Os_OSX, jar, native, native32, native64, QString());
			QCOMPARE(jar, getStorage("com/paulscode/codecwav/20101023/codecwav-20101023.jar"));
			QCOMPARE(native, {});
			QCOMPARE(native32, {});
			QCOMPARE(native64, {});
		}
		{
			QStringList failedFiles;
			auto dls = test->getDownloads(Os_Linux, cache.get(), failedFiles, QString());
			QCOMPARE(dls.size(), 1);
			QCOMPARE(failedFiles, {});
			QCOMPARE(dls[0]->m_url, QUrl("https://libraries.minecraft.net/com/paulscode/codecwav/20101023/codecwav-20101023.jar"));
		}
		test->setHint("local");
		{
			QStringList jar, native, native32, native64;
			test->getApplicableFiles(Os_OSX, jar, native, native32, native64, QString("data"));
			QCOMPARE(jar, {QFileInfo("data/codecwav-20101023.jar").absoluteFilePath()});
			QCOMPARE(native, {});
			QCOMPARE(native32, {});
			QCOMPARE(native64, {});
		}
		{
			QStringList failedFiles;
			auto dls = test->getDownloads(Os_Linux, cache.get(), failedFiles, QString("data"));
			QCOMPARE(dls.size(), 0);
			QCOMPARE(failedFiles, {});
		}
	}
	void test_onenine_local_override()
	{
		auto test = readMojangJson("data/lib-simple.json");
		test->setHint("local");
		{
			QStringList jar, native, native32, native64;
			test->getApplicableFiles(Os_OSX, jar, native, native32, native64, QString("data"));
			QCOMPARE(jar, {QFileInfo("data/codecwav-20101023.jar").absoluteFilePath()});
			QCOMPARE(native, {});
			QCOMPARE(native32, {});
			QCOMPARE(native64, {});
		}
		{
			QStringList failedFiles;
			auto dls = test->getDownloads(Os_Linux, cache.get(), failedFiles, QString("data"));
			QCOMPARE(dls.size(), 0);
			QCOMPARE(failedFiles, {});
		}
	}
	void test_onenine_native()
	{
		auto test = readMojangJson("data/lib-native.json");
		QStringList jar, native, native32, native64;
		test->getApplicableFiles(Os_OSX, jar, native, native32, native64, QString());
		QCOMPARE(jar, QStringList());
		QCOMPARE(native, getStorage("org/lwjgl/lwjgl/lwjgl-platform/2.9.4-nightly-20150209/lwjgl-platform-2.9.4-nightly-20150209-natives-osx.jar"));
		QCOMPARE(native32, {});
		QCOMPARE(native64, {});
		QStringList failedFiles;
		auto dls = test->getDownloads(Os_OSX, cache.get(), failedFiles, QString());
		QCOMPARE(dls.size(), 1);
		QCOMPARE(failedFiles, {});
		QCOMPARE(dls[0]->m_url, QUrl("https://libraries.minecraft.net/org/lwjgl/lwjgl/lwjgl-platform/2.9.4-nightly-20150209/lwjgl-platform-2.9.4-nightly-20150209-natives-osx.jar"));
	}
	void test_onenine_native_arch()
	{
		auto test = readMojangJson("data/lib-native-arch.json");
		QStringList jar, native, native32, native64;
		test->getApplicableFiles(Os_Windows, jar, native, native32, native64, QString());
		QCOMPARE(jar, {});
		QCOMPARE(native, {});
		QCOMPARE(native32, getStorage("tv/twitch/twitch-platform/5.16/twitch-platform-5.16-natives-windows-32.jar"));
		QCOMPARE(native64, getStorage("tv/twitch/twitch-platform/5.16/twitch-platform-5.16-natives-windows-64.jar"));
		QStringList failedFiles;
		auto dls = test->getDownloads(Os_Windows, cache.get(), failedFiles, QString());
		QCOMPARE(dls.size(), 2);
		QCOMPARE(failedFiles, {});
		QCOMPARE(dls[0]->m_url, QUrl("https://libraries.minecraft.net/tv/twitch/twitch-platform/5.16/twitch-platform-5.16-natives-windows-32.jar"));
		QCOMPARE(dls[1]->m_url, QUrl("https://libraries.minecraft.net/tv/twitch/twitch-platform/5.16/twitch-platform-5.16-natives-windows-64.jar"));
	}
private:
	std::unique_ptr<HttpMetaCache> cache;
	QString dataDir;
};

QTEST_GUILESS_MAIN(LibraryTest)

#include "Library_test.moc"