From 0e32b32fa3655e8a3e028ebb5ac838164a606132 Mon Sep 17 00:00:00 2001
From: Eelco Dolstra <edolstra@gmail.com>
Date: Wed, 29 May 2019 20:57:08 +0200
Subject: [PATCH] nix flake check: Check defaultPackage, devShell and packages

---
 src/nix/flake.cc | 38 ++++++++++++++++++++++++++------------
 1 file changed, 26 insertions(+), 12 deletions(-)

diff --git a/src/nix/flake.cc b/src/nix/flake.cc
index f0231e263..19e97aed9 100644
--- a/src/nix/flake.cc
+++ b/src/nix/flake.cc
@@ -262,6 +262,19 @@ struct CmdFlakeCheck : FlakeCommand, MixJSON
         auto state = getEvalState();
         auto flake = resolveFlake();
 
+        auto checkDerivation = [&](const std::string & attrPath, Value & v) {
+            try {
+                auto drvInfo = getDerivation(*state, v, false);
+                if (!drvInfo)
+                    throw Error("flake attribute '%s' is not a derivation", attrPath);
+                // FIXME: check meta attributes
+                return drvInfo->queryDrvPath();
+            } catch (Error & e) {
+                e.addPrefix(fmt("while checking flake attribute '" ANSI_BOLD "%s" ANSI_NORMAL "':\n", attrPath));
+                throw;
+            }
+        };
+
         PathSet drvPaths;
 
         {
@@ -281,20 +294,21 @@ struct CmdFlakeCheck : FlakeCommand, MixJSON
 
                         if (name == "checks") {
                             state->forceAttrs(vProvide);
-                            for (auto & aCheck : *vProvide.attrs) {
-                                try {
-                                    auto drvInfo = getDerivation(*state, *aCheck.value, false);
-                                    if (!drvInfo)
-                                        throw Error("flake output 'check.%s' is not a derivation", aCheck.name);
-                                    drvPaths.insert(drvInfo->queryDrvPath());
-                                    // FIXME: check meta attributes?
-                                } catch (Error & e) {
-                                    e.addPrefix(fmt("while checking flake check '" ANSI_BOLD "%s" ANSI_NORMAL "':\n", aCheck.name));
-                                    throw;
-                                }
-                            }
+                            for (auto & aCheck : *vProvide.attrs)
+                                drvPaths.insert(checkDerivation(
+                                        name + "." + (std::string) aCheck.name, *aCheck.value));
                         }
 
+                        else if (name == "packages") {
+                            state->forceAttrs(vProvide);
+                            for (auto & aCheck : *vProvide.attrs)
+                                checkDerivation(
+                                    name + "." + (std::string) aCheck.name, *aCheck.value);
+                        }
+
+                        else if (name == "defaultPackage" || name == "devShell")
+                            checkDerivation(name, vProvide);
+
                     } catch (Error & e) {
                         e.addPrefix(fmt("while checking flake output '" ANSI_BOLD "%s" ANSI_NORMAL "':\n", name));
                         throw;