mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-10-31 08:39:34 +08:00
151 lines
5.0 KiB
C++
151 lines
5.0 KiB
C++
/*
|
||
* 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;
|
||
}
|