mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-14 17:00:52 +08:00
116 lines
2.5 KiB
C++
116 lines
2.5 KiB
C++
#include "stdafx.h"
|
|
#include "AddBOM.h"
|
|
|
|
CAddBOM::CAddBOM(void)
|
|
: m_nMsgAdding(0)
|
|
, m_nMsgAdded(0)
|
|
, m_hWnd(0)
|
|
, m_sPath(_T(""))
|
|
{
|
|
}
|
|
|
|
CAddBOM::~CAddBOM(void)
|
|
{
|
|
}
|
|
|
|
void CAddBOM::Init(HWND hWnd, CString &sPath)
|
|
{
|
|
m_hWnd = hWnd;
|
|
m_sPath = sPath;
|
|
}
|
|
void CAddBOM::OnAdding(int nMsg)
|
|
{
|
|
m_nMsgAdding = nMsg;
|
|
}
|
|
|
|
void CAddBOM::OnAdded(int nMsg)
|
|
{
|
|
m_nMsgAdded = nMsg;
|
|
}
|
|
|
|
void* CAddBOM::run(void)
|
|
{
|
|
ScanAdd();
|
|
::PostMessage(m_hWnd, m_nMsgAdded, 0, 0);
|
|
return NULL;
|
|
}
|
|
|
|
void CAddBOM::ScanAdd(void)
|
|
{
|
|
if (m_sPath.GetLength() == 0) {
|
|
MessageBox(NULL, "文件路径为空！", "Error", 0);
|
|
return;
|
|
}
|
|
|
|
acl::scan_dir scan;
|
|
if (!scan.open(m_sPath.GetString())) {
|
|
MessageBox(NULL, "打开文件路径失败！", "Error", 0);
|
|
return;
|
|
}
|
|
|
|
logger("open %s and start scanning for adding BOM ...",
|
|
m_sPath.GetString());
|
|
|
|
const char* pFile;
|
|
while ((pFile = scan.next_file(true)) != NULL) {
|
|
acl::string path(pFile);
|
|
if (path.end_with("resource.h", false)) {
|
|
logger(">>skip file: %s", pFile);
|
|
} else if (path.end_with(".c") || path.end_with(".h") ||
|
|
path.end_with(".cpp") || path.end_with(".cxx") ||
|
|
path.end_with(".hpp") || path.end_with(".hxx") ||
|
|
path.end_with(".java") || path.end_with(".txt") ||
|
|
path.end_with(".php") || path.end_with(".html") ||
|
|
path.end_with(".js") || path.end_with(".css") ||
|
|
path.end_with(".d") || path.end_with(".py") ||
|
|
path.end_with(".perl") || path.end_with(".cs") ||
|
|
path.end_with(".as") || path.end_with(".go") ||
|
|
path.end_with(".rust") || path.end_with(".erl")) {
|
|
|
|
AddBOM(path);
|
|
} else {
|
|
logger(">>skip file: %s", pFile);
|
|
}
|
|
}
|
|
|
|
logger("%s: all text files have been added BOM!", m_sPath.GetString());
|
|
}
|
|
|
|
bool CAddBOM::AddBOM(const acl::string& filePath)
|
|
{
|
|
acl::string buf;
|
|
if (!acl::ifstream::load(filePath, &buf)) {
|
|
logger_error("load %s error %s",
|
|
filePath.c_str(), acl::last_serror());
|
|
return false;
|
|
}
|
|
if (buf.size() >= 3) {
|
|
if (buf[0] == (char) 0xEF && buf[1] == (char) 0xBB
|
|
&& buf[2] == (char) 0xBF) {
|
|
|
|
logger("%s: has BOM header!", filePath.c_str());
|
|
return true;
|
|
}
|
|
}
|
|
|
|
acl::string buf2(buf.size() + 4);
|
|
buf2 << (char) 0xEF << (char) 0xBB << (char) 0xBF;
|
|
buf2 += buf;
|
|
|
|
acl::ofstream fp;
|
|
if (!fp.open_write(filePath, true)) {
|
|
logger_error("open_write %s error %s",
|
|
filePath.c_str(), acl::last_serror());
|
|
return false;
|
|
}
|
|
|
|
if (fp.write(buf2) != (int) buf2.size()) {
|
|
logger_error("write to %s error %s",
|
|
filePath.c_str(), acl::last_serror());
|
|
return false;
|
|
}
|
|
|
|
logger("%s: add BOM ok!", filePath.c_str());
|
|
return true;
|
|
}
|