From c4f98941ed7e5e07786d92fc0be4435878f9b3cd Mon Sep 17 00:00:00 2001
From: Eelco Dolstra <e.dolstra@tudelft.nl>
Date: Mon, 4 Aug 2008 12:29:04 +0000
Subject: * nix-env --dry-run: show the total size of the substituter  
 downloads.

---
 src/libstore/misc.cc | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'src/libstore/misc.cc')

diff --git a/src/libstore/misc.cc b/src/libstore/misc.cc
index 1759f521cc29..85197adcf0db 100644
--- a/src/libstore/misc.cc
+++ b/src/libstore/misc.cc
@@ -46,8 +46,11 @@ Path findOutput(const Derivation & drv, string id)
 
 
 void queryMissing(const PathSet & targets,
-    PathSet & willBuild, PathSet & willSubstitute, PathSet & unknown)
+    PathSet & willBuild, PathSet & willSubstitute, PathSet & unknown,
+    unsigned long long & downloadSize)
 {
+    downloadSize = 0;
+    
     PathSet todo(targets.begin(), targets.end()), done;
 
     while (!todo.empty()) {
@@ -86,6 +89,7 @@ void queryMissing(const PathSet & targets,
             SubstitutablePathInfo info;
             if (store->querySubstitutablePathInfo(p, info)) {
                 willSubstitute.insert(p);
+                downloadSize += info.downloadSize;
                 todo.insert(info.references.begin(), info.references.end());
             } else
                 unknown.insert(p);
-- 
cgit 1.4.1