#!/usr/bin/env bash

if [[ "$(xprop -root | grep -i "openbox")" > /dev/null ]]; then
    CUR_WM="openbox"
else
    CUR_WM=$(xprop -root | grep "_NET_WM_NAME(UTF8_STRING)" | cut -d " " -f 3 | sed 's/"//g' | awk '{print tolower($0)}')
fi

ANS=$(echo " lock| logout| reboot| shutdown" | rofi -sep "|" -dmenu -i -p 'System: ' "" -width 20 -hide-scrollbar -font "Ubuntu 11" -eh 1 -line-padding 4 -padding 20 -no-config -lines 4 -color-enabled -color-window "#1F2326, #F1F1F1, #1F2326" -color-normal "#1F2326, #F1F1F1, #1F2326, #4E88CF, #1F2326")

if [[ "$ANS" == *lock ]]; then
  i3lock-fancy
elif [[ "$ANS" == *logout ]]; then
    if [[ "$CUR_WM" == "i3" ]]; then
        i3-msg exit
    elif [[ "$CUR_WM" == "bspwm" ]]; then
        for window_id in $(bspc query -W); do
            bspc window $window_id -c
        done
        killall sxhkd
        bspc quit
    elif [[ "$CUR_WM" == "openbox" ]]; then
        openbox --exit
    else
        echo "Unknown WM... Exiting"
        exit 1
    fi
elif [[ "$ANS" == *reboot ]]; then
  systemctl reboot
elif [[ "$ANS" == *shutdown ]]; then
  systemctl poweroff
fi