#!/usr/bin/env bash

# ***This script was made by Clay Gomera (Drake)***
#  - Description: A simple screenshot dmenu script
#  - Dependencies: scrot, dmenu, notify-send

## CREATING SCREENSHOT FOLDER ##
mkdir -p "$HOME/pictures/screenshots"
cd "$HOME/pictures/screenshots" || exit 0

## CHOICES ##
cho1="entire screen"
cho2="entire screen with delay"
cho3="focused window"
cho4="select area"
chos="$cho1\n$cho2\n$cho3\n$cho4"

## DELAY OPTIONS ##
del1="3 sec delay"
del2="5 sec delay"
del3="10 sec delay"
dels="$del1\n$del2\n$del3"

## DELAY FUNCTION ##
delays() {
    del=$(echo -e "$dels" | dmenu -i -fn "mononoki Nerd Font-10" -nb \#1d2021 -nf \#fbf1c7 -sb \#cc241d -sf \#fbf1c7 -p "Select: ");
        case $del in
            $del1)
                scrot -d 3 && notify-send "screenshot saved";;
            $del2)
                scrot -d 5 && notify-send "screenshot saved";;
            $del3)
                scrot -d 10 && notify-send "screenshot saved"
        esac
}

## ENTIRE SCREEN FUNCTION ##
screen() {
        scrot && notify-send "screenshot saved"
}

## FOCUSED WINDOW FUNCTION
window() {
        scrot -u -b && notify-send "screenshot saved."
}

## SELECTED AREA FUNCTION ##
area() {
        scrot -s && notify-send "screenshot saved."
}

## MAIN ACTION ##
choice=$(echo -e "$chos" | dmenu -i -fn "mononoki Nerd Font-10" -nb \#1d2021 -nf \#fbf1c7 -sb \#cc241d -sf \#fbf1c7 -p "Select: ")
case $choice in
    $cho1*)
        screen;;
    $cho2*)
        delays;;
    $cho3*)
        window;;
    $cho4*)
        area
esac