2015-05-21 13:21:38 +00:00
|
|
|
|
#include "store-api.hh"
|
2015-04-22 12:21:52 +00:00
|
|
|
|
#include "hash.hh"
|
|
|
|
|
#include "shared.hh"
|
|
|
|
|
#include "globals.hh"
|
|
|
|
|
|
|
|
|
|
#include <iostream>
|
|
|
|
|
|
|
|
|
|
using namespace nix;
|
|
|
|
|
|
2015-05-05 23:43:24 +00:00
|
|
|
|
std::string gen = "old";
|
2015-04-22 12:21:52 +00:00
|
|
|
|
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)
|
|
|
|
|
{
|
2015-05-21 13:04:05 +00:00
|
|
|
|
if (access(dir.c_str(), R_OK) != 0) return;
|
|
|
|
|
|
|
|
|
|
bool canWrite = access(dir.c_str(), W_OK) == 0;
|
|
|
|
|
|
2015-04-22 12:21:52 +00:00
|
|
|
|
for (auto & i : readDirectory(dir)) {
|
|
|
|
|
checkInterrupt();
|
|
|
|
|
|
2015-05-21 13:04:05 +00:00
|
|
|
|
auto path = dir + "/" + i.name;
|
2015-05-21 12:09:34 +00:00
|
|
|
|
auto type = i.type == DT_UNKNOWN ? getFileType(path) : i.type;
|
2015-04-22 12:21:52 +00:00
|
|
|
|
|
2015-05-21 13:04:05 +00:00
|
|
|
|
if (type == DT_LNK && canWrite) {
|
2015-04-22 12:21:52 +00:00
|
|
|
|
auto link = readLink(path);
|
|
|
|
|
if (link.find("link") != string::npos) {
|
|
|
|
|
printMsg(lvlInfo, format("removing old generations of profile %1%") % path);
|
|
|
|
|
|
2015-05-06 15:16:35 +00:00
|
|
|
|
auto args = Strings{"-p", path, "--delete-generations", gen};
|
2015-05-21 13:04:05 +00:00
|
|
|
|
if (dryRun) args.push_back("--dry-run");
|
2015-05-06 15:16:35 +00:00
|
|
|
|
runProgramSimple(settings.nixBinDir + "/nix-env", args);
|
2015-04-22 12:21:52 +00:00
|
|
|
|
}
|
|
|
|
|
} 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;
|
|
|
|
|
gen = 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.
|
2015-05-21 13:21:38 +00:00
|
|
|
|
if (!dryRun) {
|
|
|
|
|
store = openStore(false);
|
|
|
|
|
GCOptions options;
|
|
|
|
|
options.action = GCOptions::gcDeleteDead;
|
|
|
|
|
GCResults results;
|
|
|
|
|
PrintFreed freed(true, results);
|
|
|
|
|
store->collectGarbage(options, results);
|
|
|
|
|
}
|
2015-04-22 12:21:52 +00:00
|
|
|
|
});
|
|
|
|
|
}
|