forked from lix-project/lix
4ca5a9dcfd
Also, make sure --delete-older-than doesn't delete the current generation.
101 lines
2.8 KiB
C++
101 lines
2.8 KiB
C++
#include "store-api.hh"
|
||
#include "profiles.hh"
|
||
#include "shared.hh"
|
||
#include "globals.hh"
|
||
|
||
#include <iostream>
|
||
|
||
using namespace nix;
|
||
|
||
std::string deleteOlderThan;
|
||
bool dryRun = false;
|
||
|
||
void runProgramSimple(Path program, const Strings & args)
|
||
{
|
||
checkInterrupt();
|
||
|
||
/* Fork. */
|
||
Pid pid = startProcess([&]() {
|
||
Strings args_(args);
|
||
args_.push_front(program);
|
||
auto cargs = stringsToCharPtrs(args_);
|
||
|
||
execv(program.c_str(), (char * *) &cargs[0]);
|
||
|
||
throw SysError(format("executing ‘%1%’") % program);
|
||
});
|
||
|
||
pid.wait(true);
|
||
}
|
||
|
||
|
||
/* If `-d' was specified, remove all old generations of all profiles.
|
||
* Of course, this makes rollbacks to before this point in time
|
||
* impossible. */
|
||
|
||
void removeOldGenerations(std::string dir)
|
||
{
|
||
if (access(dir.c_str(), R_OK) != 0) return;
|
||
|
||
bool canWrite = access(dir.c_str(), W_OK) == 0;
|
||
|
||
for (auto & i : readDirectory(dir)) {
|
||
checkInterrupt();
|
||
|
||
auto path = dir + "/" + i.name;
|
||
auto type = i.type == DT_UNKNOWN ? getFileType(path) : i.type;
|
||
|
||
if (type == DT_LNK && canWrite) {
|
||
auto link = readLink(path);
|
||
if (link.find("link") != string::npos) {
|
||
printMsg(lvlInfo, format("removing old generations of profile %1%") % path);
|
||
if (deleteOlderThan != "")
|
||
deleteGenerationsOlderThan(path, deleteOlderThan, dryRun);
|
||
else
|
||
deleteOldGenerations(path, dryRun);
|
||
}
|
||
} else if (type == DT_DIR) {
|
||
removeOldGenerations(path);
|
||
}
|
||
}
|
||
}
|
||
|
||
int main(int argc, char * * argv)
|
||
{
|
||
bool removeOld = false;
|
||
Strings extraArgs;
|
||
|
||
return handleExceptions(argv[0], [&]() {
|
||
initNix();
|
||
|
||
parseCmdLine(argc, argv, [&](Strings::iterator & arg, const Strings::iterator & end) {
|
||
if (*arg == "--help")
|
||
showManPage("nix-collect-garbage");
|
||
else if (*arg == "--version")
|
||
printVersion("nix-collect-garbage");
|
||
else if (*arg == "--delete-old" || *arg == "-d") removeOld = true;
|
||
else if (*arg == "--delete-older-than") {
|
||
removeOld = true;
|
||
deleteOlderThan = getArg(*arg, arg, end);
|
||
}
|
||
else if (*arg == "--dry-run") dryRun = true;
|
||
else
|
||
extraArgs.push_back(*arg);
|
||
return true;
|
||
});
|
||
|
||
auto profilesDir = settings.nixStateDir + "/profiles";
|
||
if (removeOld) removeOldGenerations(profilesDir);
|
||
|
||
// Run the actual garbage collector.
|
||
if (!dryRun) {
|
||
store = openStore(false);
|
||
GCOptions options;
|
||
options.action = GCOptions::gcDeleteDead;
|
||
GCResults results;
|
||
PrintFreed freed(true, results);
|
||
store->collectGarbage(options, results);
|
||
}
|
||
});
|
||
}
|