laurent/nabarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(rss): improve dispatching of tmdb items to appropriate pvr's (#33)

Browse Source
This commit is contained in:
l3uddz
2021-11-21 13:24:13 +00:00
committed by GitHub
parent 685a9a075d
commit a4d8a0e4cc
5 changed files with 80 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
rss/process.go
View File

@@ -24,8 +24,16 @@ func (j *rssJob) process() error {
return nil
}
for p, _ := range items {
j.queueItemWithPvrs(&items[p])
queuedItems := 0
for p := range items {
if j.queueItemWithPvrs(&items[p]) {
queuedItems++
}
}
if queuedItems == 0 {
j.log.Debug().Msg("Refreshed, no valid items to queue")
return nil
}
j.log.Info().
@@ -34,17 +42,21 @@ func (j *rssJob) process() error {
return nil
}
func (j *rssJob) queueItemWithPvrs(item *media.FeedItem) {
func (j *rssJob) queueItemWithPvrs(item *media.FeedItem) bool {
queued := false
for _, pvr := range j.pvrs {
switch {
case (item.TvdbId != "" || item.TmdbId != "") && pvr.Type() == "sonarr":
case (item.TvdbId != "" || item.TmdbId != "") && util.ContainsTvCategory(item.Categories) && pvr.Type() == "sonarr":
// tvdbId/tmdbId is present, queue with sonarr
pvr.QueueFeedItem(item)
case (item.ImdbId != "" || item.TmdbId != "") && pvr.Type() == "radarr":
queued = true
case (item.ImdbId != "" || item.TmdbId != "") && util.ContainsMovieCategory(item.Categories) && pvr.Type() == "radarr":
// imdbId is present, queue with radarr
pvr.QueueFeedItem(item)
queued = true
}
}
return queued
}
func (j *rssJob) getFeed() ([]media.FeedItem, error) {
@@ -97,6 +109,8 @@ func (j *rssJob) getFeed() ([]media.FeedItem, error) {
// process feed item attributes
for _, a := range i.Attributes {
switch strings.ToLower(a.Name) {
case "category":
b.Channel.Items[p].Categories = append(b.Channel.Items[p].Categories, a.Value)
case "language":
b.Channel.Items[p].Language = a.Value
case "tvdb", "tvdbid", "thetvdb":