mirror of
https://github.com/Tptr-lateworker/RealESR_Tools.git
synced 2025-01-07 03:17:34 +08:00
Add files via upload
This commit is contained in:
lateworker
GitHub
parent
ed0ff85c65
commit
c85e34c785
253
include/path.h
Normal file
253
include/path.h
Normal file
@ -0,0 +1,253 @@
|
||||
#include <iostream>
|
||||
#include <sys/stat.h>
|
||||
#include <vector>
|
||||
#include <queue>
|
||||
#include <io.h>
|
||||
|
||||
#ifndef PATH_BASIC
|
||||
#define PATH_BASIC
|
||||
inline std::string pathQuote(std::string x) { return "\"" + x + "\""; }
|
||||
inline void pathDelete(std::string &data) {
|
||||
while (data.front() == '\"') data.erase(data.begin());
|
||||
while (data.back() == '\"') data.erase(data.end() - 1);
|
||||
while (data.front() == '\\' || data.front() == '/') data.erase(data.begin());
|
||||
while (data.back() == '\\' || data.back() == '/') data.erase(data.end() - 1);
|
||||
for (size_t i = 1; i < data.size(); ++i)
|
||||
if ((data[i] == '\\' && data[i - 1] == data[i]) || (data[i] == '/' && data[i - 1] == data[i]))
|
||||
data.erase(data.begin() + i), --i;
|
||||
}
|
||||
inline void pathSplit(std::string x, std::string &path, std::string &name) {
|
||||
name = path = "";
|
||||
size_t pos = x.find_last_of("\\");
|
||||
if (pos == std::string::npos)
|
||||
path = ".", name = x;
|
||||
else {
|
||||
name = x.substr(pos);
|
||||
pathDelete(name);
|
||||
path = x.substr(0, pos);
|
||||
pathDelete(path);
|
||||
}
|
||||
}
|
||||
inline std::string pathMerge(const std::string &path, const std::string &name) { return path + "\\" + name; }
|
||||
#endif
|
||||
|
||||
#ifndef PATH_H
|
||||
#define PATH_H
|
||||
inline bool pathExist(std::string path, struct stat *buffer = nullptr) {
|
||||
struct stat _buffer;
|
||||
if (buffer == nullptr) buffer = &_buffer;
|
||||
return !stat(path.c_str(), buffer);
|
||||
}
|
||||
namespace path {
|
||||
struct FileAttribute {
|
||||
signed attrib;
|
||||
FileAttribute() { attrib = 0; }
|
||||
FileAttribute(const signed &_attrib) { attrib = _attrib; }
|
||||
inline void operator= (const unsigned &o) { attrib = o; }
|
||||
inline bool is_normal() { return attrib == _A_NORMAL; }
|
||||
inline bool is_read_only() { return attrib & _A_RDONLY; }
|
||||
inline bool is_hidden() { return attrib & _A_HIDDEN; }
|
||||
inline bool is_system() { return attrib & _A_SYSTEM; }
|
||||
inline bool is_folder() { return attrib & _A_SUBDIR; }
|
||||
inline bool is_archive() { return attrib & _A_ARCH; }
|
||||
inline bool is_file() { return !(attrib & _A_SUBDIR); }
|
||||
};
|
||||
struct TraverseData {
|
||||
FileAttribute attrib;
|
||||
unsigned long size;
|
||||
std::string path, name;
|
||||
inline std::string fullpath() const { return pathMerge(path, name); }
|
||||
};
|
||||
struct TraverseMode {
|
||||
bool forFile, forDir, IntoSubdir, includingEmptyDir;
|
||||
bool operator== (const TraverseMode &o) const {
|
||||
return forFile == o.forFile && forDir == o.forDir && IntoSubdir == o.IntoSubdir;
|
||||
}
|
||||
};
|
||||
#define TraverseForAll {true, true, true, true}
|
||||
std::queue<TraverseData> traverse_temp;
|
||||
|
||||
// 参数传递版本
|
||||
bool _traverse(const std::string &path, std::vector<TraverseData> &results, const std::string &selector, const TraverseMode &mode);
|
||||
bool _traverseFile(const std::string &path, std::vector<TraverseData> &results, const std::string &selector, const TraverseMode &mode) {
|
||||
_finddata_t fileInfo;
|
||||
std::string currentPath = path + "\\" + selector;
|
||||
intptr_t handle = _findfirst(currentPath.c_str(), &fileInfo);
|
||||
if (handle == -1) return false;
|
||||
FileAttribute newAttrib;
|
||||
TraverseData fileData, dirData;
|
||||
do {
|
||||
newAttrib = fileInfo.attrib;
|
||||
if (newAttrib.is_folder()) continue;
|
||||
fileData = {
|
||||
newAttrib,
|
||||
fileInfo.size,
|
||||
path, fileInfo.name
|
||||
};
|
||||
if (!mode.includingEmptyDir) {
|
||||
while (!traverse_temp.empty()) {
|
||||
dirData = traverse_temp.front();
|
||||
traverse_temp.pop();
|
||||
if (path.find(dirData.path + "\\" + dirData.name) != std::string::npos)
|
||||
results.push_back(dirData);
|
||||
}
|
||||
if (!mode.forFile) return false;
|
||||
}
|
||||
if (mode.forFile) {
|
||||
results.push_back(fileData);
|
||||
}
|
||||
} while (!_findnext(handle, &fileInfo));
|
||||
return true;
|
||||
}
|
||||
bool _traverseDir(const std::string &path, std::vector<TraverseData> &results, const std::string &selector, const TraverseMode &mode) {
|
||||
_finddata_t fileInfo;
|
||||
std::string currentPath = path + "\\*.*";
|
||||
intptr_t handle = _findfirst(currentPath.c_str(), &fileInfo);
|
||||
if (handle == -1) return false;
|
||||
bool returnValue = false;
|
||||
std::string newpath;
|
||||
FileAttribute newAttrib;
|
||||
TraverseData dirData;
|
||||
do {
|
||||
newAttrib = fileInfo.attrib;
|
||||
if (!newAttrib.is_folder() || strcmp(fileInfo.name, "..") == 0 || strcmp(fileInfo.name, ".") == 0)
|
||||
continue;
|
||||
dirData = {
|
||||
newAttrib,
|
||||
fileInfo.size,
|
||||
path, fileInfo.name
|
||||
};
|
||||
if (mode.forDir) {
|
||||
if (mode.includingEmptyDir) {
|
||||
results.push_back(dirData);
|
||||
} else {
|
||||
traverse_temp.push(dirData);
|
||||
}
|
||||
}
|
||||
newpath = path + "\\" + fileInfo.name;
|
||||
if (mode.IntoSubdir) {
|
||||
if (_traverse(newpath, results, selector, mode))
|
||||
returnValue = true;
|
||||
} else returnValue = true;
|
||||
} while (!_findnext(handle, &fileInfo));
|
||||
return returnValue;
|
||||
}
|
||||
bool _traverse(const std::string &path, std::vector<TraverseData> &results, const std::string &selector, const TraverseMode &mode) {
|
||||
bool returnValue_traverseFile = false;
|
||||
bool returnValue_traverseDir = false;
|
||||
if (mode.forFile || (!mode.includingEmptyDir && !traverse_temp.empty()))
|
||||
returnValue_traverseFile = _traverseFile(path, results, selector, mode);
|
||||
if (mode.forDir || mode.IntoSubdir)
|
||||
returnValue_traverseDir = _traverseDir(path, results, selector, mode);
|
||||
return returnValue_traverseFile || returnValue_traverseDir;
|
||||
}
|
||||
bool traverse(std::string path, std::vector<TraverseData> &results, std::string selector = "*.*", TraverseMode mode = TraverseForAll) {
|
||||
pathDelete(path);
|
||||
return _traverse(path, results, selector, mode);
|
||||
}
|
||||
|
||||
// 回调函数版本
|
||||
typedef void (*callback)(TraverseData data);
|
||||
bool _traverseC(const std::string &path, const callback &fileFunc, const callback &dirFunc, const std::string &selector, const TraverseMode &mode);
|
||||
bool _traverseFileC(const std::string &path, const callback &fileFunc, const callback &dirFunc, const std::string &selector, const TraverseMode &mode) {
|
||||
_finddata_t fileInfo;
|
||||
std::string currentPath = path + "\\" + selector;
|
||||
intptr_t handle = _findfirst(currentPath.c_str(), &fileInfo);
|
||||
if (handle == -1) return false;
|
||||
FileAttribute newAttrib;
|
||||
TraverseData fileData, dirData;
|
||||
do {
|
||||
newAttrib = fileInfo.attrib;
|
||||
if (newAttrib.is_folder()) continue;
|
||||
fileData = {
|
||||
newAttrib,
|
||||
fileInfo.size,
|
||||
path, fileInfo.name
|
||||
};
|
||||
if (!mode.includingEmptyDir) {
|
||||
while (!traverse_temp.empty()) {
|
||||
dirData = traverse_temp.front();
|
||||
traverse_temp.pop();
|
||||
if (path.find(dirData.path + "\\" + dirData.name) != std::string::npos)
|
||||
if (dirFunc != nullptr)
|
||||
dirFunc(dirData);
|
||||
}
|
||||
if (!mode.forFile) return false;
|
||||
}
|
||||
if (mode.forFile) {
|
||||
if (fileFunc != nullptr)
|
||||
fileFunc(fileData);
|
||||
}
|
||||
} while (!_findnext(handle, &fileInfo));
|
||||
return true;
|
||||
}
|
||||
bool _traverseDirC(const std::string &path, const callback &fileFunc, const callback &dirFunc, const std::string &selector, const TraverseMode &mode) {
|
||||
_finddata_t fileInfo;
|
||||
std::string currentPath = path + "\\*.*";
|
||||
intptr_t handle = _findfirst(currentPath.c_str(), &fileInfo);
|
||||
if (handle == -1) return false;
|
||||
bool returnValue = false;
|
||||
std::string newpath;
|
||||
FileAttribute newAttrib;
|
||||
TraverseData dirData;
|
||||
do {
|
||||
newAttrib = fileInfo.attrib;
|
||||
if (!newAttrib.is_folder() || strcmp(fileInfo.name, "..") == 0 || strcmp(fileInfo.name, ".") == 0)
|
||||
continue;
|
||||
dirData = {
|
||||
newAttrib,
|
||||
fileInfo.size,
|
||||
path, fileInfo.name
|
||||
};
|
||||
if (mode.forDir) {
|
||||
if (mode.includingEmptyDir) {
|
||||
if (dirFunc != nullptr)
|
||||
dirFunc(dirData);
|
||||
} else {
|
||||
traverse_temp.push(dirData);
|
||||
}
|
||||
}
|
||||
newpath = path + "\\" + fileInfo.name;
|
||||
if (mode.IntoSubdir) {
|
||||
if (_traverseC(newpath, fileFunc, dirFunc, selector, mode))
|
||||
returnValue = true;
|
||||
} else returnValue = true;
|
||||
} while (!_findnext(handle, &fileInfo));
|
||||
return returnValue;
|
||||
}
|
||||
bool _traverseC(const std::string &path, const callback &fileFunc, const callback &dirFunc, const std::string &selector, const TraverseMode &mode) {
|
||||
bool returnValue_traverseFile = false;
|
||||
bool returnValue_traverseDir = false;
|
||||
if (mode.forFile || (!mode.includingEmptyDir && !traverse_temp.empty()))
|
||||
returnValue_traverseFile = _traverseFileC(path, fileFunc, dirFunc, selector, mode);
|
||||
if (mode.forDir || mode.IntoSubdir)
|
||||
returnValue_traverseDir = _traverseDirC(path, fileFunc, dirFunc, selector, mode);
|
||||
return returnValue_traverseFile || returnValue_traverseDir;
|
||||
}
|
||||
bool traverseC(std::string path, callback fileFunc = nullptr, callback dirFunc = nullptr, std::string selector = "*.*", TraverseMode mode = TraverseForAll) {
|
||||
pathDelete(path);
|
||||
return _traverseC(path, fileFunc, dirFunc, selector, mode);
|
||||
}
|
||||
|
||||
FileAttribute getFileAttribute(const std::string &path) {
|
||||
_finddata_t fileInfo;
|
||||
_findfirst(path.c_str(), &fileInfo);
|
||||
return fileInfo.attrib;
|
||||
}
|
||||
|
||||
struct Time {
|
||||
long time_access, time_modify, time_create;
|
||||
inline void operator= (const struct stat &o) {
|
||||
time_access = o.st_atime;
|
||||
time_modify = o.st_mtime;
|
||||
time_create = o.st_ctime;
|
||||
}
|
||||
};
|
||||
bool getTimeData(std::string path, Time &timeData) {
|
||||
struct stat _buffer;
|
||||
if (!pathExist(path, &_buffer)) return true;
|
||||
timeData = _buffer;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
54
include/system.h
Normal file
54
include/system.h
Normal file
@ -0,0 +1,54 @@
|
||||
#include <iostream>
|
||||
|
||||
#ifndef PATH_BASIC
|
||||
#define PATH_BASIC
|
||||
inline std::string pathQuote(std::string x) { return "\"" + x + "\""; }
|
||||
inline void pathDelete(std::string &data) {
|
||||
while (data.front() == '\"') data.erase(data.begin());
|
||||
while (data.back() == '\"') data.erase(data.end() - 1);
|
||||
while (data.front() == '\\' || data.front() == '/') data.erase(data.begin());
|
||||
while (data.back() == '\\' || data.back() == '/') data.erase(data.end() - 1);
|
||||
for (size_t i = 1; i < data.size(); ++i)
|
||||
if ((data[i] == '\\' && data[i - 1] == data[i]) || (data[i] == '/' && data[i - 1] == data[i]))
|
||||
data.erase(data.begin() + i), --i;
|
||||
}
|
||||
inline void pathSplit(std::string x, std::string &path, std::string &name) {
|
||||
name = path = "";
|
||||
size_t pos = x.find_last_of("\\");
|
||||
if (pos == std::string::npos)
|
||||
path = ".", name = x;
|
||||
else {
|
||||
name = x.substr(pos);
|
||||
pathDelete(name);
|
||||
path = x.substr(0, pos);
|
||||
pathDelete(path);
|
||||
}
|
||||
}
|
||||
inline std::string pathMerge(const std::string &path, const std::string &name) { return path + "\\" + name; }
|
||||
#endif
|
||||
|
||||
#ifndef SYSTEM_H
|
||||
#define SYSTEM_H
|
||||
inline void makedir(std::string path) {
|
||||
system(("md " + pathQuote(path) + " >nul 2>&1").c_str());
|
||||
}
|
||||
inline void removedir(std::string path) {
|
||||
system(("rd " + pathQuote(path) + " >nul 2>&1").c_str());
|
||||
}
|
||||
inline void copyfile(std::string file_path, std::string direction_path) {
|
||||
bool ret = system(("xcopy /q/g/k/r/h/y/-i " + pathQuote(file_path) + " " + pathQuote(direction_path) + " >nul 2>&1").c_str());
|
||||
if (ret) {
|
||||
std::string _path, _name;
|
||||
pathSplit(direction_path, _path, _name);
|
||||
makedir(_path);
|
||||
system(("copy /d/y " + pathQuote(file_path) + " " + pathQuote(direction_path) + " >nul 2>&1").c_str());
|
||||
}
|
||||
}
|
||||
inline void movefile(std::string file_path, std::string direction_path) {
|
||||
system(("move /y " + pathQuote(file_path) + " " + pathQuote(direction_path) + " >nul 2>&1").c_str());
|
||||
}
|
||||
inline void executefile(std::string file) { system(file.c_str()); }
|
||||
inline void deletefile(std::string file) {
|
||||
system(("del /f/s/q " + pathQuote(file) + " >nul 2>&1").c_str());
|
||||
}
|
||||
#endif
|
||||
Loading…
Reference in New Issue
Block a user