From 57e90afb80ae23a61356aa885df69b5b5ee957ec Mon Sep 17 00:00:00 2001
From: John Ericson <John.Ericson@Obsidian.Systems>
Date: Fri, 12 Feb 2021 21:51:36 +0000
Subject: [PATCH] New "indexed" installable syntax: `<drvPath>!<outputName>`

Being conservative and only doing a single output name for now.
---
 src/libcmd/installables.cc     | 42 +++++++++++++++++++++++++++++++++-
 src/nix/nix.md                 | 10 ++++++++
 tests/build-explicit-output.sh | 17 ++++++++++++++
 tests/build.sh                 |  5 ++--
 tests/local.mk                 |  1 +
 5 files changed, 72 insertions(+), 3 deletions(-)
 create mode 100644 tests/build-explicit-output.sh

diff --git a/src/libcmd/installables.cc b/src/libcmd/installables.cc
index b68c5f6a75f6..a74fee882b51 100644
--- a/src/libcmd/installables.cc
+++ b/src/libcmd/installables.cc
@@ -349,6 +349,31 @@ struct InstallableStorePath : Installable
     }
 };
 
+struct InstallableIndexedStorePath : Installable
+{
+    ref<Store> store;
+    BuildableReqFromDrv req;
+
+    InstallableIndexedStorePath(ref<Store> store, BuildableReqFromDrv && req)
+        : store(store), req(std::move(req))
+    { }
+
+    std::string what() override
+    {
+        return req.to_string(*store);
+    }
+
+    Buildables toBuildables() override
+    {
+        std::map<std::string, std::optional<StorePath>> outputs;
+        for (auto & output : req.outputs)
+            outputs.insert_or_assign(output, std::nullopt);
+        return {
+            Buildable { BuildableFromDrv { req.drvPath, std::move(outputs) } }
+        };
+    }
+};
+
 Buildables InstallableValue::toBuildables()
 {
     Buildables res;
@@ -638,7 +663,22 @@ std::vector<std::shared_ptr<Installable>> SourceExprCommand::parseInstallables(
                 ex = std::current_exception();
             }
 
-            if (s.find('/') != std::string::npos) {
+            auto found = s.rfind('!');
+            if (found != std::string::npos) {
+                try {
+                    result.push_back(std::make_shared<InstallableIndexedStorePath>(
+                        store,
+                        BuildableReqFromDrv::parse(*store, s)));
+                    continue;
+                } catch (BadStorePath &) {
+                } catch (...) {
+                    if (!ex)
+                        ex = std::current_exception();
+                }
+            }
+
+            found = s.find('/');
+            if (found != std::string::npos) {
                 try {
                     result.push_back(std::make_shared<InstallableStorePath>(store, store->followLinksToStorePath(s)));
                     continue;
diff --git a/src/nix/nix.md b/src/nix/nix.md
index d10de7c01b11..22cc9d4767df 100644
--- a/src/nix/nix.md
+++ b/src/nix/nix.md
@@ -94,6 +94,16 @@ the Nix store. Here are the recognised types of installables:
   If you want to operate on the store derivation itself, pass the
   `--derivation` flag.
 
+* **Indexed store derivations**: `/nix/store/p7gp6lxdg32h4ka1q398wd9r2zkbbz2v-hello-2.10.drv!out`
+
+  Store derivations can be indexed with a specific output name. This
+  allows finer control versus just specifying a derivation (without
+  `--derivation`) and getting all the outputs.
+
+  This is especially useful for (currently unstable) floating content
+  addressed derivations, which do not have precomputed output paths that
+  can be used instead.
+
 * **Nix attributes**: `--file /path/to/nixpkgs hello`
 
   When the `-f` / `--file` *path* option is given, installables are
diff --git a/tests/build-explicit-output.sh b/tests/build-explicit-output.sh
new file mode 100644
index 000000000000..17930c2c09ab
--- /dev/null
+++ b/tests/build-explicit-output.sh
@@ -0,0 +1,17 @@
+source common.sh
+
+drv=$(nix eval -f multiple-outputs.nix --raw a.drvPath)
+if nix build "$drv!not-an-output" --json; then
+    fail "'not-an-output' should fail to build"
+fi
+
+nix build "$drv!first" --json | jq --exit-status '
+  (.[0] |
+    (.drvPath | match(".*multiple-outputs-a.drv")) and
+    (.outputs |
+      .first and
+	  (has("second") | not)))
+'
+# TODO use
+#      (.first | match(".*multiple-outputs-a-first")) and
+# once we make it put the result paths in the buildables.
diff --git a/tests/build.sh b/tests/build.sh
index aa54b88eb8ab..5a2819336770 100644
--- a/tests/build.sh
+++ b/tests/build.sh
@@ -4,8 +4,9 @@ expectedJSONRegex='\[\{"drvPath":".*multiple-outputs-a.drv","outputs":\{"first":
 nix build -f multiple-outputs.nix --json a.all b.all | jq --exit-status '
   (.[0] |
     (.drvPath | match(".*multiple-outputs-a.drv")) and
-    (.outputs.first | match(".*multiple-outputs-a-first")) and
-    (.outputs.second | match(".*multiple-outputs-a-second")))
+    (.outputs |
+      (.first | match(".*multiple-outputs-a-first")) and
+      (.second | match(".*multiple-outputs-a-second"))))
   and (.[1] |
     (.drvPath | match(".*multiple-outputs-b.drv")) and
     (.outputs.out | match(".*multiple-outputs-b")))
diff --git a/tests/local.mk b/tests/local.mk
index 747f38a29433..f1aa52006c82 100644
--- a/tests/local.mk
+++ b/tests/local.mk
@@ -42,6 +42,7 @@ nix_tests = \
   flakes.sh \
   text-hashed-output.sh \
   build.sh \
+  build-explicit-output.sh \
   compute-levels.sh \
   ca/build.sh \
   ca/substitute.sh \