feat(lists): integrate Omegabrr (#1885)

* feat(lists): integrate Omegabrr * feat(lists): add missing lists index * feat(lists): add db repo * feat(lists): add db migrations * feat(lists): labels * feat(lists): url lists and more arrs * fix(lists): db migrations client_id wrong type * fix(lists): db fields * feat(lists): create list form wip * feat(lists): show in list and create * feat(lists): update and delete * feat(lists): trigger via webhook * feat(lists): add webhook handler * fix(arr): encode json to pointer * feat(lists): rename endpoint to lists * feat(lists): fetch tags from arr * feat(lists): process plaintext lists * feat(lists): add background refresh job * run every 6th hour with a random start delay between 1-35 seconds * feat(lists): refresh on save and improve logging * feat(lists): cast arr client to pointer * feat(lists): improve error handling * feat(lists): reset shows field with match release * feat(lists): filter opts all lists * feat(lists): trigger on update if enabled * feat(lists): update option for lists * feat(lists): show connected filters in list * feat(lists): missing listSvc dep * feat(lists): cleanup * feat(lists): typo arr list * feat(lists): radarr include original * feat(lists): rename ExcludeAlternateTitle to IncludeAlternateTitle * fix(lists): arr client type conversion to pointer * fix(actions): only log panic recover if err not nil * feat(lists): show spinner on save * feat(lists): show icon in filters list * feat(lists): change icon color in filters list * feat(lists): delete relations on filter delete
2025-07-23 16:59:12 +00:00 · 2024-12-25 13:23:37 +01:00 · 2024-12-25 13:23:37 +01:00 · 221bc35371
commit 221bc35371
parent b68ae334ca
77 changed files with 5025 additions and 254 deletions
--- a/internal/list/process_arr_sonarr.go
+++ b/internal/list/process_arr_sonarr.go
@ -0,0 +1,147 @@
+package list
+
+import (
+	"context"
+	"sort"
+	"strings"
+
+	"github.com/autobrr/autobrr/internal/domain"
+	"github.com/autobrr/autobrr/pkg/arr"
+	"github.com/autobrr/autobrr/pkg/arr/sonarr"
+	"github.com/autobrr/autobrr/pkg/errors"
+
+	"github.com/rs/zerolog"
+)
+
+func (s *service) sonarr(ctx context.Context, list *domain.List) error {
+	var arrType string
+	if list.Type == domain.ListTypeWhisparr {
+		arrType = "whisparr"
+	} else {
+		arrType = "sonarr"
+	}
+
+	l := s.log.With().Str("list", list.Name).Str("type", arrType).Int("client", list.ClientID).Logger()
+
+	l.Debug().Msgf("gathering titles...")
+
+	titles, err := s.processSonarr(ctx, list, &l)
+	if err != nil {
+		return err
+	}
+
+	l.Debug().Msgf("got %d filter titles", len(titles))
+
+	if len(titles) == 0 {
+		l.Debug().Msgf("no titles found to update for list: %v", list.Name)
+		return nil
+	}
+
+	joinedTitles := strings.Join(titles, ",")
+
+	l.Trace().Str("titles", joinedTitles).Msgf("found %d titles", len(joinedTitles))
+
+	filterUpdate := domain.FilterUpdate{Shows: &joinedTitles}
+
+	if list.MatchRelease {
+		filterUpdate.Shows = &nullString
+		filterUpdate.MatchReleases = &joinedTitles
+	}
+
+	for _, filter := range list.Filters {
+		l.Debug().Msgf("updating filter: %v", filter.ID)
+
+		filterUpdate.ID = filter.ID
+
+		if err := s.filterSvc.UpdatePartial(ctx, filterUpdate); err != nil {
+			return errors.Wrap(err, "error updating filter: %v", filter.ID)
+		}
+
+		l.Debug().Msgf("successfully updated filter: %v", filter.ID)
+	}
+
+	return nil
+}
+
+func (s *service) processSonarr(ctx context.Context, list *domain.List, logger *zerolog.Logger) ([]string, error) {
+	downloadClient, err := s.downloadClientSvc.GetClient(ctx, int32(list.ClientID))
+	if err != nil {
+		return nil, errors.Wrap(err, "could not get client with id %d", list.ClientID)
+	}
+
+	if !downloadClient.Enabled {
+		return nil, errors.New("client %s %s not enabled", downloadClient.Type, downloadClient.Name)
+	}
+
+	client := downloadClient.Client.(*sonarr.Client)
+
+	var tags []*arr.Tag
+	if len(list.TagsExclude) > 0 || len(list.TagsInclude) > 0 {
+		t, err := client.GetTags(ctx)
+		if err != nil {
+			logger.Debug().Msg("could not get tags")
+		}
+		tags = t
+	}
+
+	shows, err := client.GetAllSeries(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	logger.Debug().Msgf("found %d shows to process", len(shows))
+
+	titleSet := make(map[string]struct{})
+	var processedTitles int
+
+	for _, show := range shows {
+		if !list.ShouldProcessItem(show.Monitored) {
+			continue
+		}
+
+		//if !s.shouldProcessItem(show.Monitored, list) {
+		//	continue
+		//}
+
+		if len(list.TagsInclude) > 0 {
+			if len(show.Tags) == 0 {
+				continue
+			}
+			if !containsTag(tags, show.Tags, list.TagsInclude) {
+				continue
+			}
+		}
+
+		if len(list.TagsExclude) > 0 {
+			if containsTag(tags, show.Tags, list.TagsExclude) {
+				continue
+			}
+		}
+
+		processedTitles++
+
+		titles := processTitle(show.Title, list.MatchRelease)
+		for _, title := range titles {
+			titleSet[title] = struct{}{}
+		}
+
+		if list.IncludeAlternateTitles {
+			for _, title := range show.AlternateTitles {
+				altTitles := processTitle(title.Title, list.MatchRelease)
+				for _, altTitle := range altTitles {
+					titleSet[altTitle] = struct{}{}
+				}
+			}
+		}
+	}
+
+	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 titles and created %d release titles", len(shows), processedTitles, len(uniqueTitles))
+
+	return uniqueTitles, nil
+}