lidarr-dl/sabnzbd/audio.bash
2023-07-07 15:17:03 -04:00

269 lines
7.7 KiB
Bash

#!/usr/bin/with-contenv bash
TITLESHORT="APP"
ScriptVersion="1.3"
set -e
set -o pipefail
touch "/config/scripts/audio.txt"
exec &> >(tee -a "/config/scripts/audio.txt")
Main () {
# SETTINGS
AudioVerification="TRUE" # TRUE = ENABLED, Verifies FLAC/MP3 files for errors (fixes MP3's, deletes bad FLAC files)
ConversionFormat="FLAC" # SET TO: OPUS or AAC or MP3 or ALAC or FLAC - converts lossless FLAC files to set format
ConversionBitrate="160" # Set to desired bitrate when converting to OPUS/AAC/MP3 format types
ReplaygainTagging="false" # TRUE = ENABLED, adds replaygain tags for compatible players (FLAC ONLY)
BeetsTagging="TRUE" # TRUE = ENABLED, enables tagging with beets
DetectNonSplitAlubms="TRUE" # TRUE = ENABLED :: Uses "MaxFileSize" to detect and mark download as failed if detected
MaxFileSize="153600k" # M = MB, G = GB :: Set size threshold for detecting single file albums
#============FUNCTIONS============
settings () {
echo "Configuration:"
echo "Script Version: $ScriptVersion"
echo "Remove Non Audio Files: ENABLED"
echo "Duplicate File CleanUp: ENABLED"
if [ "${AudioVerification}" = TRUE ]; then
echo "Audio Verification: ENABLED"
else
echo "Audio Verification: DISABLED"
fi
echo "Format: $ConversionFormat"
if [ "${ConversionFormat}" = FLAC ]; then
echo "Bitrate: lossless"
echo "Replaygain Tagging: ENABLED"
AudioFileExtension="flac"
elif [ "${ConversionFormat}" = ALAC ]; then
echo "Bitrate: lossless"
AudioFileExtension="m4a"
else
echo "Conversion Bitrate: ${ConversionBitrate}k"
if [ "${ConversionFormat}" = MP3 ]; then
AudioFileExtension="mp3"
elif [ "${ConversionFormat}" = AAC ]; then
AudioFileExtension="m4a"
elif [ "${ConversionFormat}" = OPUS ]; then
AudioFileExtension="opus"
fi
fi
if [ "$RequireAudioQualityMatch" = "true" ]; then
echo "Audio Quality Match Verification: ENABLED (.$AudioFileExtension)"
else
echo "Audio Quality Match Verification: DISABLED"
fi
if [ "${DetectNonSplitAlubms}" = TRUE ]; then
echo "Detect Non Split Alubms: ENABLED"
echo "Max File Size: $MaxFileSize"
else
echo "DetectNonSplitAlubms: DISABLED"
fi
echo "Processing: $1"
}
AudioQualityMatch () {
if [ "$RequireAudioQualityMatch" == "true" ]; then
find "$1" -type f -not -iname "*.$AudioFileExtension" -delete
if [ $(find "$1" -type f -iname "*.$AudioFileExtension" | wc -l) -gt 0 ]; then
echo "Verifying Audio Quality Match: PASSED (.$AudioFileExtension)"
else
echo "Verifying Audio Quality Match"
echo "ERROR: Audio Qualty Check Failed, missing required file extention (.$AudioFileExtension)"
exit 1
fi
fi
}
clean () {
if [ $(find "$1" -type f -regex ".*/.*\.\(flac\|mp3\|m4a\|alac\|ogg\|opus\)" | wc -l) -gt 0 ]; then
find "$1" -type f -not -regex ".*/.*\.\(flac\|mp3\|m4a\|alac\|ogg\|opus\)" -delete
find "$1" -mindepth 2 -type f -exec mv "{}" "$1"/ \;
find "$1" -mindepth 1 -type d -delete
else
echo "ERROR: NO AUDIO FILES FOUND" && exit 1
fi
}
detectsinglefilealbums () {
if [ $(find "$1" -type f -regex ".*/.*\.\(flac\|mp3\|m4a\|alac\|ogg\|opus\)" -size +${MaxFileSize} | wc -l) -gt 0 ]; then
echo "ERROR: Non split album detected"
exit 1
fi
}
verify () {
if [ $(find "$1" -iname "*.flac" | wc -l) -gt 0 ]; then
verifytrackcount=$(find "$1"/ -iname "*.flac" | wc -l)
echo "Verifying: $verifytrackcount Tracks"
if ! [ -x "$(command -v flac)" ]; then
echo "ERROR: FLAC verification utility not installed (ubuntu: apt-get install -y flac)"
else
for fname in "$1"/*.flac; do
filename="$(basename "$fname")"
if flac -t --totally-silent "$fname"; then
echo "Verified Track: $filename"
else
echo "ERROR: Track Verification Failed: \"$filename\""
rm -rf "$1"/*
sleep 0.1
exit 1
fi
done
fi
fi
}
conversion () {
converttrackcount=$(find "$1"/ -name "*.flac" | wc -l)
targetformat="$ConversionFormat"
bitrate="$ConversionBitrate"
if [ "${ConversionFormat}" = OPUS ]; then
options="-acodec libopus -ab ${bitrate}k -application audio -vbr off"
extension="opus"
targetbitrate="${bitrate}k"
fi
if [ "${ConversionFormat}" = AAC ]; then
options="-acodec aac -ab ${bitrate}k -movflags faststart"
extension="m4a"
targetbitrate="${bitrate}k"
fi
if [ "${ConversionFormat}" = MP3 ]; then
options="-acodec libmp3lame -ab ${bitrate}k"
extension="mp3"
targetbitrate="${bitrate}k"
fi
if [ "${ConversionFormat}" = ALAC ]; then
options="-acodec alac -movflags faststart"
extension="m4a"
targetbitrate="lossless"
fi
if [ "${ConversionFormat}" = FLAC ]; then
options="-acodec flac"
extension="flac"
targetbitrate="lossless"
fi
if [ -x "$(command -v ffmpeg)" ]; then
if [ "${ConversionFormat}" = FLAC ]; then
sleep 0.1
elif [ $(find "$1"/ -name "*.flac" | wc -l) -gt 0 ]; then
echo "Converting: $converttrackcount Tracks (Target Format: $targetformat (${targetbitrate}))"
for fname in "$1"/*.flac; do
filename="$(basename "${fname%.flac}")"
if [ "${ConversionFormat}" = OPUS ]; then
opusenc --bitrate ${bitrate} --vbr --music "$fname" "${fname%.flac}.temp.$extension";
echo "Converted: $filename"
if [ -f "${fname%.flac}.temp.$extension" ]; then
rm "$fname"
sleep 0.1
mv "${fname%.flac}.temp.$extension" "${fname%.flac}.$extension"
fi
continue
else
echo "Conversion failed: $filename, performing cleanup..."
rm -rf "$1"/*
sleep 0.1
exit 1
fi
if ffmpeg -loglevel warning -hide_banner -nostats -i "$fname" -n -vn $options "${fname%.flac}.temp.$extension"; then
echo "Converted: $filename"
if [ -f "${fname%.flac}.temp.$extension" ]; then
rm "$fname"
sleep 0.1
mv "${fname%.flac}.temp.$extension" "${fname%.flac}.$extension"
fi
else
echo "Conversion failed: $filename, performing cleanup..."
rm -rf "$1"/*
sleep 0.1
exit 1
fi
done
fi
else
echo "ERROR: ffmpeg not installed, please install ffmpeg to use this conversion feature"
sleep 5
fi
}
replaygain () {
replaygaintrackcount=$(find "$1"/ -type f -regex ".*/.*\.\(flac\|mp3\|m4a\|alac\|ogg\|opus\)" | wc -l)
echo "Replaygain: Calculating $replaygaintrackcount Tracks"
r128gain -r -a "$1" &>/dev/null
}
beets () {
echo ""
trackcount=$(find "$1" -type f -regex ".*/.*\.\(flac\|opus\|m4a\|mp3\)" | wc -l)
echo "Matching $trackcount tracks with Beets"
if [ -f /scripts/library.blb ]; then
rm /scripts/library.blb
sleep 0.1
fi
if [ -f /scripts/beets/beets.log ]; then
rm /scripts/beets.log
sleep 0.1
fi
touch "/scripts/beets-match"
sleep 0.1
if [ $(find "$1" -type f -regex ".*/.*\.\(flac\|opus\|m4a\|mp3\)" | wc -l) -gt 0 ]; then
beet -c /config/scripts/beets-config.yaml -l /scripts/library.blb -d "$1" import -q "$1"
if [ $(find "$1" -type f -regex ".*/.*\.\(flac\|opus\|m4a\|mp3\)" -newer "/scripts/beets-match" | wc -l) -gt 0 ]; then
echo "SUCCESS: Matched with beets!"
else
rm -rf "$1"/*
echo "ERROR: Unable to match using beets to a musicbrainz release, marking download as failed..."
exit 1
fi
fi
if [ -f "/scripts/beets-match" ]; then
rm "/scripts/beets-match"
sleep 0.1
fi
}
#============START SCRIPT============
settings "$1"
clean "$1"
detectsinglefilealbums "$1"
if [ "${AudioVerification}" = TRUE ]; then
verify "$1"
fi
conversion "$1"
AudioQualityMatch "$1"
if [ "${BeetsTagging}" = TRUE ]; then
beets "$1"
fi
if [ "${ReplaygainTagging}" = TRUE ]; then
replaygain "$1"
fi
}
SECONDS=0
Main "$@"
chmod 777 "$1"
chmod 666 "$1"/*
duration=$SECONDS
echo "Post Processing Completed in $(($duration / 60 )) minutes and $(($duration % 60 )) seconds!"
exit $?