lidarr-dl/lidarr/BeetsTagger.bash

#!/usr/bin/env bash
scriptVersion="1.1"
scriptName="BeetsTagger"

#### Import Settings
source /config/extended.conf
#### Import Functions
source /config/extended/functions
#### Create Log File
logfileSetup

SECONDS=0

if [ "$lidarr_eventtype" == "Test" ]; then
	log "Tested Successfully"
	exit 0	
fi

verifyConfig
getArrAppInfo
verifyApiAccess

if [ -z "$lidarr_album_id" ]; then
	lidarr_album_id="$1"
fi

if [ "$enableBeetsTagging" != "true" ]; then
  log "Beets tagging is disabled, please enable by setting \"enableBeetsTagging=true\" in \"/config/extended.conf\""
  exit 0
fi

getAlbumArtist="$(curl -s "$arrUrl/api/v1/album/$lidarr_album_id" -H "X-Api-Key: ${arrApiKey}" | jq -r .artist.artistName)"
getAlbumArtistPath="$(curl -s "$arrUrl/api/v1/album/$lidarr_album_id" -H "X-Api-Key: ${arrApiKey}" | jq -r .artist.path)"
getTrackPath="$(curl -s "$arrUrl/api/v1/trackFile?albumId=$lidarr_album_id" -H "X-Api-Key: ${arrApiKey}" | jq -r .[].path | head -n1)"
getFolderPath="$(dirname "$getTrackPath")"
getAlbumFolderName="$(basename "$getFolderPath")"

log "Processing :: $getAlbumFolderName :: Processing Files..."

if echo "$getFolderPath" | grep "$getAlbumArtistPath" | read; then
	if [ ! -d "$getFolderPath" ]; then
		log "ERROR :: \"$getFolderPath\" Folder is missing :: Exiting..."
	fi
else 
	log "ERROR :: $getAlbumArtistPath not found within \"$getFolderPath\" :: Exiting..."
	exit
fi
ProcessWithBeets () {

	SECONDS=0
	log "$1 :: Start Processing..."
	

	# Input
	# $1 Download Folder to process
	if [ -f /config/extended/library-lidarr.blb ]; then
		rm /config/extended/library-lidarr.blb
		sleep 0.5
	fi
	if [ -f /config/extended/extended/logs/beets.log ]; then 
		rm /config/extended/extended/logs/beets.log
		sleep 0.5
	fi

	if [ -f "/config/extended/beets-lidarr-match" ]; then 
		rm "/config/extended/beets-lidarr-match"
		sleep 0.5
	fi
	touch "/config/extended/beets-lidarr-match"
	sleep 0.5

    log "$1 :: Begin matching with beets!"
	beet -c /config/extended/beets-config-lidarr.yaml -l /config/extended/library-lidarr.blb -d "$1" import -qC "$1"
	# Fix tags
	log "$1 :: Fixing Tags..."
		
	# Fix flac tags
	fixed=0
	find "$1" -type f -iname "*.flac" -print0 | while IFS= read -r -d '' file; do
		if [ $fixed == 0 ]; then
			fixed=$(( $fixed + 1 ))
			log "$1 :: Fixing Flac Tags..."
		fi
		getArtistCredit="$(ffprobe -loglevel 0 -print_format json -show_format -show_streams "$file" | jq -r ".format.tags.ARTIST_CREDIT" | sed "s/null//g" | sed "/^$/d")"
		metaflac --remove-tag=ARTIST "$file"
		metaflac --remove-tag=ALBUMARTIST "$file"
		metaflac --remove-tag=ALBUMARTIST_CREDIT "$file"
		metaflac --remove-tag=ALBUMARTISTSORT "$file"
		metaflac --remove-tag=ALBUM_ARTIST "$file"
		metaflac --remove-tag="ALBUM ARTIST" "$file"
		metaflac --remove-tag=ARTISTSORT "$file"
		metaflac --remove-tag=COMPOSERSORT "$file"
		metaflac --set-tag=ALBUMARTIST="$getAlbumArtist" "$file"
		if [ ! -z "$getArtistCredit" ]; then
       			metaflac --set-tag=ARTIST="$getArtistCredit" "$file"
			else
			metaflac --set-tag=ARTIST="$getAlbumArtist" "$file"
		fi
	done
		
	log "$1 :: Fixing Tags Complete!"	
	

	if [ -f "/config/extended/beets-lidarr-match" ]; then 
		rm "/config/extended/beets-lidarr-match"
		sleep 0.5
	fi

	if [ -f /config/extended/library-lidarr.blb ]; then
		rm /config/extended/library-lidarr.blb
		sleep 0.5
	fi
	if [ -f /config/extended/logs/beets.log ]; then 
		rm /config/extended/logs/beets.log
		sleep 0.5
	fi

	duration=$SECONDS
	log "$1 :: Finished in $(($duration / 60 )) minutes and $(($duration % 60 )) seconds!"
}

NotifyPlex () {
	# Process item with PlexNotify.bash if plexToken is configured
	if [ ! -z "$plexToken" ]; then
		# update plex
		log "$1 :: Using PlexNotify.bash to update Plex...."
		bash /config/extended/PlexNotify.bash "$2"
	fi
}

ProcessWithBeets "$getFolderPath"
MetadataPostProcess "$lidarr_album_id"
NotifyPlex "$getAlbumArtist" "$getAlbumArtistPath"
exit
exit