about summary refs log tree commit diff
path: root/third_party/nix/src/nix-copy-closure/nix-copy-closure.cc
blob: 3cc29ae337ff74ffed3a886e3e8de09173c04052 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#include "legacy.hh"
#include "shared.hh"
#include "store-api.hh"

using namespace nix;

static int _main(int argc, char** argv) {
  {
    auto gzip = false;
    auto toMode = true;
    auto includeOutputs = false;
    auto dryRun = false;
    auto useSubstitutes = NoSubstitute;
    std::string sshHost;
    PathSet storePaths;

    parseCmdLine(
        argc, argv, [&](Strings::iterator& arg, const Strings::iterator& end) {
          if (*arg == "--help")
            showManPage("nix-copy-closure");
          else if (*arg == "--version")
            printVersion("nix-copy-closure");
          else if (*arg == "--gzip" || *arg == "--bzip2" || *arg == "--xz") {
            if (*arg != "--gzip")
              printMsg(lvlError, format("Warning: '%1%' is not implemented, "
                                        "falling back to gzip") %
                                     *arg);
            gzip = true;
          } else if (*arg == "--from")
            toMode = false;
          else if (*arg == "--to")
            toMode = true;
          else if (*arg == "--include-outputs")
            includeOutputs = true;
          else if (*arg == "--show-progress")
            printMsg(lvlError, "Warning: '--show-progress' is not implemented");
          else if (*arg == "--dry-run")
            dryRun = true;
          else if (*arg == "--use-substitutes" || *arg == "-s")
            useSubstitutes = Substitute;
          else if (sshHost.empty())
            sshHost = *arg;
          else
            storePaths.insert(*arg);
          return true;
        });

    initPlugins();

    if (sshHost.empty()) throw UsageError("no host name specified");

    auto remoteUri = "ssh://" + sshHost + (gzip ? "?compress=true" : "");
    auto to = toMode ? openStore(remoteUri) : openStore();
    auto from = toMode ? openStore() : openStore(remoteUri);

    PathSet storePaths2;
    for (auto& path : storePaths)
      storePaths2.insert(from->followLinksToStorePath(path));

    PathSet closure;
    from->computeFSClosure(storePaths2, closure, false, includeOutputs);

    copyPaths(from, to, closure, NoRepair, NoCheckSigs, useSubstitutes);

    return 0;
  }
}

static RegisterLegacyCommand s1("nix-copy-closure", _main);