#!/bin/sh

if [ ! -t 0 ]; then
	set -- "$(awk '{print tolower($2)}')"
elif [ $# -eq 0 ]; then
	printf '%s\n' '󰌾 Lock' '󰍃 Logout' '⏾ Suspend' '󰋊 Hibernate' '󰜉 Reboot' '󰐥 Poweroff'
	exit 0
fi

case "$1" in
lock)
	loginctl lock-session "${XDG_SESSION_ID-}"
	;;
logout)
	loginctl terminate-session "${XDG_SESSION_ID-}"
	;;
suspend)
	systemctl suspend
	;;
hibernate)
	systemctl hibernate
	;;
reboot)
	systemctl reboot
	;;
poweroff)
	systemctl poweroff
	;;
*)
	echo "Invalid argument: $*"
	exit 1
	;;
esac