NixOS · Jan 31, 2020 · Jan 31, 2020 · Jan 31, 2020
Showing with 89 additions and 56 deletions.

+34 −0 src/libutil/util.hh

+5 −10 src/nix-store/nix-store.cc

+37 −32 src/nix/flake.cc

+1 −6 src/nix/why-depends.cc

+4 −4 tests/dependencies.sh

+8 −4 tests/flakes.sh
diff --git a/src/libutil/util.hh b/src/libutil/util.hh
@@ -458,6 +458,13 @@ void ignoreException();
 #define ANSI_BLUE "\e[34;1m"
 
 
+/* Tree formatting. */
+constexpr char treeConn[] = "├───";
+constexpr char treeLast[] = "└───";
+constexpr char treeLine[] = "│   ";
+constexpr char treeNull[] = "    ";
+
+
 /* Truncate a string to 'width' printable characters. If 'filterAll'
    is true, all ANSI escape sequences are filtered out. Otherwise,
    some escape sequences (such as colour setting) are copied but not
@@ -583,4 +590,31 @@ extern PathFilter defaultPathFilter;
 AutoCloseFD createUnixDomainSocket(const Path & path, mode_t mode);
 
 
+// A Rust/Python-like enumerate() iterator adapter.
+// Borrowed from http://reedbeta.com/blog/python-like-enumerate-in-cpp17.
+template <typename T,
+          typename TIter = decltype(std::begin(std::declval<T>())),
+          typename = decltype(std::end(std::declval<T>()))>
+constexpr auto enumerate(T && iterable)
+{
+    struct iterator
+    {
+        size_t i;
+        TIter iter;
+        bool operator != (const iterator & other) const { return iter != other.iter; }
+        void operator ++ () { ++i; ++iter; }
+        auto operator * () const { return std::tie(i, *iter); }
+    };
+
+    struct iterable_wrapper
+    {
+        T iterable;
+        auto begin() { return iterator{ 0, std::begin(iterable) }; }
+        auto end() { return iterator{ 0, std::end(iterable) }; }
+    };
+
+    return iterable_wrapper{ std::forward<T>(iterable) };
+}
+
+
 }
diff --git a/src/nix-store/nix-store.cc b/src/nix-store/nix-store.cc
@@ -229,12 +229,6 @@ static StorePathSet maybeUseOutputs(const StorePath & storePath, bool useOutput,
 /* Some code to print a tree representation of a derivation dependency
    graph.  Topological sorting is used to keep the tree relatively
    flat. */
-
-const string treeConn = "+---";
-const string treeLine = "|   ";
-const string treeNull = "    ";
-
-
 static void printTree(const StorePath & path,
     const string & firstPad, const string & tailPad, StorePathSet & done)
 {
@@ -254,10 +248,11 @@ static void printTree(const StorePath & path,
     auto sorted = store->topoSortPaths(info->references);
     reverse(sorted.begin(), sorted.end());
 
-    for (auto i = sorted.begin(); i != sorted.end(); ++i) {
-        auto j = i; ++j;
-        printTree(*i, tailPad + treeConn,
-            j == sorted.end() ? tailPad + treeNull : tailPad + treeLine,
+    for (const auto &[n, i] : enumerate(sorted)) {
+        bool last = n + 1 == sorted.size();
+        printTree(i,
+            tailPad + (last ? treeLast : treeConn),
+            tailPad + (last ? treeNull : treeLine),
             done);
     }
 }

diff --git a/src/nix/flake.cc b/src/nix/flake.cc
@@ -110,37 +110,6 @@ static nlohmann::json flakeToJson(const Store & store, const Flake & flake)
     return j;
 }
 
-#if 0
-// FIXME: merge info CmdFlakeInfo?
-struct CmdFlakeDeps : FlakeCommand
-{
-    std::string description() override
-    {
-        return "list informaton about dependencies";
-    }
-
-    void run(nix::ref<nix::Store> store) override
-    {
-        auto evalState = getEvalState();
-
-        std::queue<LockedFlake> todo;
-        todo.push(lockFlake());
-
-        stopProgressBar();
-
-        while (!todo.empty()) {
-            auto lockedFlake = std::move(todo.front());
-            todo.pop();
-
-            for (auto & info : lockedFlake.flakeDeps) {
-                printFlakeInfo(*store, info.second.flake);
-                todo.push(info.second);
-            }
-        }
-    }
-};
-#endif
-
 struct CmdFlakeUpdate : FlakeCommand
 {
     std::string description() override
@@ -150,7 +119,6 @@ struct CmdFlakeUpdate : FlakeCommand
 
     void run(nix::ref<nix::Store> store) override
     {
-        auto evalState = getEvalState();
         lockFlake();
     }
 };
@@ -214,6 +182,42 @@ struct CmdFlakeInfo : FlakeCommand, MixJSON
     }
 };
 
+struct CmdFlakeListInputs : FlakeCommand, MixJSON
+{
+    std::string description() override
+    {
+        return "list flake inputs";
+    }
+
+    void run(nix::ref<nix::Store> store) override
+    {
+        auto flake = lockFlake();
+
+        stopProgressBar();
+
+        if (json)
+            std::cout << ((LockedInputs &) flake.lockFile).toJson() << "\n";
+        else {
+            std::cout << fmt("%s\n", flake.flake.resolvedRef);
+
+            std::function<void(const LockedInputs & inputs, const std::string & prefix)> recurse;
+
+            recurse = [&](const LockedInputs & inputs, const std::string & prefix)
+            {
+                for (const auto & [i, input] : enumerate(inputs.inputs)) {
+                    //auto tree2 = tree.child(i + 1 == inputs.inputs.size());
+                    bool last = i + 1 == inputs.inputs.size();
+                    std::cout << fmt("%s" ANSI_BOLD "%s" ANSI_NORMAL ": %s\n",
+                        prefix + (last ? treeLast : treeConn), input.first, input.second.ref);
+                    recurse(input.second, prefix + (last ? treeNull : treeLine));
+                }
+            };
+
+            recurse(flake.lockFile, "");
+        }
+    }
+};
+
 struct CmdFlakeCheck : FlakeCommand
 {
     bool build = true;
@@ -685,6 +689,7 @@ struct CmdFlake : virtual MultiCommand, virtual Command
                 {"list", []() { return make_ref<CmdFlakeList>(); }},
                 {"update", []() { return make_ref<CmdFlakeUpdate>(); }},
                 {"info", []() { return make_ref<CmdFlakeInfo>(); }},
+                {"list-inputs", []() { return make_ref<CmdFlakeListInputs>(); }},
                 {"check", []() { return make_ref<CmdFlakeCheck>(); }},
                 {"add", []() { return make_ref<CmdFlakeAdd>(); }},
                 {"remove", []() { return make_ref<CmdFlakeRemove>(); }},

diff --git a/src/nix/why-depends.cc b/src/nix/why-depends.cc
@@ -143,11 +143,6 @@ struct CmdWhyDepends : SourceExprCommand
            and `dependency`. */
         std::function<void(Node &, const string &, const string &)> printNode;
 
-        const string treeConn = "╠═══";
-        const string treeLast = "╚═══";
-        const string treeLine = "║   ";
-        const string treeNull = "    ";
-
         struct BailOut { };
 
         printNode = [&](Node & node, const string & firstPad, const string & tailPad) {
@@ -157,7 +152,7 @@ struct CmdWhyDepends : SourceExprCommand
             std::cout << fmt("%s%s%s%s" ANSI_NORMAL "\n",
                 firstPad,
                 node.visited ? "\e[38;5;244m" : "",
-                firstPad != "" ? "=> " : "",
+                firstPad != "" ? "→ " : "",
                 pathS);
 
             if (node.path == dependencyPath && !all

diff --git a/tests/dependencies.sh b/tests/dependencies.sh
@@ -6,7 +6,7 @@ drvPath=$(nix-instantiate dependencies.nix)
 
 echo "derivation is $drvPath"
 
-nix-store -q --tree "$drvPath" | grep '   +---.*builder1.sh'
+nix-store -q --tree "$drvPath" | grep '│   └───.*builder1.sh'
 
 # Test Graphviz graph generation.
 nix-store -q --graph "$drvPath" > $TEST_ROOT/graph
@@ -22,9 +22,9 @@ nix-store -q --graph "$outPath" > $TEST_ROOT/graph
 if test -n "$dot"; then
     # Does it parse?
     $dot < $TEST_ROOT/graph
-fi    
+fi
 
-nix-store -q --tree "$outPath" | grep '+---.*dependencies-input-2'
+nix-store -q --tree "$outPath" | grep '───.*dependencies-input-2'
 
 echo "output path is $outPath"
 
@@ -49,4 +49,4 @@ nix-store -q --referrers-closure "$input2OutPath" | grep "$outPath"
 
 # Check that the derivers are set properly.
 test $(nix-store -q --deriver "$outPath") = "$drvPath"
-nix-store -q --deriver "$input2OutPath" | grep -q -- "-input-2.drv" 
+nix-store -q --deriver "$input2OutPath" | grep -q -- "-input-2.drv"
diff --git a/tests/flakes.sh b/tests/flakes.sh
@@ -114,7 +114,7 @@ cat > $registry <<EOF
 EOF
 
 # Test 'nix flake list'.
-(( $(nix flake list | wc -l) == 6 ))
+[[ $(nix flake list | wc -l) == 6 ]]
 
 # Test 'nix flake info'.
 nix flake info flake1 | grep -q 'URL: .*flake1.*'
@@ -361,11 +361,11 @@ nix build -o $TEST_ROOT/result flake4/removeXyzzy#sth
 
 # Testing the nix CLI
 nix flake add flake1 flake3
-(( $(nix flake list | wc -l) == 7 ))
+[[ $(nix flake list | wc -l) == 7 ]]
 nix flake pin flake1
-(( $(nix flake list | wc -l) == 7 ))
+[[ $(nix flake list | wc -l) == 7 ]]
 nix flake remove flake1
-(( $(nix flake list | wc -l) == 6 ))
+[[ $(nix flake list | wc -l) == 6 ]]
 
 # Test 'nix flake init'.
 (cd $flake7Dir && nix flake init)
@@ -617,3 +617,7 @@ nix flake update $flake3Dir
 
 nix flake update $flake3Dir --update-input flake2/flake1
 [[ $(jq .inputs.flake2.inputs.flake1.url $flake3Dir/flake.lock) =~ flake1.*rev=$hash2 ]]
+
+# Test 'nix flake list-inputs'.
+[[ $(nix flake list-inputs $flake3Dir | wc -l) == 5 ]]
+nix flake list-inputs $flake3Dir --json | jq .