From e1c0b41f7fef1dca952c6076f081b4b0a3883628 Mon Sep 17 00:00:00 2001
From: Benedikt Heine <bebe@bebehei.de>
Date: Tue, 7 Nov 2017 20:07:01 +0100
Subject: [PATCH] Use swaylock in wayland environment

---
 i3/lockscreen.sh | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/i3/lockscreen.sh b/i3/lockscreen.sh
index 8f79ec9..0c9d3f6 100755
--- a/i3/lockscreen.sh
+++ b/i3/lockscreen.sh
@@ -5,6 +5,12 @@
 LOCK_TIME=${LOCK_TIME:-3}
 LOCK_NOTIFY_TIME=${LOCK_NOTIFY_TIME:-15}
 
+if [ -n "${WAYLAND_DISPLAY}" ]; then
+	LOCK_CMD="swaylock"
+else
+	LOCK_CMD="i3lock"
+fi
+
 PROGNAME=$(basename $0)
 
 OUTPUT_IMAGE="/tmp/i3lock.png"
@@ -36,7 +42,9 @@ checkfull(){
 lock(){
 	scrot -z $OUTPUT_IMAGE
 	convert $OUTPUT_IMAGE -resize 20% -level 0%,100%,0.9 -blur 0x2 -resize 500% $OUTPUT_IMAGE
-	i3lock -i $OUTPUT_IMAGE -t
+	${LOCK_CMD} \
+	  -t \
+	  -i $OUTPUT_IMAGE
 	rm $OUTPUT_IMAGE
 }