forked from lix-project/lix
bbe97dff8b
Most functions now take a StorePath argument rather than a Path (which is just an alias for std::string). The StorePath constructor ensures that the path is syntactically correct (i.e. it looks like <store-dir>/<base32-hash>-<name>). Similarly, functions like buildPaths() now take a StorePathWithOutputs, rather than abusing Path by adding a '!<outputs>' suffix. Note that the StorePath type is implemented in Rust. This involves some hackery to allow Rust values to be used directly in C++, via a helper type whose destructor calls the Rust type's drop() function. The main issue is the dynamic nature of C++ move semantics: after we have moved a Rust value, we should not call the drop function on the original value. So when we move a value, we set the original value to bitwise zero, and the destructor only calls drop() if the value is not bitwise zero. This should be sufficient for most types. Also lots of minor cleanups to the C++ API to make it more modern (e.g. using std::optional and std::string_view in some places).
94 lines
3.3 KiB
C++
94 lines
3.3 KiB
C++
#include "command.hh"
|
|
#include "store-api.hh"
|
|
#include "references.hh"
|
|
|
|
using namespace nix;
|
|
|
|
struct CmdMakeContentAddressable : StorePathsCommand
|
|
{
|
|
CmdMakeContentAddressable()
|
|
{
|
|
realiseMode = Build;
|
|
}
|
|
|
|
std::string description() override
|
|
{
|
|
return "rewrite a path or closure to content-addressable form";
|
|
}
|
|
|
|
Examples examples() override
|
|
{
|
|
return {
|
|
Example{
|
|
"To create a content-addressable representation of GNU Hello (but not its dependencies):",
|
|
"nix make-content-addressable nixpkgs.hello"
|
|
},
|
|
Example{
|
|
"To compute a content-addressable representation of the current NixOS system closure:",
|
|
"nix make-content-addressable -r /run/current-system"
|
|
},
|
|
};
|
|
}
|
|
void run(ref<Store> store, StorePaths storePaths) override
|
|
{
|
|
auto paths = store->topoSortPaths(storePathsToSet(storePaths));
|
|
|
|
std::reverse(paths.begin(), paths.end());
|
|
|
|
std::map<StorePath, StorePath> remappings;
|
|
|
|
for (auto & path : paths) {
|
|
auto pathS = store->printStorePath(path);
|
|
auto oldInfo = store->queryPathInfo(path);
|
|
auto oldHashPart = storePathToHash(pathS);
|
|
|
|
StringSink sink;
|
|
store->narFromPath(path, sink);
|
|
|
|
StringMap rewrites;
|
|
|
|
StorePathSet references;
|
|
bool hasSelfReference = false;
|
|
for (auto & ref : oldInfo->references) {
|
|
if (ref == path)
|
|
hasSelfReference = true;
|
|
else {
|
|
auto i = remappings.find(ref);
|
|
auto replacement = i != remappings.end() ? i->second.clone() : ref.clone();
|
|
// FIXME: warn about unremapped paths?
|
|
if (replacement != ref)
|
|
rewrites.insert_or_assign(store->printStorePath(ref), store->printStorePath(replacement));
|
|
references.insert(std::move(replacement));
|
|
}
|
|
}
|
|
|
|
*sink.s = rewriteStrings(*sink.s, rewrites);
|
|
|
|
HashModuloSink hashModuloSink(htSHA256, oldHashPart);
|
|
hashModuloSink((unsigned char *) sink.s->data(), sink.s->size());
|
|
|
|
auto narHash = hashModuloSink.finish().first;
|
|
|
|
ValidPathInfo info(store->makeFixedOutputPath(true, narHash, path.name(), references, hasSelfReference));
|
|
info.references = std::move(references);
|
|
if (hasSelfReference) info.references.insert(info.path.clone());
|
|
info.narHash = narHash;
|
|
info.narSize = sink.s->size();
|
|
info.ca = makeFixedOutputCA(true, info.narHash);
|
|
|
|
printError("rewrote '%s' to '%s'", pathS, store->printStorePath(info.path));
|
|
|
|
auto source = sinkToSource([&](Sink & nextSink) {
|
|
RewritingSink rsink2(oldHashPart, storePathToHash(store->printStorePath(info.path)), nextSink);
|
|
rsink2((unsigned char *) sink.s->data(), sink.s->size());
|
|
rsink2.flush();
|
|
});
|
|
|
|
store->addToStore(info, *source);
|
|
|
|
remappings.insert_or_assign(std::move(path), std::move(info.path));
|
|
}
|
|
}
|
|
};
|
|
|
|
static auto r1 = registerCommand<CmdMakeContentAddressable>("make-content-addressable");
|