719528a06b
Some checks failed
Build / build (push) Failing after 13m30s
Items from RSS feeds that have no media ID (tmdb/imdb/tvdb) are now enriched automatically using TMDB's search API. The release name is parsed to extract a clean title and year, then searched against TMDB to retrieve the TMDB ID before validation. - Add media/tmdb package with SearchMovies, SearchShows, and ExtractTitleAndYear (parses torrent release names) - Add EnrichFeedItemWithTmdbId to media.Client, called in rss/process.go before the ID validation switch - Add --run-now flag to nabarr run to trigger all feeds immediately - Wire media.Client through rss.Client and rssJob - Merge feature/add-tag-option (tag support for Sonarr/Radarr) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
101 lines
1.9 KiB
Go
101 lines
1.9 KiB
Go
package rss
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/robfig/cron/v3"
|
|
|
|
"github.com/l3uddz/nabarr/cmd/nabarr/pvr"
|
|
"github.com/l3uddz/nabarr/util"
|
|
)
|
|
|
|
func (c *Client) AddJob(feed feedItem) error {
|
|
// prepare job
|
|
if feed.Cron == "" {
|
|
feed.Cron = "*/15 * * * *"
|
|
}
|
|
|
|
if feed.CacheDuration == 0 {
|
|
feed.CacheDuration = (24 * time.Hour) * 28
|
|
}
|
|
|
|
l := c.log.With().
|
|
Str("feed_name", feed.Name).
|
|
Logger()
|
|
|
|
// create job
|
|
job := &rssJob{
|
|
name: feed.Name,
|
|
log: l,
|
|
http: util.NewRetryableHttpClient(60*time.Second, nil, &l),
|
|
mediaClient: c.media,
|
|
|
|
url: feed.URL,
|
|
pvrs: make(map[string]pvr.PVR, 0),
|
|
|
|
attempts: 0,
|
|
errors: make([]error, 0),
|
|
|
|
cron: c.cron,
|
|
cache: c.cache,
|
|
cacheDuration: feed.CacheDuration,
|
|
cacheFiltersHash: "",
|
|
}
|
|
|
|
// add pvrs
|
|
for _, p := range feed.Pvrs {
|
|
po, exists := c.pvrs[p]
|
|
if !exists {
|
|
return fmt.Errorf("pvr object does not exist: %v", p)
|
|
}
|
|
job.pvrs[p] = po
|
|
job.cacheFiltersHash += po.GetFiltersHash()
|
|
}
|
|
|
|
// schedule job
|
|
if id, err := c.cron.AddJob(feed.Cron, cron.NewChain(
|
|
cron.SkipIfStillRunning(cron.DiscardLogger)).Then(job),
|
|
); err != nil {
|
|
return fmt.Errorf("add job: %w", err)
|
|
} else {
|
|
job.jobID = id
|
|
}
|
|
|
|
c.jobs = append(c.jobs, job)
|
|
job.log.Info().Msg("Initialised")
|
|
return nil
|
|
}
|
|
|
|
func (j *rssJob) Run() {
|
|
// increase attempt counter
|
|
j.attempts++
|
|
|
|
// run job
|
|
err := j.process()
|
|
|
|
// handle job response
|
|
switch {
|
|
case err == nil:
|
|
// job completed successfully
|
|
j.attempts = 0
|
|
j.errors = j.errors[:0]
|
|
return
|
|
|
|
default:
|
|
j.log.Warn().
|
|
Err(err).
|
|
Int("attempts", j.attempts).
|
|
Msg("Unexpected error occurred")
|
|
j.errors = append(j.errors, err)
|
|
}
|
|
|
|
if j.attempts > 5 {
|
|
j.log.Error().
|
|
Errs("error", j.errors).
|
|
Int("attempts", j.attempts).
|
|
Msg("Consecutive errors occurred while refreshing rss, job has been stopped...")
|
|
j.cron.Remove(j.jobID)
|
|
}
|
|
}
|