#!/bin/sh

# 
# Takes a screenshot with grimshot, in clipboard or in given directory
#

MODE=$(printf "copy\nsave" | dmenu)

# If dmenu was cancelled, exit program
if [ $? -ne 0 ]; then
    exit 1
fi

case "$MODE" in 
  *save*)
    cd $HOME
    while true; do
        selected_dir=$(ls | dmenu -i)
        if [ -z "$selected_dir" ]; then
            exit 0
        fi

        if [ -d "$selected_dir" ]; then
            cd "$selected_dir"
            continue
        fi 

        if [ -e "$selected_dir" ]; then
            notify-send "Error: File exists!"
            exit 0
        fi 

        break
    done

    grimshot save area "$PWD/$selected_dir.png" && notify-send "$PWD/$selected_dir.png" "saved" 
    ;;
  *copy*)
    grimshot copy area && notify-send "Screenshot copied" 
    ;;
esac