ZLMediaKit/tests/bom.cpp
2020-05-18 15:31:49 +08:00

151 lines
5.0 KiB
C++
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/*
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
*
* This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
*
* Use of this source code is governed by MIT license that can be found in the
* LICENSE file in the root of the source tree. All contributing project authors
* may be found in the AUTHORS file in the root of the source tree.
*/
#include <stdlib.h>
#include <memory.h>
#if !defined(_WIN32)
#include <dirent.h>
#endif //!defined(_WIN32)
#include <set>
#include "Util/CMD.h"
#include "Util/util.h"
#include "Util/logger.h"
#include "Util/File.h"
#include "Util/uv_errno.h"
using namespace std;
using namespace toolkit;
class CMD_main : public CMD {
public:
CMD_main() {
_parser.reset(new OptionParser(nullptr));
(*_parser) << Option('r',/*该选项简称,如果是\x00则说明无简称*/
"rm",/*该选项全称,每个选项必须有全称不得为null或空字符串*/
Option::ArgNone,/*该选项后面必须跟值*/
nullptr,/*该选项默认值*/
false,/*该选项是否必须赋值如果没有默认值且为ArgRequired时用户必须提供该参数否则将抛异常*/
"是否删除或添加bom,默认添加bom头",/*该选项说明文字*/
nullptr);
(*_parser) << Option('f',/*该选项简称,如果是\x00则说明无简称*/
"filter",/*该选项全称,每个选项必须有全称不得为null或空字符串*/
Option::ArgRequired,/*该选项后面必须跟值*/
"c,cpp,cxx,c,h,hpp",/*该选项默认值*/
true,/*该选项是否必须赋值如果没有默认值且为ArgRequired时用户必须提供该参数否则将抛异常*/
"文件后缀过滤器",/*该选项说明文字*/
nullptr);
(*_parser) << Option('i',/*该选项简称,如果是\x00则说明无简称*/
"in",/*该选项全称,每个选项必须有全称不得为null或空字符串*/
Option::ArgRequired,/*该选项后面必须跟值*/
nullptr,/*该选项默认值*/
true,/*该选项是否必须赋值如果没有默认值且为ArgRequired时用户必须提供该参数否则将抛异常*/
"文件夹或文件",/*该选项说明文字*/
nullptr);
}
virtual ~CMD_main() {}
virtual const char *description() const {
return "添加或删除bom";
}
};
static const char s_bom[] = "\xEF\xBB\xBF";
void add_or_rm_bom(const char *file,bool rm_bom){
auto file_str = File::loadFile(file);
if(rm_bom){
file_str.erase(0, sizeof(s_bom) - 1);
}else{
file_str.insert(0,s_bom,sizeof(s_bom) - 1);
}
File::saveFile(file_str,file);
}
void process_file(const char *file,bool rm_bom){
std::shared_ptr<FILE> fp(fopen(file, "rb+"), [](FILE *fp) {
if (fp) {
fclose(fp);
}
});
if (!fp) {
WarnL << "打开文件失败:" << file << " " << get_uv_errmsg();
return;
}
bool have_bom = rm_bom;
char buf[sizeof(s_bom) - 1] = {0};
if (sizeof(buf) == fread(buf,1,sizeof(buf),fp.get())) {
have_bom = (memcmp(s_bom, buf, sizeof(s_bom) - 1) == 0);
}
if (have_bom == !rm_bom) {
// DebugL << "无需" << (rm_bom ? "删除" : "添加") << "bom:" << file;
return;
}
fp = nullptr;
add_or_rm_bom(file,rm_bom);
InfoL << (rm_bom ? "删除" : "添加") << "bom:" << file;
}
/// 这个程序是为了统一添加或删除utf-8 bom头
int main(int argc, char *argv[]) {
CMD_main cmd_main;
try {
cmd_main.operator()(argc, argv);
} catch (std::exception &ex) {
cout << ex.what() << endl;
return -1;
}
bool rm_bom = cmd_main.hasKey("rm");
string path = cmd_main["in"];
string filter = cmd_main["filter"];
auto vec = split(filter,",");
set<string> filter_set;
for(auto ext : vec){
filter_set.emplace(ext);
}
bool no_filter = filter_set.find("*") != filter_set.end();
//设置日志
Logger::Instance().add(std::make_shared<ConsoleChannel>());
File::scanDir(path, [&](const string &path, bool isDir) {
if (isDir) {
return true;
}
if (!no_filter) {
//开启了过滤器
auto pos = strstr(path.data(), ".");
if (pos == nullptr) {
//没有后缀
return true;
}
auto ext = pos + 1;
if (filter_set.find(ext) == filter_set.end()) {
//后缀不匹配
return true;
}
}
//该文件匹配
process_file(path.data(), rm_bom);
return true;
}, true);
return 0;
}