mpvc-fzf

#!/usr/bin/env sh
#
# @file mpvc-fzf
# @description mpvc fzf, mpv terminal user interface
# @author gmt4 <gmt4 at github.com> (c) Copyright 2022 GPLv2+
# @url github.com/gmt4/mpvc
# SPDX-License-Identifier: GPL-2.0-or-later
#

PROGNAME=${0##*/}
PROGDIR=${0%/*}
PROGVERSION="v1.6"
PROGAUTHOR=gmt4
PROGURL="https://github.com/gmt4/mpvc"

set -euf

quiet()  { "$@" > /dev/null 2>&1; }
mpvcfzf_warn() { echo "$@" >&2; }
mpvcfzf_awk0() { awk '{print $NF}'; }
mpvcfzf_awk1() { awk '{print $1}'; }
mpvc() { "$PROGDIR/mpvc" "$@"; } # mpvc-web?

# Set defaults
mpvcfzf_defaults()
{
    XDG_CONFIG_HOME=${XDG_CONFIG_HOME:-"$HOME/.config"}
    MPVC_CONFIG_DIR="${MPVC_CONFIG_DIR:-$XDG_CONFIG_HOME/mpvc}"
    MPVC_CONFIG="${MPVC_CONFIG:-$MPVC_CONFIG_DIR/mpvc.conf}"
    MPVC_YTDL_ARCHIVE=${MPVC_YTDL_ARCHIVE:-}
    MPVC_YTDL_OUTPUT=${MPVC_YTDL_OUTPUT:-}
    MPVC_TERM=${MPVC_TERM:-xterm}

    MPVC_FZF_SOMAFM_FILE="${MPVC_FZF_SOMAFM_FILE:-$MPVC_CONFIG_DIR/somafm-channels.json}"
    MPVC_FZF_RADIOAPI_FILE="${MPVC_FZF_RADIOAPI_FILE:-$MPVC_CONFIG_DIR/radioapi-stations.json}"
    #mpvc -q
}

mpvcfzf_config()
{
    mpvcfzf_defaults
    if [ ! -d "$MPVC_CONFIG_DIR" ]; then mkdir -p "$MPVC_CONFIG_DIR"; fi
    # if present load user config
    if [ -r "$MPVC_CONFIG" ]; then . "$MPVC_CONFIG"; fi
    if ! quiet command -v fzf; then
        mpvcfzf_warn "$PROGNAME: Warning: No fzf, Install to get fuzzy media search & play support!"
        sleep 1
    fi
}

mpvcfzf_curl()
{
    curl -fsSL -A 'Mozilla/5.0' --retry 1 "$@"
}

mpvcfzf_getvideo_id()
{
    echo "${1##*/}"
}

mpvcfzf_cparse()
{
    awk -v video=1 -v playlist=0 '
        /^.*<p dir="auto">/ {
            l=$0; gsub("^.*<p dir=\"auto\">","", l); gsub("</p>.*$","", l); vtitle=l
        }
        /<p class="video-data" dir="auto">/ {
            gsub("^ *<p class=\"video-data\" dir=\"auto\">",""); gsub(" views",""); gsub("</p>",""); shared=$0
        }
        /^ *<p class="length">/ {
            gsub("^ *<p class=\"length\">",""); gsub("</p>",""); vlength=$0
        }

        /^ *<a title="Audio mode/ {
            gsub("^ *<a title=\"Audio mode\" href=\"",""); gsub("\">","");
            gsub("/watch\\?v=", "https://yewtu.be/"); gsub("&listen=1","");
            if (video)
                printf "%s %4s %7s %s\n", $0, shared, vlength, vtitle
        }

        /^ *<a href="\/playlist/ {
            gsub("^ *<a href=\"",""); gsub("\">.*","");
            gsub("/playlist", "https://yewtu.be/playlist");
            if (playlist)
                printf "%s %7s %s\n", $0, vlength, vtitle
        }

        END {
            if (NR == 0)
            {
                print "Error: No results found, maybe yewtu.be is down, try switching to a different instance with:"
                print "       url_host=invidious.projectsegfau.lt mpvc-fzf"
            }
        }
    '
}

mpvcfzf_csearch()
{
    echo "$@"
    mpvcfzf_curl "$@" | mpvcfzf_cparse
}

mpvcfzf_csearch_play()
{
    mpvcfzf_csearch "$1" |
        mpvcfzf_preview --prompt="(csearch) " |
            mpvcfzf_awk1
}

mpvcfzf_isearch_once()
{
    mpvcfzf_isearch "$@" | awk 'NR==2 {sub(" +", " # "); print}'
}

mpvcfzf_isearch()
{
    url_host="${url_host:-yewtu.be}"
    url_sort="sort=relevance" # relevance views date
    url_query="q=$(for f in $@; do printf "%s+" "$f"; done)"
    url_search="https://${url_host}/search?$url_query&$url_sort"

    mpvcfzf_csearch "$url_search&page=1" "$url_search&page=2"
}

mpvcfzf_related_search()
{
    id=${1:-$(mpvc get filename)}
    id=$(mpvcfzf_getvideo_id "$id")
    output=$(mpvcfzf_curl "https://yewtu.be/redirect?referer=/watch?v=$id")
    echo "$output" |
        awk '
            /<div class="video-card-row">/ { c++; next }
            c && c-- {
                sub("^ *<a href=\"/watch\\?v=","https://yewtu.be/");
                sub("&listen=false\"><p dir=\"auto\">"," ");
                sub("</p></a>$","");
                print $0
            }'
}

mpvcfzf_related_play()
{
    mpvcfzf_related_search "$1" |
        mpvcfzf_preview --prompt="(similar) " |
            mpvcfzf_awk1
}

mpvcfzf_related_once()
{
    mpvcfzf_related_search "$(mpvcfzf_isearch_once "${1:-$(mpvc get media-title)}" | mpvcfzf_awk1)"
}

mpvcfzf_lyrics()
{
    id=${1:-$(mpvc get filename)}
    id=$(mpvcfzf_getvideo_id "$id")
    output=$(mpvcfzf_curl "https://yewtu.be/redirect?referer=/watch?v=$id")
    echo "$output" |
        awk -v id="$id" '
            c==0 && /<meta name="description"/  {c=1}
            c==1 && /^[Ll][Yy][Rr][Ii][Cc][Ss]/ {c=2}
            c==2 && /<meta name="keywords"/     {c=0;exit}
            c==2                                {print}
            END                                 {if (c==1) print "Lyrics: No lyrics found for",id}
        '
}

mpvcfzf_lyrics_ddg()
{
    echo "https://ddg.gg?ia=lyrics&iax=lyrics&q=lyrics $(mpvc -f '%title%')"
}

mpvcfzf_nextvideo()
{
    id=$(mpvcfzf_getvideo_id "$1")
    output=$(mpvcfzf_curl "https://yewtu.be/watch?v=$id")
    next_video=$(echo "$output" | awk '/next_video/ {gsub("[\",]",""); print $2}')
    next_url="https://yewtu.be/$next_video"
    echo "$next_url"
}

mpvcfzf_icache_ytdlro() { mpvc getr ytdl-raw-options; }
mpvcfzf_icache_archive() {
    if [ -n "$MPVC_YTDL_ARCHIVE" ]; then
        echo "$MPVC_YTDL_ARCHIVE"
    else
        mpvcfzf_icache_ytdlro | jq -r ".data.download-archive"
    fi
}
mpvcfzf_icache_output() { mpvcfzf_icache_ytdlro | jq -r ".data.output"; }
mpvcfzf_icache_output_dir() {
    if [ -n "$MPVC_YTDL_OUTPUT" ]; then
        echo "$MPVC_YTDL_OUTPUT"
    else
        MPVC_YTDL_OUTPUT=$(mpvcfzf_icache_output)
        MPVC_YTDL_OUTPUT=${MPVC_YTDL_OUTPUT%/*}
        echo "$MPVC_YTDL_OUTPUT" | sed "s|^~|$HOME|"
    fi
}

mpvcfzf_icache()
{
    # check for media present in local cache
    #ytdl_archive="$(mpvcfzf_icache_archive)"
    ytdl_output="$(mpvcfzf_icache_output_dir)"
    ytdl_dir="$ytdl_output"

    if [ -z "$ytdl_output" ] || [ -z "$ytdl_output" ]; then
        #quiet mpvc get ytdl-raw-options || cat
        mpvcfzf_warn "$PROGNAME: Error cache not enabled: undefined ytdl-raw-options"
        if cat; then return; fi
    fi

    while read -r line
    do
        url=$(echo "$line" | awk '/https:/ { gsub("watch\\?v=",""); print $1 }')
        vid=$(basename "$url")
        file=$(find "$ytdl_dir/" 2>/dev/null | awk "/$vid/ { print; exit }")
        if [ -n "$url" ] && [ -e "$file" ]; then
            echo "$file"
        else
            echo "$line"
        fi
    done
}

mpvcfzf_fetch()
{
    if [ $# -lt 1 ]; then mpvcfzf_usage; fi

    ytdl_opts=""
    ytdlp=$(command -v youtube-dl || command -v yt-dlp)
    if [ ! -x "$ytdlp" ]; then
        mpvcfzf_warn "$PROGNAME: Warning: No yt-dlp, Install to get ytdl support!"
    fi

    mpvc -q --mpv

    ytdl_json=$(mpvc getr ytdl-raw-options)
    if [ -z "$ytdl_json" ]; then
        mpvcfzf_warn "$PROGNAME: Error ytdl-raw-options not set. Check docs/mpv.conf"
        return
    fi
    for option in $(echo "$ytdl_json" | jq -r ".data | keys | .[]")
    do
        value=$(echo "$ytdl_json" | jq -r ".data.\"$option\"")
        if [ -n "$value" ]
        then
            ytdl_opts="$ytdl_opts --$option=$value"
        else
            ytdl_opts="$ytdl_opts --$option"
        fi
    done
    "$ytdlp" $ytdl_opts ${YTDL_OPTS:-} "$@"
}

#
# fzf wrappers
#

mpvcfzf_preview()
{
    MPVC_FZF_BIND_HELP="C-u:go C-x:ls C-v:yt C-\\:fs C-s:chapter C-t:add C-o:play C-r:reload C-spc:toggle"
    # --bind='ctrl-d:execute(mpvc {q})'
    # --bind='left:execute-silent(mpvc -q seek -20 &)' \
    # --bind='right:execute-silent(mpvc -q seek +20 &)' \

    # fzf keys: ctrl + [dorstxv\]
    fzf \
        --ansi \
        --no-sort \
        --multi \
        \
        --layout=reverse \
        --header="$PROGNAME help: $MPVC_FZF_BIND_HELP" \
        --bind='ctrl-a:toggle-all' \
        --bind='ctrl-s:execute(mpvc-fzf -c)' \
        --bind='ctrl-x:execute(mpvc-fzf -f {q})' \
        --bind='ctrl-y:execute(mpvc-fzf -b {+1})' \
        --bind='ctrl-u:execute(mpvc-fzf -b {+1})' \
        --bind='ctrl-v:execute(mpvc-fzf -p {q})' \
        --bind='ctrl-\:execute(mpvc-fzf -l)' \
        --bind='ctrl-t:execute-silent(mpvc add {+1} &)' \
        --bind='ctrl-space:execute-silent(mpvc toggle &)' \
        \
        --preview-window="up:$(mpvc statussz)" \
        --preview="mpvc" \
        "$@"
}

mpvcfzf_iplay()
{
    mpvcfzf_isearch "$1" |
        mpvcfzf_preview --prompt="(youtube) " |
            mpvcfzf_awk1
}

mpvcfzf_local()
{
    find . -type f,d | sort |
        mpvcfzf_preview --prompt="(local) " |
            mpvc load
}

mpvcfzf_playlist()
{
    len=$(mpvc gets playlist-count)
    (if [ -z "$len" ]; then echo; else mpvc fullplaylist; fi) |
        mpvcfzf_preview --prompt="(playlist) " \
        --bind='ctrl-o:execute-silent(mpvc play {1})' \
        --bind='ctrl-r:reload(mpvc fullplaylist)'
}

mpvcfzf_chapterlist()
{
    len=$(mpvc gets chapter-list/count)
    (if [ -z "$len" ]; then echo; else mpvc chapter-list-nr; fi) |
        mpvcfzf_preview --prompt="(chapter) " \
        --bind='ctrl-o:execute-silent(mpvc set chapter {1})' \
        --bind='ctrl-r:reload(mpvc chapter-list-nr)'
}

mpvcfzf_stashlist()
{
    mpvc stash list | mpvcfzf_awk0 |
        mpvcfzf_preview --prompt="(stash) " \
        --bind='ctrl-o:execute-silent(mpvc stash apply {+} &)' \
        --bind='ctrl-r:reload(mpvc stash list | mpvc-fzf -0)'
}

mpvcfzf_ddg_play()
{
    mpvcfzf_ddg "$@" |
        mpvcfzf_preview --prompt="(ddg) " |
            mpvcfzf_awk1 |
                mpvc load
}

mpvcfzf_urlget()
{
    url=${1:-https://localhost:8443/ytdl-archive/}
    mpvcfzf_curl "$url" |
        awk -v url="$url" '
            /<a href=".*">.*<\/a>/ {
                match(url, "https?://[^/]+/");
                host = substr($0, RSTART, RLENGTH)
                href=$0; sub("^.*href=\"", "", href) sub("\".*$", "", href)
                if (href ~ "^/") href = host"/"href
                print (href !~ host) ? url"/"href : href
            }
        '
}

mpvcfzf_urlbrowser()
{
    url=${1:-https://localhost:8443/ytdl-archive/}
    mpvcfzf_urlget "$url" |
        mpvcfzf_preview --prompt="(urlbrowse) "
}

mpvcfzf_cmds()
{
    mpvcfzf_preview --prompt '(cmds)' --bind='ctrl-o:execute(sh -c {})' "$@";
}

mpvcfzf_equalizer()
{
    MPVC_FZF_BIND_HELP="C-x:list C-v:yt-dlp C-\\:local C-s:chapter C-t:add C-o:play C-r:reload C-spc:toggle"
    mpvc-equalizer preset |
        mpvcfzf_preview --prompt="(equalizer) " \
        --header="$PROGNAME help: $MPVC_FZF_BIND_HELP" \
        --bind='ctrl-x:execute-silent(mpvc-equalizer dec {1} &)' \
        --bind='ctrl-v:execute-silent(mpvc-equalizer inc {1} &)' \
        --bind='ctrl-space:execute-silent(mpvc-equalizer reset)' \
        --bind='ctrl-o:execute-silent(mpvc-equalizer preset {-1} | mpvc-equalizer load)' \
        --bind='ctrl-s:reload(mpvc-equalizer bars)' \
        --bind='ctrl-r:reload(mpvc-equalizer preset)' \
        "$@"
}

mpvcfzf_ddg()
{
    url_search="https://duckduckgo.com/html?q=$*&iax=videos"

    echo "$url_search"
    mpvcfzf_curl "$url_search" |
        awk '
        function urldecode(url)
        {
            for (y=0; y < 127; y++)
            {
                if (y != 37)
                    gsub(sprintf("%%%02x|%%%02X", y, y), y==38 ? "\\&" : sprintf("%c", y), url);
                gsub(/%25/, "%", url);
            }
            return url
        }

        /result__a/ {
            gsub("<a rel=\"nofollow\" class=\"result__a\" href=\"//duckduckgo.com/l/\\?uddg=", "");
            gsub("&amp", "\\&");
            gsub("rut=[0-9a-f]+","");
            gsub("\">", " ");
            gsub("</a>","");

            gsub("%3A",":"); gsub("%2F","/"); gsub("%3F","?"); gsub("%3D","=");
            gsub("^ *","");
            #print urldecode($0)
            print
        }'
}

mpvcfzf_lofi()
{
    mpvcfzf_urlbrowser "${1:-https://lofigirl.com/wp-content/uploads/2023/06/}"
}

mpvcfzf_somafm()
{
    chan=${1:-}
    if [ -n "$chan" ]; then
        mpvcfzf_somafm_search | awk -F';' -v chan="$chan" '{ if(match($0,chan))print $1}' | mpvc load
    else
        mpvcfzf_somafm_search | column -s ';' -t | mpvcfzf_preview --prompt="(somafm) "
    fi
}

mpvcfzf_somafm_search()
{
    url="https://somafm.com/channels.json"
    file="$MPVC_FZF_SOMAFM_FILE"
    [ ! -s "$file" ] && mpvcfzf_curl -o "$file" "$url"
    jq -r --argjson quality 3 '
        .channels | sort_by(.listeners | tonumber) | reverse | .[] |
            (.playlists[$quality]|.url) + ";" + .listeners + " ls "+.genre+";"+ .title' "$file"
}

mpvcfzf_radioapi()
{
    chan=${1:-}
    if [ -n "$chan" ]; then
        mpvcfzf_radioapi_search | awk -F';' -v chan="$chan" '{ if(match($0,chan))print $1}' | mpvc load
    else
        mpvcfzf_radioapi_search | column -s ';' -t | mpvcfzf_preview --prompt="(radioapi) "
    fi
}

mpvcfzf_radioapi_search()
{
    url="https://nl1.api.radio-browser.info/json/stations/topvote"
    file="$MPVC_FZF_RADIOAPI_FILE"
    [ ! -s "$file" ] && mpvcfzf_curl --retry 2 -d hidebroken=true -d limit=4096 -o "$file" "$url"
    jq -r -c '.[] | select( (.url|length) < 80 ) | (.url+";"+(.votes|tostring)+" "+.name)' "$file"
}

mpvcfzf_now_geturls() { mpvc saven; }
mpvcfzf_now_getvids() { echo "$@" | awk '/youtube|yewtu/ { sub("^.*/",""); sub(".*=",""); sub("^","\\&v="); printf $0; }'; }

mpvcfzf_now()
{
    date=$(date -Idate)
    now="https://gmt4.github.io/mpvc/now/"
    urls=$(mpvcfzf_now_geturls)
    vids=$(mpvcfzf_now_getvids "$urls")
    list=$(mpvc fullplaylist)

    echo "$now?t=$date$vids"
    if [ "$1" != "short" ]
    then
        echo
        gawk -v u="$urls" -v l="$list" 'END{ split(u,au,"\n"); split(l,al,"\n"); for(u in au) {print au[u], al[u]}; }' /dev/null
    fi
}

mpvcfzf_launch_term()
{
    if quiet command -v "$MPVC_TERM"; then "$MPVC_TERM" -e "$0" "$@"; fi;
}

mpvcfzf_version()
{
    echo "usage: $PROGNAME opts # @version $PROGVERSION (c) $PROGAUTHOR $PROGURL"
}

mpvcfzf_usage()
{
    cat <<EOF
usage: $PROGNAME opts # @version $PROGVERSION (c) $PROGAUTHOR $PROGURL
 -a|stash    : Search & play previously saved stashes
 -b|browse   : Browse the provided ytdl-archive URL with fzf
 -c|chapters : Start fzf to manage the current mpv chapterlist
 -d|dir      : Set the WD to the media directory given as argument
 -e|eqz      : Start fzf to manage the equalizer settings
 -f|playlist : Start fzf to manage the current mpv playist
 -g|fetch    : Fetch the given YT URL, and store locally
 -G|Fetch    : Search on Invidious, fetch, and store locally
 -i|lyrics   : Search given media lyrics on Invidious
 -k|dplay    : Search & play DuckDuckGo videos
 -K|dsearch  : Search DuckDuckGo videos
 -l|local    : Search & play local media
 -s|search   : Search on Invidious
 -p|splay    : Search & play media found using Invidious
 -x|launch   : Starts $PROGNAME in a new $MPVC_TERM (config \$MPVC_TERM) [combine -x with other opts]
 -y|related  : Search related media on Invidious
 -Y|Related  : Search & play related media using Invidious
 -z|relatedo : Search current media-title on Invidious and return related media
 -v|version  : Return the $PROGNAME version.

    now      : Return a shareable URL to the "now listening" playlist
    lofi     : Search & play Lo-Fi channels
    somafm   : Search & play SomaFM channels
    radioapi : Search & play Radio-Browser API channels
*tips: If unsure about where to begin, start: $PROGNAME -p 'kupla mirage'
EOF
    exit;
}
main()
{
    mpvcfzf_config
    if [ $# -lt 1 ]; then mpvcfzf_usage; fi

    for arg in "$@";
    do
        arg1="${2:-}"

        case "$arg" in
        -0|awk0)    shift; mpvcfzf_awk0; return ;;
        -1|awk1)    shift; mpvcfzf_awk1; return ;;

        -a|stash)   shift; mpvcfzf_stashlist; return ;;
        -b|browse)  shift; mpvcfzf_urlbrowser "$arg1"; return ;;
        -B|urlget)  shift; mpvcfzf_urlget "$arg1"; return ;;
        -c|chapters)shift; mpvcfzf_chapterlist; return ;;
        -C|cache)   shift; mpvcfzf_icache; return;;
        -CC|cmds)   shift; mpvcfzf_cmds "$@"; return;;

        -d|dir)     shift; MPVC_DIR="$arg1"; cd "$MPVC_DIR" || return;;
        -e|eqz)     shift; mpvcfzf_equalizer "$@"; return ;;
        -f|playlist)shift; mpvcfzf_playlist; return ;;
        -F|preview) shift; mpvcfzf_preview "$@"; return ;;
        -g|fetch)   shift; mpvcfzf_fetch "$arg1"; return ;;
        -G|Fetch)   shift; mpvcfzf_fetch "$(mpvcfzf_iplay "$1")"; return ;;
        -i|lyrics)  shift; mpvcfzf_lyrics "$arg1"; return ;;
        -L|Lyrics)  shift; mpvcfzf_lyrics_ddg; return ;;
        -k|dplay)   shift; mpvcfzf_ddg_play "$arg1"; return ;;
        -K|dsearch) shift; mpvcfzf_ddg "$arg1"; return ;;
        -h|help)    shift; mpvcfzf_usage; return ;;
        -l|local)   shift; mpvcfzf_local; return ;;
        -n|next)    shift; mpvcfzf_nextvideo "$arg1"; return ;;

        -o|osearch) shift; mpvcfzf_isearch_once "$arg1" ;; # lucky
        -O|Osearch) shift; mpvcfzf_isearch_once "$arg1" | mpvcfzf_awk1 | mpvc load ;; # lucky
        -OO)        shift; mpvcfzf_icache_output_dir ;;

        -p|splay)   shift; mpvcfzf_iplay "$arg1" | mpvc load; return ;;
        -P|Splay)   shift; mpvcfzf_iplay "$arg1" | mpvc loadc; return ;;
        -s|search)  shift; mpvcfzf_isearch "$arg1"; return;;
        -u|url)     shift; mpvcfzf_csearch "$arg1"; return ;;
        -U|Url)     shift; mpvcfzf_csearch_play "$arg1"; return ;;
        -x|launch)  shift; mpvcfzf_launch_term "$@"; exit ;;
        -y|related) shift; mpvcfzf_related_search "$arg1"; return ;;
        -Y|Related) shift; mpvcfzf_related_play "$arg1"; return ;;
        -z|relatedo)shift; mpvcfzf_related_once "$arg1"; return ;;
        -v|version) shift; mpvcfzf_version; return ;;

        now)        shift; mpvcfzf_now "${1:-}"; return ;;
        lofi)       shift; mpvcfzf_lofi "${1:-}"; return ;;
        somafm)     shift; mpvcfzf_somafm "${1:-}"; return ;;
        radioapi)   shift; mpvcfzf_radioapi "${1:-}"; return ;;
        #*)         shift; mpvcfzf_usage; return ;;
        esac
    done
}

main "$@"