diff --git a/root/scripts/audio-pp.bash b/root/scripts/audio-pp.bash new file mode 100644 index 0000000..e20928f --- /dev/null +++ b/root/scripts/audio-pp.bash @@ -0,0 +1,252 @@ +#!/usr/bin/with-contenv bash + +# SETTINGS +AudioVerification="${AUDIO_VERIFY}" # TRUE = ENABLED, Verifies FLAC/MP3 files for errors (fixes MP3's, deletes bad FLAC files) +ConversionFormat="${AUDIO_FORMAT}" # SET TO: OPUS or AAC or MP3 or ALAC or FLAC - converts lossless FLAC files to set format +ConversionBitrate="${AUDIO_BITRATE}" # Set to desired bitrate when converting to OPUS/AAC/MP3 format types +ReplaygainTagging="${AUDIO_REPLAYGAIN}" # TRUE = ENABLED, adds replaygain tags for compatible players (FLAC ONLY) +DetectNonSplitAlubms="${AUDIO_DSFA}" # TRUE = ENABLED :: Uses "MaxFileSize" to detect and mark download as failed if detected +MaxFileSize="${AUDIO_DSFAS}" # M = MB, G = GB :: Set size threshold for detecting single file albums +RequireBeetsMatch="${AUDIO_REQUIREBEETSTAGGING}" # true = enabled :: skips importing files that could not be matched using beets + +#============FUNCTIONS============ + +settings () { + +echo "Configuration:" +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" +elif [ "${ConversionFormat}" = ALAC ]; then + echo "Bitrate: lossless" +else + echo "Conversion Bitrate: ${ConversionBitrate}k" +fi +if [ "${DetectNonSplitAlubms}" = TRUE ]; then + echo "Detect Non Split Alubms: ENABLED" + echo "Max File Size: $MaxFileSize" +else + echo "DetectNonSplitAlubms: DISABLED" +fi +if [ "${TagWithBeets}" = TRUE ]; then + echo "Tag with Beets: ENABLED" +else + echo "Tag with Beets: DISABLED" +fi +if [ "${RequireBeetsMatch}" = true ]; then + echo "Require Beets Match: ENABLED" +else + echo "Require Beets Match: DISABLED" +fi +echo "Processing: $1" + +} + +clean () { + if find "$1" -type f -iregex ".*/.*\.\(flac\|mp3\|m4a\|alac\|ogg\|opus\)" | read; then + if find "$1" -type f -not -iregex ".*/.*\.\(flac\|mp3\|m4a\|alac\|ogg\|opus\)" | read; then + find "$1" -type f -not -iregex ".*/.*\.\(flac\|mp3\|m4a\|alac\|ogg\|opus\)" -delete + fi + 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 +} + +duplicatefilecleanup () { + duplicate="FALSE" + if find "$1" -type f -mindepth 1 -iname "*([0-9]).*" | read; then + find "$1" -type f -mindepth 1 -iname "*([0-9]).*" -delete + duplicate="TRUE" + fi + + if find "$1" -type f -mindepth 1 -iname "*.[0-9].*" | read; then + find "$1" -type f -mindepth 1 -iname "*.[0-9].*" -delete + duplicate="TRUE" + fi + + if find "$1" -type f -mindepth 1 -iname "*.flac" | read; then + if find "$1"/* -type f -not -iname "*.flac" | read; then + find "$1"/* -type f -not -iname "*.flac" -delete + duplicate="TRUE" + fi + fi + if [ "${duplicate}" = TRUE ]; then + echo "DUPLICATE FILE CLEANUP" + echo "DUPLICATE FILE CLEANUP COMPLETE" + fi +} + +detectsinglefilealbums () { + if find "$1" -type f -iregex ".*/.*\.\(flac\|mp3\|m4a\|alac\|ogg\|opus\)" -size +${MaxFileSize} | read; then + echo "ERROR: Non split album detected" && exit 1 + fi +} + +verify () { + if find "$1" -iname "*.flac" | read; 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 + if find "$1" -iname "*.mp3" | read; then + verifytrackcount=$(find "$1"/ -iname "*.mp3" | wc -l) + echo "" + echo "Verifying: $verifytrackcount Tracks" + if ! [ -x "$(command -v mp3val)" ]; then + echo "MP3VAL verification utility not installed (ubuntu: apt-get install -y mp3val)" + else + for fname in "$1"/*.mp3; do + filename="$(basename "$fname")" + if mp3val -f -nb "$fname" > /dev/null; then + echo "Verified Track: $filename" + 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" | read; then + echo "Converting: $converttrackcount Tracks (Target Format: $targetformat (${targetbitrate}))" + for fname in "$1"/*.flac; do + filename="$(basename "${fname%.flac}")" + 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 () { + if ! [ -x "$(command -v flac)" ]; then + echo "ERROR: METAFLAC replaygain utility not installed (ubuntu: apt-get install -y flac)" + elif find "$1" -iname "*.flac" | read; then + replaygaintrackcount=$(find "$1"/ -iname "*.flac" | wc -l) + echo "Replaygain: Calculating $replaygaintrackcount Tracks" + find "$1" -iname "*.flac" -exec metaflac --add-replay-gain "{}" + && echo "Replaygain: $replaygaintrackcount Tracks Tagged" + fi +} + +beets () { + trackcount=$(find "$1" -type f -iregex ".*/.*\.\(flac\|opus\|m4a\|mp3\)" | wc -l) + echo "Matching $trackcount tracks with Beets" + + touch "$1/beets-match" + sleep 0.1 + + if find "$1" -type f -iregex ".*/.*\.\(flac\|opus\|m4a\|mp3\)" | read; then + beet -c "/config/scripts/configs/beets-config.yaml" -l "$1/beets-library.blb" -d "$1" import -q "$1" + if find "$1" -type f -iregex ".*/.*\.\(flac\|opus\|m4a\|mp3\)" -newer "$1/beets-match" | read; then + echo "SUCCESS: Matched with beets!" + else + if [ "$RequireBeetsMatch" = true ]; then + echo "ERROR: RequireBeetsMatch enabled, performing cleanup" + rm -rf "$1"/* + sleep 0.1 + exit 1 + else + echo "ERROR: Unable to match using beets, fallback to lidarr import matching..." + fi + fi + fi + + if [ -f "$1/beets-match" ]; then + rm "$1/beets-match" + sleep 0.1 + fi + + if [ -f "$1/beets-library.blb" ]; then + rm "$1/beets-library.blb" + sleep 0.1 + fi +} + +#============START SCRIPT============ + +settings "$1" +clean "$1" +duplicatefilecleanup "$1" +detectsinglefilealbums "$1" + +if [ "${AudioVerification}" = TRUE ]; then + verify "$1" +fi + +conversion "$1" + +if [ "${ReplaygainTagging}" = TRUE ]; then + replaygain "$1" +fi + +echo "" +echo "Post Processing Complete!" && exit 0 +#============END SCRIPT============