feat(filters): improve rejection handling (#1776)

* feat(filters): improve rejection handling

* fix(filters): rejection tests

* fix(filters): size check error rejection
This commit is contained in:
ze0s 2024-10-20 22:58:55 +02:00 committed by GitHub
parent ae779e5461
commit f029de233f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 586 additions and 311 deletions

View file

@ -52,6 +52,10 @@ type FilterDownloads struct {
TotalCount int TotalCount int
} }
func (f *FilterDownloads) String() string {
return fmt.Sprintf("Hour: %d, Day: %d, Week: %d, Month: %d, Total: %d", f.HourCount, f.DayCount, f.WeekCount, f.MonthCount, f.TotalCount)
}
type FilterMaxDownloadsUnit string type FilterMaxDownloadsUnit string
const ( const (
@ -164,6 +168,7 @@ type Filter struct {
Indexers []Indexer `json:"indexers"` Indexers []Indexer `json:"indexers"`
Downloads *FilterDownloads `json:"-"` Downloads *FilterDownloads `json:"-"`
Rejections []string `json:"-"` Rejections []string `json:"-"`
RejectReasons *RejectionReasons `json:"-"`
} }
type FilterExternal struct { type FilterExternal struct {
@ -354,156 +359,156 @@ func (f *Filter) Sanitize() error {
return nil return nil
} }
func (f *Filter) CheckFilter(r *Release) ([]string, bool) { func (f *Filter) CheckFilter(r *Release) (*RejectionReasons, bool) {
f.RejectReasons = NewRejectionReasons()
// max downloads check. If reached return early so other filters can be checked as quick as possible. // max downloads check. If reached return early so other filters can be checked as quick as possible.
if f.MaxDownloads > 0 && !f.checkMaxDownloads() { if f.MaxDownloads > 0 && !f.checkMaxDownloads() {
f.addRejectionF("max downloads (%d) this (%v) reached", f.MaxDownloads, f.MaxDownloadsUnit) f.RejectReasons.Addf("max downloads", fmt.Sprintf("[max downloads] reached %d per %s", f.MaxDownloads, f.MaxDownloadsUnit), f.Downloads.String(), fmt.Sprintf("reached %d per %s", f.MaxDownloads, f.MaxDownloadsUnit))
return f.Rejections, false return f.RejectReasons, false
} }
if len(f.Bonus) > 0 && !sliceContainsSlice(r.Bonus, f.Bonus) { if len(f.Bonus) > 0 && !sliceContainsSlice(r.Bonus, f.Bonus) {
r.addRejectionF("bonus not matching. got: %v want: %v", r.Bonus, f.Bonus) f.RejectReasons.Add("bonus", r.Bonus, f.Bonus)
} }
if f.Freeleech && r.Freeleech != f.Freeleech { if f.Freeleech && r.Freeleech != f.Freeleech {
f.addRejection("wanted: freeleech") f.RejectReasons.Add("freeleech", r.Freeleech, f.Freeleech)
} }
if f.FreeleechPercent != "" && !checkFreeleechPercent(r.FreeleechPercent, f.FreeleechPercent) { if f.FreeleechPercent != "" && !checkFreeleechPercent(r.FreeleechPercent, f.FreeleechPercent) {
f.addRejectionF("freeleech percent not matching. got: %v want: %v", r.FreeleechPercent, f.FreeleechPercent) f.RejectReasons.Add("freeleech percent", r.FreeleechPercent, f.FreeleechPercent)
} }
if len(f.Origins) > 0 && !containsSlice(r.Origin, f.Origins) { if len(f.Origins) > 0 && !containsSlice(r.Origin, f.Origins) {
f.addRejectionF("origin not matching. got: %v want: %v", r.Origin, f.Origins) f.RejectReasons.Add("match origin", r.Origin, f.Origins)
} }
if len(f.ExceptOrigins) > 0 && containsSlice(r.Origin, f.ExceptOrigins) { if len(f.ExceptOrigins) > 0 && containsSlice(r.Origin, f.ExceptOrigins) {
f.addRejectionF("except origin not matching. got: %v unwanted: %v", r.Origin, f.ExceptOrigins) f.RejectReasons.Add("except origin", r.Origin, f.ExceptOrigins)
} }
// title is the parsed title // title is the parsed title
if f.Shows != "" && !contains(r.Title, f.Shows) { if f.Shows != "" && !contains(r.Title, f.Shows) {
f.addRejectionF("shows not matching. got: %v want: %v", r.Title, f.Shows) f.RejectReasons.Add("shows", r.Title, f.Shows)
} }
if f.Seasons != "" && !containsIntStrings(r.Season, f.Seasons) { if f.Seasons != "" && !containsIntStrings(r.Season, f.Seasons) {
f.addRejectionF("season not matching. got: %d want: %v", r.Season, f.Seasons) f.RejectReasons.Add("season", r.Season, f.Seasons)
} }
if f.Episodes != "" && !containsIntStrings(r.Episode, f.Episodes) { if f.Episodes != "" && !containsIntStrings(r.Episode, f.Episodes) {
f.addRejectionF("episodes not matching. got: %d want: %v", r.Episode, f.Episodes) f.RejectReasons.Add("episodes", r.Episode, f.Episodes)
} }
// matchRelease // matchRelease
// match against regex // match against regex
if f.UseRegex { if f.UseRegex {
if f.MatchReleases != "" && !matchRegex(r.TorrentName, f.MatchReleases) { if f.MatchReleases != "" && !matchRegex(r.TorrentName, f.MatchReleases) {
f.addRejectionF("match release regex not matching. got: %v want: %v", r.TorrentName, f.MatchReleases) f.RejectReasons.Add("match releases: REGEX", r.TorrentName, f.MatchReleases)
} }
if f.ExceptReleases != "" && matchRegex(r.TorrentName, f.ExceptReleases) { if f.ExceptReleases != "" && matchRegex(r.TorrentName, f.ExceptReleases) {
f.addRejectionF("except releases regex: unwanted release. got: %v want: %v", r.TorrentName, f.ExceptReleases) f.RejectReasons.Add("except releases: REGEX", r.TorrentName, f.ExceptReleases)
} }
} else { } else {
if f.MatchReleases != "" && !containsFuzzy(r.TorrentName, f.MatchReleases) { if f.MatchReleases != "" && !containsFuzzy(r.TorrentName, f.MatchReleases) {
f.addRejectionF("match release not matching. got: %v want: %v", r.TorrentName, f.MatchReleases) f.RejectReasons.Add("match releases", r.TorrentName, f.MatchReleases)
} }
if f.ExceptReleases != "" && containsFuzzy(r.TorrentName, f.ExceptReleases) { if f.ExceptReleases != "" && containsFuzzy(r.TorrentName, f.ExceptReleases) {
f.addRejectionF("except releases: unwanted release. got: %v want: %v", r.TorrentName, f.ExceptReleases) f.RejectReasons.Add("except releases", r.TorrentName, f.ExceptReleases)
} }
} }
if f.MatchReleaseGroups != "" && !contains(r.Group, f.MatchReleaseGroups) { if f.MatchReleaseGroups != "" && !contains(r.Group, f.MatchReleaseGroups) {
f.addRejectionF("release groups not matching. got: %v want: %v", r.Group, f.MatchReleaseGroups) f.RejectReasons.Add("match release groups", r.Group, f.MatchReleaseGroups)
} }
if f.ExceptReleaseGroups != "" && contains(r.Group, f.ExceptReleaseGroups) { if f.ExceptReleaseGroups != "" && contains(r.Group, f.ExceptReleaseGroups) {
f.addRejectionF("unwanted release group. got: %v unwanted: %v", r.Group, f.ExceptReleaseGroups) f.RejectReasons.Add("except release groups", r.Group, f.ExceptReleaseGroups)
} }
// check raw releaseTags string // check raw releaseTags string
if f.UseRegexReleaseTags { if f.UseRegexReleaseTags {
if f.MatchReleaseTags != "" && !matchRegex(r.ReleaseTags, f.MatchReleaseTags) { if f.MatchReleaseTags != "" && !matchRegex(r.ReleaseTags, f.MatchReleaseTags) {
f.addRejectionF("match release tags regex not matching. got: %v want: %v", r.ReleaseTags, f.MatchReleaseTags) f.RejectReasons.Add("match release tags: REGEX", r.ReleaseTags, f.MatchReleaseTags)
} }
if f.ExceptReleaseTags != "" && matchRegex(r.ReleaseTags, f.ExceptReleaseTags) { if f.ExceptReleaseTags != "" && matchRegex(r.ReleaseTags, f.ExceptReleaseTags) {
f.addRejectionF("except release tags regex: unwanted release. got: %v want: %v", r.ReleaseTags, f.ExceptReleaseTags) f.RejectReasons.Add("except release tags: REGEX", r.ReleaseTags, f.ExceptReleaseTags)
} }
} else { } else {
if f.MatchReleaseTags != "" && !containsFuzzy(r.ReleaseTags, f.MatchReleaseTags) { if f.MatchReleaseTags != "" && !containsFuzzy(r.ReleaseTags, f.MatchReleaseTags) {
f.addRejectionF("match release tags not matching. got: %v want: %v", r.ReleaseTags, f.MatchReleaseTags) f.RejectReasons.Add("match release tags", r.ReleaseTags, f.MatchReleaseTags)
} }
if f.ExceptReleaseTags != "" && containsFuzzy(r.ReleaseTags, f.ExceptReleaseTags) { if f.ExceptReleaseTags != "" && containsFuzzy(r.ReleaseTags, f.ExceptReleaseTags) {
f.addRejectionF("except release tags: unwanted release. got: %v want: %v", r.ReleaseTags, f.ExceptReleaseTags) f.RejectReasons.Add("except release tags", r.ReleaseTags, f.ExceptReleaseTags)
} }
} }
if f.MatchUploaders != "" && !contains(r.Uploader, f.MatchUploaders) { if f.MatchUploaders != "" && !contains(r.Uploader, f.MatchUploaders) {
f.addRejectionF("uploaders not matching. got: %v want: %v", r.Uploader, f.MatchUploaders) f.RejectReasons.Add("match uploaders", r.Uploader, f.MatchUploaders)
} }
if f.ExceptUploaders != "" && contains(r.Uploader, f.ExceptUploaders) { if f.ExceptUploaders != "" && contains(r.Uploader, f.ExceptUploaders) {
f.addRejectionF("unwanted uploaders. got: %v unwanted: %v", r.Uploader, f.ExceptUploaders) f.RejectReasons.Add("except uploaders", r.Uploader, f.ExceptUploaders)
} }
if len(f.MatchLanguage) > 0 && !sliceContainsSlice(r.Language, f.MatchLanguage) { if len(f.MatchLanguage) > 0 && !sliceContainsSlice(r.Language, f.MatchLanguage) {
f.addRejectionF("language not matching. got: %v want: %v", r.Language, f.MatchLanguage) f.RejectReasons.Add("match language", r.Language, f.MatchLanguage)
} }
if len(f.ExceptLanguage) > 0 && sliceContainsSlice(r.Language, f.ExceptLanguage) { if len(f.ExceptLanguage) > 0 && sliceContainsSlice(r.Language, f.ExceptLanguage) {
f.addRejectionF("language unwanted. got: %v want: %v", r.Language, f.ExceptLanguage) f.RejectReasons.Add("except language", r.Language, f.ExceptLanguage)
} }
if len(f.Resolutions) > 0 && !containsSlice(r.Resolution, f.Resolutions) { if len(f.Resolutions) > 0 && !containsSlice(r.Resolution, f.Resolutions) {
f.addRejectionF("resolution not matching. got: %v want: %v", r.Resolution, f.Resolutions) f.RejectReasons.Add("resolution", r.Resolution, f.Resolutions)
} }
if len(f.Codecs) > 0 && !sliceContainsSlice(r.Codec, f.Codecs) { if len(f.Codecs) > 0 && !sliceContainsSlice(r.Codec, f.Codecs) {
f.addRejectionF("codec not matching. got: %v want: %v", r.Codec, f.Codecs) f.RejectReasons.Add("codec", r.Codec, f.Codecs)
} }
if len(f.Sources) > 0 && !containsSlice(r.Source, f.Sources) { if len(f.Sources) > 0 && !containsSlice(r.Source, f.Sources) {
f.addRejectionF("source not matching. got: %v want: %v", r.Source, f.Sources) f.RejectReasons.Add("source", r.Source, f.Sources)
} }
if len(f.Containers) > 0 && !containsSlice(r.Container, f.Containers) { if len(f.Containers) > 0 && !containsSlice(r.Container, f.Containers) {
f.addRejectionF("container not matching. got: %v want: %v", r.Container, f.Containers) f.RejectReasons.Add("container", r.Container, f.Containers)
} }
// HDR is parsed into the Codec slice from rls
if len(f.MatchHDR) > 0 && !matchHDR(r.HDR, f.MatchHDR) { if len(f.MatchHDR) > 0 && !matchHDR(r.HDR, f.MatchHDR) {
f.addRejectionF("hdr not matching. got: %v want: %v", r.HDR, f.MatchHDR) f.RejectReasons.Add("match hdr", r.HDR, f.MatchHDR)
} }
// HDR is parsed into the Codec slice from rls
if len(f.ExceptHDR) > 0 && matchHDR(r.HDR, f.ExceptHDR) { if len(f.ExceptHDR) > 0 && matchHDR(r.HDR, f.ExceptHDR) {
f.addRejectionF("hdr unwanted. got: %v want: %v", r.HDR, f.ExceptHDR) f.RejectReasons.Add("except hdr", r.HDR, f.ExceptHDR)
} }
// Other is parsed into the Other slice from rls // Other is parsed into the Other slice from rls
if len(f.MatchOther) > 0 && !sliceContainsSlice(r.Other, f.MatchOther) { if len(f.MatchOther) > 0 && !sliceContainsSlice(r.Other, f.MatchOther) {
f.addRejectionF("match other not matching. got: %v want: %v", r.Other, f.MatchOther) f.RejectReasons.Add("match other", r.Other, f.MatchOther)
} }
// Other is parsed into the Other slice from rls // Other is parsed into the Other slice from rls
if len(f.ExceptOther) > 0 && sliceContainsSlice(r.Other, f.ExceptOther) { if len(f.ExceptOther) > 0 && sliceContainsSlice(r.Other, f.ExceptOther) {
f.addRejectionF("except other unwanted. got: %v unwanted: %v", r.Other, f.ExceptOther) f.RejectReasons.Add("except other", r.Other, f.ExceptOther)
} }
if f.Years != "" && !containsIntStrings(r.Year, f.Years) { if f.Years != "" && !containsIntStrings(r.Year, f.Years) {
f.addRejectionF("year not matching. got: %d want: %v", r.Year, f.Years) f.RejectReasons.Add("year", r.Year, f.Years)
} }
if f.Months != "" && !containsIntStrings(r.Month, f.Months) { if f.Months != "" && !containsIntStrings(r.Month, f.Months) {
f.addRejectionF("month not matching. got: %d want: %v", r.Month, f.Months) f.RejectReasons.Add("month", r.Month, f.Months)
} }
if f.Days != "" && !containsIntStrings(r.Day, f.Days) { if f.Days != "" && !containsIntStrings(r.Day, f.Days) {
f.addRejectionF("day not matching. got: %d want: %v", r.Day, f.Days) f.RejectReasons.Add("day", r.Day, f.Days)
} }
if f.MatchCategories != "" { if f.MatchCategories != "" {
@ -513,7 +518,7 @@ func (f *Filter) CheckFilter(r *Release) ([]string, bool) {
categories = append(categories, r.Category) categories = append(categories, r.Category)
} }
if !contains(r.Category, f.MatchCategories) && !containsAny(categories, f.MatchCategories) { if !contains(r.Category, f.MatchCategories) && !containsAny(categories, f.MatchCategories) {
f.addRejectionF("category not matching. got: %v want: %v", strings.Join(categories, ","), f.MatchCategories) f.RejectReasons.Add("match category", strings.Join(categories, ","), f.MatchCategories)
} }
} }
@ -524,121 +529,126 @@ func (f *Filter) CheckFilter(r *Release) ([]string, bool) {
categories = append(categories, r.Category) categories = append(categories, r.Category)
} }
if contains(r.Category, f.ExceptCategories) && containsAny(categories, f.ExceptCategories) { if contains(r.Category, f.ExceptCategories) && containsAny(categories, f.ExceptCategories) {
f.addRejectionF("category unwanted. got: %v unwanted: %v", strings.Join(categories, ","), f.ExceptCategories) f.RejectReasons.Add("except category", strings.Join(categories, ","), f.ExceptCategories)
} }
} }
// music related
if len(f.MatchReleaseTypes) > 0 && !containsSlice(r.Category, f.MatchReleaseTypes) { if len(f.MatchReleaseTypes) > 0 && !containsSlice(r.Category, f.MatchReleaseTypes) {
f.addRejectionF("release type not matching. got: %v want: %v", r.Category, f.MatchReleaseTypes) f.RejectReasons.Add("release type", r.Category, f.MatchReleaseTypes)
} }
if (f.MinSize != "" || f.MaxSize != "") && !f.checkSizeFilter(r) { if f.MinSize != "" && !f.checkSizeFilter(r) {
f.addRejectionF("size not matching. got: %v want min: %v max: %v", r.Size, f.MinSize, f.MaxSize) f.RejectReasons.Add("min size", r.Size, f.MinSize)
}
if f.MaxSize != "" && !f.checkSizeFilter(r) {
f.RejectReasons.Add("max size", r.Size, f.MaxSize)
} }
if f.Tags != "" { if f.Tags != "" {
if f.TagsMatchLogic == "ALL" && !containsAll(r.Tags, f.Tags) { if f.TagsMatchLogic == "ALL" && !containsAll(r.Tags, f.Tags) {
f.addRejectionF("tags not matching. got: %v want(all): %v", r.Tags, f.Tags) f.RejectReasons.Add("match tags: ALL", r.Tags, f.Tags)
} else if !containsAny(r.Tags, f.Tags) { // TagsMatchLogic is set to "" by default, this makes sure that "" and "ANY" are treated the same way. } else if !containsAny(r.Tags, f.Tags) { // TagsMatchLogic is set to "" by default, this makes sure that "" and "ANY" are treated the same way.
f.addRejectionF("tags not matching. got: %v want: %v", r.Tags, f.Tags) f.RejectReasons.Add("match tags: ANY", r.Tags, f.Tags)
} }
} }
if f.ExceptTags != "" { if f.ExceptTags != "" {
if f.ExceptTagsMatchLogic == "ALL" && containsAll(r.Tags, f.ExceptTags) { if f.ExceptTagsMatchLogic == "ALL" && containsAll(r.Tags, f.ExceptTags) {
f.addRejectionF("tags unwanted. got: %v don't want: %v", r.Tags, f.ExceptTags) f.RejectReasons.Add("except tags: ALL", r.Tags, f.ExceptTags)
} else if containsAny(r.Tags, f.ExceptTags) { // ExceptTagsMatchLogic is set to "" by default, this makes sure that "" and "ANY" are treated the same way. } else if containsAny(r.Tags, f.ExceptTags) { // ExceptTagsMatchLogic is set to "" by default, this makes sure that "" and "ANY" are treated the same way.
f.addRejectionF("tags unwanted. got: %v don't want: %v", r.Tags, f.ExceptTags) f.RejectReasons.Add("except tags: ANY", r.Tags, f.ExceptTags)
} }
} }
if len(f.Artists) > 0 && !contains(r.Artists, f.Artists) { if len(f.Artists) > 0 && !contains(r.Artists, f.Artists) {
f.addRejectionF("artists not matching. got: %v want: %v", r.Artists, f.Artists) f.RejectReasons.Add("artists", r.Artists, f.Artists)
} }
if len(f.Albums) > 0 && !contains(r.Title, f.Albums) { if len(f.Albums) > 0 && !contains(r.Title, f.Albums) {
f.addRejectionF("albums not matching. got: %v want: %v", r.Title, f.Albums) f.RejectReasons.Add("albums", r.Title, f.Albums)
} }
// Perfect flac requires Cue, Log, Log Score 100, FLAC and 24bit Lossless // Perfect flac requires Cue, Log, Log Score 100, FLAC and 24bit Lossless
if f.PerfectFlac && !f.isPerfectFLAC(r) { if f.PerfectFlac && !f.isPerfectFLAC(r) {
f.addRejectionF("wanted: perfect flac. got: %v", r.Audio) f.RejectReasons.Add("perfect flac", r.Audio, "Cue, Log, Log Score 100, FLAC and 24bit Lossless")
} }
if len(f.Formats) > 0 && !sliceContainsSlice(r.Audio, f.Formats) { if len(f.Formats) > 0 && !sliceContainsSlice(r.Audio, f.Formats) {
f.addRejectionF("formats not matching. got: %v want: %v", r.Audio, f.Formats) f.RejectReasons.Add("formats", r.Audio, f.Formats)
} }
if len(f.Quality) > 0 && !containsMatchBasic(r.Audio, f.Quality) { if len(f.Quality) > 0 && !containsMatchBasic(r.Audio, f.Quality) {
f.addRejectionF("quality not matching. got: %v want: %v", r.Audio, f.Quality) f.RejectReasons.Add("quality", r.Audio, f.Quality)
} }
if len(f.Media) > 0 && !containsSlice(r.Source, f.Media) { if len(f.Media) > 0 && !containsSlice(r.Source, f.Media) {
f.addRejectionF("media not matching. got: %v want: %v", r.Source, f.Media) f.RejectReasons.Add("media", r.Source, f.Media)
} }
if f.Cue && !containsAny(r.Audio, "Cue") { if f.Cue && !containsAny(r.Audio, "Cue") {
f.addRejection("wanted: cue") f.RejectReasons.Add("cue", r.Audio, "Cue")
} }
if f.Log && !containsAny(r.Audio, "Log") { if f.Log && !containsAny(r.Audio, "Log") {
f.addRejection("wanted: log") f.RejectReasons.Add("log", r.Audio, "Log")
} }
if f.Log && f.LogScore != 0 && r.LogScore != f.LogScore { if f.Log && f.LogScore != 0 && r.LogScore != f.LogScore {
f.addRejectionF("log score. got: %v want: %v", r.LogScore, f.LogScore) f.RejectReasons.Add("log score", r.LogScore, f.LogScore)
} }
// check description string // check description string
if f.UseRegexDescription { if f.UseRegexDescription {
if f.MatchDescription != "" && !matchRegex(r.Description, f.MatchDescription) { if f.MatchDescription != "" && !matchRegex(r.Description, f.MatchDescription) {
f.addRejectionF("match description regex not matching. got: %v want: %v", r.Description, f.MatchDescription) f.RejectReasons.Add("match description: REGEX", r.Description, f.MatchDescription)
} }
if f.ExceptDescription != "" && matchRegex(r.Description, f.ExceptDescription) { if f.ExceptDescription != "" && matchRegex(r.Description, f.ExceptDescription) {
f.addRejectionF("except description regex: unwanted release. got: %v want: %v", r.Description, f.ExceptDescription) f.RejectReasons.Add("except description: REGEX", r.Description, f.ExceptDescription)
} }
} else { } else {
if f.MatchDescription != "" && !containsFuzzy(r.Description, f.MatchDescription) { if f.MatchDescription != "" && !containsFuzzy(r.Description, f.MatchDescription) {
f.addRejectionF("match description not matching. got: %v want: %v", r.Description, f.MatchDescription) f.RejectReasons.Add("match description", r.Description, f.MatchDescription)
} }
if f.ExceptDescription != "" && containsFuzzy(r.Description, f.ExceptDescription) { if f.ExceptDescription != "" && containsFuzzy(r.Description, f.ExceptDescription) {
f.addRejectionF("except description: unwanted release. got: %v want: %v", r.Description, f.ExceptDescription) f.RejectReasons.Add("except description", r.Description, f.ExceptDescription)
} }
} }
// Min and Max Seeders/Leechers is only for Torznab feeds // Min and Max Seeders/Leechers is only for Torznab feeds
if f.MinSeeders > 0 { if f.MinSeeders > 0 {
if f.MinSeeders > r.Seeders { if f.MinSeeders > r.Seeders {
f.addRejectionF("min seeders not matcing. got: %d want %d", r.Seeders, f.MinSeeders) f.RejectReasons.Add("min seeders", r.Seeders, f.MinSeeders)
} }
} }
if f.MaxSeeders > 0 { if f.MaxSeeders > 0 {
if f.MaxSeeders < r.Seeders { if f.MaxSeeders < r.Seeders {
f.addRejectionF("max seeders not matcing. got: %d want %d", r.Seeders, f.MaxSeeders) f.RejectReasons.Add("max seeders", r.Seeders, f.MaxSeeders)
} }
} }
if f.MinLeechers > 0 { if f.MinLeechers > 0 {
if f.MinLeechers > r.Leechers { if f.MinLeechers > r.Leechers {
f.addRejectionF("min leechers not matcing. got: %d want %d", r.Leechers, f.MinLeechers) f.RejectReasons.Add("min leechers", r.Leechers, f.MinLeechers)
} }
} }
if f.MaxLeechers > 0 { if f.MaxLeechers > 0 {
if f.MaxLeechers < r.Leechers { if f.MaxLeechers < r.Leechers {
f.addRejectionF("max leechers not matcing. got: %d want %d", r.Leechers, f.MaxLeechers) f.RejectReasons.Add("max leechers", r.Leechers, f.MaxLeechers)
} }
} }
if len(f.Rejections) > 0 { if f.RejectReasons.Len() > 0 {
return f.Rejections, false return f.RejectReasons, false
} }
return nil, true return f.RejectReasons, true
} }
func (f *Filter) checkMaxDownloads() bool { func (f *Filter) checkMaxDownloads() bool {
@ -699,7 +709,7 @@ func (f *Filter) checkSizeFilter(r *Release) bool {
sizeOK, err := f.CheckReleaseSize(r.Size) sizeOK, err := f.CheckReleaseSize(r.Size)
if err != nil { if err != nil {
f.addRejectionF("size: error checking release size against filter: %+v", err) f.RejectReasons.Add("size: ERROR", fmt.Sprintf("error checking release size against filter: %v", err), f.MinSize)
return false return false
} }
@ -736,35 +746,6 @@ func (f *Filter) IsPerfectFLAC(r *Release) ([]string, bool) {
return rejections, len(rejections) == 0 return rejections, len(rejections) == 0
} }
func (f *Filter) addRejection(reason string) {
f.Rejections = append(f.Rejections, reason)
}
func (f *Filter) AddRejectionF(format string, v ...interface{}) {
f.addRejectionF(format, v...)
}
func (f *Filter) addRejectionF(format string, v ...interface{}) {
f.Rejections = append(f.Rejections, fmt.Sprintf(format, v...))
}
// ResetRejections reset rejections
func (f *Filter) resetRejections() {
f.Rejections = []string{}
}
func (f *Filter) RejectionsString(trim bool) string {
if len(f.Rejections) > 0 {
out := strings.Join(f.Rejections, ", ")
if trim && len(out) > 1024 {
out = out[:1024]
}
return out
}
return ""
}
func matchRegex(tag string, filterList string) bool { func matchRegex(tag string, filterList string) bool {
if tag == "" { if tag == "" {
return false return false
@ -1121,12 +1102,12 @@ func (f *Filter) CheckReleaseSize(releaseSize uint64) (bool, error) {
} }
if minBytes != nil && releaseSize <= *minBytes { if minBytes != nil && releaseSize <= *minBytes {
f.addRejectionF("release size %d bytes smaller than filter min size %d bytes", releaseSize, *minBytes) f.RejectReasons.Addf("release size", "release size %d bytes is smaller than filter min size %d bytes", releaseSize, *minBytes)
return false, nil return false, nil
} }
if maxBytes != nil && releaseSize >= *maxBytes { if maxBytes != nil && releaseSize >= *maxBytes {
f.addRejectionF("release size %d bytes is larger than filter max size %d bytes", releaseSize, *maxBytes) f.RejectReasons.Addf("release size", "release size %d bytes is larger than filter max size %d bytes", releaseSize, *maxBytes)
return false, nil return false, nil
} }

File diff suppressed because one or more lines are too long

View file

@ -0,0 +1,140 @@
package domain
import (
"encoding/json"
"fmt"
"strings"
)
type Rejection struct {
key string
got any
want any
format string
}
type RejectionReasons struct {
data []Rejection
}
func (r *RejectionReasons) Len() int {
return len(r.data)
}
func NewRejectionReasons() *RejectionReasons {
return &RejectionReasons{
data: make([]Rejection, 0),
}
}
func (r *RejectionReasons) String() string {
if len(r.data) == 0 {
return ""
}
builder := strings.Builder{}
for i, rejection := range r.data {
if i > 0 {
builder.WriteString(", ")
}
if rejection.format != "" {
fmt.Fprintf(&builder, rejection.format, rejection.key, rejection.got, rejection.want)
continue
}
fmt.Fprintf(&builder, "[%s] not matching: got %v want: %v", rejection.key, rejection.got, rejection.want)
}
return builder.String()
}
func (r *RejectionReasons) StringTruncated() string {
if len(r.data) == 0 {
return ""
}
builder := strings.Builder{}
for i, rejection := range r.data {
got := rejection.got
switch v := rejection.got.(type) {
case string:
if len(v) > 1024 {
got = v[:1024]
}
}
want := rejection.want
switch v := rejection.want.(type) {
case string:
if len(v) > 1024 {
want = v[:1024]
}
}
if i > 0 {
builder.WriteString(", ")
}
fmt.Fprintf(&builder, "[%s] not matching: got %v want: %v", rejection.key, got, want)
}
return builder.String()
}
func (r *RejectionReasons) WriteString() string {
var output []string
for _, rejection := range r.data {
output = append(output, fmt.Sprintf("[%s] not matching: got %v want: %v", rejection.key, rejection.got, rejection.want))
}
return strings.Join(output, ", ")
}
func (r *RejectionReasons) WriteJSON() ([]byte, error) {
var output map[string]string
for _, rejection := range r.data {
output[rejection.key] = fmt.Sprintf("[%s] not matching: got %v want: %v", rejection.key, rejection.got, rejection.want)
}
return json.Marshal(output)
}
func (r *RejectionReasons) Add(key string, got any, want any) {
r.data = append(r.data, Rejection{
key: key,
got: got,
want: want,
})
}
func (r *RejectionReasons) Addf(key string, format string, got any, want any) {
r.data = append(r.data, Rejection{
key: key,
format: format,
got: got,
want: want,
})
}
func (r *RejectionReasons) AddTruncated(key string, got any, want any) {
switch wanted := want.(type) {
case string:
if len(wanted) > 1024 {
want = wanted[:1024]
}
case []string:
for i, s := range wanted {
if len(s) > 1024 {
wanted[i] = s[:1024]
}
}
}
r.Add(key, got, want)
}
// Clear rejections
func (r *RejectionReasons) Clear() {
r.data = make([]Rejection, 0)
}

File diff suppressed because one or more lines are too long

View file

@ -620,35 +620,6 @@ func (r *Release) HasMagnetUri() bool {
const MagnetURIPrefix = "magnet:?" const MagnetURIPrefix = "magnet:?"
func (r *Release) addRejection(reason string) {
r.Rejections = append(r.Rejections, reason)
}
func (r *Release) AddRejectionF(format string, v ...interface{}) {
r.addRejectionF(format, v...)
}
func (r *Release) addRejectionF(format string, v ...interface{}) {
r.Rejections = append(r.Rejections, fmt.Sprintf(format, v...))
}
// ResetRejections reset rejections between filter checks
func (r *Release) resetRejections() {
r.Rejections = []string{}
}
func (r *Release) RejectionsString(trim bool) string {
if len(r.Rejections) > 0 {
out := strings.Join(r.Rejections, ", ")
if trim && len(out) > 1024 {
out = out[:1024]
}
return out
}
return ""
}
// MapVars map vars from regex captures to fields on release // MapVars map vars from regex captures to fields on release
func (r *Release) MapVars(def *IndexerDefinition, varMap map[string]string) error { func (r *Release) MapVars(def *IndexerDefinition, varMap map[string]string) error {

View file

@ -386,8 +386,8 @@ func (s *service) CheckFilter(ctx context.Context, f *domain.Filter, release *do
} }
rejections, matchedFilter := f.CheckFilter(release) rejections, matchedFilter := f.CheckFilter(release)
if len(rejections) > 0 { if rejections.Len() > 0 {
l.Debug().Msgf("(%s) for release: %v rejections: (%s)", f.Name, release.TorrentName, f.RejectionsString(true)) l.Debug().Msgf("(%s) for release: %v rejections: (%s)", f.Name, release.TorrentName, rejections.StringTruncated())
return false, nil return false, nil
} }
@ -415,9 +415,9 @@ func (s *service) CheckFilter(ctx context.Context, f *domain.Filter, release *do
l.Trace().Msgf("failed smart episode check: %s", f.Name) l.Trace().Msgf("failed smart episode check: %s", f.Name)
if params.IsDailyEpisode() { if params.IsDailyEpisode() {
f.AddRejectionF("smart episode check: not new: (%s) Daily: %d-%d-%d", release.Title, release.Year, release.Month, release.Day) f.RejectReasons.Add("smart episode", fmt.Sprintf("not new (%s) daily: %d-%d-%d", release.Title, release.Year, release.Month, release.Day), fmt.Sprintf("expected newer than (%s) daily: %d-%d-%d", release.Title, release.Year, release.Month, release.Day))
} else { } else {
f.AddRejectionF("smart episode check: not new: (%s) season: %d ep: %d", release.Title, release.Season, release.Episode) f.RejectReasons.Add("smart episode", fmt.Sprintf("not new (%s) season: %d ep: %d", release.Title, release.Season, release.Episode), fmt.Sprintf("expected newer than (%s) season: %d ep: %d", release.Title, release.Season, release.Episode))
} }
return false, nil return false, nil
@ -566,7 +566,7 @@ func (s *service) RunExternalFilters(ctx context.Context, f *domain.Filter, exte
if exitCode != external.ExecExpectStatus { if exitCode != external.ExecExpectStatus {
s.log.Trace().Msgf("filter.Service.CheckFilter: external script unexpected exit code. got: %d want: %d", exitCode, external.ExecExpectStatus) s.log.Trace().Msgf("filter.Service.CheckFilter: external script unexpected exit code. got: %d want: %d", exitCode, external.ExecExpectStatus)
f.AddRejectionF("external script unexpected exit code. got: %d want: %d", exitCode, external.ExecExpectStatus) f.RejectReasons.Add("external script exit code", exitCode, external.ExecExpectStatus)
return false, nil return false, nil
} }
@ -579,7 +579,7 @@ func (s *service) RunExternalFilters(ctx context.Context, f *domain.Filter, exte
if statusCode != external.WebhookExpectStatus { if statusCode != external.WebhookExpectStatus {
s.log.Trace().Msgf("filter.Service.CheckFilter: external webhook unexpected status code. got: %d want: %d", statusCode, external.WebhookExpectStatus) s.log.Trace().Msgf("filter.Service.CheckFilter: external webhook unexpected status code. got: %d want: %d", statusCode, external.WebhookExpectStatus)
f.AddRejectionF("external webhook unexpected status code. got: %d want: %d", statusCode, external.WebhookExpectStatus) f.RejectReasons.Add("external webhook status code", statusCode, external.WebhookExpectStatus)
return false, nil return false, nil
} }
} }

View file

@ -398,7 +398,7 @@ func TestIndexersParseAndFilter(t *testing.T) {
//match, err := filterSvc.CheckFilter(ctx, filter, rls) //match, err := filterSvc.CheckFilter(ctx, filter, rls)
rejections, matchedFilter := filter.CheckFilter(rls) rejections, matchedFilter := filter.CheckFilter(rls)
assert.Len(t, rejections, len(filterT.rejections)) assert.Equal(t, rejections.Len(), len(filterT.rejections))
assert.Equal(t, filterT.match, matchedFilter) assert.Equal(t, filterT.match, matchedFilter)
}) })
} }

View file

@ -206,9 +206,9 @@ func (s *service) processFilters(ctx context.Context, filters []*domain.Filter,
} }
if !match { if !match {
l.Trace().Msgf("release.Process: indexer: %s, filter: %s release: %s, no match. rejections: %s", release.Indexer.Name, release.FilterName, release.TorrentName, f.RejectionsString(false)) l.Trace().Msgf("release.Process: indexer: %s, filter: %s release: %s, no match. rejections: %s", release.Indexer.Name, release.FilterName, release.TorrentName, f.RejectReasons.String())
l.Debug().Msgf("filter %s rejected release: %s", f.Name, f.RejectionsString(true)) l.Debug().Msgf("filter %s rejected release: %s", f.Name, release.TorrentName)
continue continue
} }