#!/bin/sh
lock() {

# Add a small delay to prevent suspend races

i3lock && sleep 1

}

case "$1" in
           lock)
                  lock
                  ;; 
           logout)
                  i3-msg exit
                  ;; 
           suspend)
                  lock && systemctl suspend
                  ;; 
           reboot)
                  systemctl reboot
                  ;; 
           shutdown)
                  systemctl poweroff
                  ;; 
          *)
             echo "Usage: $0 {lock | logout | suspend | reboot | shutdown}"
             exit 2
esac

exit 0