/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

#include <catch2/catch_all.hpp>

#include "Base/Logger.hpp"
#include "IO/FsUtils.hpp"
#include "IO/AppFolders.hpp"

#include <filesystem>
#include <fstream>

using namespace OpenVulkano;

TEST_CASE("DeleteEmptyDirs", "[FsUtils]")
{
	Logger::SetupLogger("", "tests.log");
	auto baseDir = AppFolders::GetAppTempDir() / "testEmptyDirs";
	std::filesystem::create_directories(baseDir / "emptyDir");
	std::filesystem::create_directories(baseDir / "nonEmptyDir");
	std::ofstream(baseDir / "nonEmptyDir/file.txt").put('x');

	REQUIRE(FsUtils::DeleteEmptyDirs(baseDir));
	REQUIRE(!std::filesystem::exists(baseDir / "emptyDir"));
	REQUIRE(std::filesystem::exists(baseDir / "nonEmptyDir"));

	std::filesystem::create_directories(baseDir / "emptyDir");
	REQUIRE(FsUtils::DeleteEmptyDirs(baseDir, false));
	REQUIRE(!std::filesystem::exists(baseDir / "emptyDir"));

	std::filesystem::remove_all(baseDir);
}

TEST_CASE("DeleteEmptySubTrees", "[FsUtils]")
{
	Logger::SetupLogger("", "tests.log");
	auto baseDir = AppFolders::GetAppTempDir() / "testEmptySubtrees";
	std::filesystem::create_directories(baseDir / "emptyTree" / "subDir1");
	std::filesystem::create_directories(baseDir / "emptyTree" / "subDir2");

	REQUIRE(FsUtils::DeleteEmptySubTrees(baseDir));
	REQUIRE(!std::filesystem::exists(baseDir / "emptyTree"));

	std::filesystem::create_directories(baseDir / "partialTree" / "subDir1");
	std::filesystem::create_directories(baseDir / "partialTree" / "subDir2");
	std::ofstream(baseDir / "partialTree" / "subDir2/file.txt").put('x');

	REQUIRE(FsUtils::DeleteEmptySubTrees(baseDir / "partialTree"));

	REQUIRE(!std::filesystem::exists(baseDir / "partialTree" / "subDir1"));
	REQUIRE(std::filesystem::exists(baseDir / "partialTree" / "subDir2"));
	REQUIRE(std::filesystem::exists(baseDir / "partialTree" / "subDir2/file.txt"));

	std::filesystem::remove(baseDir / "partialTree" / "subDir2/file.txt");

	REQUIRE(!std::filesystem::exists(baseDir / "partialTree" / "subDir1"));
	REQUIRE(std::filesystem::exists(baseDir / "partialTree" / "subDir2"));
	REQUIRE(!std::filesystem::exists(baseDir / "partialTree" / "subDir2/file.txt"));

	REQUIRE(FsUtils::DeleteEmptySubTrees(baseDir / "partialTree"));

	REQUIRE(!std::filesystem::exists(baseDir / "partialTree" / "subDir1"));
	REQUIRE(!std::filesystem::exists(baseDir / "partialTree" / "subDir2"));

	std::filesystem::remove_all(baseDir);
}

TEST_CASE("IsTreeEmpty", "[FsUtils]")
{
	Logger::SetupLogger("", "tests.log");
	auto baseDir = AppFolders::GetAppTempDir() / "testTreeEmpty";
	std::filesystem::create_directories(baseDir / "emptyDir");
	std::filesystem::create_directories(baseDir / "nonEmptyDir");
	std::ofstream(baseDir / "nonEmptyDir/file.txt").put('x');

	REQUIRE(FsUtils::IsTreeEmpty(baseDir / "emptyDir"));
	REQUIRE(!FsUtils::IsTreeEmpty(baseDir / "nonEmptyDir"));

	std::filesystem::remove_all(baseDir);
}

TEST_CASE("GetDirSize", "[FsUtils]")
{
	Logger::SetupLogger("", "tests.log");
	auto baseDir = AppFolders::GetAppTempDir() / "testDirSize";
	std::filesystem::create_directories(baseDir / "dir");
	std::ofstream(baseDir / "dir/file1.txt").put('x');
	std::ofstream(baseDir / "dir/file2.txt").put('y');

	REQUIRE(FsUtils::GetDirSize(baseDir / "dir") == 2);

	std::filesystem::remove_all(baseDir);
}

TEST_CASE("ToValidFileName", "[FsUtils]")
{
	REQUIRE(FsUtils::ToValidFileName("file<name>") == "file＜name＞");
	REQUIRE(FsUtils::ToValidFileName("file:name") == "file꞉name");
	REQUIRE(FsUtils::ToValidFileName("file\"name\"") == "file＂name＂");
	REQUIRE(FsUtils::ToValidFileName("file/name\\") == "file⁄name⧹");
	REQUIRE(FsUtils::ToValidFileName("file|name") == "file｜name");
	REQUIRE(FsUtils::ToValidFileName("file?name*") == "file？name∗");

	REQUIRE(FsUtils::ToValidFileName("file~name") == "filename");
	REQUIRE(FsUtils::ToValidFileName("file`name") == "filename");
	REQUIRE(FsUtils::ToValidFileName("file!@#$%^&name") == "filename");
	REQUIRE(FsUtils::ToValidFileName("file(name{name}") == "filenamename");
	REQUIRE(FsUtils::ToValidFileName("file=name") == "filename");
	REQUIRE(FsUtils::ToValidFileName("file+name,name") == "filenamename");

	REQUIRE(FsUtils::ToValidFileName("  file name  ") == "file name");
	REQUIRE(FsUtils::ToValidFileName("..file name..") == "file name");
	REQUIRE(FsUtils::ToValidFileName(".file.") == "file");

	REQUIRE(FsUtils::ToValidFileName("file    name") == "file name");
	REQUIRE(FsUtils::ToValidFileName("file  name") == "file name");

	REQUIRE(FsUtils::ToValidFileName("") == "unnamed_file");
	REQUIRE(FsUtils::ToValidFileName("    ") == "unnamed_file");
	REQUIRE(FsUtils::ToValidFileName("...") == "unnamed_file");

	std::string long_name(300, 'a');
	REQUIRE(FsUtils::ToValidFileName(long_name).length() == 255);

	std::string utf8_name = "name_" + std::string(250, 'a') + "∗";
	REQUIRE(FsUtils::ToValidFileName(utf8_name).back() == 'a');

	REQUIRE(FsUtils::ToValidFileName("<>:/\\|?*") == "＜＞꞉⁄⧹｜？∗");
	REQUIRE(FsUtils::ToValidFileName("~~~~") == "unnamed_file");
	REQUIRE(FsUtils::ToValidFileName("..") == "unnamed_file");

	REQUIRE(FsUtils::ToValidFileName("file:name<>") == "file꞉name＜＞");
	REQUIRE(FsUtils::ToValidFileName(" file<name>?") == "file＜name＞？");
}