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

#include <fstream>
#include <iostream>
#include <filesystem>

#include <string>
#include <debugapi.h>

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

#define INPUT_FILE_NAME "MyZip.txt"
#define OUTPUT_FILE_NAME "MyZip.zip"

bool FileReader();

int main()
{
	std::printf("==================== Hello Zip! ====================\r\n");
	int iRet = 0;

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

	return iRet;
}
bool searchDir()
{

}


bool FileReader()
{
	LocalFileHeader LFHeader{};
	CentralDirectory CentDir{};
	EoCDRecord EOCDR{};

	using namespace std;

	ifstream fileIn(INPUT_FILE_NAME, ios::binary);
	bool iRet = fileIn.is_open();
	if (!iRet)
	{
		printf("FileIn open failed");
	}
	else
	{
		// path class obj 생성.
		filesystem::path pathObj(INPUT_FILE_NAME);

		// file valid check
		if (!filesystem::exists(pathObj))
		{
			string strErr = "File not found : " + pathObj.string();
			OutputDebugStringA(strErr.c_str());
			return false;
		}
		if (!filesystem::is_regular_file(pathObj))
		{
			string strErr = "File is not regular file : " + pathObj.string();
			OutputDebugStringA(strErr.c_str());

			// directory recursive search
			
			if (filesystem::is_directory(pathObj))
			{
				filesystem::directory_iterator itrDir(pathObj);

				while (itrDir != filesystem::end(itrDir))
				{
					// get entry one by one
					const filesystem::directory_entry& currentEntry = *itrDir;

					if (filesystem::is_directory(currentEntry))
					{
						// TODO : 다시 위로 돌아가야 함 + 재귀적으로 돌아가면서 처리
						// Do Something...
					}
					else if(filesystem::is_regular_file(currentEntry))
					{
						// TODO : 파일 Archiving logic
						ZipArchiver();
					}
					else
					{
						// 애초에 제 3의 Type 이 있나?
						OutputDebugStringA("Wrong Type of File");
					}

					// step forward entry
					itrDir++;
				}
			}
			else
			{
				// 이런 경우가 있으려나? 일단 혹시 모르니까
				string strErr = "Something went Wrong..!!";
				OutputDebugStringA(strErr.c_str());

				return false;
			}
		}
		else
		{
			ZipArchiver();
		}
	}

	// TODO : 어떤 식으로 Return 할 건지 하나만 정하기
	return iRet;
}

bool ZipArchiver()
{
	// 파일 크기 계산
	// TODO : Buffering 으로 구현. 현재는 파일 통짜로 읽음.
	fileIn.seekg(0, ios::end);	// seekg (offset, 목표점)
	streamsize fileInSize = fileIn.tellg();
	fileIn.seekg(0, ios::beg);	// beg == begin

	// 메모리 할당
	vector<uint8_t> fileInBuffer(fileInSize);
	auto* pInBuffer = reinterpret_cast<char*>(fileInBuffer.data());

	// Read
	fileIn.read(pInBuffer, fileInSize);
	// ZIP 포맷 채워넣기
	// TODO : 클래스화 대상
	{
		// Get CRC value
		uint32_t myCRC = getCRC32(pInBuffer, fileInSize);

		// Put PK Signature
		LFHeader.signature = PK_SIGNATURE_LF_HEADER;
		CentDir.signature = PK_SIGNATURE_CENT_DIR;
		EOCDR.signature = PK_SIGNATURE_EO_CDR;

		// TOOD : Temp Format 채우기.
		CentDir.versionMadeBy = 0; // Temp
		LFHeader.versionNeededToExtract = 0; // Temp
		CentDir.versionNeededToExtract = 0; // Temp
		LFHeader.generalBitFlag = 0; // Temp
		CentDir.generalBitFlag = 0; // Temp

		// Put compressed method (uncompressed)
		LFHeader.compressionMethod = METHOD_UNCOMPRESSED;
		CentDir.compressionMethod = METHOD_UNCOMPRESSED;

		// lastMode[Time|Date] => Not Used.

		// Put CRC value
		LFHeader.crc32 = myCRC;
		CentDir.crc32 = myCRC;

		// Put File size
		LFHeader.sizeOfUncompressed = (uint32_t)fileInSize;
		CentDir.sizeOfUncompressed = (uint32_t)fileInSize;
		LFHeader.sizeOfCompressed = LFHeader.sizeOfUncompressed;
		CentDir.sizeOfCompressed = CentDir.sizeOfUncompressed;

		// Put File name length
		uint16_t fileNameLeng = static_cast<uint16_t>(strlen(INPUT_FILE_NAME));
		LFHeader.lengthOfFileName = fileNameLeng;
		CentDir.lengthOfFileName = fileNameLeng;

		// Extra Field 는 사용하지 않음
		// LFHeader.lengthOfExtraField = NOT_USED;
		// CentDir.lengthOfExtraField = NOT_USED;

		// TODO : CentDir Temp format 채우기.
		CentDir.lengthOfFileComment = 0;
		CentDir.diskNumStart = 0; // Temp
		CentDir.attributeInternal = 0; // Temp
		CentDir.attributeExternal = 0; // Temp
		// CentDir.offsetLocalFileHeader -> 실제 파일 Write 중에 작성.

		// TODO : EOCDR Temp format 채우기.
		EOCDR.diskNumber = 0; // Temp
		EOCDR.diskNumber_CentralDirStart = 0; // Temp
		EOCDR.numOfCentralDir = 1;
		EOCDR.numOfEntries = 1;
		EOCDR.sizeOfCentralDir = sizeof(CentDir) + CentDir.lengthOfFileName;
		// EOCDR.offsetCentralDir -> 실제 파일 Write 중에 작성.
		EOCDR.lengthOfCommentLength = 0;
	}

	// Zip arhcive 파일 쓰기
	{
		ofstream fileOut(OUTPUT_FILE_NAME, ios::binary);
		iRet = fileOut.is_open();
		if (!iRet)
		{
			printf("FileOut open failed");
		}
		else
		{
			// 1. Local File Header, file name (var), extra field (var)
			CentDir.offsetLocalFileHeader = static_cast<uint32_t>(fileOut.tellp());
			// TODO : sizeof 의 값이 스펙과 정확히 일치한다는 보장 필요 (현재는 compiler 에 종속적)
			//		: offsetof((LocalFileHeader, crc32) 으로 멤버별 static_assert 로 변경할 지 고민.
			//		: 아니면 필드 단위 직접쓰기로 struct 분해하기?
			fileOut.write(reinterpret_cast<const char*>(&LFHeader), sizeof(LFHeader));
			fileOut.write(INPUT_FILE_NAME, LFHeader.lengthOfFileName);
			//fileOut.write(NOT_USED, LFHeader.LFHeader.lengthOfExtraField);

			// 2. File Data
			fileOut.write(pInBuffer, fileInSize);

			// 3. CentralDirectory, file name (var)
			EOCDR.offsetCentralDir = static_cast<uint32_t>(fileOut.tellp());
			// TODO : sizeof 의 값이 스펙과 정확히 일치한다는 보장 필요 (현재는 compiler 에 종속적)
			fileOut.write(reinterpret_cast<const char*>(&CentDir), sizeof(CentDir));
			fileOut.write(INPUT_FILE_NAME, CentDir.lengthOfFileName);

			// 4. End of CentralDirectory Record
			// TODO : sizeof 의 값이 스펙과 정확히 일치한다는 보장 필요 (현재는 compiler 에 종속적)
			fileOut.write(reinterpret_cast<const char*>(&EOCDR), sizeof(EOCDR));

			fileOut.close();
		}
	}

	return iRet;
}