Auto suspend after 30 seconds

Also add a nim version.

2 files changed, 84 insertions, 3 deletions

diff --git a/linux-bin/zzz.c b/linux-bin/zzz.c
index 8bc3c22..479dd25 100644
--- a/linux-bin/zzz.c
+++ b/linux-bin/zzz.c
@@ -4,8 +4,10 @@
 #include <fcntl.h>
 #include <string.h>
 #include <sys/types.h>
+#include <sys/wait.h>
 
 #define POWER_STATE_F "/sys/power/state"
+#define SLEEP_TIMEOUT 30
 
 void
 die(const char *msg)
@@ -34,8 +36,6 @@ suspend()
 	n = write(fh, cmd, 3);
 	if (n == -1)
 		err("Writing to "POWER_STATE_F);
-	
-	exit(0);
 }
 
 int
@@ -84,7 +84,19 @@ main(int argc, char **argv)
 		execlp(lockProg, NULL);
 		err("execlp");
 	default:
-		suspend();
+		for(;;) {
+			ret = 0;
+			ret = waitpid(pid, NULL, WNOHANG);
+
+			if (ret == 0) {
+				suspend();
+				sleep(SLEEP_TIMEOUT);
+				continue;
+			}
+
+			puts("exiting");
+			exit(0);
+		}
 	}
 
 }
diff --git a/linux-bin/zzz.nim b/linux-bin/zzz.nim
new file mode 100644
index 0000000..eac2797
--- /dev/null
+++ b/linux-bin/zzz.nim
@@ -0,0 +1,69 @@
+# Just a copy of the C version.
+# I don't actually use this at the moment, but it was a nice small
+# introduction of using nim for something.
+import std/strformat
+import std/envvars
+import std/os
+import posix
+import posix_utils as putils
+
+let power_state_f = "/sys/power/state"
+
+proc suspend() =
+    writeFile(power_state_f, "mem")
+
+let uid = posix.getuid()
+let gid = posix.getgid()
+let euid = posix.geteuid()
+
+var lockProg = "slock"
+let lockProgWl = "swaylock"
+
+if euid != 0:
+    echo "This program must be run with setuid"
+    system.quit(1)
+
+if getEnv("XDG_SESSION_TYPE") == "wayland":
+    lockProg = lockProgWl
+
+for i in 1..os.paramCount():
+    if paramStr(i) == "-l":
+        if i+1 <= os.paramCount():
+            lockProg = paramStr(i + 1)
+        else:
+            echo "-l requires an argument"
+            system.quit(1)
+    else:
+        let s = paramStr(i)
+        echo &"unknown argument {s}"
+        system.quit(1)
+
+var pid = posix.fork()
+if pid == -1:
+    echo "unknown error"
+    system.quit(1)
+elif pid == 0:
+    var ret = posix.setuid(uid)
+    if ret == -1:
+        echo "Failed to setuid"
+        system.quit(1)
+    ret = posix.setgid(gid)
+    if ret == -1:
+        echo "Failed to setuid"
+        system.quit(1)
+
+    ret = execlp(lockprog, "")
+    echo &"excelp: {ret}"
+else:
+    suspend()
+    while true:
+        var info : cint
+        let ret = waitpid(pid, info, WNOHANG)
+
+        if ret == 0:
+            suspend()
+            sleep(10000)
+            continue
+
+        echo "exiting"
+        system.quit(0)