#include <cstdint>
#include <cstdio>
#include <iostream>
#include <cstring>
#include <filesystem>
#include <vector>

#include <Windows.h>
#include <debugapi.h>

#include "ZipDefine.h"
#include "CRC.h"
#include "MyZip.h"

static uint32_t g_recursive_counter = 0;

bool ZipArchiver(std::string strInputPath, bool bIsFinal = false)
{
	MyZip zip;
	bool bRet = false;

	bRet = zip.AddFileToZip(strInputPath);
	if(bIsFinal)
		bRet = zip.FinalizeZip();

	return bRet;
}

bool FileReader(std::string strInputPath)
{
	using namespace std;
	bool bRet = true;

	// path class obj 생성.
	filesystem::path pathObj(strInputPath);

	// file valid check
	if (!filesystem::exists(pathObj))
	{
		string strDbg = "[ERROR] File not found : " + pathObj.string();
		strDbg += "\r\n";
		OutputDebugStringA(strDbg.c_str());
		
		bRet = false;
	}
	enum class eFILETYPE
	{
		isFile, isDir, ETC
	};

	// distinguish file type
	eFILETYPE eFileDist;
	if (filesystem::is_directory(pathObj)) eFileDist = eFILETYPE::isDir;
	else if (filesystem::is_regular_file(pathObj)) eFileDist = eFILETYPE::isFile;
	else eFileDist = eFILETYPE::ETC;

	const filesystem::path& rootPath = pathObj;
	switch (eFileDist)
	{
		// case 1 : input file
		case eFILETYPE::isFile:
		{
			bool bIsFinal = g_recursive_counter > 0 ? false : true;
			bRet = ZipArchiver(pathObj.string(), bIsFinal );

			break;
		}
		// case 2 : input directory
		case eFILETYPE::isDir:
		{
			// make iterator to traversal
			filesystem::directory_iterator itrDir(pathObj);
			while (itrDir != filesystem::end(itrDir))
			{
				// get entry one by one
				const filesystem::directory_entry& currentEntry = *itrDir;
				const filesystem::path& currentAbsPath = currentEntry.path();
				if (filesystem::is_directory(currentEntry))
				{
					// 있으면 좋을거같아서 만든 변수 "g_recursive_counter"
					g_recursive_counter++;
					// TODO : export to debugging log function
					string strDbg = "[INFO] Current Entry (Dir)\t: " + currentAbsPath.string();
					strDbg += "\t Counter : " + to_string(g_recursive_counter);
					strDbg += "\r\n";
					OutputDebugStringA(strDbg.c_str());

					// recursive traversal
					bRet = FileReader(currentAbsPath.string());
					g_recursive_counter--;
				}
				else if (filesystem::is_regular_file(currentEntry))
				{
					// TODO : export to debugging log function
					string strDbg = "[INFO] Current Entry (File)\t: " + currentAbsPath.string();
					strDbg += "\t Counter : " + to_string(g_recursive_counter);
					strDbg += "\r\n";
					OutputDebugStringA(strDbg.c_str());

					// archiving file
					auto relativePath = filesystem::relative(currentAbsPath, rootPath);
					bRet = ZipArchiver(relativePath.string());
				}
				else
				{
					OutputDebugStringA("[ERROR] Wrong Type of File");
					bRet = false;
				}

				// step forward entry
				itrDir++;
			}

			break;
		}
		// case 3 : ERROR
		case eFILETYPE::ETC:
		{
			// 이런 경우가 있으려나? 일단 혹시 모르니까
			string strErr = "[ERROR] Something went Wrong..!!";
			OutputDebugStringA(strErr.c_str());

			bRet = false;
			break;
		}
		default:
			break;
	};

	if(bRet)
	{
		bRet = ZipArchiver(strInputPath, true);
	}

	// TODO : 실패 상황 별 다른 코드 부여 + 코드 별 처리 어떻게 할 건지?
	return bRet;
}

int main(int argc, char* argv[])
{
	std::printf("==================== Hello Zip! ====================\r\n");
	int iRet = 0;

	// Main Zip archive method
	if (!FileReader(std::string(INPUT_FILE_NAME)))
	{
		iRet = -1;
		std::printf("[ ERROR ] Somthing went Wrong... \r\n");
	}
	else
	{
		std::printf("==================== Good Bye Zip! ====================\r\n");
	}

	return iRet;
}