1 files changed, 56 insertions, 44 deletions

diff --git a/src/nix/diff-closures.cc b/src/nix/diff-closures.cc
index 56ddb575b..4199dae0f 100644
--- a/src/nix/diff-closures.cc
+++ b/src/nix/diff-closures.cc
@@ -6,7 +6,7 @@
 
 #include <regex>
 
-using namespace nix;
+namespace nix {
 
 struct Info
 {
@@ -52,6 +52,60 @@ std::string showVersions(const std::set<std::string> & versions)
     return concatStringsSep(", ", versions2);
 }
 
+void printClosureDiff(
+    ref<Store> store,
+    const StorePath & beforePath,
+    const StorePath & afterPath,
+    std::string_view indent)
+{
+    auto beforeClosure = getClosureInfo(store, beforePath);
+    auto afterClosure = getClosureInfo(store, afterPath);
+
+    std::set<std::string> allNames;
+    for (auto & [name, _] : beforeClosure) allNames.insert(name);
+    for (auto & [name, _] : afterClosure) allNames.insert(name);
+
+    for (auto & name : allNames) {
+        auto & beforeVersions = beforeClosure[name];
+        auto & afterVersions = afterClosure[name];
+
+        auto totalSize = [&](const std::map<std::string, std::map<StorePath, Info>> & versions)
+        {
+            uint64_t sum = 0;
+            for (auto & [_, paths] : versions)
+                for (auto & [path, _] : paths)
+                    sum += store->queryPathInfo(path)->narSize;
+            return sum;
+        };
+
+        auto beforeSize = totalSize(beforeVersions);
+        auto afterSize = totalSize(afterVersions);
+        auto sizeDelta = (int64_t) afterSize - (int64_t) beforeSize;
+        auto showDelta = abs(sizeDelta) >= 8 * 1024;
+
+        std::set<std::string> removed, unchanged;
+        for (auto & [version, _] : beforeVersions)
+            if (!afterVersions.count(version)) removed.insert(version); else unchanged.insert(version);
+
+        std::set<std::string> added;
+        for (auto & [version, _] : afterVersions)
+            if (!beforeVersions.count(version)) added.insert(version);
+
+        if (showDelta || !removed.empty() || !added.empty()) {
+            std::vector<std::string> items;
+            if (!removed.empty() || !added.empty())
+                items.push_back(fmt("%s → %s", showVersions(removed), showVersions(added)));
+            if (showDelta)
+                items.push_back(fmt("%s%+.1f KiB" ANSI_NORMAL, sizeDelta > 0 ? ANSI_RED : ANSI_GREEN, sizeDelta / 1024.0));
+            std::cout << fmt("%s%s: %s\n", indent, name, concatStringsSep(", ", items));
+        }
+    }
+}
+
+}
+
+using namespace nix;
+
 struct CmdDiffClosures : SourceExprCommand
 {
     std::string _before, _after;
@@ -85,49 +139,7 @@ struct CmdDiffClosures : SourceExprCommand
         auto beforePath = toStorePath(store, Realise::Outputs, operateOn, before);
         auto after = parseInstallable(store, _after);
         auto afterPath = toStorePath(store, Realise::Outputs, operateOn, after);
-
-        auto beforeClosure = getClosureInfo(store, beforePath);
-        auto afterClosure = getClosureInfo(store, afterPath);
-
-        std::set<std::string> allNames;
-        for (auto & [name, _] : beforeClosure) allNames.insert(name);
-        for (auto & [name, _] : afterClosure) allNames.insert(name);
-
-        for (auto & name : allNames) {
-            auto & beforeVersions = beforeClosure[name];
-            auto & afterVersions = afterClosure[name];
-
-            auto totalSize = [&](const std::map<std::string, std::map<StorePath, Info>> & versions)
-            {
-                uint64_t sum = 0;
-                for (auto & [_, paths] : versions)
-                    for (auto & [path, _] : paths)
-                        sum += store->queryPathInfo(path)->narSize;
-                return sum;
-            };
-
-            auto beforeSize = totalSize(beforeVersions);
-            auto afterSize = totalSize(afterVersions);
-            auto sizeDelta = (int64_t) afterSize - (int64_t) beforeSize;
-            auto showDelta = abs(sizeDelta) >= 8 * 1024;
-
-            std::set<std::string> removed, unchanged;
-            for (auto & [version, _] : beforeVersions)
-                if (!afterVersions.count(version)) removed.insert(version); else unchanged.insert(version);
-
-            std::set<std::string> added;
-            for (auto & [version, _] : afterVersions)
-                if (!beforeVersions.count(version)) added.insert(version);
-
-            if (showDelta || !removed.empty() || !added.empty()) {
-                std::vector<std::string> items;
-                if (!removed.empty() || !added.empty())
-                    items.push_back(fmt("%s → %s", showVersions(removed), showVersions(added)));
-                if (showDelta)
-                    items.push_back(fmt("%s%+.1f KiB" ANSI_NORMAL, sizeDelta > 0 ? ANSI_RED : ANSI_GREEN, sizeDelta / 1024.0));
-                std::cout << fmt("%s: %s\n", name, concatStringsSep(", ", items));
-            }
-        }
+        printClosureDiff(store, beforePath, afterPath, "");
     }
 };