fix(arrs): deduplicate titles (#91)

autobrr · Aug 24, 2024 · 81446c6 · 81446c6
1 parent f867a2a
commit 81446c6
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 17 deletions.
diff --git a/internal/processor/radarr.go b/internal/processor/radarr.go
@@ -2,6 +2,7 @@ package processor
 
 import (
 	"context"
+	"sort"
 	"strings"
 	"time"
 
@@ -89,7 +90,7 @@ func (s Service) processRadarr(ctx context.Context, cfg *domain.ArrConfig, logge
 
 	logger.Debug().Msgf("found %d movies to process", len(movies))
 
-	var titles []string
+	titleSet := make(map[string]struct{})
 	var monitoredTitles int
 
 	for _, movie := range movies {
@@ -119,20 +120,22 @@ func (s Service) processRadarr(ctx context.Context, cfg *domain.ArrConfig, logge
 		monitoredTitles++
 
 		// Taking the international title and the original title and appending them to the titles array.
-		titlesMap := make(map[string]bool) // Initialize a map to keep track of unique titles.
-		t := m.Title
-		ot := m.OriginalTitle
-
-		for _, title := range []string{t, ot} {
-			if title != "" && !titlesMap[title] {
-				titlesMap[title] = true
-				titles = append(titles, processTitle(title, cfg.MatchRelease)...)
+		for _, title := range []string{m.Title, m.OriginalTitle} {
+			if title != "" {
+				for _, t := range processTitle(title, cfg.MatchRelease) {
+					titleSet[t] = struct{}{}
+				}
 			}
 		}
+	}
 
+	uniqueTitles := make([]string, 0, len(titleSet))
+	for title := range titleSet {
+		uniqueTitles = append(uniqueTitles, title)
 	}
 
-	logger.Debug().Msgf("from a total of %d movies we found %d monitored and created %d release titles", len(movies), monitoredTitles, len(titles))
+	sort.Strings(uniqueTitles)
+	logger.Debug().Msgf("from a total of %d movies we found %d monitored and created %d release titles", len(movies), monitoredTitles, len(uniqueTitles))
 
-	return titles, nil
+	return uniqueTitles, nil
 }
diff --git a/internal/processor/sonarr.go b/internal/processor/sonarr.go
@@ -97,7 +97,7 @@ func (s Service) processSonarr(ctx context.Context, cfg *domain.ArrConfig, logge
 
 	logger.Debug().Msgf("found %d shows to process", len(shows))
 
-	var titles []string
+	titleSet := make(map[string]struct{})
 	var monitoredTitles int
 
 	for _, show := range shows {
@@ -129,17 +129,28 @@ func (s Service) processSonarr(ctx context.Context, cfg *domain.ArrConfig, logge
 		//titles = append(titles, rls.MustNormalize(s.Title))
 		//titles = append(titles, rls.MustClean(s.Title))
 
-		titles = append(titles, processTitle(s.Title, cfg.MatchRelease)...)
+		titles := processTitle(s.Title, cfg.MatchRelease)
+		for _, title := range titles {
+			titleSet[title] = struct{}{}
+		}
 
 		if !cfg.ExcludeAlternateTitles {
 			for _, title := range s.AlternateTitles {
-				titles = append(titles, processTitle(title.Title, cfg.MatchRelease)...)
+				altTitles := processTitle(title.Title, cfg.MatchRelease)
+				for _, altTitle := range altTitles {
+					titleSet[altTitle] = struct{}{}
+				}
 			}
 		}
 	}
 
-	sort.Strings(titles)
-	logger.Debug().Msgf("from a total of %d shows we found %d monitored and created %d release titles", len(shows), monitoredTitles, len(titles))
+	uniqueTitles := make([]string, 0, len(titleSet))
+	for title := range titleSet {
+		uniqueTitles = append(uniqueTitles, title)
+	}
+
+	sort.Strings(uniqueTitles)
+	logger.Debug().Msgf("from a total of %d shows we found %d monitored and created %d release titles", len(shows), monitoredTitles, len(uniqueTitles))
 
-	return titles, nil
+	return uniqueTitles, nil
 }