Implemented json flag for `nix flake info`

3 files changed, 27 insertions, 3 deletions

diff --git a/src/nix/command.cc b/src/nix/command.cc
index 5967ab36c..e1e32aaae 100644
--- a/src/nix/command.cc
+++ b/src/nix/command.cc
@@ -27,6 +27,15 @@ void StoreCommand::run()
     run(getStore());
 }
 
+JsonFormattable::JsonFormattable()
+{
+    mkFlag()
+        .longName("json-formattable")
+        .shortName('j')
+        .description("output will be printed as json")
+        .handler([&]() { jsonFormatting = true; });
+}
+
 StorePathsCommand::StorePathsCommand(bool recursive)
     : recursive(recursive)
 {
diff --git a/src/nix/command.hh b/src/nix/command.hh
index b3248222e..5c2f8c304 100644
--- a/src/nix/command.hh
+++ b/src/nix/command.hh
@@ -26,6 +26,13 @@ private:
     std::shared_ptr<Store> _store;
 };
 
+struct JsonFormattable : virtual Command
+{
+    bool jsonFormatting = false;;
+
+    JsonFormattable();
+};
+
 struct Buildable
 {
     Path drvPath; // may be empty
diff --git a/src/nix/flake.cc b/src/nix/flake.cc
index 099425688..22e5b297c 100644
--- a/src/nix/flake.cc
+++ b/src/nix/flake.cc
@@ -4,6 +4,7 @@
 #include "shared.hh"
 #include "progress-bar.hh"
 #include "eval.hh"
+#include <nlohmann/json.hpp>
 
 using namespace nix;
 
@@ -33,7 +34,7 @@ struct CmdFlakeList : StoreCommand, MixEvalArgs
     }
 };
 
-struct CmdFlakeInfo : FlakeCommand
+struct CmdFlakeInfo : FlakeCommand, JsonFormattable
 {
     std::string name() override
     {
@@ -49,8 +50,15 @@ struct CmdFlakeInfo : FlakeCommand
     {
         auto evalState = std::make_shared<EvalState>(searchPath, store);
         nix::Flake flake = nix::getFlake(*evalState, FlakeRef(flakeUri));
-        std::cout << "Location:    " << flake.path << "\n";
-        std::cout << "Description: " << flake.description << "\n";
+        if (jsonFormatting) {
+            nlohmann::json j;
+            j["location"] = flake.path;
+            j["description"] = flake.description;
+            std::cout << j.dump(4) << std::endl;
+        } else {
+            std::cout << "Location:    " << flake.path << "\n";
+            std::cout << "Description: " << flake.description << "\n";
+        }
     }
 };