124 lines
4.4 KiB
C++
124 lines
4.4 KiB
C++
#include <algorithm>
|
||
#include <filesystem>
|
||
#include <iostream>
|
||
#include <map>
|
||
#include <string>
|
||
#include <vector>
|
||
|
||
namespace fs = std::filesystem;
|
||
|
||
class AdvancedNfoFileChecker {
|
||
private:
|
||
std::vector<std::string> videoExtensions = {".avi", ".m2ts", ".mkv", ".mp4", ".ts"};
|
||
std::map<std::string, bool> checkResults;
|
||
|
||
public:
|
||
bool isVideoFile(const fs::path &filePath) {
|
||
std::string extension = filePath.extension().string();
|
||
std::transform(extension.begin(), extension.end(), extension.begin(), ::tolower);
|
||
return std::find(videoExtensions.begin(), videoExtensions.end(), extension) != videoExtensions.end();
|
||
}
|
||
|
||
bool hasNfoFile(const fs::path &videoPath) {
|
||
fs::path nfoPath = videoPath;
|
||
nfoPath.replace_extension(".nfo");
|
||
return fs::exists(nfoPath);
|
||
}
|
||
|
||
void checkFolderRecursive(const std::string &folderPath, bool recursive = false) {
|
||
checkResults.clear();
|
||
int totalVideoFiles = 0;
|
||
int missingNfoCount = 0;
|
||
|
||
try {
|
||
auto iterator = fs::directory_iterator(folderPath);
|
||
for (const auto &entry : iterator) {
|
||
if (entry.is_regular_file()) {
|
||
fs::path filePath = entry.path();
|
||
|
||
if (isVideoFile(filePath)) {
|
||
totalVideoFiles++;
|
||
bool hasNfo = hasNfoFile(filePath);
|
||
checkResults[filePath.string()] = hasNfo;
|
||
|
||
if (!hasNfo) {
|
||
missingNfoCount++;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
printResults(totalVideoFiles, missingNfoCount);
|
||
|
||
} catch (const fs::filesystem_error &ex) {
|
||
std::cerr << "错误：无法访问文件夹 '" << folderPath << "'" << std::endl;
|
||
std::cerr << "详细信息: " << ex.what() << std::endl;
|
||
}
|
||
}
|
||
|
||
private:
|
||
void printResults(int totalVideoFiles, int missingNfoCount) {
|
||
std::cout << "检查结果：" << std::endl;
|
||
std::cout << "==========================================" << std::endl;
|
||
std::cout << "总视频文件数: " << totalVideoFiles << std::endl;
|
||
std::cout << "缺少.nfo的文件数: " << missingNfoCount << std::endl;
|
||
std::cout << "完整率: " << (totalVideoFiles - missingNfoCount) * 100 / std::max(1, totalVideoFiles) << "%"
|
||
<< std::endl;
|
||
std::cout << "==========================================" << std::endl;
|
||
|
||
if (missingNfoCount > 0) {
|
||
std::cout << "缺少.nfo文件的视频文件：" << std::endl;
|
||
for (const auto &[filepath, hasNfo] : checkResults) {
|
||
if (!hasNfo) {
|
||
std::cout << " ✗ " << fs::path(filepath).filename().string() << std::endl;
|
||
}
|
||
}
|
||
} else if (totalVideoFiles > 0) {
|
||
std::cout << "✓ 所有视频文件都有对应的.nfo文件" << std::endl;
|
||
} else {
|
||
std::cout << "! 在指定文件夹中未找到视频文件" << std::endl;
|
||
}
|
||
}
|
||
};
|
||
|
||
int main(int argc, char *argv[]) {
|
||
AdvancedNfoFileChecker checker;
|
||
|
||
std::string folderPath;
|
||
bool recursive = false;
|
||
|
||
// 解析命令行参数
|
||
for (int i = 1; i < argc; ++i) {
|
||
std::string arg = argv[i];
|
||
if (arg == "-r" || arg == "--recursive") {
|
||
recursive = true;
|
||
} else if (arg == "-h" || arg == "--help") {
|
||
std::cout << "用法: " << argv[0] << " [文件夹路径] [-r]" << std::endl;
|
||
std::cout << "选项:" << std::endl;
|
||
std::cout << " -r, --recursive 递归检查子文件夹" << std::endl;
|
||
std::cout << " -h, --help 显示帮助信息" << std::endl;
|
||
return 0;
|
||
} else {
|
||
folderPath = arg;
|
||
}
|
||
}
|
||
|
||
if (folderPath.empty()) {
|
||
std::cout << "请输入要检查的文件夹路径（留空使用当前目录）: ";
|
||
std::getline(std::cin, folderPath);
|
||
if (folderPath.empty()) {
|
||
folderPath = ".";
|
||
}
|
||
}
|
||
|
||
std::cout << "正在检查文件夹: " << fs::absolute(folderPath) << std::endl;
|
||
if (recursive) {
|
||
std::cout << "模式: 递归检查（包含子文件夹）" << std::endl;
|
||
}
|
||
std::cout << "支持的视频格式: .avi, .m2ts, .mkv, .mp4, .ts" << std::endl;
|
||
std::cout << "==========================================" << std::endl;
|
||
|
||
checker.checkFolderRecursive(folderPath, recursive);
|
||
|
||
return 0;
|
||
} |