Added wrapper functions for devenv shell, nix-shell and nix develop commands that starts the environment in zsh automatically

1 files changed, 27 insertions, 0 deletions

diff --git a/modules/tools/zsh/nix-aliases.sh b/modules/tools/zsh/nix-aliases.sh
new file mode 100644
index 0000000..0a2f453
--- /dev/null
+++ b/modules/tools/zsh/nix-aliases.sh
@@ -0,0 +1,27 @@
+nix-shell() {
+  for arg in "$@"; do
+    if [[ "$arg" == "--run" || "$arg" == "--command" ]]; then
+      command nix-shell "$@"
+      return
+    fi
+  done
+
+  command nix-shell "$@" --run "$SHELL"
+}
+
+nix() {
+  if [[ "$1" == "develop" ]]; then
+    local args=("${@:2}")
+
+    for arg in "${args[@]}"; do
+      if [[ "$arg" == "-c" || "$arg" == "--command" ]]; then
+        command nix "$@"
+        return
+      fi
+    done
+
+    command nix develop "${args[@]}" -c "$SHELL"
+  else
+    command nix "$@"
+  fi
+}