mkDevShell: split env var setting and printing helpers

1 files changed, 17 insertions, 5 deletions

diff --git a/lib/mk-dev-shell.nix b/lib/mk-dev-shell.nix
index e6e916a..0e2a265 100644
--- a/lib/mk-dev-shell.nix
+++ b/lib/mk-dev-shell.nix
@@ -9,17 +9,26 @@ with pkgs.lib;
 let
   exportEnvVar = k: v: ''
     export ${escapeShellArg k}="${v}"
+  '';
+
+  exportEnvVarsHook = envVars: ''
+    ${pkgs.lib.concatStringsSep "\n" (
+      pkgs.lib.mapAttrsToList exportEnvVar envVars
+    )}
+  '';
+
+  printEnvVar = k: v: ''
     echo ${escapeShellArg k}=${escapeShellArg v}
   '';
 
-  envVarsHook = envVars: ''
+  printEnvVarsHook = envVars: ''
     echo -e "\nDEVSHELL ENVIRONMENT VARIABLES:"
     ${pkgs.lib.concatStringsSep "\n" (
-      pkgs.lib.mapAttrsToList exportEnvVar envVars
+      pkgs.lib.mapAttrsToList printEnvVar envVars
     )}
   '';
 
-  listToolsHook = tools: ''
+  printToolsHook = tools: ''
     echo -e "\nDEVSHELL COMMANDS:"
     ls "${pkgs.symlinkJoin { name = "env"; paths = tools; }}/bin"
   '';
@@ -33,8 +42,11 @@ let
 in pkgs.mkShell {
   packages = tools;
   shellHook = ''
-    ${optionalString (envVars != {}) (envVarsHook envVars)}
-    ${optionalString (tools != []) (listToolsHook tools)}
+    ${exportEnvVarsHook envVars}
+
+    ${optionalString (envVars != {}) (printEnvVarsHook envVars)}
+    ${optionalString (tools != []) (printToolsHook tools)}
+
     ${if (shell != null) then shell else startUserShellHook}
     exit $?
   '';