#!/usr/bin/env bash

if [ ! -d $HOME/Pictures/Screenshots ]
then
        mkdir -p $HOME/Pictures/Screenshots
fi

fullscreen() {
        filename=$HOME/Pictures/Screenshots/$(date +%Y%m%d_%H:%M:%S_Fullscreen).png
        grim $filename
        cat $filename | wl-copy         && notify-send -i $filename "Fullscreen Screenshot taken"
        }

selecting() {
        filename=$HOME/Pictures/Screenshots/$(date +%Y%m%d_%H:%M:%S_Selected).png
        grim -g "$(slurp -b e3eaf699)" $filename         && notify-send -i $filename "Selective Screenshot taken"
        cat $filename | wl-copy
        }

wofirun() {
        case "$(printf "fullscreen\nselect\n" | wofi --dmenu -i -H 500 -W 250 -x 0 -y 0 )" in
                'fullscreen')fullscreen;;
                'select')selecting;;
        esac
}

helpscreen() { 
        echo "Usage: screenshot [-f] [-s] [-w] [-h] "
        echo "Options:"
        echo "-f : Fullscreen screenshot"
        echo "-s : Select a region for the screenshot"
        echo "-w : Shows a wofi prompt for screenshot"
        echo "-h : Shows this fucking help menu"
        echo "If no flags are selected then -f is chosen"
}
case $1 in
        -f)fullscreen;;
        -s)selecting;;
        -h)helpscreen;;
        -w)wofirun;;
        *)fullscreen;;
esac
