#!/bin/bash
scriptVersion="1.4"
scriptName="Video"

#### Import Settings
source /config/extended.conf

log () {
  m_time=`date "+%F %T"`
  echo $m_time" :: $scriptName :: $scriptVersion :: "$1
}

set -e
set -o pipefail

# auto-clean up log file to reduce space usage
if [ -f "/config/scripts/video.txt" ]; then
  find /config/scripts -type f -name "video.txt" -size +1024k -delete
fi

touch "/config/scripts/video.txt"
exec &> >(tee -a "/config/scripts/video.txt")

function Configuration {
	log "SABnzbd Job: $jobname"
	log "SABnzbd Category: $category"
    log "SABnzbd Download ID: $downloadId"
	log "Script Versiion: $scriptVersion"
	log "CONFIGURATION VERIFICATION"
	log "##########################"
	log "Preferred Audio/Subtitle Languages: ${videoLanguages}"
	if [ "${requireLanguageMatch}" = "true" ]; then
		log "Require Matching Language :: Enabled"
	else
		log "Require Matching Language :: Disabled"
	fi
	
	if [ ${enableSma} = true ]; then
		log "Sickbeard MP4 Automator (SMA): ENABLED"
		if [ ${enableSmaTagging} = true ]; then
			tagging="-a"
			log "Sickbeard MP4 Automator (SMA): Tagging: ENABLED"
		else
			tagging="-nt"
			log "Sickbeard MP4 Automator (SMA): Tagging: DISABLED"
		fi
	else
		log "Sickbeard MP4 Automator (SMA): DISABLED"
	fi
	
	if [ -z "enableSmaTagging" ]; then
		enableSmaTagging=FALSE
	fi
}

VideoLanguageCheck () {

	count=0
	fileCount=$(find "$1" -type f -regex ".*/.*\.\(m4v\|wmv\|mkv\|mp4\|avi\)" | wc -l)
	log "Processing ${fileCount} video files..."
	find "$1" -type f -regex ".*/.*\.\(m4v\|wmv\|mkv\|mp4\|avi\)" -print0 | while IFS= read -r -d '' file; do
		count=$(($count+1))
		baseFileName="${file%.*}"
		fileName="$(basename "$file")"
		extension="${fileName##*.}"
		log "$count of $fileCount :: Processing $fileName"
		videoData=$(ffprobe -v quiet -print_format json -show_streams "$file")
		videoAudioTracksCount=$(echo "${videoData}" | jq -r ".streams[] | select(.codec_type==\"audio\") | .index" | wc -l)
		videoSubtitleTracksCount=$(echo "${videoData}" | jq -r ".streams[] | select(.codec_type==\"subtitle\") | .index" | wc -l)
		log "$count of $fileCount :: $videoAudioTracksCount Audio Tracks Found!"
		log "$count of $fileCount :: $videoSubtitleTracksCount Subtitle Tracks Found!"
		videoAudioLanguages=$(echo "${videoData}" | jq -r ".streams[] | select(.codec_type==\"audio\") | .tags.language")
		videoSubtitleLanguages=$(echo "${videoData}" | jq -r ".streams[] | select(.codec_type==\"subtitle\") | .tags.language")

		# Language Check
		log "$count of $fileCount :: Checking for preferred languages \"$videoLanguages\""
		preferredLanguage=false
		IFS=',' read -r -a filters <<< "$videoLanguages"
		for filter in "${filters[@]}"
		do
			videoAudioTracksLanguageCount=$(echo "${videoData}" | jq -r ".streams[] | select(.codec_type==\"audio\") | select(.tags.language==\"${filter}\") | .index" | wc -l)
			videoSubtitleTracksLanguageCount=$(echo "${videoData}" | jq -r ".streams[] | select(.codec_type==\"subtitle\") | select(.tags.language==\"${filter}\") | .index" | wc -l)
			log "$count of $fileCount :: $videoAudioTracksLanguageCount \"$filter\" Audio Tracks Found!"
			log "$count of $fileCount :: $videoSubtitleTracksLanguageCount \"$filter\" Subtitle Tracks Found!"			
			if [ "$preferredLanguage" == "false" ]; then
				if echo "$videoAudioLanguages" | grep -i "$filter" | read; then
					preferredLanguage=true
				elif echo "$videoSubtitleLanguages" | grep -i "$filter" | read; then
					preferredLanguage=true
				fi
			fi
		done

		if [ "$preferredLanguage" == "false" ]; then
			if [ ${enableSma} = true ]; then
				if [ "$smaProcessComplete" == "false" ]; then
					return
				fi
			fi
			if [ "$requireLanguageMatch" == "true" ]; then
				log "$count of $fileCount :: ERROR :: No matching languages found in $(($videoAudioTracksCount + $videoSubtitleTracksCount)) Audio/Subtitle tracks"
				log "$count of $fileCount :: ERROR :: Disable "
				rm "$file" && log "INFO: deleted: $fileName"
			fi
		fi

		
		
		log "$count of $fileCount :: Processing complete for: ${fileName}!"

	done

}

VideoFileCheck () {
	# check for video files
	if find "$1" -type f -regex ".*/.*\.\(m4v\|wmv\|mkv\|mp4\|avi\)" | read; then
		sleep 0.1
	else
		log "ERROR: No video files found for processing"
		exit 1
	fi
}

ArrWaitForTaskCompletion () {
  alerted=no
  until false
  do
    taskCount=$(curl -s "$arrUrl/api/v3/command?apikey=${arrApiKey}" | jq -r '.[] | select(.status=="started") | .name' | wc -l)
	if [ "$taskCount" -ge "1" ]; then
	  if [ "$alerted" == "no" ]; then
		alerted=yes
		log "$count of $fileCount :: STATUS :: ARR APP BUSY :: Pausing/waiting for all active Arr app tasks to end..."
	  fi
	  sleep 2
	else
	  break
	fi
  done
}

VideoSmaProcess (){
	count=0
	fileCount=$(find "$1" -type f -regex ".*/.*\.\(m4v\|wmv\|mkv\|mp4\|avi\)" | wc -l)
	log "Processing ${fileCount} video files..."
	find "$1" -type f -regex ".*/.*\.\(m4v\|wmv\|mkv\|mp4\|avi\)" -print0 | while IFS= read -r -d '' file; do
		count=$(($count+1))
		baseFileName="${file%.*}"
		fileName="$(basename "$file")"
		extension="${fileName##*.}"
		log "$count of $fileCount :: Processing $fileName"
		if [ -f "$file" ]; then	
			if [ -f /config/scripts/sma/config/sma.log ]; then
				rm /config/scripts/sma/config/sma.log
			fi
			log "$count of $fileCount :: Processing with SMA..."
			if [ -f "/config/scripts/sma.ini" ]; then
				if [ ${enableSmaTagging} = true ]; then
				arrItemId=""
				arrItemData=""
				arrUrl=""
				arrApiKey=""
				log "$count of $fileCount :: Getting Media ID"
				if echo $category | grep radarr | read; then
				    if echo $category | grep radarr4k | read; then
					  arrUrl="$radarr4kArrUrl"
					  arrApiKey="$radarr4kArrApiKey"
					else
					  arrUrl="$radarrArrUrl"
					  arrApiKey="$radarrArrApiKey"
					fi
					log "$count of $fileCount :: Refreshing Radarr app Queue"
     				refreshQueue=$(curl -s "$arrUrl/api/v3/command" -X POST -H 'Content-Type: application/json' -H "X-Api-Key: $arrApiKey" --data-raw '{"name":"RefreshMonitoredDownloads"}')
					ArrWaitForTaskCompletion
					arrItemId=$(curl -s "$arrUrl/api/v3/queue?page=1&pageSize=50&sortDirection=ascending&sortKey=timeleft&includeUnknownMovieItems=true&apikey=$arrApiKey" | jq -r --arg id "$downloadId" '.records[] | select(.downloadId==$id) | .movieId')
					arrItemData=$(curl -s "$arrUrl/api/v3/movie/$arrItemId?apikey=$arrApiKey")
					onlineSourceId="$(echo "$arrItemData" | jq -r ".tmdbId")"
					log "$count of $fileCount :: Radarr Movie ID = $arrItemId"
					log "$count of $fileCount :: TMDB ID = $onlineSourceId"
					onlineData="-tmdb $onlineSourceId"
				fi
				if echo $category | grep sonarr | read; then
				    if echo $category | grep sonarr4k | read; then
					  arrUrl="$sonarr4kArrUrl"
					  arrApiKey="$sonarr4kArrApiKey"
					else
					  arrUrl="$sonarrArrUrl"
					  arrApiKey="$sonarrArrApiKey"
					fi
					log "$count of $fileCount :: Refreshing Sonarr app Queue"
					refreshQueue=$(curl -s "$arrUrl/api/v3/command" -X POST -H 'Content-Type: application/json' -H "X-Api-Key: $arrApiKey" --data-raw '{"name":"RefreshMonitoredDownloads"}')
					ArrWaitForTaskCompletion
					arrQueueItemData=$(curl -s "$arrUrl/api/v3/queue?page=1&pageSize=50&sortDirection=ascending&sortKey=timeleft&includeUnknownSeriesItems=true&apikey=$arrApiKey" | jq -r --arg id "$downloadId" '.records[] | select(.downloadId==$id)')
					arrSeriesId="$(echo $arrQueueItemData | jq -r .seriesId)"
					arrEpisodeId="$(echo $arrQueueItemData | jq -r .episodeId)"
					arrSeriesData=$(curl -s "$arrUrl/api/v3/series/$arrSeriesId?apikey=$arrApiKey")
					arrEpisodeData=$(curl -s "$arrUrl/api/v3/episode/$arrEpisodeId?apikey=$arrApiKey")
					onlineSourceId="$(echo "$arrSeriesData" | jq -r ".tvdbId")"
					seasonNumber="$(echo "$arrEpisodeData" | jq -r ".seasonNumber")"
					episodeNumber="$(echo "$arrEpisodeData" | jq -r ".episodeNumber")"
					log "$count of $fileCount :: Sonarr Show ID = $arrSeriesId"
					log "$count of $fileCount :: TVDB ID = $onlineSourceId"
					onlineSource="-tvdb"
					onlineData="-tvdb $onlineSourceId -s $seasonNumber -e $episodeNumber"
				fi
			else
			  onlineSourceId=""
			  onlineData=""
			fi
			
			# Manual run of Sickbeard MP4 Automator
				if python3 /config/scripts/sma/manual.py --config "/config/scripts/sma.ini" -i "$file" $tagging $onlineData; then
					log "$count of $fileCount :: Complete!"
				else
					log "$count of $fileCount :: ERROR :: SMA Processing Error"
					rm "$file" && log "INFO: deleted: $fileName"
				fi
			else
				log "$count of $fileCount :: ERROR :: SMA Processing Error"
				log "$count of $fileCount :: ERROR :: \"/config/scripts/sma.ini\" configuration file is missing..."
				rm "$file" && log "INFO: deleted: $fileName"
			fi
		fi
	done
	smaProcessComplete="true"
}

function Main {
	SECONDS=0
	error=0
	folderpath="$1"
	jobname="$3"
	category="$5"
	smaProcessComplete="false"
	downloadId="$SAB_NZO_ID"

	if [ "$category"  == "radarr" ]; then
	  arrUrl="$radarrArrUrl"
      arrApiKey="$radarrArrApiKey"
    fi
	if [ "$category"  == "radarr4k" ]; then
	  arrUrl="$radarr4kArrUrl"
      arrApiKey="$radarr4kArrApiKey"
    fi
	if [ "$category"  == "sonarr" ]; then
	  arrUrl="$sonarrArrUrl"
      arrApiKey="$sonarrArrApiKey"
    fi
	if [ "$category"  == "sonarr4k" ]; then
	  arrUrl="$sonarr4kArrUrl"
      arrApiKey="$sonarr4kArrApiKey"
    fi

	Configuration
	VideoFileCheck "$folderpath"
	VideoLanguageCheck "$folderpath"
	VideoFileCheck "$folderpath"
	if [ ${enableSma} = true ]; then
		VideoSmaProcess "$folderpath" "$category"
	fi
	VideoFileCheck "$folderpath"
	VideoLanguageCheck "$folderpath"	
	VideoFileCheck "$folderpath"

	duration=$SECONDS
	echo "Post Processing Completed in $(($duration / 60 )) minutes and $(($duration % 60 )) seconds!"
}


Main "$@" 

exit $?