#!/bin/sh

set -e

command -v scrot > /dev/null ||
	{ echo "scrot not installed" ; exit 1 ; }
command -v rofi > /dev/null &&
    menu_exec='rofi -dmenu' ||
    menu_exec='dmenu'

choice="$(
$menu_exec <<EOF
Crop to clipboard
Crop to file
Window to clipboard
Window to file
Fullscreen to clipboard
Fullscreen to file
EOF
)"

screenshot_dir="$HOME/screenshots"
[ ! -d "$screenshot_dir" ] && mkdir "$screenshot_dir"

scrot_wrapper() {
    # shellcheck disable=SC2016
    file_format="$screenshot_dir/"'%Y-%m-%d_$wx$h.png'
    if [ "$1" = '-' ]
    then
        file_format='-'  # Keep stdout if not saving to a file
        shift
    else
        # optipng optimizes png size without loss (doesn't work if we output directly to stdout)
        # shellcheck disable=SC2016
        command -v optipng && optipng_arg='optipng -o4 $f'
    fi
    scrot --silent --quality 95 --compression 9 --file "$file_format" -e "$optipng_arg" "$@"
}

xclip_wrapper() {
    xclip -selection clipboard -target image/png
}

case "$choice" in
    'Crop to clipboard')
        scrot_wrapper - --select | xclip_wrapper
        ;;
    'Crop to file')
        scrot_wrapper --select
        ;;
    'Window to clipboard')
        scrot_wrapper - --window "$(xdo id -p PID)" | xclip_wrapper
        ;;
    'Window to file')
        scrot_wrapper --window "$(xdo id -p PID)"
        ;;
    'Fullscreen to clipboard')
        scrot_wrapper - | xclip_wrapper
        ;;
    'Fullscreen to file')
        scrot_wrapper
        ;;
esac