Modify the logic of lock script

[max/saltfiles.git] / states / sway / files / lock

diff --git a/states/sway/files/lock b/states/sway/files/lock
index 75498371adad556c7610e686b82b84d51c1508de..050e67a7c308afbd5cbe6e66c210f5641c14b32e 100644 (file)
--- a/states/sway/files/lock
+++ b/states/sway/files/lock
@@ -1,22 +1,33 @@
 #!/bin/sh
 
 NAG_MSG="Smartcard is still plugged in, not locking! \
-Lock again to force locking."
+Remove now to lock."
+NOLOCK_MSG="Smartcard was not removed, did not lock!"
 
 LOCK_CMD="swaylock "
 LOCK_CMD+=" -i HDMI-A-2:~/Pictures/lockscreen_1080_1.png"
 LOCK_CMD+=" -i HDMI-A-3:~/Pictures/lockscreen_1080_2.png"
 
-if [ $(pgrep -c lock.{{ grains['host'] }}) -gt 1 ]; then
-       pkill -f "$NAG_MSG" 
-       $LOCK_CMD
+if [ $(pgrep -c lock.magnesium) -gt 1 ]; then
+       exit
 else
        gpg --card-status > /dev/null
        if [ $? -ne 2 ]; then
-               gpg --card-status
-               echo $?
-               swaynag -m "$NAG_MSG" --type warning
-       else
-               $LOCK_CMD
+               swaynag -m "$NAG_MSG" --type warning&
+
+               counter=0
+               while [ $? -ne 2 ]; do
+                       if [ $counter -gt 20 ]; then
+                               pkill -f "$NAG_MSG"
+                               swaynag -m "$NOLOCK_MSG" --type error&
+                               exit
+                       fi
+
+                       sleep 1
+                       counter=$(bc <<<"${counter}+1")
+                       gpg --card-status > /dev/null
+               done
+               pkill -f "$NAG_MSG"
        fi
+       $LOCK_CMD
 fi