mirror of
https://github.com/idanoo/autobrr
synced 2025-07-22 16:29:12 +00:00
0391629862
* chore: update copyright year in license headers * Revert "chore: update copyright year in license headers" This reverts commit 3e58129c431b9a491089ce36b908f9bb6ba38ed3. * chore: update copyright year in license headers * fix: sort go imports * fix: add missing license headers
116 lines
2.8 KiB
Go
116 lines
2.8 KiB
Go
// Copyright (c) 2021 - 2025, Ludvig Lundgren and the autobrr contributors.
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
package list
|
|
|
|
import (
|
|
"context"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
"github.com/autobrr/autobrr/pkg/arr/readarr"
|
|
"github.com/autobrr/autobrr/pkg/errors"
|
|
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
func (s *service) readarr(ctx context.Context, list *domain.List) error {
|
|
l := s.log.With().Str("list", list.Name).Str("type", "readarr").Int("client", list.ClientID).Logger()
|
|
|
|
l.Debug().Msgf("gathering titles...")
|
|
|
|
titles, err := s.processReadarr(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{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) processReadarr(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.(*readarr.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
|
|
//}
|
|
|
|
books, err := client.GetBooks(ctx, "")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
logger.Debug().Msgf("found %d books to process", len(books))
|
|
|
|
var titles []string
|
|
var processedTitles int
|
|
|
|
for _, book := range books {
|
|
if !list.ShouldProcessItem(book.Monitored) {
|
|
continue
|
|
}
|
|
|
|
//if len(list.TagsInclude) > 0 {
|
|
// if len(book.Tags) == 0 {
|
|
// continue
|
|
// }
|
|
// if !containsTag(tags, book.Tags, list.TagsInclude) {
|
|
// continue
|
|
// }
|
|
//}
|
|
//
|
|
//if len(list.TagsExclude) > 0 {
|
|
// if containsTag(tags, book.Tags, list.TagsExclude) {
|
|
// continue
|
|
// }
|
|
//}
|
|
|
|
processedTitles++
|
|
|
|
titles = append(titles, processTitle(book.Title, list.MatchRelease)...)
|
|
}
|
|
|
|
sort.Strings(titles)
|
|
logger.Debug().Msgf("from a total of %d books we found %d titles and created %d release titles", len(books), processedTitles, len(titles))
|
|
|
|
return titles, nil
|
|
}
|