RandomNinjaAtk
4 years ago
committed by
GitHub
parent
2f3e9900a8
commit
3a67ea6597
@ -0,0 +1,344 @@
|
||||
#!/usr/bin/with-contenv bash
|
||||
|
||||
set -e
|
||||
|
||||
# start
|
||||
|
||||
echo ""
|
||||
|
||||
# check for video files
|
||||
if find "$1" -type f -iregex ".*/.*\.\(mkv\|mp4\|avi\)" | read; then
|
||||
sleep 0.1
|
||||
else
|
||||
echo "ERROR: No video files found for processing"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo "Script Configuration:"
|
||||
echo "Required Audio/Subtitle Language: ${VIDEO_LANG}"
|
||||
if [ ${VIDEO_MKVCLEANER} = TRUE ]; then
|
||||
echo "Video Post Processing with MKV Cleaner: ENABLED"
|
||||
else
|
||||
echo "Video Post Processing with MKV Cleaner: DISABLED"
|
||||
fi
|
||||
if [ ${VIDEO_SMA} = TRUE ]; then
|
||||
echo "Video Post Processing with SMA: ENABLED"
|
||||
if [ ${VIDEO_SMA_TAGGING} = TRUE ]; then
|
||||
echo "Video Post Processing with SMA Tagging: ENABLED"
|
||||
fi
|
||||
else
|
||||
echo "Video Post Processing with SMA: DISABLED"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
if [ ${VIDEO_SMA} = TRUE ]; then
|
||||
touch "$1/sma-conversion-check"
|
||||
elif [ ${VIDEO_MKVCLEANER} = TRUE ]; then
|
||||
touch "$1/sma-conversion-check"
|
||||
fi
|
||||
|
||||
if [ -z "VIDEO_SMA_TAGGING" ]; then
|
||||
VIDEO_SMA_TAGGING=FALSE
|
||||
fi
|
||||
|
||||
if [ ${VIDEO_SMA_TAGGING} = TRUE ]; then
|
||||
tagging="-a"
|
||||
else
|
||||
tagging="-nt"
|
||||
fi
|
||||
|
||||
filecount=$(find "$1" -type f -iregex ".*/.*\.\(mkv\|mp4\|avi\)" | wc -l)
|
||||
echo "Processing ${filecount} video files..."
|
||||
count=0
|
||||
find "$1" -type f -iregex ".*/.*\.\(mkv\|mp4\|avi\)" -print0 | while IFS= read -r -d '' video; do
|
||||
count=$(($count+1))
|
||||
echo ""
|
||||
echo "===================================================="
|
||||
basefilename="${video%.*}"
|
||||
filename="$(basename "$video")"
|
||||
extension="${filename##*.}"
|
||||
echo "Begin processing $count of $filecount: $filename"
|
||||
echo "Checking for audio/subtitle tracks"
|
||||
tracks=$(mkvmerge -J "$video")
|
||||
if [ ! -z "${tracks}" ]; then
|
||||
# video tracks
|
||||
VideoTrack=$(echo "${tracks}" | jq ".tracks[] | select(.type==\"video\") | .id")
|
||||
VideoTrackCount=$(echo "${tracks}" | jq ".tracks[] | select(.type==\"video\") | .id" | wc -l)
|
||||
# video preferred language
|
||||
VideoTrackLanguage=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"video\") and select(.properties.language==\"${VIDEO_LANG}\")) | .id")
|
||||
# audio tracks
|
||||
AudioTracks=$(echo "${tracks}" | jq ".tracks[] | select(.type==\"audio\") | .id")
|
||||
AudioTracksCount=$(echo "${tracks}" | jq ".tracks[] | select(.type==\"audio\") | .id" | wc -l)
|
||||
# audio preferred language
|
||||
AudioTracksLanguage=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"audio\") and select(.properties.language==\"${VIDEO_LANG}\")) | .id")
|
||||
AudioTracksLanguageCount=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"audio\") and select(.properties.language==\"${VIDEO_LANG}\")) | .id" | wc -l)
|
||||
AudioTracksLanguageFound=$(echo "${tracks}" | jq ".tracks[] | select(.type==\"audio\") | .properties.language")
|
||||
# audio unkown laguage
|
||||
AudioTracksLanguageUND=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"audio\") and select(.properties.language==\"und\")) | .id")
|
||||
AudioTracksLanguageUNDCount=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"audio\") and select(.properties.language==\"und\")) | .id" | wc -l)
|
||||
AudioTracksLanguageNull=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"audio\") and select(.properties.language==null)) | .id")
|
||||
AudioTracksLanguageNullCount=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"audio\") and select(.properties.language==null)) | .id" | wc -l)
|
||||
# audio foreign language
|
||||
AudioTracksLanguageForeignCount=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"audio\") and select(.properties.language!=\"${VIDEO_LANG}\")) | .id" | wc -l)
|
||||
# subtitle tracks
|
||||
SubtitleTracks=$(echo "${tracks}" | jq ".tracks[] | select(.type==\"subtitles\") | .id")
|
||||
SubtitleTracksCount=$(echo "${tracks}" | jq ".tracks[] | select(.type==\"subtitles\") | .id" | wc -l)
|
||||
# subtitle preferred langauge
|
||||
SubtitleTracksLanguage=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"subtitles\") and select(.properties.language==\"${VIDEO_LANG}\")) | .id")
|
||||
SubtitleTracksLanguageCount=$(echo "${tracks}" | jq ".tracks[] | select((.type==\"subtitles\") and select(.properties.language==\"${VIDEO_LANG}\")) | .id" | wc -l)
|
||||
SubtitleTracksLanguageFound=$(echo "${tracks}" | jq ".tracks[] | select(.type==\"subtitles\") | .properties.language")
|
||||
|
||||
else
|
||||
echo "ERROR: ffprobe failed to read tracks and set values"
|
||||
rm "$video" && echo "INFO: deleted: $video"
|
||||
fi
|
||||
|
||||
# Check for video track
|
||||
if [ -z "${VideoTrack}" ]; then
|
||||
echo "ERROR: no video track found"
|
||||
rm "$video" && echo "INFO: deleted: $filename"
|
||||
continue
|
||||
else
|
||||
echo "$VideoTrackCount video track found!"
|
||||
fi
|
||||
|
||||
# Check for audio track
|
||||
if [ -z "${AudioTracks}" ]; then
|
||||
echo "ERROR: no audio tracks found"
|
||||
rm "$video" && echo "INFO: deleted: $filename"
|
||||
continue
|
||||
else
|
||||
echo "$AudioTracksCount audio tracks found!"
|
||||
fi
|
||||
|
||||
# Check for audio track
|
||||
if [ ! -z "${SubtitleTracks}" ]; then
|
||||
echo "$SubtitleTracksCount subtitle tracks found!"
|
||||
fi
|
||||
|
||||
if [ ! -z "$AudioTracksLanguage" ] || [ ! -z "$SubtitleTracksLanguage" ]; then
|
||||
if [ ${VIDEO_MKVCLEANER} = TRUE ] || [ ${VIDEO_SMA} = TRUE ]; then
|
||||
if [ ! -z "$AudioTracksLanguage" ] || [ ! -z "$SubtitleTracksLanguage" ] || [ ! -z "$AudioTracksLanguageUND" ] || [ ! -z "$AudioTracksLanguageNull" ]; then
|
||||
sleep 0.1
|
||||
else
|
||||
echo "Checking for \"${VIDEO_LANG}\" video/audio/subtitle tracks"
|
||||
echo "ERROR: No \"${VIDEO_LANG}\" or \"Unknown\" audio tracks found..."
|
||||
echo "ERROR: No \"${VIDEO_LANG}\" subtitle tracks found..."
|
||||
# rm "$video" && echo "INFO: deleted: $filename"
|
||||
exit 1
|
||||
continue
|
||||
fi
|
||||
else
|
||||
echo "Checking for \"${VIDEO_LANG}\" video/audio/subtitle tracks"
|
||||
if [ ! -z "${AudioTracks}" ]; then
|
||||
echo "INFO: ${AudioTracksLanguageFound} audio track found!"
|
||||
fi
|
||||
if [ ! -z "${SubtitleTracks}" ]; then
|
||||
echo "INFO: ${SubtitleTracksLanguageFound} subtitle track found!"
|
||||
fi
|
||||
echo "ERROR: No \"${VIDEO_LANG}\" audio or subtitle tracks found..."
|
||||
# rm "$video" && echo "INFO: deleted: $filename"
|
||||
exit 1
|
||||
continue
|
||||
fi
|
||||
else
|
||||
if [ ! ${VIDEO_MKVCLEANER} = TRUE ] || [ ! ${VIDEO_SMA} = TRUE ]; then
|
||||
echo "Checking for \"${VIDEO_LANG}\" video/audio/subtitle tracks"
|
||||
if [ ! -z "$AudioTracksLanguage" ]; then
|
||||
echo "$AudioTracksLanguageCount \"${VIDEO_LANG}\" audio track found..."
|
||||
fi
|
||||
if [ ! -z "$SubtitleTracksLanguage" ]; then
|
||||
echo "$SubtitleTracksLanguageCount \"${VIDEO_LANG}\" subtitle track found..."
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ ${VIDEO_MKVCLEANER} = TRUE ]; then
|
||||
echo "Begin processing with MKV Cleaner..."
|
||||
echo "Checking for \"${VIDEO_LANG}\" video/audio/subtitle tracks"
|
||||
# Check for unwanted audio tracks and remove/re-label as needed...
|
||||
if [ ! -z "$AudioTracksLanguage" ] || [ ! -z "$AudioTracksLanguageUND" ] || [ ! -z "$AudioTracksLanguageNull" ]; then
|
||||
if [ $AudioTracksCount -ne $AudioTracksLanguageCount ]; then
|
||||
RemoveAudioTracks="true"
|
||||
if [ ! -z "$AudioTracksLanguage" ]; then
|
||||
MKVaudio=" -a ${VIDEO_LANG}"
|
||||
echo "$AudioTracksLanguageCount audio tracks found!"
|
||||
unwantedaudiocount=$(($AudioTracksCount-$AudioTracksLanguageCount))
|
||||
if [ $AudioTracksLanguageCount -ne $AudioTracksCount ]; then
|
||||
unwantedaudio="true"
|
||||
fi
|
||||
elif [ ! -z "$AudioTracksLanguageUND" ]; then
|
||||
for I in $AudioTracksLanguageUND
|
||||
do
|
||||
OUT=$OUT" -a $I --language $I:${VIDEO_LANG}"
|
||||
done
|
||||
MKVaudio="$OUT"
|
||||
echo "$AudioTracksLanguageUNDCount \"unknown\" audio tracks found, re-tagging as \"${VIDEO_LANG}\""
|
||||
unwantedaudiocount=$(($AudioTracksCount-$AudioTracksLanguageUNDCount))
|
||||
if [ $AudioTracksLanguageUNDCount -ne $AudioTracksCount ]; then
|
||||
unwantedaudio="true"
|
||||
fi
|
||||
elif [ ! -z "$AudioTracksLanguageNull" ]; then
|
||||
for I in $AudioTracksLanguageNull
|
||||
do
|
||||
OUT=$OUT" -a $I --language $I:${VIDEO_LANG}"
|
||||
done
|
||||
MKVaudio="$OUT"
|
||||
echo "$AudioTracksLanguageNullCount \"unknown\" audio tracks found, re-tagging as \"${VIDEO_LANG}\""
|
||||
unwantedaudiocount=$(($AudioTracksCount-$AudioTracksLanguageNullCount))
|
||||
if [ $AudioTracksLanguageNullCount -ne $AudioTracksCount ]; then
|
||||
unwantedaudio="true"
|
||||
fi
|
||||
fi
|
||||
else
|
||||
echo "$AudioTracksLanguageCount audio tracks found!"
|
||||
RemoveAudioTracks="false"
|
||||
MKVaudio=""
|
||||
fi
|
||||
elif [ -z "$SubtitleTracksLanguage" ]; then
|
||||
if [ ! -z "${AudioTracks}" ]; then
|
||||
echo "INFO: ${AudioTracksLanguageFound} audio track found!"
|
||||
fi
|
||||
if [ ! -z "${SubtitleTracks}" ]; then
|
||||
echo "INFO: ${SubtitleTracksLanguageFound} subtitle track found!"
|
||||
fi
|
||||
echo "ERROR: no \"${VIDEO_LANG}\" audio/subtitle tracks found!"
|
||||
# rm "$video" && echo "INFO: deleted: $filename"
|
||||
exit 1
|
||||
continue
|
||||
else
|
||||
foreignaudio="true"
|
||||
RemoveAudioTracks="false"
|
||||
MKVaudio=""
|
||||
fi
|
||||
|
||||
# Check for unwanted subtitle tracks...
|
||||
if [ ! -z "$SubtitleTracks" ]; then
|
||||
if [ $SubtitleTracksCount -ne $SubtitleTracksLanguageCount ]; then
|
||||
RemoveSubtitleTracks="true"
|
||||
MKVSubtitle=" -s ${VIDEO_LANG}"
|
||||
if [ ! -z "$SubtitleTracksLanguage" ]; then
|
||||
echo "$SubtitleTracksLanguageCount subtitle tracks found!"
|
||||
fi
|
||||
unwantedsubtitlecount=$(($SubtitleTracksCount-$SubtitleTracksLanguageCount))
|
||||
if [ $SubtitleTracksLanguageCount -ne $SubtitleTracksCount ]; then
|
||||
unwantedsubtitle="true"
|
||||
fi
|
||||
else
|
||||
echo "$SubtitleTracksLanguageCount subtitle tracks found!"
|
||||
RemoveSubtitleTracks="false"
|
||||
MKVSubtitle=""
|
||||
fi
|
||||
else
|
||||
RemoveSubtitleTracks="false"
|
||||
MKVSubtitle=""
|
||||
fi
|
||||
|
||||
# Correct video language, if needed...
|
||||
if [ -z "$VideoTrackLanguage" ]; then
|
||||
if [ ! -z "$AudioTracksLanguage" ] || [ ! -z "$AudioTracksLanguageUND" ] || [ ! -z "$AudioTracksLanguageNull" ]; then
|
||||
SetVideoLanguage="true"
|
||||
if [ "${RemoveAudioTracks}" = true ] || [ "${RemoveSubtitleTracks}" = true ]; then
|
||||
echo "$VideoTrackCount \"unknown\" video language track found, re-tagging as \"${VIDEO_LANG}\""
|
||||
fi
|
||||
MKVvideo=" -d ${VideoTrack} --language ${VideoTrack}:${VIDEO_LANG}"
|
||||
else
|
||||
foreignvideo="true"
|
||||
SetVideoLanguage="false"
|
||||
MKVvideo=""
|
||||
fi
|
||||
else
|
||||
echo "$VideoTrackCount video track found!"
|
||||
SetVideoLanguage="false"
|
||||
MKVvideo=""
|
||||
fi
|
||||
|
||||
# Display foreign audio track counts
|
||||
if [ "$foreignaudio" = true ] || [ "$foreignvideo" = true ]; then
|
||||
echo "Checking for \"foreign\" audio/video tracks"
|
||||
if [ "$foreignvideo" = true ]; then
|
||||
echo "$VideoTrackCount video track found!"
|
||||
foreignvideo="false"
|
||||
fi
|
||||
if [ "$foreignaudio" = true ]; then
|
||||
echo "$AudioTracksLanguageForeignCount audio tracks found!"
|
||||
foreignaudio="false"
|
||||
fi
|
||||
fi
|
||||
|
||||
# Display unwanted audio/subtitle track counts
|
||||
if [ "$unwantedaudio" = true ] || [ "$unwantedsubtitle" = true ]; then
|
||||
echo "Checking for unwanted \"not: ${VIDEO_LANG}\" audio/subtitle tracks"
|
||||
if [ "$unwantedaudio" = true ]; then
|
||||
echo "$unwantedaudiocount audio tracks to remove..."
|
||||
unwantedaudio="false"
|
||||
fi
|
||||
if [ "$unwantedsubtitle" = true ]; then
|
||||
echo "$unwantedsubtitlecount subtitle tracks to remove..."
|
||||
unwantedsubtitle="false"
|
||||
fi
|
||||
fi
|
||||
skip="false"
|
||||
if [ "${RemoveAudioTracks}" = false ] && [ "${RemoveSubtitleTracks}" = false ]; then
|
||||
if find "$video" -type f -iname "*.mkv" | read; then
|
||||
echo "INFO: Video passed all checks, no processing needed"
|
||||
touch "$video"
|
||||
if [ ${VIDEO_SMA} = TRUE ]; then
|
||||
skip="true"
|
||||
fi
|
||||
else
|
||||
echo "INFO: Video passed all checks, but is in the incorrect container, repackaging as mkv..."
|
||||
MKVvideo=" -d ${VideoTrack} --language ${VideoTrack}:${VIDEO_LANG}"
|
||||
MKVaudio=" -a ${VIDEO_LANG}"
|
||||
MKVSubtitle=" -s ${VIDEO_LANG}"
|
||||
fi
|
||||
fi
|
||||
if [ $skip = false ]; then
|
||||
if mkvmerge --no-global-tags --title "" -o "${basefilename}.merged.mkv"${MKVvideo}${MKVaudio}${MKVSubtitle} "$video"; then
|
||||
echo "SUCCESS: mkvmerge complete"
|
||||
echo "INFO: Options used:${MKVvideo}${MKVaudio}${MKVSubtitle}"
|
||||
# cleanup temp files and rename
|
||||
mv "$video" "$video.original" && echo "INFO: Renamed source file"
|
||||
mv "${basefilename}.merged.mkv" "${basefilename}.mkv" && echo "INFO: Renamed temp file"
|
||||
rm "$video.original" && echo "INFO: Deleted source file"
|
||||
extension="mkv"
|
||||
else
|
||||
echo "ERROR: mkvmerge failed"
|
||||
rm "$video" && echo "INFO: deleted: $video"
|
||||
rm "${basefilename}.merged.mkv" && echo "INFO: deleted: ${basefilename}.merged.mkv"
|
||||
continue
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
if [ ${VIDEO_SMA} = TRUE ]; then
|
||||
if [ -f "${basefilename}.${extension}" ]; then
|
||||
echo ""
|
||||
echo "Begin processing with Sickbeard MP4 Automator..."
|
||||
echo ""
|
||||
# Manual run of Sickbeard MP4 Automator
|
||||
if python3 /usr/local/sma/manual.py --config "$2" -i "${basefilename}.${extension}" $tagging; then
|
||||
echo "Processing complete for: ${filename}!"
|
||||
else
|
||||
echo "ERROR: Sickbeard MP4 Automator Processing Error"
|
||||
rm "$video" && echo "INFO: deleted: $filename"
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
echo "===================================================="
|
||||
done
|
||||
|
||||
if [ ${VIDEO_SMA} = TRUE ] || [ ${VIDEO_MKVCLEANER} = TRUE ]; then
|
||||
find "$1" -type f ! -newer "$1/sma-conversion-check" ! -name "sma-conversion-check" -delete
|
||||
# check for video files
|
||||
if find "$1" -type f -iregex ".*/.*\.\(mkv\|mp4\)" | read; then
|
||||
echo "Post Processing Complete!"
|
||||
else
|
||||
echo "ERROR: Post Processing failed, no video files found..."
|
||||
exit 1
|
||||
fi
|
||||
if [ -f "$1/sma-conversion-check" ]; then
|
||||
rm "$1/sma-conversion-check"
|
||||
fi
|
||||
fi
|
||||
|
||||
exit $?
|
||||
Loading…
Reference in new issue