Prioritize correctly between different matching files.

[utils.git] / planime

#!/bin/bash

isnum() {
    grep -xq '[0-9]\+' <<<"$1"
}

scorefile() {
    if [ "${1##*.}" = mkv ]; then
	echo 10
    elif [ "${1##*.}" = ogm ]; then
	echo 5
    else
	echo 0
    fi
}

findfile() {
    local file tail eq eqt m matches max score
    matches=()
    for file in "$1"*; do
	tail="${file#$1}"
	eq="${tail%.*}"
	m=n
	if [ "${eq%% *}" -eq "$2" ] 2>/dev/null; then
	    if [[ "$eq" == *\ * ]]; then
		eqt="${eq#* }"
	    else
		eqt=
	    fi
	    m=y
	elif [ "${eq:0:${#2}}" = "$2" ]; then
	    eqt="${eq:${#2}}"
	    if [ "${eqt:0:1}" = " " -o -z "$eqt" ]; then
		eqt="${eqt# }"
		m=y
	    fi
	fi
	if [ "$m" = y ]; then
	    if [ "$eqt" = "$3" -o "$eqt" = "($3)" ]; then
		matches=("${matches[@]}" "$file")
	    fi
	fi
    done
    if [ ${#matches[@]} -lt 1 ]; then return 1; fi
    max=
    for m in "${matches[@]}"; do
	score="$(scorefile "$m")"
	if [ -z "$max" ] || [ "$score" -gt "$max" ]; then
	    max="$score"
	    file="$m"
	fi
    done
    echo "$file"
    return 0
}

origargs=("$0" "$@")
cmdline=(mplayer -fs -ass)
unset pretend

while [ "${1:0:1}" = - ]; do
    a="$1"
    shift
    case "$a" in
	-h)
	    echo "usage: planime [-htC] [-s PAR VAL] [NAME-QUAL] [EP|.] [TYPE-QUAL]" >&2
	    exit 0
	    ;;
	-t)
	    cmdline=("${cmdline[@]}" -ao oss:/dev/dsp1)
	    ;;
	-C)
	    pretend=y
	    ;;
	-A)
	    while :; do
		a="$1"
		shift
		if [ "$a" = \; ]; then
		    break;
		fi
		cmdline=("${cmdline[@]}" "$a")
	    done
	    ;;
	-s)
	    savepar="$1"
	    shift
	    saveval="$1"
	    shift
	    ;;
    esac
done

base="$(basename "$(pwd)")"
nextep=n
file=
tqual=
if [ $# -eq 0 ]; then
    nextep=y
    if [ -r curser ]; then
	base="$base $(<curser)"
    fi
elif [ $# -eq 1 ]; then
    if [ -r "$1" ]; then
	file="$1"
    else
	if findfile "$base - " "$1" >/dev/null; then
	    ep="$1"
	else
	    nextep=y
	    base="$base $1"
	fi
    fi
elif [ $# -eq 2 ]; then
    if findfile "$base - " "$1" "$2" >/dev/null; then
	ep="$1"
	tqual="$2"
    else
	if [ "$1" = . -a -r curser ]; then
	    base="$base $(<curser)"
	else
	    base="$base $1"
	fi
	ep="$2"
    fi
else
    if [ "$1" = . -a -r curser ]; then
	base="$base $(<curser)"
    else
	base="$base $1"
    fi
    ep="$2"
    tqual="$3"
fi
if [ "$ep" = . ]; then nextep=y; fi
if [ "$nextep" = y -a -r nextep ]; then
    ep="$(<nextep)"
    if ! isnum "$ep"; then
	echo "planime: nextep is non-numeric" >&2
	exit 1
    fi
fi
if [ -z "$file" ]; then
    file="$(findfile "$base - " "$ep" "$tqual")"
fi

if [ -z "$file" -o ! -r "$file" ]; then
    echo "planime: no matching file found" >&2
    exit 1
fi

case "${file##*.}" in
    ogm)
	aid=1
	sid=0
	;;
    mkv)
	alang=jpn
	slang=eng
	;;
esac

ifile=".${file}.info"

if [ -n "$savepar" ]; then
    if [ -r "$ifile" ]; then
	egrep -v "^${savepar} " "$ifile" >"$ifile.new"
	mv -f "$ifile.new" "$ifile"
    fi
    echo "$savepar $saveval" >>"$ifile"
    exit 0
fi

unset delay

if [ -r "$ifile" ]; then
    exec 3<&0
    exec 0<"$ifile"
    while read par arg; do
	if [ "$par" = delay ]; then
	    cmdline=("${cmdline[@]}" -delay "$arg")
	elif [ "$par" = aspect ]; then
	    cmdline=("${cmdline[@]}" -aspect "$arg")
	elif [ "$par" = volmod ]; then
	    cmdline=("${cmdline[@]}" -af volume="$arg")
	elif [ "$par" = alang ]; then
	    unset alang aid
	    alang="$arg"
	elif [ "$par" = aid ]; then
	    unset alang aid
	    aid="$arg"
	elif [ "$par" = slang ]; then
	    unset slang sid
	    slang="$arg"
	elif [ "$par" = sid ]; then
	    unset slang sid
	    sid="$arg"
	fi
    done
    exec 0<&3
    exec 3<&-
fi

if [ -n "$alang" ]; then
    cmdline=("${cmdline[@]}" -alang "$alang")
elif [ -n "$aid" ]; then
    cmdline=("${cmdline[@]}" -aid "$aid")
fi
if [ -n "$slang" ]; then
    cmdline=("${cmdline[@]}" -slang "$slang")
elif [ -n "$sid" ]; then
    cmdline=("${cmdline[@]}" -sid "$sid")
fi

if [ "`hostname`" = pc6 -o "`hostname`" = pc7 -a "$DISPLAY" = :1 ]; then chwp=y; fi

if [ -n "$pretend" ]; then
    echo "${cmdline[@]}" "$file"
else
    if [ "$chwp" = y ]; then (sleep 2; randomwp) & fi
    "${cmdline[@]}" "$file"
    
    if [ "$nextep" = y ]; then
	echo "0. Save and continue (or Space)"
	echo "1. Continue without saving"
	echo "2. Save and exit (or Enter)"
	echo "3. Exit without saving (or any key)"
	IFS= read -sn1 c
	save=n
	cont=n
	case "$c" in
	    0|" ")
		save=y
		cont=y
		;;
	    1)
		cont=y
		;;
	    2|"")
		save=y
		;;
	esac
	if [ "$save" = y ]; then
	    let ep++
	    echo "$ep" >nextep
	fi
	if [ "$cont" = y ]; then
	    exec "${origargs[@]}"
	else
	    echo "nextep is $ep"
	fi
    fi
fi