elcleaner

#!/usr/bin/env bash

usage() {
cat - <<EOF
Usage: ${0##*/} -i <INPUT FILE> -o <OUTPUT FILE>

-h|--help:          show this help message
-i|--input:         file name for input
-o|--output:        file name for output
-d|--directory:     directory for output file (default: .)
-a|--aformat:       audio format of main original track (used for track title, e.g. "dtshd")
-c|--cformat:       channel format/layout of main original track (used for track title, e.g. "5.1")
-al|--alangs:       audio languages to keep (valid ISO 639 language codes, comma separated, default: eng)
-sl|--slangs:       subtitle languages to keep (valid ISO 639 language codes, comma separated, default: eng)
-at|--atracks:      comma separated list of audio track numbers to keep
-st|--stracks:      comma separated list of subtitle track numbers to keep
-sc|--stripcompat:  strip ac3 compatibility tracks if found
-fo|--firstonly:    only keep the first audio track
-nc|--nochapters:   don't include chapters from original
-ka|--keepattach:   keep attachments (useful for subtitle fonts)
-nm|--nomediainfo:  do not show mediainfo output summary at the end
--tracknames:       a list like '<track num>:"<name string>",<track num>:"<name string>"'
--debug:            print debugging information

EOF
}

truncate() {
    echo "$@" | $AWK -v len=60 '{ if (length($0) > len) print substr($0, 1, len-3) "..."; else print; }'
}


# Check for tool dependencies
SELF_FULLPATH=`realpath $0`
SELF_DIR=`dirname $SELF_FULLPATH`
# Override default tool list
TOOLS=(mkvmerge mkvpropedit jq sed awk column mediainfo tr bc base64)
# Does not depend on any other scripts
SCRIPTS=NONE
if [ ! -r $SELF_DIR/toolchecker ]; then
    echo "toolchecker not found."
    exit 1
else
    . $SELF_DIR/toolchecker
fi

while [ $# -gt 0 ]; do
    OPT="$1"
    case "${OPT}" in
    -i|--input)
        shift
        INPUT="${1}"
        shift;;
    -o|--output)
        shift
        OUTPUT="${1}"
        shift;;
    -d|--directory)
        shift
        OUTPUT_DIR="${1}"
        shift;;
    -al|--alangs|-at|--atracks)
        shift
        AUDIO_TRACKS="${1}"
        shift;;
    -sl|--slangs|-st|--stracks)
        shift
        SUBTITLE_TRACKS="${1}"
        shift;;
    -a|--aformat)
        shift
        AFORMAT="${1}"
        shift;;
    -c|--cformat)
        shift
        CFORMAT="${1}"
        shift;;
    -sc|--stripcompat)
        STRIPCOMPAT=1
        shift;;
    -fo|--firstonly)
        FIRSTONLY=1
        shift;;
    -nc|--nochapters)
        NOCHAPTERS=1
        shift;;
    -ka|--keepattach)
        KEEPATTACH=1
        shift;;
    -nm|--nomediainfo)
        NOMI=1
        shift;;
    --tracknames)
        shift
        TRACKNAMES="$1"
        shift;;
    --debug)
        DEBUG=1
        shift;;
    -h|--help)
        usage
        exit;;
    *)
        echo "$0: invalid option ${1}"
        echo "$0 -h for usage options"
        usage
        exit 1;;
    esac
done

if [ -z "$INPUT" ]; then
    echo "No input file provided."
    usage
    exit 1
fi

if [ -n "$TRACKNAMES" ] && [ -n "$AFORMAT" ]; then
    echo "Can't do both tracknames and aformat cause I'm lazy, pick one, try again"
    exit 1
fi

if [ -z "$OUTPUT_DIR" ] && [ -n "$OUTPUT" ]; then
    OUTPUT="${OUTPUT}"
elif [ -n "$OUTPUT_DIR" ] && [ -z "$OUTPUT" ]; then
    OUTPUT_FILE=$(basename "${INPUT}")
    OUTPUT="${OUTPUT_DIR%/}/${OUTPUT_FILE}"
elif [ -n "$OUTPUT_DIR" ] && [ -n "$OUTPUT" ]; then
    OUTPUT="${OUTPUT_DIR%/}/${OUTPUT}"
elif [ -z "$OUTPUT_DIR" ] && [ -z "$OUTPUT" ]; then
    OUTPUT="${INPUT:0: -4}_1.mkv"
fi

if [ -f "${OUTPUT}" ]; then
    echo "${OUTPUT} already exists. Exiting."
    exit 1
fi

if [ -z "$AUDIO_TRACKS" ]; then
    AUDIO_TRACKS="eng"
fi

if [ $FIRSTONLY ]; then
    AUDIO_TRACKS="1"
fi

if [ -z "$SUBTITLE_TRACKS" ]; then
    SUBTITLE_TRACKS="eng"
fi

if [ $DEBUG ]; then
    echo "INPUT: ${INPUT}"
    echo "OUTPUT: ${OUTPUT}"
    echo "OUTPUT_DIR: ${OUTPUT_DIR}"
    echo "AFORMAT: ${AFORMAT}"
    echo "CFORMAT: ${CFORMAT}"
    echo "AUDIO_TRACKS: ${AUDIO_TRACKS}"
    echo "SUBTITLE_TRACKS: ${SUBTITLE_TRACKS}"

    JSON=`$MKVMERGE -F json --identify "$INPUT"`
    track_array=()
    track_array+=("ID;SELECTOR;TYPE;CODEC;LANG;DEFAULT;FORCED;NAME")
    track_array+=("--;--------;----;-----;----;-------;------;----")
    while read -r line; do
        ID=`echo "$line" | $JQ -c '.id'`
        TYPE=`echo "$line" | $JQ -c '.type' | $SED 's/\"//g'`
        CODEC=`echo "$line" | $JQ -c '.codec' | $SED 's/\"//g'`
        NAME=`echo "$line" | $JQ -c '.properties.track_name' | $SED 's/\"//g'`
        LANGUAGE=`echo "$line" | $JQ -c '.properties.language' | $SED 's/\"//g'`
        DEFAULT=`echo "$line" | $JQ -c '.properties.default_track'`
        FORCED=`echo "$line" | $JQ -c '.properties.forced_track'`
        SELECTOR="${TYPE:0:1}${!TYPE}"
        track_array+=("${ID};${SELECTOR};${TYPE};${CODEC};${LANGUAGE};${DEFAULT};${FORCED};$(truncate $NAME)")
    done < <(echo "$JSON" | $JQ -c '.tracks[]')
fi

TRACKS=`$MEDIAINFO --output=JSON "$INPUT" | ${JQ} -r '.media.track[] | select (."@type"=="Audio") | @base64' | head -n 1`
CODECID=$(echo "${TRACKS}" | ${BASE64} --decode | ${JQ} -r '.CodecID') 
CLAYOUT=$(echo "${TRACKS}" | ${BASE64} --decode | ${JQ} -r '.ChannelLayout') 
COMPRESSION=$(echo "${TRACKS}" | ${BASE64} --decode | ${JQ} -r '.Compression_Mode')
[ -n "${CFORMAT}" ] || CFORMAT=$(echo "${CLAYOUT} 0.0" | ${SED} -e 's~LFE~.1~g' -e 's~Cb\|C\|Cs\|M\|L\|Lt\|R\|Rt\|Rss\|Lss\|Rsr\|Lsr\|Rb\|Lb\|Rs\|Ls\|Rw\|Lw~1~g' | ${TR} -s ' ' '+' | ${BC})
[ $DEBUG ] && echo "CODECID: ${CODECID}"
[ $DEBUG ] && echo "CLAYOUT: ${CLAYOUT}"
[ $DEBUG ] && echo "CFORMAT: ${CFORMAT}"
[ $DEBUG ] && echo "COMPRESSION: ${COMPRESSION}"
if [ -z "${AFORMAT}" ]; then
    case "${CODECID}" in
        "A_EAC3") AFORMAT="eac3";;
        "A_AC3") AFORMAT="ac3";;
        "A_AAC") AFORMAT="aac";;
        "A_TRUEHD") AFORMAT="atmos";;
        "A_DTS") [ "$COMPRESSION" == "Lossy" ] && AFORMAT="dts" || AFORMAT="dtsx";;
        "A_OPUS"|"A_AAC")
            SOURCEDIR=${PWD##*/}
            if [[ ${SOURCEDIR} =~ "DTS-HD.MA" ]]; then
                AFORMAT="dtshd"
            elif [[ ${SOURCEDIR} =~ "DTS-HD.HRA" ]]; then
                AFORMAT="dtshra"
            elif [[ ${SOURCEDIR} =~ "TrueHD" ]]; then
                AFORMAT="truehd"
            elif [[ ${SOURCEDIR} =~ "FLAC" ]]; then
                AFORMAT="lpcm"
            elif [[ ${SOURCEDIR} =~ "LPCM" ]]; then
                AFORMAT="lpcm"
            elif [[ ${SOURCEDIR} =~ "Opus" ]]; then
                AFORMAT="opus"
            elif [[ ${SOURCEDIR} =~ "AAC" ]]; then
                AFORMAT="AAC"
            else
                read -p "Enter original aformat: " AFORMAT
            fi;;
    esac
fi

case "$AFORMAT" in
    dtsx)   TRACKNAME="DTS-X MA"; [ -n "$CFORMAT" ] && echo "No channel format with object-based codecs, ignoring";;
    atmos)  TRACKNAME="Dolby TrueHD Atmos"; [ -n "$CFORMAT" ] && echo "No channel format with object-based codecs, ignoring";;
    truehd) [ -n "$CFORMAT" ] && TRACKNAME="Opus from TrueHD ${CFORMAT}" || NOCFORMAT=1;;
    dtshd)  [ -n "$CFORMAT" ] && TRACKNAME="Opus from DTS-HD MA ${CFORMAT}" || NOCFORMAT=1;;
    dtshra) [ -n "$CFORMAT" ] && TRACKNAME="Opus from DTS-HD HRA ${CFORMAT}" || NOCFORMAT=1;;
    flac)   [ -n "$CFORMAT" ] && TRACKNAME="Opus from FLAC ${CFORMAT}" || NOCFORMAT=1;;
    lpcm)   [ -n "$CFORMAT" ] && TRACKNAME="Opus from LPCM ${CFORMAT}" || NOCFORMAT=1;;
    ac3)    [ -n "$CFORMAT" ] && TRACKNAME="Dolby AC-3 ${CFORMAT}" || NOCFORMAT=1;;
    eac3)   [ -n "$CFORMAT" ] && TRACKNAME="Dolby EAC-3 ${CFORMAT}" || NOCFORMAT=1;;
    opus)   [ -n "$CFORMAT" ] && TRACKNAME="Opus ${CFORMAT}" || NOCFORMAT=1;;
    aac)    [ -n "$CFORMAT" ] && TRACKNAME="AAC ${CFORMAT}" || NOCFORMAT=1;;
    dts)    [ -n "$CFORMAT" ] && TRACKNAME="DTS ${CFORMAT}" || NOCFORMAT=1;;
    nope)   TRACKNAME=""; echo "Not updating primary audio track title.";; # No value provided, do not need output here, just a no-op
    "")     TRACKNAME=""; echo "Not updating primary audio track title.";; # No value provided, do not need output here, just a no-op
    *)      echo "Unrecognized format shorthand, text as entered, with channel format, if provided"; [ -n "$CFORMAT" ] && TRACKNAME="${AFORMAT} ${CFORMAT}" || TRACKNAME="${AFORMAT}";;
esac

if [ $NOCFORMAT ]; then
    echo "Must provide channel format for audio format ${AFORMAT}"
    exit 1
fi

printf '%s\n' "${track_array[@]}" | $COLUMN -t -s ';'


IFS=""

## I'm breaking this up so I can play more games with the order later if I need to
MERGEOPTIONS=()
MERGEOPTIONS+=(-o "${OUTPUT}")
[ $KEEPATTACH ] || MERGEOPTIONS+=(--no-attachments)
MERGEOPTIONS+=(--no-track-tags
--no-global-tags
--no-buttons)
MERGEOPTIONS+=(-a ${AUDIO_TRACKS})
MERGEOPTIONS+=(-s ${SUBTITLE_TRACKS})
## the space at the end is important, it unsets the track name entirely
MERGEOPTIONS+=(--track-name 0:\ )
MERGEOPTIONS+=(--language 0:eng)
#MERGEOPTIONS+=(--language 1:en)
[ $STRIPCOMPAT ] && MERGEOPTIONS+=(-a !2)
[ $NOCHAPTERS ] && MERGEOPTIONS+=(--no-chapters)
MERGEOPTIONS+=("${INPUT}")

[ "$TRACKNAME" != "" ] && PROPTIONS+=(--edit track:a1 --set "name=${TRACKNAME}")
if [ -n "$TRACKNAMES" ]; then
    IFS=$'\n'
    for ITEM in $(echo "$TRACKNAMES" | tr ',' '\n'); do
        sleep 1
        #PROPTIONS+=(--track-name ${ITEM})
    done
    IFS=""
fi
unset IFS

if [ $DEBUG ]; then
    echo "COMMANDS:"
    echo "$MKVMERGE $(printf " %q" "${MERGEOPTIONS[@]}")"
    echo "$MKVPROPEDIT "${OUTPUT}" $(printf " %q" "${PROPTIONS[@]}")"
    read -p "Press y to continue, any other key to exit..." CONFIRM
    case $CONFIRM in
        [Yy] ) true;;
        * ) exit;;
    esac
fi

echo "## Merging..."
set -x
($MKVMERGE "${MERGEOPTIONS[@]}")
{ set +x; } 2>/dev/null

if [ $? -eq 0 ]; then
    echo "## Merge complete."
else
    echo "<> merge error"
    exit 1
fi

echo "## Setting audio track names"
set -x
(${MKVPROPEDIT} "${OUTPUT}" "${PROPTIONS[@]}")
{ set +x; } 2>/dev/null

if [ $? -eq 0 ]; then
    echo "## Track naming complete."
else
    echo "<> track naming error"
    exit 1
fi


echo "## Theraputic tag strip on output file"
set -x
$MKVPROPEDIT "${OUTPUT}" --tags all:"" --delete title --edit track:v1 --delete name
{ set +x; } 2>/dev/null
if [ $? -eq 0 ]; then
    echo "## Tags stripped."
    echo "## Adding back stats"
    set -x
    $MKVPROPEDIT --add-track-statistics-tags "${OUTPUT}"
    { set +x; } 2>/dev/null
    if [ $? -eq 0 ]; then
        echo "## Stats added"
    else
        echo "## Stats adding error"
        exit 1
    fi
else
    echo "<> Tag stripping error"
    exit 1
fi

[ $NOMI ] || $MEDIAINFO "${OUTPUT}"