aboutsummaryrefslogtreecommitdiff
path: root/lib/mk-dev-shell.nix
blob: 87b9dbb3cf8e5fddb573dde3f5d5f47eb5695e05 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
{ pkgs
, tools ? []
, envVars ? {}
, shell ? null
}:

with pkgs.lib;

let
  mapAttrsToLines = mapping: attrs:
    pkgs.lib.concatStringsSep "\n" (pkgs.lib.mapAttrsToList mapping attrs);

  exportEnvVar = k: v: ''
    export ${escapeShellArg k}="${v}"
  '';

  exportEnvVarsHook = mapAttrsToLines exportEnvVar;

  fmt = rec {
    codeBlock = code: text: ''"\e[${code}m"${text}"\e[0m"'';
    keyword = codeBlock "1;36";
    section = codeBlock "4;35";
    printSectionTitle = title: ''echo -e "\n\n"${section title}"\n"'';
  };

  printEnvVar = k: v: ''
    echo -e ${fmt.keyword (escapeShellArg "$" + k)}: ${escapeShellArg v}
  '';

  printEnvVarsHook = envVars: ''
    ${fmt.printSectionTitle "DEVSHELL ENVIRONMENT VARIABLES"}
    ${mapAttrsToLines printEnvVar envVars}
  '';

  printToolsHook = tools: ''
    ${fmt.printSectionTitle "DEVSHELL TOOLS"}
    ls "${pkgs.symlinkJoin { name = "env"; paths = tools; }}/bin"
  '';

  # Use the default user shell instead of Bash
  startUserShellHook = ''
    $(${pkgs.finger_bsd}/bin/finger $USER \
      | ${pkgs.gnugrep}/bin/grep -oP 'Shell: \K.*')
  '';

in pkgs.mkShell {
  packages = tools;
  shellHook = ''
    ${exportEnvVarsHook envVars}

    ${optionalString (envVars != {}) (printEnvVarsHook envVars)}
    ${optionalString (tools != []) (printToolsHook tools)}
    echo

    ${if (shell != null) then shell else startUserShellHook}
    exit $?
  '';
}