From a876fe32b44dac8d6db23e7cd6b6f96b75ba0bbb Mon Sep 17 00:00:00 2001
From: Linnnus <linnnus@users.noreply.github.com>
Date: Sat, 16 Nov 2024 12:46:23 +0100
Subject: pkgs: Add disable-sleep

---
 pkgs/disable-sleep/default.nix | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 pkgs/disable-sleep/default.nix

(limited to 'pkgs/disable-sleep/default.nix')

diff --git a/pkgs/disable-sleep/default.nix b/pkgs/disable-sleep/default.nix
new file mode 100644
index 0000000..07c3bb8
--- /dev/null
+++ b/pkgs/disable-sleep/default.nix
@@ -0,0 +1,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 "$@"
+  '';
+}
-- 
cgit v1.2.3