diff options
author | Lluís Batlle i Rossell <viric@viric.name> | 2013-04-11T17·52+0200 |
---|---|---|
committer | Eelco Dolstra <eelco.dolstra@logicblox.com> | 2013-05-03T09·08+0200 |
commit | 5cc2fc46ecd4ad38a42591943fc29d412ad5cfc4 (patch) | |
tree | f0ace74fe6f1763892de2b7358041195922bb134 /scripts/nix-copy-closure.in | |
parent | 3628b61ce0aaff43daf37d086697c7c3f9c8ef13 (diff) |
Adding ETA support to the --show-progress in nix-copy-closure
Based on https://github.com/NixOS/nix/pull/6 from shlevy
Diffstat (limited to 'scripts/nix-copy-closure.in')
-rwxr-xr-x | scripts/nix-copy-closure.in | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/scripts/nix-copy-closure.in b/scripts/nix-copy-closure.in index 1a1684ed00ac..7fb38ac2239d 100755 --- a/scripts/nix-copy-closure.in +++ b/scripts/nix-copy-closure.in @@ -4,6 +4,7 @@ use Nix::SSH; use Nix::Config; use Nix::Store; use Nix::CopyClosure; +use List::Util qw(sum); if (scalar @ARGV < 1) { @@ -61,7 +62,7 @@ while (@ARGV) { $includeOutputs = 1; } elsif ($arg eq "--show-progress") { - $progressViewer = "@pv@"; + $progressViewer = "pv"; } elsif ($arg eq "--dry-run") { $dryRun = 1; @@ -104,6 +105,11 @@ else { # Copy FROM the remote machine. close READ or die "nix-store on remote machine `$sshHost' failed: $?"; + my $missingSize = 0; + if ($progressViewer ne "") { + $missingSize = sum (split ' ', `set -f; ssh @sshOpts $sshHost nix-store -q --size @missing`) or die; + } + # Export the store paths on the remote machine and import them locally. if (scalar @missing > 0) { print STDERR "copying ", scalar @missing, " missing paths from ‘$sshHost’...\n"; @@ -113,7 +119,7 @@ else { # Copy FROM the remote machine. } $compressor = "| $compressor" if $compressor ne ""; $decompressor = "$decompressor |" if $decompressor ne ""; - $progressViewer = "$progressViewer |" if $progressViewer ne ""; + $progressViewer = "$progressViewer -s $missingSize |" if $progressViewer ne ""; my $extraOpts = $sign ? "--sign" : ""; system("set -f; ssh $sshHost @sshOpts 'nix-store --export $extraOpts @missing $compressor' | $progressViewer $decompressor $Nix::Config::binDir/nix-store --import > /dev/null") == 0 or die "copying store paths from remote machine `$sshHost' failed: $?"; |