From 024a8ed3822867c5f9b9498bd8b27fa5dc180846 Mon Sep 17 00:00:00 2001
From: Eelco Dolstra <e.dolstra@tudelft.nl>
Date: Wed, 21 Feb 2007 23:14:53 +0000
Subject: [PATCH] * New command `nix-copy-closure' to copy a closure to a Nix
 store on   another machine through ssh.  E.g.,

    $ nix-copy-closure xyzzy $(which svn)

  copies the closure of Subversion to machine `xyzzy'.  This is like
  `nix-pack-closure $(which svn) | ssh xyzzy', but it's much more
  efficient since it only copies those paths that are missing on the
  target machine.
---
 scripts/Makefile.am         |  4 ++-
 scripts/nix-copy-closure.in | 55 +++++++++++++++++++++++++++++++++++++
 2 files changed, 58 insertions(+), 1 deletion(-)
 create mode 100644 scripts/nix-copy-closure.in

diff --git a/scripts/Makefile.am b/scripts/Makefile.am
index 2c27f832ed..e16b36d611 100644
--- a/scripts/Makefile.am
+++ b/scripts/Makefile.am
@@ -1,7 +1,8 @@
 bin_SCRIPTS = nix-collect-garbage \
   nix-pull nix-push nix-prefetch-url \
   nix-install-package nix-channel nix-build \
-  nix-pack-closure nix-unpack-closure
+  nix-pack-closure nix-unpack-closure \
+  nix-copy-closure
 
 noinst_SCRIPTS = nix-profile.sh generate-patches.pl find-runtime-roots.pl
 
@@ -30,4 +31,5 @@ EXTRA_DIST = nix-collect-garbage.in \
   download-using-manifests.pl.in \
   generate-patches.pl.in \
   nix-pack-closure.in nix-unpack-closure.in \
+  nix-copy-closure.in \
   find-runtime-roots.pl.in
diff --git a/scripts/nix-copy-closure.in b/scripts/nix-copy-closure.in
new file mode 100644
index 0000000000..ec2df88289
--- /dev/null
+++ b/scripts/nix-copy-closure.in
@@ -0,0 +1,55 @@
+#! @perl@ -w
+
+my $binDir = $ENV{"NIX_BIN_DIR"};
+$binDir = "@bindir@" unless defined $binDir;
+
+
+# Get the target host.
+my $sshHost = shift @ARGV;
+
+
+# !!! Copied from nix-pack-closure, should put this in a module.
+my %storePathsSeen;
+my @storePaths = ();
+
+while (@ARGV) {
+    my $storePath = shift @ARGV;
+
+    # $storePath might be a symlink to the store, so resolve it.
+    $storePath = (`$binDir/nix-store --query --resolve '$storePath'`
+        or die "cannot resolve `$storePath'");
+    chomp $storePath;
+
+    # Get the closure of this path.
+    my $pid = open(READ,
+        "$binDir/nix-store --query --requisites '$storePath'|") or die;
+    
+    while (<READ>) {
+        chomp;
+        die "bad: $_" unless /^\//;
+        if (!defined $storePathsSeen{$_}) {
+            push @storePaths, $_;
+            $storePathsSeen{$_} = 1;
+        }
+    }
+
+    close READ or die "nix-store failed: $?";
+}
+
+
+# Ask the remote host which paths are invalid.
+open(READ, "-|", "ssh", $sshHost, "nix-store", "--check-validity", "--print-invalid", @storePaths);
+my @missing = ();
+while (<READ>) {
+    chomp;
+    print STDERR "target needs $_\n";
+    push @missing, $_;
+}
+close READ or die;
+
+
+# Export the store paths and import them on the remote machine.
+if (scalar @missing > 0) {
+    system("nix-store --export @missing | ssh $sshHost nix-store --import") == 0
+        or die "copying store paths to remote machine failed: $?";
+}