#!/bin/bash scriptVersion="1.3" 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 } 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 refreshQueue=$(curl -s "$arrUrl/api/v3/command" -X POST -H 'Content-Type: application/json' -H "X-Api-Key: $arrApiKey" --data-raw '{"name":"RefreshMonitoredDownloads"}') 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 refreshQueue=$(curl -s "$arrUrl/api/v3/command" -X POST -H 'Content-Type: application/json' -H "X-Api-Key: $arrApiKey" --data-raw '{"name":"RefreshMonitoredDownloads"}') 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 $?