#include <cstdio>
#include <iostream>
#include <cstring>
#include <fstream>

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

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

bool ZipArchiver();
uint32_t getCRC32(const char* pData, size_t length);

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

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

	return iRet;
}

bool ZipArchiver()
{
	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
	{
		// 파일 크기 계산
		// seekg(offset, 목표점)
		fileIn.seekg(0, ios::end);
		streamsize fileInSize = fileIn.tellg();
		// beg == begin
		fileIn.seekg(0, ios::beg);

		// TODO : Buffering 으로 구현. 현재는 파일 통짜로 읽음.
		// 메모리 할당
		// TODO : 자동으로 free 되는 구조로 변경
		auto pFile = (char *)malloc(fileInSize);

		// Read
		fileIn.read(pFile, fileInSize);
		// ZIP 포맷 채워넣기
		// TODO : 여기있을 필요까진 없는듯. 코드 이관 필요
		{
			// Get CRC value
			// TODO : 현재 streamsize 와 size_t 의 mismatch 존재. 해결필요.(long long <-> unsigned long long)
			uint32_t myCRC = getCRC32(pFile, 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(pFile, 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();
			}
		}

		// Free
		free(pFile);
	}

	return iRet;
}

uint32_t getCRC32(const char* pData, size_t length)
{
	uint32_t crc = 0xFFFFFFFF;
	
	// 1. 입력
	for (size_t i = 0; i < length ; i++)
	{
		// Index 추출
		// 입력과 현재 CRC를 1 Byte 씩 XOR + (1의 보수처리)
		uint8_t iTableIdx = (crc ^ pData[i]) & 0xFF;

		// 1 Byte Shift
		crc = crc >> 8;

		// Table 과 XOR 계산
		crc = crc ^ crcTable[iTableIdx];
	}

	// TODO : CRC 최적화
	//		  "_mm_crc32_u16" => MSVC 자체적으로 지원하는 최적화된 CRC 가 있는듯?

	// 1의 보수처리
	return crc ^ 0xFFFFFFFF;
}