#!/usr/bin/env bash

dir="~/.config/rofi"
uptime=$(uptime -p | sed -e 's/up //g')

rofi_command="rofi -theme $dir/powermenu.rasi"

# Options
lock=" Lock"
suspend=" Sleep"
logout=" Logout"
reboot=" Restart"
shutdown=" Shutdown"

# Confirmation
confirm_exit() {
  rofi -dmenu\
    -i\
    -no-fixed-num-lines\
    -p "Are You Sure? : "\
    -theme $dir/confirm.rasi
}

# Message
msg() {
  rofi -theme "$dir/message.rasi" -e "Available Options  -  yes / y / no / n"
}

# Variable passed to rofi
options="$lock\n$suspend\n$logout\n$reboot\n$shutdown"

chosen="$(echo -e "$options" | $rofi_command -p "Uptime: $uptime" -dmenu -selected-row 0)"
case $chosen in
    $shutdown)
      ans=$(confirm_exit &)
      if [[ $ans == "yes" || $ans == "YES" || $ans == "y" || $ans == "Y" ]]; then
        i3exit shutdown
      elif [[ $ans == "no" || $ans == "NO" || $ans == "n" || $ans == "N" ]]; then
        exit 0
      else
        msg
      fi
      ;;
    $reboot)
      ans=$(confirm_exit &)
      if [[ $ans == "yes" || $ans == "YES" || $ans == "y" || $ans == "Y" ]]; then
        i3exit reboot
      elif [[ $ans == "no" || $ans == "NO" || $ans == "n" || $ans == "N" ]]; then
        exit 0
      else
        msg
      fi
      ;;
    $lock)
      i3exit lock
      ;;
    $suspend)
      ans=$(confirm_exit &)
      if [[ $ans == "yes" || $ans == "YES" || $ans == "y" || $ans == "Y" ]]; then
        i3exit suspend
      elif [[ $ans == "no" || $ans == "NO" || $ans == "n" || $ans == "N" ]]; then
        exit 0
      else
        msg
      fi
      ;;
    $logout)
      ans=$(confirm_exit &)
      if [[ $ans == "yes" || $ans == "YES" || $ans == "y" || $ans == "Y" ]]; then
        i3exit logout
      elif [[ $ans == "no" || $ans == "NO" || $ans == "n" || $ans == "N" ]]; then
        exit 0
      else
        msg
      fi
      ;;
esac