pkgs/disable-sleep/default.nix


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

{
  writeShellApplication,
  human-sleep,
}:
writeShellApplication {
  name = "disable-sleep";

  runtimeInputs = [human-sleep];

  text = ''
    set -ue

    if [ "$(id -u)" -ne 0 ]; then
            echo "Acquiring root access..."
            exec sudo "$0" "$@"
    fi

    cleanup() {
            echo "Re-enabling sleep..."
            pmset -a disablesleep 0
    }

    echo "Disabling sleep..."
    pmset -a disablesleep 1
    trap cleanup EXIT

    echo "Waiting..."
    human-sleep "$@"
  '';
}