#!/bin/bash

### From http://www.archlinux.org/index.php/i3

_logfile="$LOGDIR/i3/i3exit.log"
LOCKSCREEN="$LIBDIR/wallpaper/lockscreen"

_fallback_color="000000"

touch "$_logfile"

log()
{
    echo "$*"
    echo "[$(date +%FT%T)] $*" >> "$_logfile"
}

lock()
{
    set -x
    playing=0
    # if dbus-send --print-reply --dest=org.mpris.MediaPlayer2.spotify /org/mpris/MediaPlayer2 org.freedesktop.DBus.Properties.Get string:'org.mpris.MediaPlayer2.Player' string:'PlaybackStatus'|grep -q Playing ; then
    #     playing=1
    # fi
    # echo $playing
    # (( $playing )) && dbus-send --print-reply --dest=org.mpris.MediaPlayer2.spotify /org/mpris/MediaPlayer2 org.mpris.MediaPlayer2.Player.Pause
    if [[ -f "$LOCKSCREEN" ]] ; then
        i3lock --nofork --show-failed-attempts --ignore-empty-password \
            --image "$LOCKSCREEN"
    else
        i3lock --nofork --show-failed-attempts --ignore-empty-password \
            --color "$_fallback_color"
    fi
    # (( $playing )) && dbus-send --print-reply --dest=org.mpris.MediaPlayer2.spotify /org/mpris/MediaPlayer2 org.mpris.MediaPlayer2.Player.Play
}

screen_off() {
    :
    xset dpms force off
}

reset_screen() {
    xset -dpms
    xset s off
}

lock_and_screen_off() {
    lock &
    _pid=$!
    systemctl --user --no-block kill --signal SIGUSR1 dunst_user
    screen_off
    xset dpms 0 0 5
    wait $_pid
    systemctl --user --no-block kill --signal SIGUSR2 dunst_user
    reset_screen
}

disable_lid_switch() {
    grep "^${ACPI_LID_NAME}.*enabled" /proc/acpi/wakeup && echo " ${ACPI_LID_NAME}" | sudo tee /proc/acpi/wakeup
}

signal="$1"
log "[I] Received signal \"$signal\"."

case "$signal" in
    lock)
        log "[I] Locking session."
        lock_and_screen_off
        ;;
    logout)
        log "[I] Exiting i3."
        i3-msg exit
        ;;
    suspend)
        log "[I] Suspending."
        lock &
        disable_lid_switch
        sleep 0.1
        systemctl suspend
        ;;
    hibernate)
        log "[I] Hibernating."
        disable_lid_switch
        sudo systemctl hibernate
        ;;
    reboot)
        log "[I] Rebooting."
        systemctl reboot
        ;;
    shutdown)
        log "[I] Shutting down."
        systemctl poweroff
        ;;
    screen-off)
        log "[I] Turning screen off."
        screen_off
        ;;
    *)
        echo "Usage: $0 {lock|logout|suspend|hibernate|reboot|shutdown}"
        log "[E] Signal \"$signal\" unknown. Aborting."
        exit 2
esac

log "[I] Done."
exit 0