/*
 * This file is part of VoidArchiveTool.
 *
 * Copyright (C) 2025 Yanczi
 *
 * Void Archive Toolis 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 3 of the License, or
 * (at your option) any later version.
 *
 * This program 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 program.  If not, see <https://www.gnu.org/licenses/>.
 */

#include "CreateCargo.h"

CreateCargo::CreateCargo()
	: signature(SIGNATURE)
	, extension(EXTENSION)
	, version(VERSION)
	, methodFlags(0)
	, offset(0)
	, hppKey(false)
{
	// TODO Auto-generated constructor stub
}

CreateCargo::~CreateCargo() {
	// TODO Auto-generated destructor stub
}


//-----------------------------------------------------------------------------
//   Punk wejścia do tworzenia archivum
//-----------------------------------------------------------------------------
bool CreateCargo::Create(const std::string& path, const int16_t& flag)
{
	cargoFile = path + "." + extension;
	catalogPath = path;
	methodFlags = flag;

	//Sprawdzanie pakowanego kontentu
	if (!std::filesystem::is_directory(path))
	{
		std::cerr << "Error: The specified directory is a file!" << std::endl;
		return false;
	}

	if (!std::filesystem::exists(path))
	{
		std::cerr << "Error: The specified directory does not exist!" << std::endl;
		return false;
	}

	// Pobieranie listy plików wyjątków
	if (flag == -1)
	{
		std::string filterFile = path + ".json";
		if (!std::filesystem::exists(filterFile))
		{
			std::cerr << "Error: Missing " << filterFile << " file!" << std::endl;
			return false;
		}
		GetFilters(filterFile);
	}

	//Pobieranie listy plików do spakowania
	std::cout << "Creating a file list..." << std::endl;
	if (!GetFileList(path))
	{
		std::cerr << "Error: The specified directory contains no files!" << std::endl;
		return false;
	}

	// Utworzenie kontenera
	cargo.open(cargoFile, std::ios::binary);

	//Rozpocznij process zapisywania danych do kontenera
	if (!WriteCargo())
	{
		std::cerr << "Error: Failed to create container!" << std::endl;
		return false;
	}

	// Zapisywanie klucza szyfrującego
	if (flag == 2 || flag == 3 || encList.size() > 0)
	{
		crypt.saveKey(catalogPath, hppKey);
	}

	return true;
}

//-----------------------------------------------------------------------------
//   Tworzenie listy plików do spakowania
//-----------------------------------------------------------------------------
bool CreateCargo::GetFileList(const std::string& path)
{
	for (const auto& entry : std::filesystem::directory_iterator(path))
	{
		std::string tmpPath = entry.path().string();
		if (std::filesystem::is_directory(tmpPath))
		{
			GetFileList(tmpPath);
		}
		else
		{
			std::string fileRef = RemoveStartPath(PathToUnixLike(tmpPath));
			PathConf pc;
			if (methodFlags > -1)
			{
				pc.path = PathToUnixLike(tmpPath);
				pc.parameter = methodFlags;
				filesPaths.push_back(pc);
			}
			else
			{
				if (!FindOnTheList(ignoreList, fileRef) || !CheckFileExtension(fileRef, ignoreList))
				{
					if (FindOnTheList(zipList, fileRef) || CheckFileExtension(fileRef, zipList))
					{
						pc.parameter = FindOnTheList(encList, fileRef) || CheckFileExtension(fileRef, encList) ? 3 : 1;
					}
					else
					{
						pc.parameter = FindOnTheList(encList, fileRef) || CheckFileExtension(fileRef, encList) ? 2 : 0;
					}
					pc.path = PathToUnixLike(tmpPath);
					std::cout << pc.path << " - " << pc.parameter << std::endl;
					filesPaths.push_back(pc);
				}
			}
		}
	}

	return filesPaths.size() > 0 ? true : false;
}

//-----------------------------------------------------------------------------
//   Konwersja ścieżki na unix like
//-----------------------------------------------------------------------------
std::string CreateCargo::PathToUnixLike(std::string path)
{
	std::replace(path.begin(), path.end(), '\\', '/');
	return path;
}

//-----------------------------------------------------------------------------
//   Kasowanie wskazanego katalogu ze ścieżki
//-----------------------------------------------------------------------------
std::string CreateCargo::RemoveStartPath(const std::filesystem::path& path)
{
	std::filesystem::path cleanPath;

	auto it = path.begin();
	if (it != path.end()) { ++it; }

	for (; it != path.end(); ++it)
	{
		cleanPath /= *it;
	}

	return cleanPath.string();
}

//-----------------------------------------------------------------------------
//   Tworzenie nagłowka pliku
//-----------------------------------------------------------------------------
CargoHead CreateCargo::CreateCargoHead(const uint32_t& filesLen, const uint64_t& table)
{
	CargoHead ch;

	ch.signature = signature;
	ch.version = version;
	ch.files = filesLen;
	ch.table = table;

	return ch;
}

//-----------------------------------------------------------------------------
//   Sprawdza czy plik znajduje się na liście
//-----------------------------------------------------------------------------
void CreateCargo::computingBytes(const int16_t& flag, std::vector<char>& input, std::vector<char>& output)
{
	//Flaga aktywna sprawdza czy plik jest na liście. Jeśli jest to zwraca surowedane
	//Przeciwnie kompresuje dane
	ChunkManager cm(crypt);

	switch (flag)
	{
	case 1:
		output = cm.chunked(input, true, false);
		break;

	case 2:
		output = cm.chunked(input, false, true);
		break;

	case 3:
		output = cm.chunked(input, true, true);
		break;

	case 4:
		output = cm.chunked(input, false, false);
		break;

	default:
		output = std::move(input);
		break;
	}
}

//-----------------------------------------------------------------------------
//   Przygotowanie nagłówków i plików
//-----------------------------------------------------------------------------
std::vector<FilesTable> CreateCargo::ComputingHeadFiles()
{
	//Utwórz header TMP. Zabezpiecza Pierwsze bajty na właściwy nagłówek
	CargoHead cargoHead = CreateCargoHead(0, 0);
	offset += cargoHead.signature.length() + sizeof(cargoHead.version) + sizeof(cargoHead.files) + sizeof(cargoHead.table);

	//Zapisanie tymczasowego nagłowka jako rezerwacja miejsca
	cargo.write(cargoHead.signature.data(), cargoHead.signature.length());
	cargo.write(reinterpret_cast<const char*>(&cargoHead.version), sizeof(cargoHead.version));
	cargo.write(reinterpret_cast<const char*>(&cargoHead.files), sizeof(cargoHead.files));
	cargo.write(reinterpret_cast<const char*>(&cargoHead.table), sizeof(cargoHead.table));

	std::vector<FilesTable> filesTable;

	//Tworzenie nagłówków plików jednocześnie zapisywanie plików
	for (const auto& file : filesPaths)
	{
		std::string path = PathToUnixLike(RemoveStartPath(file.path));
		std::ifstream f(file.path, std::ios::binary | std::ios::ate);

		//Obliczanie rozmiaru pliku
		size_t size = f.tellg();
		f.seekg(0, std::ios::beg);

		//Wczytanie pliku do pamięci
		std::vector<char> buffer(size);
		f.read(buffer.data(), size);
		f.close();

		//Tworzenie hashu CRC
		const uint64_t crc = XXH64(buffer.data(), buffer.size(), VERSION);

		//Kompresjia
		std::vector<char> pakBuffer;
		computingBytes(file.parameter, buffer, pakBuffer);

		FilesTable ft;
		ft.nameFile = path;
		ft.nameLen = path.length();
		ft.hashName = fnv64(path);
		ft.offset = offset;
		ft.size = pakBuffer.size();
		ft.flag = file.parameter;
		ft.crc = crc;

		cargo.write(reinterpret_cast<const char*>(pakBuffer.data()), pakBuffer.size());

		filesTable.push_back(ft);
		offset += pakBuffer.size();
	}
	return filesTable;
}

//-----------------------------------------------------------------------------
//   Wczytanie filtrów wyjątków
//-----------------------------------------------------------------------------
void CreateCargo::GetFilters(const std::string& filterFile)
{
	std::cout << "Downloading the exception list" << std::endl;

	std::ifstream file(filterFile);
	nlohmann::json jslist;
	file >> jslist;
	file.close();

	// Lista plików do skompresowania
	zipList = jslist[KEY_ZIP].get<std::vector<std::string>>();

	// Lista plików do zaszyfrowania
	encList = jslist[KEY_ENCRYPT].get<std::vector<std::string>>();

	// Lista plików do pominięcia
	ignoreList = jslist[KEY_IGNORE].get<std::vector<std::string>>();

	hppKey = jslist["keyhpp"];
}

//-----------------------------------------------------------------------------
//   Znajdź wskazany element na liście
//-----------------------------------------------------------------------------
bool CreateCargo::FindOnTheList(const std::vector<std::string>& list, const std::string& element)
{
	auto it = std::find(list.begin(), list.end(), element);
	return it == list.end() ? false : true;
}

//-----------------------------------------------------------------------------
//   Sprawdzanie rozszeżeń plików
//-----------------------------------------------------------------------------
bool CreateCargo::CheckFileExtension(const std::string& p, const std::vector<std::string>& patterns) {
	std::filesystem::path _p = p;
	std::string ext =  "*" + UpperString(_p.extension().string());

	for (const auto& e : patterns)
	{
		std::string element = UpperString(e);
		if (element == ext)
		{
			return true;
		}
	}

	return false;
}

//-----------------------------------------------------------------------------
//   Zamień cały ciąg na duże litery
//-----------------------------------------------------------------------------
std::string CreateCargo::UpperString(std::string s) {
	std::transform(s.begin(), s.end(), s.begin(),
		[](unsigned char c) { return static_cast<char>(std::toupper(c)); });
	return s;
}

//-----------------------------------------------------------------------------
//   Wygenerój FNV-1a HASH
//-----------------------------------------------------------------------------
uint64_t CreateCargo::fnv64(const std::string& data)
{
	const uint64_t fnvOffset = 14695981039346656037u;
	const uint64_t fnvPrime = 1099511628211u;

	uint64_t hash = fnvOffset;
	for (unsigned char c : data)
	{
		hash ^= c;
		hash *= fnvPrime;
	}

	return hash;
}

//-----------------------------------------------------------------------------
//   Trworzenie archiwum
//-----------------------------------------------------------------------------
bool CreateCargo::WriteCargo()
{
	std::cout << "Packing files..." << std::endl;

	uint32_t filesLen = filesPaths.size();

	//Przygotowanie nagłówków plików i przetworzenie danych
	std::vector<FilesTable> filesHead = ComputingHeadFiles();

	//std::ofstream cargo(cargoFile, std::ios::binary);

	if (!cargo.is_open())
	{
		std::cerr << "ERROR: Failed to create container " << cargoFile << std::endl;
		return false;
	}

	for (const auto& head : filesHead)
	{
		// Ścieżka do plików
		cargo.write(reinterpret_cast<const char*>(&head.nameLen), sizeof(head.nameLen));
		cargo.write(head.nameFile.data(), head.nameLen);

		cargo.write(reinterpret_cast<const char*>(&head.hashName), sizeof(head.hashName));
		cargo.write(reinterpret_cast<const char*>(&head.offset), sizeof(head.offset));
		cargo.write(reinterpret_cast<const char*>(&head.size), sizeof(head.size));
		cargo.write(reinterpret_cast<const char*>(&head.crc), sizeof(head.crc));
		cargo.write(reinterpret_cast<const char*>(&head.flag), sizeof(head.flag));
	}

	//Cofnij się na początek pliku
	cargo.seekp(std::ios::beg);

	//Utwórz header
	CargoHead cargoHead = CreateCargoHead(filesLen, offset);

	//Nadpisz tymczasowy nagłówek
	cargo.write(cargoHead.signature.data(), cargoHead.signature.length());
	cargo.write(reinterpret_cast<const char*>(&cargoHead.version), sizeof(cargoHead.version));
	cargo.write(reinterpret_cast<const char*>(&cargoHead.files), sizeof(cargoHead.files));
	cargo.write(reinterpret_cast<const char*>(&cargoHead.table), sizeof(cargoHead.table));

	cargo.close();

	std::cout << "The container was successfully created! " << cargoFile << std::endl;

	return true;
}
