From 71e30717151a94d395f52ba46aa4d096d99561a5 Mon Sep 17 00:00:00 2001
From: RandomNinjaAtk <RandomNinjaAtk@users.noreply.github.com>
Date: Sat, 8 Jul 2023 13:23:13 -0400
Subject: [PATCH] Update AutoExtras.service

---
 sonarr/AutoExtras.service | 112 +++++++++++++++++++++++++-------------
 1 file changed, 75 insertions(+), 37 deletions(-)

diff --git a/sonarr/AutoExtras.service b/sonarr/AutoExtras.service
index 6a61049..e0ed8a0 100644
--- a/sonarr/AutoExtras.service
+++ b/sonarr/AutoExtras.service
@@ -1,51 +1,89 @@
 #!/usr/bin/env bash
 scriptVersion="1.0.1"
 
-if [ -z "$arrUrl" ] || [ -z "$arrApiKey" ]; then
-  arrUrlBase="$(cat /config/config.xml | xq | jq -r .Config.UrlBase)"
-  if [ "$arrUrlBase" == "null" ]; then
-    arrUrlBase=""
-  else
-    arrUrlBase="/$(echo "$arrUrlBase" | sed "s/\///g")"
+
+getArrAppInfo () {
+  # Get Arr App information
+  if [ -z "$arrUrl" ] || [ -z "$arrApiKey" ]; then
+    arrUrlBase="$(cat /config/config.xml | xq | jq -r .Config.UrlBase)"
+    if [ "$arrUrlBase" == "null" ]; then
+      arrUrlBase=""
+    else
+      arrUrlBase="/$(echo "$arrUrlBase" | sed "s/\///g")"
+    fi
+    arrName="$(cat /config/config.xml | xq | jq -r .Config.InstanceName)"
+    arrApiKey="$(cat /config/config.xml | xq | jq -r .Config.ApiKey)"
+    arrPort="$(cat /config/config.xml | xq | jq -r .Config.Port)"
+    arrUrl="http://127.0.0.1:${arrPort}${arrUrlBase}"
   fi
-  arrApiKey="$(cat /config/config.xml | xq | jq -r .Config.ApiKey)"
-  arrPort="$(cat /config/config.xml | xq | jq -r .Config.Port)"
-  arrUrl="http://127.0.0.1:${arrPort}${arrUrlBase}"
-fi
+}
+
+verifyApiAccess () {
+  until false
+  do
+    arrApiTest=""
+    arrApiVersion=""
+    if [ "$arrName" == "Sonarr" ] || [ "$arrName" == "Radarr" ]; then
+      arrApiVersion="v3"
+    elif [ "$arrName" == "Lidarr" ] || [ "$arrName" == "Readarr" ]; then
+      arrApiVersion="v1"
+    fi
+    arrApiTest=$(curl -s "$arrUrl/api/$arrApiVersion/system/status?apikey=$arrApiKey" | jq -r .instanceName)
+    if [ "$arrApiTest" == "$arrName" ]; then
+      break
+    else
+      log "$arrName is not ready, sleeping until valid response..."
+      sleep 1
+    fi
+  done
+}
+
 
 log () {
   m_time=`date "+%F %T"`
   echo $m_time" :: AutoExtras :: $scriptVersion :: "$1
 }
 
-# auto-clean up log file to reduce space usage
-if [ -f "/config/logs/AutoExtras.txt" ]; then
-	find /config/logs -type f -name "AutoExtras.txt" -size +1024k -delete
-fi
+AutoExtrasProcess () {
+  # auto-clean up log file to reduce space usage
+  if [ -f "/config/logs/AutoExtras.txt" ]; then
+  	find /config/logs -type f -name "AutoExtras.txt" -size +1024k -delete
+  fi
+  
+  if [ ! -f "/config/logs/AutoExtras.txt" ]; then
+      touch "/config/logs/AutoExtras.txt"
+      chmod 666 "/config/logs/AutoExtras.txt"
+  fi
+  exec &> >(tee -a "/config/logs/AutoExtras.txt")
+  
+  sonarrSeriesList=$(curl -s --header "X-Api-Key:"${arrApiKey} --request GET  "$arrUrl/api/v3/series")
+  sonarrSeriesTotal=$(echo "${sonarrSeriesList}"  | jq -r '.[].id' | wc -l)
+  sonarrSeriesIds=$(echo "${sonarrSeriesList}" | jq -r '.[].id')
+  
+  loopCount=0
+  for id in $(echo $sonarrSeriesIds); do
+      loopCount=$(( $loopCount + 1 ))
+      arrSeriesData="$(echo "$sonarrSeriesList" | jq -r ".[] | select(.id==$id)")"
+      arrSeriesPath="$(echo "$arrSeriesData" | jq -r ".path")"
+      arrSeriesTitle="$(echo "$arrSeriesData" | jq -r ".title")"
+      if [ -d "$arrSeriesPath" ]; then
+        log "$loopCount of $sonarrSeriesTotal :: $id :: $arrSeriesTitle :: Processing with Extras.bash"
+        bash /config/extended/scripts/Extras.bash "$id"
+      else
+        log "$loopCount of $sonarrSeriesTotal :: $id :: $arrSeriesTitle :: Series folder does not exist, skipping..."
+        continue
+      fi
+  done
+}
 
-if [ ! -f "/config/logs/AutoExtras.txt" ]; then
-    touch "/config/logs/AutoExtras.txt"
-    chmod 666 "/config/logs/AutoExtras.txt"
-fi
-exec &> >(tee -a "/config/logs/AutoExtras.txt")
-
-sonarrSeriesList=$(curl -s --header "X-Api-Key:"${arrApiKey} --request GET  "$arrUrl/api/v3/series")
-sonarrSeriesTotal=$(echo "${sonarrSeriesList}"  | jq -r '.[].id' | wc -l)
-sonarrSeriesIds=$(echo "${sonarrSeriesList}" | jq -r '.[].id')
-
-loopCount=0
-for id in $(echo $sonarrSeriesIds); do
-    loopCount=$(( $loopCount + 1 ))
-    arrSeriesData="$(echo "$sonarrSeriesList" | jq -r ".[] | select(.id==$id)")"
-    arrSeriesPath="$(echo "$arrSeriesData" | jq -r ".path")"
-    arrSeriesTitle="$(echo "$arrSeriesData" | jq -r ".title")"
-    if [ -d "$arrSeriesPath" ]; then
-      log "$loopCount of $sonarrSeriesTotal :: $id :: $arrSeriesTitle :: Processing with Extras.bash"
-      bash /config/extended/scripts/Extras.bash "$id"
-    else
-      log "$loopCount of $sonarrSeriesTotal :: $id :: $arrSeriesTitle :: Series folder does not exist, skipping..."
-      continue
-    fi
+echo "Starting Script...."
+for (( ; ; )); do
+	let i++
+  getArrAppInfo
+  verifyApiAccess
+	AutoExtrasProcess
+	echo "Script sleeping for 24 hours..."
+	sleep 24h
 done
 
 exit