Add completion for --update-input

1 files changed, 6 insertions, 1 deletions

diff --git a/src/nix/flake.cc b/src/nix/flake.cc
index 33bcc9604..57ce996cd 100644
--- a/src/nix/flake.cc
+++ b/src/nix/flake.cc
@@ -20,7 +20,7 @@
 using namespace nix;
 using namespace nix::flake;
 
-class FlakeCommand : virtual Args, public EvalCommand, public MixFlakeOptions
+class FlakeCommand : virtual Args, public MixFlakeOptions
 {
     std::string flakeUrl = ".";
 
@@ -53,6 +53,11 @@ public:
     {
         return flake::lockFlake(*getEvalState(), getFlakeRef(), lockFlags);
     }
+
+    std::optional<FlakeRef> getFlakeRefForCompletion() override
+    {
+        return getFlakeRef();
+    }
 };
 
 static void printFlakeInfo(const Store & store, const Flake & flake)