Update sonarr-hunter.sh

Author: Admin9705

sonarr-hunter.sh

@@ -5,167 +5,187 @@
 # ---------------------------
 # Use environment variables if provided; otherwise, fall back to defaults.
 API_KEY=${API_KEY:-"your-api-key"}
-API_URL=${API_URL:-"http://your-radarr-address:7878"}
+API_URL=${API_URL:-"http://your-address:8989"}
 
-# How many movies to process before restarting the search cycle
-MAX_MOVIES=${MAX_MOVIES:-1}
+# How many shows to process before restarting the search cycle
+MAX_SHOWS=${MAX_SHOWS:-1}
 
-# Sleep duration in seconds after fully processing a movie (300=5min)
-SLEEP_DURATION=${SLEEP_DURATION:-300}
+# Sleep duration in seconds after finding a show with missing episodes
+# (900=15min, 600=10min, etc.)
+SLEEP_DURATION=${SLEEP_DURATION:-900}
 
-# Shorter sleep duration between each processed movie (30=0.5min)
-REFRESH_DURATION=${REFRESH_DURATION:-30}
-
-# Set to true to pick movies randomly, false to go in order
+# Set to true to pick shows randomly, false to go in order
 RANDOM_SELECTION=${RANDOM_SELECTION:-true}
 
 # ---------------------------
 # Main infinite loop
 # ---------------------------
 while true; do
-  echo "Retrieving missing movies from Radarr..."
-
-  # Fetch all monitored movies that do not have files
-  MISSING_MOVIES_JSON=$(curl -s \
+  echo "Retrieving missing episodes data from Sonarr..."
+  # Get all shows first
+  SHOWS_JSON=$(curl -s \
     -H "X-Api-Key: $API_KEY" \
-    "$API_URL/api/v3/movie" | \
-    jq '[.[] | select(.monitored == true and .hasFile == false)]'
-  )
+    "$API_URL/api/v3/series")
 
-  if [ -z "$MISSING_MOVIES_JSON" ]; then
-    echo "ERROR: Unable to retrieve movie data from Radarr. Retrying in 60 seconds..."
+  # If the above command fails or returns nothing, wait and retry
+  if [ -z "$SHOWS_JSON" ]; then
+    echo "ERROR: Unable to retrieve series data from Sonarr. Retrying in 60 seconds..."
     sleep 60
     continue
   fi
 
-  TOTAL_MISSING=$(echo "$MISSING_MOVIES_JSON" | jq 'length')
-  if [ "$TOTAL_MISSING" -eq 0 ]; then
-    echo "No missing movies found. Waiting 60 seconds before checking again..."
+  # Filter to only get shows with missing episodes
+  INCOMPLETE_SHOWS_JSON=$(echo "$SHOWS_JSON" | jq '[.[] | select(has("statistics") and .statistics.episodeCount > .statistics.episodeFileCount)]')
+  
+  # Count how many incomplete shows are in the list
+  TOTAL_INCOMPLETE=$(echo "$INCOMPLETE_SHOWS_JSON" | jq 'length')
+  if [ "$TOTAL_INCOMPLETE" -eq 0 ]; then
+    echo "No shows with missing episodes found in Sonarr. Waiting 60 seconds before checking again..."
     sleep 60
     continue
   fi
+  echo "Found $TOTAL_INCOMPLETE show(s) with missing episodes."
 
-  echo "Found $TOTAL_MISSING missing movie(s)."
   echo "Using ${RANDOM_SELECTION:+random}${RANDOM_SELECTION:-sequential} selection."
-  echo "Will process up to $MAX_MOVIES movies with a longer sleep of $SLEEP_DURATION seconds after the final one."
+  echo "Will process up to ${MAX_SHOWS:-all} shows with ${SLEEP_DURATION}s pause between each."
 
-  MOVIES_PROCESSED=0
+  SHOWS_PROCESSED=0
   ALREADY_CHECKED=()
 
   while true; do
-    # Check if we've reached the maximum number of movies to process
-    if [ "$MAX_MOVIES" -gt 0 ] && [ "$MOVIES_PROCESSED" -ge "$MAX_MOVIES" ]; then
-      echo "Reached maximum number of movies ($MAX_MOVIES). Restarting search cycle..."
+    # Check if we've reached the maximum number of shows to process
+    if [ "$MAX_SHOWS" -gt 0 ] && [ "$SHOWS_PROCESSED" -ge "$MAX_SHOWS" ]; then
+      echo "Reached maximum number of shows to process ($MAX_SHOWS). Restarting search cycle..."
       break
     fi
 
-    # Check if we've checked all missing movies
-    if [ ${#ALREADY_CHECKED[@]} -eq "$TOTAL_MISSING" ]; then
-      echo "All missing movies have been checked. Waiting before starting a new cycle..."
+    # Check if we've checked all incomplete shows
+    if [ ${#ALREADY_CHECKED[@]} -eq "$TOTAL_INCOMPLETE" ] || [ "$TOTAL_INCOMPLETE" -eq 0 ]; then
+      echo "All shows with missing episodes have been checked. Waiting before starting a new cycle..."
       sleep 60
       break
     fi
 
-    # Select next movie index based on selection method
-    if [ "$RANDOM_SELECTION" = true ] && [ "$TOTAL_MISSING" -gt 1 ]; then
+    # Select next show index based on selection method
+    if [ "$RANDOM_SELECTION" = true ] && [ "$TOTAL_INCOMPLETE" -gt 1 ]; then
+      # Keep generating random indices until we find one we haven't checked yet
       while true; do
-        INDEX=$((RANDOM % TOTAL_MISSING))
+        INDEX=$((RANDOM % TOTAL_INCOMPLETE))
         if [[ ! " ${ALREADY_CHECKED[*]} " =~ " ${INDEX} " ]]; then
           break
         fi
       done
     else
-      for ((i=0; i<TOTAL_MISSING; i++)); do
+      for ((i=0; i<TOTAL_INCOMPLETE; i++)); do
         if [[ ! " ${ALREADY_CHECKED[*]} " =~ " ${i} " ]]; then
           INDEX=$i
           break
         fi
       done
     fi
 
+    # Add this index to the list of checked indices
     ALREADY_CHECKED+=("$INDEX")
 
-    # Extract movie information
-    MOVIE=$(echo "$MISSING_MOVIES_JSON" | jq ".[$INDEX]")
-    MOVIE_ID=$(echo "$MOVIE" | jq '.id')
-    MOVIE_TITLE=$(echo "$MOVIE" | jq -r '.title')
-    MOVIE_YEAR=$(echo "$MOVIE" | jq -r '.year')
-
-    echo "Selected missing movie \"$MOVIE_TITLE ($MOVIE_YEAR)\"..."
-
-    # 1. Refresh the movie
-    echo "1. Refreshing movie information for \"$MOVIE_TITLE\"..."
+    # Extract show information
+    SHOW=$(echo "$INCOMPLETE_SHOWS_JSON" | jq ".[$INDEX]")
+    SHOW_ID=$(echo "$SHOW" | jq '.id')
+    SHOW_TITLE=$(echo "$SHOW" | jq -r '.title')
+    EPISODE_COUNT=$(echo "$SHOW" | jq '.statistics.episodeCount')
+    EPISODE_FILE_COUNT=$(echo "$SHOW" | jq '.statistics.episodeFileCount')
+    MISSING=$((EPISODE_COUNT - EPISODE_FILE_COUNT))
+    
+    echo "Selected show \"$SHOW_TITLE\" with $MISSING missing episode(s)..."
+    
+    # ---------------------------
+    # Step 1: Refresh the series to make sure Sonarr has latest information
+    # ---------------------------
+    echo "1. Refreshing series information for \"$SHOW_TITLE\" (ID: $SHOW_ID)..."
+    
     REFRESH_COMMAND=$(curl -s -X POST \
       -H "X-Api-Key: $API_KEY" \
       -H "Content-Type: application/json" \
-      -d "{\"name\":\"RefreshMovie\",\"movieIds\":[$MOVIE_ID]}" \
-      "$API_URL/api/v3/command"
-    )
-
+      -d "{\"name\":\"RefreshSeries\",\"seriesId\":$SHOW_ID}" \
+      "$API_URL/api/v3/command")
+    
+    # Check if the refresh command succeeded
     REFRESH_ID=$(echo "$REFRESH_COMMAND" | jq '.id // empty')
-    if [ -z "$REFRESH_ID" ]; then
-      echo "WARNING: Could not refresh \"$MOVIE_TITLE\". Response was:"
+    if [ -n "$REFRESH_ID" ]; then
+      echo "Refresh command accepted (ID: $REFRESH_ID)."
+      
+      # Wait for the refresh to complete
+      echo "Waiting for refresh to complete..."
+      sleep 5
+      
+      # ---------------------------
+      # Step 2: Tell Sonarr to search for missing episodes
+      # ---------------------------
+      echo "2. Telling Sonarr to perform a missing-episode search for \"$SHOW_TITLE\" (ID: $SHOW_ID)..."
+      
+      SEARCH_COMMAND=$(curl -s -X POST \
+        -H "X-Api-Key: $API_KEY" \
+        -H "Content-Type: application/json" \
+        -d "{\"name\":\"MissingEpisodeSearch\",\"seriesId\":$SHOW_ID}" \
+        "$API_URL/api/v3/command")
+      
+      # Verify the response from Sonarr
+      SEARCH_ID=$(echo "$SEARCH_COMMAND" | jq '.id // empty')
+      if [ -n "$SEARCH_ID" ]; then
+        echo "Search command accepted (ID: $SEARCH_ID)."
+        
+        # Wait a moment to let Sonarr process the command
+        echo "Waiting 2 seconds to check command status..."
+        sleep 2
+        
+        # Check the status of the command
+        COMMAND_STATUS=$(curl -s \
+          -H "X-Api-Key: $API_KEY" \
+          "$API_URL/api/v3/command/$SEARCH_ID" | jq -r '.status')
+        
+        echo "Command status: $COMMAND_STATUS"
+        SHOWS_PROCESSED=$((SHOWS_PROCESSED + 1))
+        
+        # Sleep after processing a show with missing episodes
+        echo "Show with missing episodes processed. Sleeping for $SLEEP_DURATION seconds to avoid overloading indexers..."
+        sleep "$SLEEP_DURATION"
+      else
+        echo "WARNING: Search command did not return an ID. Response was:"
+        echo "$SEARCH_COMMAND"
+        echo "Trying alternative commands..."
+        
+        # Try an alternative command for Sonarr v3
+        SEARCH_COMMAND2=$(curl -s -X POST \
+          -H "X-Api-Key: $API_KEY" \
+          -H "Content-Type: application/json" \
+          -d "{\"name\":\"EpisodeSearch\",\"seriesId\":$SHOW_ID}" \
+          "$API_URL/api/v3/command")
+        
+        SEARCH_ID2=$(echo "$SEARCH_COMMAND2" | jq '.id // empty')
+        if [ -n "$SEARCH_ID2" ]; then
+          echo "Alternative search command accepted (ID: $SEARCH_ID2)."
+          SHOWS_PROCESSED=$((SHOWS_PROCESSED + 1))
+          echo "Show with missing episodes processed. Sleeping for $SLEEP_DURATION seconds to avoid overloading indexers..."
+          sleep "$SLEEP_DURATION"
+        else
+          echo "All search commands failed. Skipping this show."
+          sleep 10
+        fi
+      fi
+    else
+      echo "WARNING: Refresh command did not return an ID. Response was:"
       echo "$REFRESH_COMMAND"
-      echo "Skipping this movie. Sleeping 10 seconds..."
+      echo "Skipping search for this show."
+      
+      # Sleep a shorter time since we didn't actually do a search
       sleep 10
-      continue
-    fi
-
-    echo "Refresh command accepted (ID: $REFRESH_ID). Waiting 5 seconds for refresh to complete..."
-    sleep 5
-
-    # 2. Search for the movie using "MoviesSearch"
-    echo "2. Searching for \"$MOVIE_TITLE\"..."
-    SEARCH_COMMAND=$(curl -s -X POST \
-      -H "X-Api-Key: $API_KEY" \
-      -H "Content-Type: application/json" \
-      -d "{\"name\":\"MoviesSearch\",\"movieIds\":[$MOVIE_ID]}" \
-      "$API_URL/api/v3/command"
-    )
-
-    SEARCH_ID=$(echo "$SEARCH_COMMAND" | jq '.id // empty')
-    if [ -n "$SEARCH_ID" ]; then
-      echo "Search command accepted (ID: $SEARCH_ID)."
-    else
-      echo "WARNING: Search command failed for \"$MOVIE_TITLE\". Response was:"
-      echo "$SEARCH_COMMAND"
-    fi
-
-    echo "Waiting 5 seconds for search operation..."
-    sleep 5
-
-    # 3. Rescan the movie folder
-    echo "3. Rescanning movie folder for \"$MOVIE_TITLE\"..."
-    RESCAN_COMMAND=$(curl -s -X POST \
-      -H "X-Api-Key: $API_KEY" \
-      -H "Content-Type: application/json" \
-      -d "{\"name\":\"RescanMovie\",\"movieIds\":[$MOVIE_ID]}" \
-      "$API_URL/api/v3/command"
-    )
-
-    RESCAN_ID=$(echo "$RESCAN_COMMAND" | jq '.id // empty')
-    if [ -n "$RESCAN_ID" ]; then
-      echo "Rescan command accepted (ID: $RESCAN_ID)."
-    else
-      echo "WARNING: Rescan command not available or failed."
-    fi
-
-    MOVIES_PROCESSED=$((MOVIES_PROCESSED + 1))
-
-    # Sleep before processing the next movie
-    if [ "$MOVIES_PROCESSED" -ge "$MAX_MOVIES" ]; then
-      echo "Processed $MOVIES_PROCESSED movies. Sleeping for $SLEEP_DURATION seconds..."
-      sleep "$SLEEP_DURATION"
-    else
-      echo "Movie refreshed. Sleeping for $REFRESH_DURATION seconds before continuing..."
-      sleep "$REFRESH_DURATION"
     fi
   done
 
-  echo "Done. Processed $MOVIES_PROCESSED missing movies in this cycle."
-
-  if [ "$MOVIES_PROCESSED" -eq 0 ]; then
-    echo "No missing movies processed this cycle. Waiting 60 seconds before starting a new cycle..."
+  echo "Done. Processed $SHOWS_PROCESSED shows with missing episodes in this cycle."
+  
+  # If we didn't find any shows to process in this cycle, wait a bit before starting a new cycle
+  if [ "$SHOWS_PROCESSED" -eq 0 ]; then
+    echo "No shows with missing episodes processed this cycle. Waiting 60 seconds before starting a new cycle..."
     sleep 60
   fi
 done