From 186a24cc4a1a2f86cfc87044f68790bb8acde176 Mon Sep 17 00:00:00 2001
From: Mukhtar Akere <akeremukhtar10@gmail.com>
Date: Fri, 7 Feb 2025 23:42:09 +0100
Subject: [PATCH] Fix repair worker

---
 pkg/arr/content.go   | 10 ++++++----
 pkg/repair/repair.go |  4 ++--
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/pkg/arr/content.go b/pkg/arr/content.go
index d2c1a86..19c0025 100644
--- a/pkg/arr/content.go
+++ b/pkg/arr/content.go
@@ -3,7 +3,6 @@ package arr
 import (
 	"encoding/json"
 	"fmt"
-	"log"
 	"net/http"
 	"strconv"
 )
@@ -15,9 +14,7 @@ func (a *Arr) GetMedia(tvId string) ([]Content, error) {
 		return nil, err
 	}
 	if resp.StatusCode == http.StatusNotFound {
-		// This is Radarr
-		log.Println("Radarr detected")
-		a.Type = Radarr
+		// This is likely Radarr
 		return GetMovies(a, tvId)
 	}
 	a.Type = Sonarr
@@ -87,6 +84,11 @@ func GetMovies(a *Arr, tvId string) ([]Content, error) {
 	if err != nil {
 		return nil, err
 	}
+	if resp.StatusCode == http.StatusNotFound {
+		// This is likely Lidarr or Readarr
+		return nil, fmt.Errorf("failed to get movies: %s", resp.Status)
+	}
+	a.Type = Radarr
 	defer resp.Body.Close()
 	var movies []Movie
 	if err = json.NewDecoder(resp.Body).Decode(&movies); err != nil {
diff --git a/pkg/repair/repair.go b/pkg/repair/repair.go
index f2c5eb4..9be58da 100644
--- a/pkg/repair/repair.go
+++ b/pkg/repair/repair.go
@@ -156,9 +156,9 @@ func (r *Repair) Start(ctx context.Context) error {
 			return nil
 		case t := <-ticker.C:
 			r.logger.Info().Msgf("Running repair at %v", t.Format("15:04:05"))
-			if err := r.Repair(r.arrs.GetAll(), []string{}); err != nil {
+			err := r.Repair(r.arrs.GetAll(), []string{})
+			if err != nil {
 				r.logger.Info().Msgf("Error during repair: %v", err)
-				continue
 			}
 
 			// If using time-of-day schedule, reset the ticker for next day