From 95abf9c40266687b5525261bad964c6856a58f18 Mon Sep 17 00:00:00 2001 From: Eelco Dolstra Date: Fri, 22 Apr 2016 14:16:48 +0200 Subject: [PATCH] Add "nix copy" command This replaces nix-push. For example, $ nix copy --to file:///tmp/cache -r $(type -p firefox) copies the closure of firefox to the specified binary cache. And $ nix copy --from file:///tmp/cache --to s3://my-cache /nix/store/abcd... copies between two binary caches. It will also replace nix-copy-closure, once we have an SSHStore class, e.g. $ nix copy --from ssh://alice@machine /nix/store/abcd... --- src/nix/copy.cc | 87 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 87 insertions(+) create mode 100644 src/nix/copy.cc diff --git a/src/nix/copy.cc b/src/nix/copy.cc new file mode 100644 index 000000000..87e6f604a --- /dev/null +++ b/src/nix/copy.cc @@ -0,0 +1,87 @@ +#include "command.hh" +#include "progress-bar.hh" +#include "shared.hh" +#include "store-api.hh" +#include "sync.hh" + +#include + +using namespace nix; + +struct CmdCopy : StorePathsCommand +{ + std::string srcUri, dstUri; + + CmdCopy() + { + mkFlag(0, "from", "store-uri", "URI of the source Nix store", &srcUri); + mkFlag(0, "to", "store-uri", "URI of the destination Nix store", &dstUri); + } + + std::string name() override + { + return "copy"; + } + + std::string description() override + { + return "copy paths between Nix stores"; + } + + Examples examples() override + { + return { + Example{ + "To copy Firefox to the local store to a binary cache in file:///tmp/cache:", + "nix copy --to file:///tmp/cache -r $(type -p firefox)" + }, + }; + } + + void run(ref store, Paths storePaths) override + { + if (srcUri.empty() && dstUri.empty()) + throw UsageError("you must pass ‘--from’ and/or ‘--to’"); + + ref srcStore = srcUri.empty() ? store : openStoreAt(srcUri); + ref dstStore = dstUri.empty() ? store : openStoreAt(dstUri); + + ProgressBar progressBar; + + std::atomic done{0}; + std::atomic total{storePaths.size()}; + + auto showProgress = [&]() { + return (format("[%d/%d copied]") % done % total).str(); + }; + + progressBar.updateStatus(showProgress()); + + storePaths.reverse(); // FIXME: assumes reverse topo sort + + for (auto & storePath : storePaths) { + checkInterrupt(); + + if (dstStore->isValidPath(storePath)) { + total--; + progressBar.updateStatus(showProgress()); + continue; + } + + auto activity(progressBar.startActivity(format("copying ‘%s’...") % storePath)); + + StringSink sink; + srcStore->exportPaths({storePath}, false, sink); + + StringSource source(*sink.s); + dstStore->importPaths(false, source, 0); + + done++; + progressBar.updateStatus(showProgress()); + } + + progressBar.done(); + } +}; + +static RegisterCommand r1(make_ref());