From 2672a28bb4ba38a8358c306b8af4897c804b690d Mon Sep 17 00:00:00 2001
From: Eelco Dolstra <edolstra@gmail.com>
Date: Thu, 27 Feb 2020 15:17:37 +0100
Subject: [PATCH] nix dev-shell: Add --command option

Note: like 'nix run', and unlike 'nix-shell', this takes an argv
vector rather than a shell command. So

  nix dev-shell -c 'echo $PATH'

doesn't work. Instead you need to do

  nix dev-shell -c bash -c 'echo $PATH'
---
 src/nix/shell.cc | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/src/nix/shell.cc b/src/nix/shell.cc
index 6b3d02b6b..bee0bddcc 100644
--- a/src/nix/shell.cc
+++ b/src/nix/shell.cc
@@ -232,6 +232,22 @@ struct Common : InstallableCommand, MixProfile
 
 struct CmdDevShell : Common, MixEnvironment
 {
+    std::vector<std::string> command;
+
+    CmdDevShell()
+    {
+        mkFlag()
+            .longName("command")
+            .shortName('c')
+            .description("command and arguments to be executed insted of an interactive shell")
+            .labels({"command", "args"})
+            .arity(ArityAny)
+            .handler([&](std::vector<std::string> ss) {
+                if (ss.empty()) throw UsageError("--command requires at least one argument");
+                command = ss;
+            });
+    }
+
     std::string description() override
     {
         return "run a bash shell that provides the build environment of a derivation";
@@ -270,6 +286,13 @@ struct CmdDevShell : Common, MixEnvironment
 
         ss << fmt("rm -f '%s'\n", rcFilePath);
 
+        if (!command.empty()) {
+            std::vector<std::string> args;
+            for (auto s : command)
+                args.push_back(shellEscape(s));
+            ss << fmt("exec %s\n", concatStringsSep(" ", args));
+        }
+
         writeFull(rcFileFd.get(), ss.str());
 
         stopProgressBar();