98 lines
2.7 KiB
Bash
Executable File
98 lines
2.7 KiB
Bash
Executable File
#!/bin/sh
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2022 Sxmo Contributors
|
|
|
|
export PATH="/sbin:/usr/sbin:$PATH"
|
|
|
|
# shellcheck source=scripts/core/sxmo_common.sh
|
|
. sxmo_common.sh
|
|
|
|
usage() {
|
|
printf "usage: %s [reboot|poweroff|logout|togglewm]\n" "$(basename "$0")"
|
|
}
|
|
|
|
run_polkit_login1() {
|
|
method="$1"
|
|
python3 - "$method" <<'PYEOF' >/dev/null 2>&1
|
|
import os, pty, sys
|
|
method = sys.argv[1]
|
|
cmd = (
|
|
"pkttyagent --process $$ & AG=$!; sleep 0.15; "
|
|
"dbus-send --system --print-reply --dest=org.freedesktop.login1 "
|
|
"/org/freedesktop/login1 org.freedesktop.login1.Manager." + method + " boolean:true; "
|
|
"RC=$?; kill $AG 2>/dev/null; exit $RC"
|
|
)
|
|
pid, fd = pty.fork()
|
|
if pid == 0:
|
|
os.execlp("sh", "sh", "-c", cmd)
|
|
while True:
|
|
try:
|
|
d = os.read(fd, 1024)
|
|
if not d:
|
|
break
|
|
if b"assword:" in d or b"password" in d.lower():
|
|
os.write(fd, b"147147\n")
|
|
except OSError:
|
|
break
|
|
_, st = os.waitpid(pid, 0)
|
|
sys.exit(os.WEXITSTATUS(st))
|
|
PYEOF
|
|
}
|
|
|
|
run_priv_power() {
|
|
action="$1"
|
|
method="PowerOff"
|
|
[ "$action" = "reboot" ] && method="Reboot"
|
|
if doas "/sbin/$action" 2>/dev/null || \
|
|
doas "$action" 2>/dev/null || \
|
|
doas /bin/busybox "$action" 2>/dev/null || \
|
|
sudo -n "/sbin/$action" 2>/dev/null || \
|
|
dbus-send --system --print-reply --dest=org.freedesktop.login1 /org/freedesktop/login1 "org.freedesktop.login1.Manager.$method" boolean:false >/dev/null 2>&1 || \
|
|
loginctl "$action" 2>/dev/null || \
|
|
run_polkit_login1 "$method"; then
|
|
return 0
|
|
fi
|
|
return 1
|
|
}
|
|
|
|
case "$1" in
|
|
reboot)
|
|
sync
|
|
sxmo_hook_power.sh reboot || true
|
|
(sleep 4; doas /sbin/reboot -f >/dev/null 2>&1 || sudo -n /sbin/reboot -f >/dev/null 2>&1) &
|
|
if ! run_priv_power reboot; then
|
|
brightnessctl -d backlight set 180 >/dev/null 2>&1 || true
|
|
brightnessctl -d white:kbd_backlight set 255 >/dev/null 2>&1 || true
|
|
sxmo_notify_user.sh "Reboot failed: permission denied"
|
|
exit 1
|
|
fi
|
|
sxmo_jobs.sh stop all >/dev/null 2>&1 || true
|
|
;;
|
|
poweroff)
|
|
sync
|
|
sxmo_hook_power.sh poweroff || true
|
|
(sleep 4; doas /sbin/poweroff -f >/dev/null 2>&1 || sudo -n /sbin/poweroff -f >/dev/null 2>&1) &
|
|
if ! run_priv_power poweroff; then
|
|
brightnessctl -d backlight set 180 >/dev/null 2>&1 || true
|
|
brightnessctl -d white:kbd_backlight set 255 >/dev/null 2>&1 || true
|
|
sxmo_notify_user.sh "Poweroff failed: permission denied"
|
|
exit 1
|
|
fi
|
|
sxmo_jobs.sh stop all >/dev/null 2>&1 || true
|
|
;;
|
|
logout)
|
|
sxmo_hook_logout.sh || true
|
|
case "$SXMO_WM" in
|
|
"i3") i3-msg exit ;;
|
|
"sway") swaymsg exit ;;
|
|
"river") riverctl exit ;;
|
|
"dwm") pkill dwm ;;
|
|
esac
|
|
;;
|
|
togglewm) sxmo_togglewm.sh ;;
|
|
*)
|
|
usage
|
|
exit 1
|
|
;;
|
|
esac
|