#!/usr/bin/env bash

# ***This script was made by Clay Gomera (Drake)***
#  - Description: A simple screenshot menu rofi script
#  - Dependencies: wofi, grim, slurp

# options
option1="  Capture the screen"
option2="  Capture region"
option3="  Exit"
options="$option1\n$option2\n$option3"

# countdown
countdown() {
    notify-send "Screenshot" "Executing in 3 seconds" -t 1000
    sleep 1
    notify-send "Screenshot" "Executing in 2 seconds" -t 1000
    sleep 1
    notify-send "Screenshot" "Executing in 1 seconds" -t 1000
    sleep 2
}

# run
choice=$(echo -e "$options" | $RUNNER -l 3 -p "[󰄀 Screenshot]  " )
case $choice in
    $option1)
        countdown
        grim - | wl-copy -t image/jpg
        notify-send "Screenshot" "Screenshot saved to clipboard"
        ;;
    $option2)
        notify-send "Screenshot" "Select a region to capture"
        grim -g "$(slurp)" - | wl-copy -t image/jpg
        notify-send "Screenshot" "Region saved to clipboard"
        ;;
    $option3)
        exit 0
        ;;
esac