#!/bin/sh

# TODO: detect term or set it globally
TERM_EMU="kitty"

die() {
    notify-send "$@"
    exit 1
}

check_exists() {
    for c in $@; do
        which "$c" > /dev/null 2>&1 || return 1
    done
}

if ! which wl-paste > /dev/null 2>&1 && [ ! -z "$WAYLAND_DISPLAY" ]; then
    die "Wayland detected and wl-paste not found"
elif ! which xsel > /dev/null 2>&1 && [ -z "$WAYLAND_DISPLAY" ]; then
    die "xsel not found"
fi

# TODO check for xsel/wl-clipboard presence ?
if [ -z "$WAYLAND_DISPLAY" ]; then
    clip="$(xsel -b -o)"
else
    clip="$(wl-paste)"
fi

urls="$(echo $clip | grep -o 'https\?://[a-zA-Z0-9~#%&_+=,.?/-]\+')"

# Check the requirements for every option
opts="rot13"
check_exists "$TERM_EMU" "mpv" "youtube-dl" &&
    opts="audio-search|$opts"
check_exists "dragon" &&
    opts="drag-n-drop|$opts"
# Enable additional features when the clipboard contains an URL
if [ -n "$urls" ] && check_exists "youtube-dl"; then
    if echo "$urls" | grep -qF "scribblehub.com"; then
        check_exists "lncrawl" && opts="crawl|$opts"
    else
        check_exists "mpv" && opts="play|$opts"
        opts="download|$opts"
    fi
fi
# Load default choice last
check_exists "qrencode" "feh" &&
    opts="qr|$opts"

# Split into multiple lines for dmenu
opts="$(printf "%s" "$opts" | sed 's/|/\n/g')"

choice="$(printf "%s" "$opts" | dmenu -p "$(printf "%s" "$clip" | cut -c 1-48)")"
case "$choice" in
    "qr")
        printf "%s" "$clip" | qrencode -o - | feh -. -Z --force-aliasing --geometry 400x400 -
        ;;
    # Download supports optional folder target
    "download"*)
        path="$HOME/mus/$(printf "%s" "$choice" | cut -d " " -f 2-)"
        [ -d "$path" ] || mkdir -p "$path"
        cd "$path" || exit
        youtube-dl -x --audio-format mp3 --no-playlist -o "%(title)s.%(ext)s" "$urls" 2>&1 | grep 'ERROR:' | tee /dev/fd/2 | xargs -n1 -d "\n" notify-send
        ;;
    "crawl")
        path="$HOME/books/"
        [ -d "$path" ] || mkdir -p "$path"
        cd "$path" || exit
        out="$(lncrawl --single --all --format epub --filename-only --suppress -s "$urls" | grep -vFe "Input is suppressed" -e "Namespace(")"
        errors="$(printf "%s" "$out" | grep -Fe " ❗ ")"
        if [ -n "$errors" ]; then
            printf "%s" "$errors" | xargs -n1 -d "\n" notify-send
        else
            printf "%s" "$out" | sed -ne '/^NOVEL: /s/NOVEL: \(.*\)/Successfully crawled "\1"/p' | xargs -0 -n1 notify-send
        fi
        ;;
    "drag-n-drop"*)
        dragon -x "$clip"
        ;;
    "play")
        playvideo "$clip"

        ;;
    "audio-search")
        "$TERM_EMU" --single-instance mpv --ytdl-format=bestaudio ytdl://ytsearch:"$clip"
        ;;
    "rot13")
        if [ -z "$WAYLAND_DISPLAY" ]; then
            xsel -b -o | tr 'A-Za-z' 'N-ZA-Mn-za-m' | xsel -b -i
        else
            wl-copy -n "$(wl-paste | tr 'A-Za-z' 'N-ZA-Mn-za-m')"
        fi
        ;;
    *)
        printf 'Nope\n'
        ;;
esac