Add command :NixShell

Add a command to call 'nix-shell' and then update the editor's
environment as if it was inside the new shell.

It's sometimes useful to fire up a nix-shell just to download a package
for using it once. It's annoying however when it's a library or any
other dependency of ':make', which can't be called from inside the
shell.

Example:

    :NixShell -p hello
    :term hello

There's no rollback command but it could be written if it's ever needed.
Calling the commande several times will nest environments the same way
as opening a shell inside a shell.

Author: Julow

plugin/nix.vim

@@ -22,4 +22,24 @@ function! nix#edit(attr)
   endif
 endfunction
 
+" Takes a list of lines of the form 'key=val'.
+function! nix#set_env(lines)
+  for l in a:lines
+    let [ key; val ] = split(l, "=", 1)
+    call setenv(key, join(val, "="))
+  endfor
+endfunction
+
+" See command :NixShell
+function! nix#add_pkg_to_env(args)
+  echo "nix-shell " . a:args
+  let l:env = systemlist("nix-shell --run env " . a:args)
+  call nix#set_env(l:env)
+  echo "Environment updated"
+endfunction
+
 command! -bang -nargs=* NixEdit call nix#edit(<q-args>)
+
+" Call 'nix-shell' with the given arguments and update the environment as if
+" we were inside the spawned shell.
+command! -nargs=* NixShell call nix#add_pkg_to_env(<q-args>)