#!/bin/sh

set -eu

readonly LOG_FILE="${HOME}/.screenshots.log"
readonly NOTIFY_TIME_MS=3000
readonly NOTIFY_TIME_ERROR_MS=5000
readonly NOTIFY_LABEL="Screenshot"
readonly SCREENSHOTS_DIR="${SCREENSHOTS_DIR:-${HOME}/Pictures/screenshots}"
readonly TIMESTAMP_FORMAT="%H-%M-%S_%Y-%m-%d"
readonly NOOPENGL="${NOOPENGL:-false}"


mkdir -p "${SCREENSHOTS_DIR}"


_notify() {
    notify-send --urgency normal \
                --expire-time "${NOTIFY_TIME_MS}" \
                "${NOTIFY_LABEL}" \
                "${1}"
}

_notify_error() {
    notify-send --urgency critical \
                --expire-time "${NOTIFY_TIME_ERROR_MS}" \
                "${NOTIFY_LABEL}" \
                "Error: ${1}"
}

_maim() {
    maim_args="--quality 10"
    if ${NOOPENGL}; then
        maim_args="${maim_args} --noopengl"
    fi

    #shellcheck disable=SC2086
    maim ${maim_args} "${@}"
}

_capture_select() {
    _maim --select
}

_capture_entire() {
    _maim
}

_capture_window() {
    _maim --window "$(xdotool getactivewindow)"
}

_copy_to_clipboard() {
    xclip -selection clipboard -t image/png
}

_save_to_file() {
    cat > "${1}"
}

_process_screenshot() {
    capture_func="${1}"
    output_func="${2}"
    success_msg="${3}"
    error_msg="${4}"
    output_arg="${5:-}"

    if [ -n "${output_arg}" ]; then
        if ! ${capture_func} | ${output_func} "${output_arg}"; then
            _notify_error "${error_msg}"
            return 1
        fi
    else
        if ! ${capture_func} | ${output_func}; then
            _notify_error "${error_msg}"
            return 1
        fi
    fi

    _notify "${success_msg}"
    return 0
}

_select_save() {
    filename="${SCREENSHOTS_DIR}/selection_$(date +"${TIMESTAMP_FORMAT}").png"
    _process_screenshot _capture_select \
                        _save_to_file \
                        "Selection saved to ${filename}" \
                        "Failed to save selection screenshot" \
                        "${filename}"
}

_select_copy() {
    _process_screenshot _capture_select \
                       _copy_to_clipboard \
                       "Selection copied to clipboard" \
                       "Failed to copy selection to clipboard"
}

_window_save() {
    filename="${SCREENSHOTS_DIR}/window_$(date +"${TIMESTAMP_FORMAT}").png"
    _process_screenshot _capture_window \
                       _save_to_file \
                       "Current window saved to ${filename}" \
                       "Failed to save Current window screenshot" \
                       "${filename}"
}

_window_copy() {
    _process_screenshot _capture_window \
                       _copy_to_clipboard \
                       "Current window copied to clipboard" \
                       "Failed to copy current window to clipboard"
}

_entire_save() {
    filename="${SCREENSHOTS_DIR}/screen_$(date +"${TIMESTAMP_FORMAT}").png"
    _process_screenshot _capture_entire \
                       _save_to_file \
                       "Entire screen saved to ${filename}" \
                       "Failed to save entire screen" \
                       "${filename}"
}

_entire_copy() {
    _process_screenshot _capture_entire \
                       _copy_to_clipboard \
                       "Entire screen copied to clipboard" \
                       "Failed to copy entire screen to clipboard"
}

main() {
    if [ $# -ne 2 ]; then
        echo "Usage: ${0} {select|window|entire} {save|copy}"
        exit 1
    fi

    readonly target="${1}"
    readonly action="${2}"

    case "${target}-${action}" in
        select-save) _select_save &>> "${LOG_FILE}" ;;
        select-copy) _select_copy &>> "${LOG_FILE}" ;;
        window-save) _window_save &>> "${LOG_FILE}" ;;
        window-copy) _window_copy &>> "${LOG_FILE}" ;;
        entire-save) _entire_save &>> "${LOG_FILE}" ;;
        entire-copy) _entire_copy &>> "${LOG_FILE}" ;;
        *)
            echo "Invalid options: target=${target}, action=${action}"
            echo "Available targets: select, window, entire"
            echo "Available actions: save, copy"
            exit 1
            ;;
    esac
}

main "$@"