94 lines
2.9 KiB
C++
94 lines
2.9 KiB
C++
#include <iostream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <boost/filesystem/operations.hpp>
|
|
#include <boost/filesystem/path.hpp>
|
|
|
|
#include "cellar.hpp"
|
|
#include "fs.hpp"
|
|
#include "output.hpp"
|
|
|
|
using namespace std;
|
|
using namespace boost;
|
|
using namespace cellar;
|
|
using namespace fs;
|
|
|
|
// these are not necessarily the final function signatures for callbacks
|
|
// this only exists so i don't have to come back and do the ductwork for
|
|
// callback functionality later
|
|
void cbnull_copy(string src, string dst) { return; }
|
|
void cbnull_remove(string src) { return; }
|
|
|
|
vector<string> fs::listdir(string path) {
|
|
vector<string> result;
|
|
filesystem::path cwd(path);
|
|
filesystem::directory_iterator iter_end;
|
|
|
|
for (filesystem::directory_iterator iter_cwd(cwd); iter_cwd != iter_end; ++iter_cwd) {
|
|
string item = iter_cwd->path().filename().native();
|
|
|
|
result.push_back(item);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool fs::recursive_copy(string src, string dst, CopyCallbackFunc callback) {
|
|
if (!filesystem::exists(dst)) {
|
|
if (dryrun) { output::statement("mkdir: " + dst); }
|
|
else {
|
|
bool success = filesystem::create_directory(dst);
|
|
if (!success) { return false; }
|
|
}
|
|
}
|
|
|
|
for (string itemrel : listdir(src)) {
|
|
string itemabs = src + "/" + itemrel;
|
|
string targetabs = dst + "/" + itemrel;
|
|
|
|
auto itemstat = filesystem::symlink_status(itemabs);
|
|
|
|
if (filesystem::is_directory(itemstat)) { recursive_copy(itemabs, targetabs, callback); }
|
|
else if (filesystem::is_symlink(itemstat)) {
|
|
auto symlinkpath = filesystem::read_symlink(itemabs);
|
|
if (dryrun) { output::statement("symlink: " + symlinkpath.native() + " => " + targetabs); }
|
|
else { filesystem::create_symlink(symlinkpath, targetabs); }
|
|
}
|
|
else {
|
|
if (dryrun) { output::statement("copy: " + itemabs + " => " + targetabs); }
|
|
else { filesystem::copy(itemabs, targetabs); }
|
|
}
|
|
|
|
callback(itemabs, targetabs);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool fs::recursive_copy(string src, string dst) { return recursive_copy(src, dst, cbnull_copy); }
|
|
|
|
bool fs::recursive_remove(string target, RemoveCallbackFunc callback) {
|
|
if (!filesystem::exists(target)) { return false; }
|
|
|
|
for (string itemrel : listdir(target)) {
|
|
string itemabs = target + "/" + itemrel;
|
|
|
|
auto itemstat = filesystem::symlink_status(itemabs);
|
|
|
|
if (filesystem::is_directory(itemstat)) { recursive_remove(itemabs, callback); }
|
|
else {
|
|
if (dryrun) { output::error("rm: " + itemabs); }
|
|
else { filesystem::remove(itemabs); }
|
|
}
|
|
|
|
callback(itemabs);
|
|
}
|
|
|
|
if (dryrun) { output::error("rm: " + target); }
|
|
else { filesystem::remove(target); }
|
|
|
|
return true;
|
|
}
|
|
|
|
bool fs::recursive_remove(string target) { return recursive_remove(target, cbnull_remove); }
|