156 lines
3.6 KiB
C++
156 lines
3.6 KiB
C++
#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;
|
|
} |