From e007f367bd605ad14ddf84d1d5ad611aa427d338 Mon Sep 17 00:00:00 2001
From: Nick Van den Broeck <nick.van.den.broeck666@gmail.com>
Date: Thu, 21 Feb 2019 06:53:01 +0100
Subject: [PATCH] Fixed minor things

---
 src/libexpr/primops/flake.cc |  6 +++---
 src/nix/build.cc             | 13 ++++++-------
 src/nix/command.hh           | 21 +++++++++++++++------
 src/nix/flake.cc             | 11 ++++++++---
 src/nix/installables.cc      |  1 +
 5 files changed, 33 insertions(+), 19 deletions(-)

diff --git a/src/libexpr/primops/flake.cc b/src/libexpr/primops/flake.cc
index df0845c24..48a036875 100644
--- a/src/libexpr/primops/flake.cc
+++ b/src/libexpr/primops/flake.cc
@@ -299,14 +299,14 @@ FlakeRegistry updateLockFile(EvalState & evalState, FlakeRef & flakeRef)
 void updateLockFile(EvalState & state, std::string path)
 {
     // 'path' is the path to the local flake repo.
-    FlakeRef flakeRef = FlakeRef(path);
+    FlakeRef flakeRef = FlakeRef("file://" + path);
     if (std::get_if<FlakeRef::IsGit>(&flakeRef.data)) {
         FlakeRegistry newLockFile = updateLockFile(state, flakeRef);
         writeRegistry(newLockFile, path + "/flake.lock");
     } else if (std::get_if<FlakeRef::IsGitHub>(&flakeRef.data)) {
-        throw UsageError("You can only update local flakes, not flakes on GitHub.");
+        throw UsageError("you can only update local flakes, not flakes on GitHub");
     } else {
-       throw UsageError("You can only update local flakes, not flakes through their FlakeId.");
+        throw UsageError("you can only update local flakes, not flakes through their FlakeId");
     }
 }
 
diff --git a/src/nix/build.cc b/src/nix/build.cc
index 12ef08679..5ab22e26c 100644
--- a/src/nix/build.cc
+++ b/src/nix/build.cc
@@ -11,7 +11,7 @@ struct CmdBuild : MixDryRun, InstallablesCommand
 {
     Path outLink = "result";
 
-    std::optional<std::string> flakeUri = std::nullopt;
+    std::optional<std::string> gitRepo = std::nullopt;
 
     CmdBuild()
     {
@@ -28,9 +28,9 @@ struct CmdBuild : MixDryRun, InstallablesCommand
             .set(&outLink, Path(""));
 
         mkFlag()
-            .longName("flake")
-            .description("update lock file of given flake")
-            .dest(&flakeUri);
+            .longName("update-lock-file")
+            .description("update the lock file")
+            .dest(&gitRepo);
     }
 
     std::string name() override
@@ -78,9 +78,8 @@ struct CmdBuild : MixDryRun, InstallablesCommand
                     }
         }
 
-        if (flakeUri) {
-            updateLockFile(*evalState, *flakeUri);
-        }
+        if(gitRepo)
+            updateLockFile(*evalState, *gitRepo);
     }
 };
 
diff --git a/src/nix/command.hh b/src/nix/command.hh
index b3248222e..c58d5d56e 100644
--- a/src/nix/command.hh
+++ b/src/nix/command.hh
@@ -34,15 +34,24 @@ struct Buildable
 
 typedef std::vector<Buildable> Buildables;
 
+struct GitRepoCommand : virtual Args
+{
+    std::string gitPath = absPath(".");
+
+    GitRepoCommand ()
+    {
+        expectArg("git-path", &gitPath, true);
+    }
+};
+
 struct FlakeCommand : virtual Args, StoreCommand, MixEvalArgs
 {
-  std::string flakeUri;
+    std::string flakeUri;
 
-public:
-  FlakeCommand()
-  {
-    expectArg("flake-uri", &flakeUri);
-  }
+    FlakeCommand()
+    {
+        expectArg("flake-uri", &flakeUri);
+    }
 };
 
 struct Installable
diff --git a/src/nix/flake.cc b/src/nix/flake.cc
index 6cef38936..a5a1d34db 100644
--- a/src/nix/flake.cc
+++ b/src/nix/flake.cc
@@ -34,7 +34,7 @@ struct CmdFlakeList : StoreCommand, MixEvalArgs
     }
 };
 
-struct CmdFlakeUpdate : FlakeCommand
+struct CmdFlakeUpdate : StoreCommand, GitRepoCommand, MixEvalArgs
 {
     std::string name() override
     {
@@ -51,7 +51,12 @@ struct CmdFlakeUpdate : FlakeCommand
         auto evalState = std::make_shared<EvalState>(searchPath, store);
 
         if (flakeUri == "") flakeUri = absPath("./flake.nix");
-        updateLockFile(*evalState, flakeUri);
+        int result = updateLockFile(*evalState, flakeUri);
+        if (result == 1) {
+            std::cout << "You can only update local flakes, not flakes on GitHub.\n";
+        } else if (result == 2) {
+            std::cout << "You can only update local flakes, not flakes through their FlakeId.\n";
+        }
     }
 };
 
@@ -77,8 +82,8 @@ struct CmdFlakeInfo : FlakeCommand, MixJSON
             j["description"] = flake.description;
             std::cout << j.dump(4) << std::endl;
         } else {
-            std::cout << "Location:    " << flake.path << "\n";
             std::cout << "Description: " << flake.description << "\n";
+            std::cout << "Location:    " << flake.path << "\n";
         }
     }
 };
diff --git a/src/nix/installables.cc b/src/nix/installables.cc
index 0453c72c2..21e9e73b8 100644
--- a/src/nix/installables.cc
+++ b/src/nix/installables.cc
@@ -234,6 +234,7 @@ Buildables build(ref<Store> store, RealiseMode mode,
     PathSet pathsToBuild;
 
     for (auto & i : installables) {
+        std::cout << i->what() << std::endl;
         for (auto & b : i->toBuildables()) {
             if (b.drvPath != "") {
                 StringSet outputNames;