forked from dolphin-emu/dolphin
60 lines
2.1 KiB
C++
60 lines
2.1 KiB
C++
// Copyright 2008 Dolphin Emulator Project
|
|
// Licensed under GPLv2+
|
|
// Refer to the license.txt file included.
|
|
|
|
#include <algorithm>
|
|
#include <functional>
|
|
|
|
#include "Common/CommonPaths.h"
|
|
#include "Common/FileSearch.h"
|
|
#include "Common/FileUtil.h"
|
|
|
|
static std::vector<std::string>
|
|
FileSearchWithTest(const std::vector<std::string>& directories, bool recursive,
|
|
std::function<bool(const File::FSTEntry&)> callback)
|
|
{
|
|
std::vector<std::string> result;
|
|
for (const std::string& directory : directories)
|
|
{
|
|
File::FSTEntry top = File::ScanDirectoryTree(directory, recursive);
|
|
|
|
std::function<void(File::FSTEntry&)> DoEntry;
|
|
DoEntry = [&](File::FSTEntry& entry) {
|
|
if (callback(entry))
|
|
result.push_back(entry.physicalName);
|
|
for (auto& child : entry.children)
|
|
DoEntry(child);
|
|
};
|
|
for (auto& child : top.children)
|
|
DoEntry(child);
|
|
}
|
|
// remove duplicates
|
|
std::sort(result.begin(), result.end());
|
|
result.erase(std::unique(result.begin(), result.end()), result.end());
|
|
return result;
|
|
}
|
|
|
|
std::vector<std::string> DoFileSearch(const std::vector<std::string>& exts,
|
|
const std::vector<std::string>& directories, bool recursive)
|
|
{
|
|
bool accept_all = std::find(exts.begin(), exts.end(), "") != exts.end();
|
|
return FileSearchWithTest(directories, recursive, [&](const File::FSTEntry& entry) {
|
|
if (accept_all)
|
|
return true;
|
|
std::string name = entry.virtualName;
|
|
std::transform(name.begin(), name.end(), name.begin(), ::tolower);
|
|
return std::any_of(exts.begin(), exts.end(), [&](const std::string& ext) {
|
|
return name.length() >= ext.length() &&
|
|
name.compare(name.length() - ext.length(), ext.length(), ext) == 0;
|
|
});
|
|
});
|
|
}
|
|
|
|
// Result includes the passed directories themselves as well as their subdirectories.
|
|
std::vector<std::string> FindSubdirectories(const std::vector<std::string>& directories,
|
|
bool recursive)
|
|
{
|
|
return FileSearchWithTest(directories, true,
|
|
[&](const File::FSTEntry& entry) { return entry.isDirectory; });
|
|
}
|