// Copyright (c) 2021 - 2023, Ludvig Lundgren and the autobrr contributors. // SPDX-License-Identifier: GPL-2.0-or-later package domain import ( "context" "regexp" "strconv" "strings" "time" "github.com/autobrr/autobrr/pkg/wildcard" "github.com/dustin/go-humanize" ) /* Works the same way as for autodl-irssi https://autodl-community.github.io/autodl-irssi/configuration/filter/ */ type FilterRepo interface { FindByID(ctx context.Context, filterID int) (*Filter, error) FindByIndexerIdentifier(ctx context.Context, indexer string) ([]Filter, error) Find(ctx context.Context, params FilterQueryParams) ([]Filter, error) ListFilters(ctx context.Context) ([]Filter, error) Store(ctx context.Context, filter Filter) (*Filter, error) Update(ctx context.Context, filter Filter) (*Filter, error) UpdatePartial(ctx context.Context, filter FilterUpdate) error ToggleEnabled(ctx context.Context, filterID int, enabled bool) error Delete(ctx context.Context, filterID int) error StoreIndexerConnection(ctx context.Context, filterID int, indexerID int) error StoreIndexerConnections(ctx context.Context, filterID int, indexers []Indexer) error DeleteIndexerConnections(ctx context.Context, filterID int) error GetDownloadsByFilterId(ctx context.Context, filterID int) (*FilterDownloads, error) } type FilterDownloads struct { HourCount int DayCount int WeekCount int MonthCount int TotalCount int } type FilterMaxDownloadsUnit string const ( FilterMaxDownloadsHour FilterMaxDownloadsUnit = "HOUR" FilterMaxDownloadsDay FilterMaxDownloadsUnit = "DAY" FilterMaxDownloadsWeek FilterMaxDownloadsUnit = "WEEK" FilterMaxDownloadsMonth FilterMaxDownloadsUnit = "MONTH" FilterMaxDownloadsEver FilterMaxDownloadsUnit = "EVER" ) type FilterQueryParams struct { Sort map[string]string Filters struct { Indexers []string } Search string } type Filter struct { ID int `json:"id"` Name string `json:"name"` Enabled bool `json:"enabled"` CreatedAt time.Time `json:"created_at"` UpdatedAt time.Time `json:"updated_at"` MinSize string `json:"min_size,omitempty"` MaxSize string `json:"max_size,omitempty"` Delay int `json:"delay,omitempty"` Priority int32 `json:"priority"` MaxDownloads int `json:"max_downloads,omitempty"` MaxDownloadsUnit FilterMaxDownloadsUnit `json:"max_downloads_unit,omitempty"` MatchReleases string `json:"match_releases,omitempty"` ExceptReleases string `json:"except_releases,omitempty"` UseRegex bool `json:"use_regex,omitempty"` MatchReleaseGroups string `json:"match_release_groups,omitempty"` ExceptReleaseGroups string `json:"except_release_groups,omitempty"` Scene bool `json:"scene,omitempty"` Origins []string `json:"origins,omitempty"` ExceptOrigins []string `json:"except_origins,omitempty"` Bonus []string `json:"bonus,omitempty"` Freeleech bool `json:"freeleech,omitempty"` FreeleechPercent string `json:"freeleech_percent,omitempty"` SmartEpisode bool `json:"smart_episode"` Shows string `json:"shows,omitempty"` Seasons string `json:"seasons,omitempty"` Episodes string `json:"episodes,omitempty"` Resolutions []string `json:"resolutions,omitempty"` // SD, 480i, 480p, 576p, 720p, 810p, 1080i, 1080p. Codecs []string `json:"codecs,omitempty"` // XviD, DivX, x264, h.264 (or h264), mpeg2 (or mpeg-2), VC-1 (or VC1), WMV, Remux, h.264 Remux (or h264 Remux), VC-1 Remux (or VC1 Remux). Sources []string `json:"sources,omitempty"` // DSR, PDTV, HDTV, HR.PDTV, HR.HDTV, DVDRip, DVDScr, BDr, BD5, BD9, BDRip, BRRip, DVDR, MDVDR, HDDVD, HDDVDRip, BluRay, WEB-DL, TVRip, CAM, R5, TELESYNC, TS, TELECINE, TC. TELESYNC and TS are synonyms (you don't need both). Same for TELECINE and TC Containers []string `json:"containers,omitempty"` MatchHDR []string `json:"match_hdr,omitempty"` ExceptHDR []string `json:"except_hdr,omitempty"` MatchOther []string `json:"match_other,omitempty"` ExceptOther []string `json:"except_other,omitempty"` Years string `json:"years,omitempty"` Artists string `json:"artists,omitempty"` Albums string `json:"albums,omitempty"` MatchReleaseTypes []string `json:"match_release_types,omitempty"` // Album,Single,EP ExceptReleaseTypes string `json:"except_release_types,omitempty"` Formats []string `json:"formats,omitempty"` // MP3, FLAC, Ogg, AAC, AC3, DTS Quality []string `json:"quality,omitempty"` // 192, 320, APS (VBR), V2 (VBR), V1 (VBR), APX (VBR), V0 (VBR), q8.x (VBR), Lossless, 24bit Lossless, Other Media []string `json:"media,omitempty"` // CD, DVD, Vinyl, Soundboard, SACD, DAT, Cassette, WEB, Other PerfectFlac bool `json:"perfect_flac,omitempty"` Cue bool `json:"cue,omitempty"` Log bool `json:"log,omitempty"` LogScore int `json:"log_score,omitempty"` MatchCategories string `json:"match_categories,omitempty"` ExceptCategories string `json:"except_categories,omitempty"` MatchUploaders string `json:"match_uploaders,omitempty"` ExceptUploaders string `json:"except_uploaders,omitempty"` MatchLanguage []string `json:"match_language,omitempty"` ExceptLanguage []string `json:"except_language,omitempty"` Tags string `json:"tags,omitempty"` ExceptTags string `json:"except_tags,omitempty"` TagsAny string `json:"tags_any,omitempty"` ExceptTagsAny string `json:"except_tags_any,omitempty"` TagsMatchLogic string `json:"tags_match_logic,omitempty"` ExceptTagsMatchLogic string `json:"except_tags_match_logic,omitempty"` MatchReleaseTags string `json:"match_release_tags,omitempty"` ExceptReleaseTags string `json:"except_release_tags,omitempty"` UseRegexReleaseTags bool `json:"use_regex_release_tags,omitempty"` ExternalScriptEnabled bool `json:"external_script_enabled,omitempty"` ExternalScriptCmd string `json:"external_script_cmd,omitempty"` ExternalScriptArgs string `json:"external_script_args,omitempty"` ExternalScriptExpectStatus int `json:"external_script_expect_status,omitempty"` ExternalWebhookEnabled bool `json:"external_webhook_enabled,omitempty"` ExternalWebhookHost string `json:"external_webhook_host,omitempty"` ExternalWebhookData string `json:"external_webhook_data,omitempty"` ExternalWebhookExpectStatus int `json:"external_webhook_expect_status,omitempty"` ActionsCount int `json:"actions_count"` Actions []*Action `json:"actions,omitempty"` Indexers []Indexer `json:"indexers"` Downloads *FilterDownloads `json:"-"` } type FilterUpdate struct { ID int `json:"id"` Name *string `json:"name,omitempty"` Enabled *bool `json:"enabled,omitempty"` MinSize *string `json:"min_size,omitempty"` MaxSize *string `json:"max_size,omitempty"` Delay *int `json:"delay,omitempty"` Priority *int32 `json:"priority,omitempty"` MaxDownloads *int `json:"max_downloads,omitempty"` MaxDownloadsUnit *FilterMaxDownloadsUnit `json:"max_downloads_unit,omitempty"` MatchReleases *string `json:"match_releases,omitempty"` ExceptReleases *string `json:"except_releases,omitempty"` UseRegex *bool `json:"use_regex,omitempty"` MatchReleaseGroups *string `json:"match_release_groups,omitempty"` ExceptReleaseGroups *string `json:"except_release_groups,omitempty"` MatchReleaseTags *string `json:"match_release_tags,omitempty"` ExceptReleaseTags *string `json:"except_release_tags,omitempty"` UseRegexReleaseTags *bool `json:"use_regex_release_tags,omitempty"` Scene *bool `json:"scene,omitempty"` Origins *[]string `json:"origins,omitempty"` ExceptOrigins *[]string `json:"except_origins,omitempty"` Bonus *[]string `json:"bonus,omitempty"` Freeleech *bool `json:"freeleech,omitempty"` FreeleechPercent *string `json:"freeleech_percent,omitempty"` SmartEpisode *bool `json:"smart_episode,omitempty"` Shows *string `json:"shows,omitempty"` Seasons *string `json:"seasons,omitempty"` Episodes *string `json:"episodes,omitempty"` Resolutions *[]string `json:"resolutions,omitempty"` // SD, 480i, 480p, 576p, 720p, 810p, 1080i, 1080p. Codecs *[]string `json:"codecs,omitempty"` // XviD, DivX, x264, h.264 (or h264), mpeg2 (or mpeg-2), VC-1 (or VC1), WMV, Remux, h.264 Remux (or h264 Remux), VC-1 Remux (or VC1 Remux). Sources *[]string `json:"sources,omitempty"` // DSR, PDTV, HDTV, HR.PDTV, HR.HDTV, DVDRip, DVDScr, BDr, BD5, BD9, BDRip, BRRip, DVDR, MDVDR, HDDVD, HDDVDRip, BluRay, WEB-DL, TVRip, CAM, R5, TELESYNC, TS, TELECINE, TC. TELESYNC and TS are synonyms (you don't need both). Same for TELECINE and TC Containers *[]string `json:"containers,omitempty"` MatchHDR *[]string `json:"match_hdr,omitempty"` ExceptHDR *[]string `json:"except_hdr,omitempty"` MatchOther *[]string `json:"match_other,omitempty"` ExceptOther *[]string `json:"except_other,omitempty"` Years *string `json:"years,omitempty"` Artists *string `json:"artists,omitempty"` Albums *string `json:"albums,omitempty"` MatchReleaseTypes *[]string `json:"match_release_types,omitempty"` // Album,Single,EP ExceptReleaseTypes *string `json:"except_release_types,omitempty"` Formats *[]string `json:"formats,omitempty"` // MP3, FLAC, Ogg, AAC, AC3, DTS Quality *[]string `json:"quality,omitempty"` // 192, 320, APS (VBR), V2 (VBR), V1 (VBR), APX (VBR), V0 (VBR), q8.x (VBR), Lossless, 24bit Lossless, Other Media *[]string `json:"media,omitempty"` // CD, DVD, Vinyl, Soundboard, SACD, DAT, Cassette, WEB, Other PerfectFlac *bool `json:"perfect_flac,omitempty"` Cue *bool `json:"cue,omitempty"` Log *bool `json:"log,omitempty"` LogScore *int `json:"log_score,omitempty"` MatchCategories *string `json:"match_categories,omitempty"` ExceptCategories *string `json:"except_categories,omitempty"` MatchUploaders *string `json:"match_uploaders,omitempty"` ExceptUploaders *string `json:"except_uploaders,omitempty"` MatchLanguage *[]string `json:"match_language,omitempty"` ExceptLanguage *[]string `json:"except_language,omitempty"` Tags *string `json:"tags,omitempty"` ExceptTags *string `json:"except_tags,omitempty"` TagsAny *string `json:"tags_any,omitempty"` ExceptTagsAny *string `json:"except_tags_any,omitempty"` TagsMatchLogic *string `json:"tags_match_logic,omitempty"` ExceptTagsMatchLogic *string `json:"except_tags_match_logic,omitempty"` ExternalScriptEnabled *bool `json:"external_script_enabled,omitempty"` ExternalScriptCmd *string `json:"external_script_cmd,omitempty"` ExternalScriptArgs *string `json:"external_script_args,omitempty"` ExternalScriptExpectStatus *int `json:"external_script_expect_status,omitempty"` ExternalWebhookEnabled *bool `json:"external_webhook_enabled,omitempty"` ExternalWebhookHost *string `json:"external_webhook_host,omitempty"` ExternalWebhookData *string `json:"external_webhook_data,omitempty"` ExternalWebhookExpectStatus *int `json:"external_webhook_expect_status,omitempty"` Actions []*Action `json:"actions,omitempty"` Indexers []Indexer `json:"indexers,omitempty"` } func (f Filter) CheckFilter(r *Release) ([]string, bool) { // reset rejections first to clean previous checks r.resetRejections() // max downloads check. If reached return early if f.MaxDownloads > 0 && !f.checkMaxDownloads(f.MaxDownloads, f.MaxDownloadsUnit) { r.addRejectionF("max downloads (%d) this (%v) reached", f.MaxDownloads, f.MaxDownloadsUnit) return r.Rejections, false } if len(f.Bonus) > 0 && !sliceContainsSlice(r.Bonus, f.Bonus) { r.addRejectionF("bonus not matching. got: %v want: %v", r.Bonus, f.Bonus) } if f.Freeleech && r.Freeleech != f.Freeleech { r.addRejection("wanted: freeleech") } if f.FreeleechPercent != "" && !checkFreeleechPercent(r.FreeleechPercent, f.FreeleechPercent) { r.addRejectionF("freeleech percent not matching. got: %v want: %v", r.FreeleechPercent, f.FreeleechPercent) } if len(f.Origins) > 0 && !containsSlice(r.Origin, f.Origins) { r.addRejectionF("origin not matching. got: %v want: %v", r.Origin, f.Origins) } if len(f.ExceptOrigins) > 0 && containsSlice(r.Origin, f.ExceptOrigins) { r.addRejectionF("except origin not matching. got: %v unwanted: %v", r.Origin, f.ExceptOrigins) } // title is the parsed title if f.Shows != "" && !contains(r.Title, f.Shows) { r.addRejectionF("shows not matching. got: %v want: %v", r.Title, f.Shows) } if f.Seasons != "" && !containsIntStrings(r.Season, f.Seasons) { r.addRejectionF("season not matching. got: %d want: %v", r.Season, f.Seasons) } if f.Episodes != "" && !containsIntStrings(r.Episode, f.Episodes) { r.addRejectionF("episodes not matching. got: %d want: %v", r.Episode, f.Episodes) } // matchRelease // match against regex if f.UseRegex { if f.MatchReleases != "" && !matchRegex(r.TorrentName, f.MatchReleases) { r.addRejectionF("match release regex not matching. got: %v want: %v", r.TorrentName, f.MatchReleases) } if f.ExceptReleases != "" && matchRegex(r.TorrentName, f.ExceptReleases) { r.addRejectionF("except releases regex: unwanted release. got: %v want: %v", r.TorrentName, f.ExceptReleases) } } else { if f.MatchReleases != "" && !containsFuzzy(r.TorrentName, f.MatchReleases) { r.addRejectionF("match release not matching. got: %v want: %v", r.TorrentName, f.MatchReleases) } if f.ExceptReleases != "" && containsFuzzy(r.TorrentName, f.ExceptReleases) { r.addRejectionF("except releases: unwanted release. got: %v want: %v", r.TorrentName, f.ExceptReleases) } } if f.MatchReleaseGroups != "" && !contains(r.Group, f.MatchReleaseGroups) { r.addRejectionF("release groups not matching. got: %v want: %v", r.Group, f.MatchReleaseGroups) } if f.ExceptReleaseGroups != "" && contains(r.Group, f.ExceptReleaseGroups) { r.addRejectionF("unwanted release group. got: %v unwanted: %v", r.Group, f.ExceptReleaseGroups) } // check raw releaseTags string if f.UseRegexReleaseTags { if f.MatchReleaseTags != "" && !matchRegex(r.ReleaseTags, f.MatchReleaseTags) { r.addRejectionF("match release tags regex not matching. got: %v want: %v", r.ReleaseTags, f.MatchReleaseTags) } if f.ExceptReleaseTags != "" && matchRegex(r.ReleaseTags, f.ExceptReleaseTags) { r.addRejectionF("except release tags regex: unwanted release. got: %v want: %v", r.ReleaseTags, f.ExceptReleaseTags) } } else { if f.MatchReleaseTags != "" && !containsFuzzy(r.ReleaseTags, f.MatchReleaseTags) { r.addRejectionF("match release tags not matching. got: %v want: %v", r.ReleaseTags, f.MatchReleaseTags) } if f.ExceptReleaseTags != "" && containsFuzzy(r.ReleaseTags, f.ExceptReleaseTags) { r.addRejectionF("except release tags: unwanted release. got: %v want: %v", r.ReleaseTags, f.ExceptReleaseTags) } } if f.MatchUploaders != "" && !contains(r.Uploader, f.MatchUploaders) { r.addRejectionF("uploaders not matching. got: %v want: %v", r.Uploader, f.MatchUploaders) } if f.ExceptUploaders != "" && contains(r.Uploader, f.ExceptUploaders) { r.addRejectionF("unwanted uploaders. got: %v unwanted: %v", r.Uploader, f.ExceptUploaders) } if len(f.MatchLanguage) > 0 && !sliceContainsSlice(r.Language, f.MatchLanguage) { r.addRejectionF("language not matching. got: %v want: %v", r.Language, f.MatchLanguage) } if len(f.ExceptLanguage) > 0 && sliceContainsSlice(r.Language, f.ExceptLanguage) { r.addRejectionF("language unwanted. got: %v want: %v", r.Language, f.ExceptLanguage) } if len(f.Resolutions) > 0 && !containsSlice(r.Resolution, f.Resolutions) { r.addRejectionF("resolution not matching. got: %v want: %v", r.Resolution, f.Resolutions) } if len(f.Codecs) > 0 && !sliceContainsSlice(r.Codec, f.Codecs) { r.addRejectionF("codec not matching. got: %v want: %v", r.Codec, f.Codecs) } if len(f.Sources) > 0 && !containsSlice(r.Source, f.Sources) { r.addRejectionF("source not matching. got: %v want: %v", r.Source, f.Sources) } if len(f.Containers) > 0 && !containsSlice(r.Container, f.Containers) { r.addRejectionF("container not matching. got: %v want: %v", r.Container, f.Containers) } // HDR is parsed into the Codec slice from rls if len(f.MatchHDR) > 0 && !matchHDR(r.HDR, f.MatchHDR) { r.addRejectionF("hdr not matching. got: %v want: %v", r.HDR, f.MatchHDR) } // HDR is parsed into the Codec slice from rls if len(f.ExceptHDR) > 0 && matchHDR(r.HDR, f.ExceptHDR) { r.addRejectionF("hdr unwanted. got: %v want: %v", r.HDR, f.ExceptHDR) } // Other is parsed into the Other slice from rls if len(f.MatchOther) > 0 && !sliceContainsSlice(r.Other, f.MatchOther) { r.addRejectionF("match other not matching. got: %v want: %v", r.Other, f.MatchOther) } // Other is parsed into the Other slice from rls if len(f.ExceptOther) > 0 && sliceContainsSlice(r.Other, f.ExceptOther) { r.addRejectionF("except other unwanted. got: %v unwanted: %v", r.Other, f.ExceptOther) } if f.Years != "" && !containsIntStrings(r.Year, f.Years) { r.addRejectionF("year not matching. got: %d want: %v", r.Year, f.Years) } if f.MatchCategories != "" { var categories []string categories = append(categories, r.Categories...) if r.Category != "" { categories = append(categories, r.Category) } if !contains(r.Category, f.MatchCategories) && !containsAny(categories, f.MatchCategories) { r.addRejectionF("category not matching. got: %v want: %v", strings.Join(categories, ","), f.MatchCategories) } } if f.ExceptCategories != "" { var categories []string categories = append(categories, r.Categories...) if r.Category != "" { categories = append(categories, r.Category) } if contains(r.Category, f.ExceptCategories) && containsAny(categories, f.ExceptCategories) { r.addRejectionF("category unwanted. got: %v unwanted: %v", strings.Join(categories, ","), f.ExceptCategories) } } if len(f.MatchReleaseTypes) > 0 && !containsSlice(r.Category, f.MatchReleaseTypes) { r.addRejectionF("release type not matching. got: %v want: %v", r.Category, f.MatchReleaseTypes) } if (f.MinSize != "" || f.MaxSize != "") && !f.checkSizeFilter(r, f.MinSize, f.MaxSize) { r.addRejectionF("size not matching. got: %v want min: %v max: %v", r.Size, f.MinSize, f.MaxSize) } if f.Tags != "" { if f.TagsMatchLogic == "ANY" && !containsAny(r.Tags, f.Tags) { r.addRejectionF("tags not matching. got: %v want: %v", r.Tags, f.Tags) } else if f.TagsMatchLogic == "ALL" && !containsAll(r.Tags, f.Tags) { r.addRejectionF("tags not matching. got: %v want(all): %v", r.Tags, f.Tags) } } if f.ExceptTags != "" { if f.ExceptTagsMatchLogic == "ANY" && containsAny(r.Tags, f.ExceptTags) { r.addRejectionF("tags unwanted. got: %v want: %v", r.Tags, f.ExceptTags) } else if f.ExceptTagsMatchLogic == "ALL" && containsAll(r.Tags, f.ExceptTags) { r.addRejectionF("tags unwanted. got: %v want(all): %v", r.Tags, f.ExceptTags) } } if len(f.Artists) > 0 && !contains(r.Artists, f.Artists) { r.addRejectionF("artists not matching. got: %v want: %v", r.Artists, f.Artists) } if len(f.Albums) > 0 && !contains(r.Title, f.Albums) { r.addRejectionF("albums not matching. got: %v want: %v", r.Title, f.Albums) } // Perfect flac requires Cue, Log, Log Score 100, FLAC and 24bit Lossless if f.PerfectFlac && !f.isPerfectFLAC(r) { r.addRejectionF("wanted: perfect flac. got: %v", r.Audio) } if len(f.Formats) > 0 && !sliceContainsSlice(r.Audio, f.Formats) { r.addRejectionF("formats not matching. got: %v want: %v", r.Audio, f.Formats) } if len(f.Quality) > 0 && !sliceContainsSlice(r.Audio, f.Quality) { r.addRejectionF("quality not matching. got: %v want: %v", r.Audio, f.Quality) } if len(f.Media) > 0 && !containsSlice(r.Source, f.Media) { r.addRejectionF("media not matching. got: %v want: %v", r.Source, f.Media) } if f.Cue && !containsAny(r.Audio, "Cue") { r.addRejection("wanted: cue") } if f.Log && !containsAny(r.Audio, "Log") { r.addRejection("wanted: log") } if f.Log && f.LogScore != 0 && r.LogScore != f.LogScore { r.addRejectionF("log score. got: %v want: %v", r.LogScore, f.LogScore) } if len(r.Rejections) > 0 { return r.Rejections, false } return nil, true } func (f Filter) checkMaxDownloads(max int, perTimeUnit FilterMaxDownloadsUnit) bool { if f.Downloads == nil { return false } switch perTimeUnit { case FilterMaxDownloadsHour: if f.Downloads.HourCount > 0 && f.Downloads.HourCount >= max { return false } case FilterMaxDownloadsDay: if f.Downloads.DayCount > 0 && f.Downloads.DayCount >= max { return false } case FilterMaxDownloadsWeek: if f.Downloads.WeekCount > 0 && f.Downloads.WeekCount >= max { return false } case FilterMaxDownloadsMonth: if f.Downloads.MonthCount > 0 && f.Downloads.MonthCount >= max { return false } case FilterMaxDownloadsEver: if f.Downloads.TotalCount > 0 && f.Downloads.TotalCount >= max { return false } default: return true } return true } // isPerfectFLAC Perfect is "CD FLAC Cue Log 100% Lossless or 24bit Lossless" func (f Filter) isPerfectFLAC(r *Release) bool { if !contains(r.Source, "CD") { return false } if !containsAny(r.Audio, "Cue") { return false } if !containsAny(r.Audio, "Log") { return false } if !containsAny(r.Audio, "Log100") { return false } if !containsAny(r.Audio, "FLAC") { return false } if !containsAnySlice(r.Audio, []string{"Lossless", "24bit Lossless"}) { return false } return true } // checkSizeFilter additional size check // for indexers that doesn't announce size, like some gazelle based // set flag r.AdditionalSizeCheckRequired if there's a size in the filter, otherwise go a head // implement API for ptp,btn,ggn to check for size if needed // for others pull down torrent and do check func (f Filter) checkSizeFilter(r *Release, minSize string, maxSize string) bool { if r.Size == 0 { r.AdditionalSizeCheckRequired = true return true } else { r.AdditionalSizeCheckRequired = false } // if r.Size parse filter to bytes and compare // handle both min and max if minSize != "" { // string to bytes minSizeBytes, err := humanize.ParseBytes(minSize) if err != nil { // log could not parse into bytes } if r.Size <= minSizeBytes { r.addRejection("size: smaller than min size") return false } } if maxSize != "" { // string to bytes maxSizeBytes, err := humanize.ParseBytes(maxSize) if err != nil { // log could not parse into bytes } if r.Size >= maxSizeBytes { r.addRejection("size: larger than max size") return false } } return true } func matchRegex(tag string, filterList string) bool { if tag == "" { return false } filters := strings.Split(filterList, ",") for _, filter := range filters { if filter == "" { continue } re, err := regexp.Compile(`(?i)(?:` + filter + `)`) if err != nil { return false } match := re.MatchString(tag) if match { return true } } return false } // checkFilterIntStrings "1,2,3-20" func containsIntStrings(value int, filterList string) bool { filters := strings.Split(filterList, ",") for _, s := range filters { s = strings.Replace(s, "%", "", -1) s = strings.Trim(s, " ") if strings.Contains(s, "-") { minMax := strings.Split(s, "-") // to int min, err := strconv.ParseInt(minMax[0], 10, 32) if err != nil { return false } max, err := strconv.ParseInt(minMax[1], 10, 32) if err != nil { return false } if min > max { // handle error return false } else { // if announcePercent is greater than min and less than max return true if value >= int(min) && value <= int(max) { return true } } } filterInt, err := strconv.ParseInt(s, 10, 32) if err != nil { return false } if int(filterInt) == value { return true } } return false } func contains(tag string, filter string) bool { return containsMatch([]string{tag}, strings.Split(filter, ",")) } func containsFuzzy(tag string, filter string) bool { return containsMatchFuzzy([]string{tag}, strings.Split(filter, ",")) } func containsSlice(tag string, filters []string) bool { return containsMatch([]string{tag}, filters) } func containsAny(tags []string, filter string) bool { return containsMatch(tags, strings.Split(filter, ",")) } func containsAll(tags []string, filter string) bool { return containsAllMatch(tags, strings.Split(filter, ",")) } func containsAnyOther(filter string, tags ...string) bool { return containsMatch(tags, strings.Split(filter, ",")) } func sliceContainsSlice(tags []string, filters []string) bool { return containsMatchBasic(tags, filters) } func containsMatchFuzzy(tags []string, filters []string) bool { for _, tag := range tags { if tag == "" { continue } tag = strings.ToLower(tag) for _, filter := range filters { if filter == "" { continue } filter = strings.ToLower(filter) filter = strings.Trim(filter, " ") // check if line contains * or ?, if so try wildcard match, otherwise try substring match a := strings.ContainsAny(filter, "?|*") if a { match := wildcard.Match(filter, tag) if match { return true } } else if strings.Contains(tag, filter) { return true } } } return false } func containsMatch(tags []string, filters []string) bool { for _, tag := range tags { if tag == "" { continue } tag = strings.ToLower(tag) for _, filter := range filters { if filter == "" { continue } filter = strings.ToLower(filter) filter = strings.Trim(filter, " ") // check if line contains * or ?, if so try wildcard match, otherwise try substring match a := strings.ContainsAny(filter, "?|*") if a { match := wildcard.Match(filter, tag) if match { return true } } else if tag == filter { return true } } } return false } func containsAllMatch(tags []string, filters []string) bool { for _, filter := range filters { if filter == "" { continue } filter = strings.ToLower(filter) filter = strings.Trim(filter, " ") found := false for _, tag := range tags { if tag == "" { continue } tag = strings.ToLower(tag) if tag == filter { found = true break } else if strings.ContainsAny(filter, "?|*") { if wildcard.Match(filter, tag) { found = true break } } } if !found { return false } } return true } func containsMatchBasic(tags []string, filters []string) bool { for _, tag := range tags { if tag == "" { continue } tag = strings.ToLower(tag) for _, filter := range filters { if filter == "" { continue } filter = strings.ToLower(filter) filter = strings.Trim(filter, " ") if tag == filter { return true } } } return false } func containsAnySlice(tags []string, filters []string) bool { for _, tag := range tags { if tag == "" { continue } tag = strings.ToLower(tag) for _, filter := range filters { if filter == "" { continue } filter = strings.ToLower(filter) filter = strings.Trim(filter, " ") // check if line contains * or ?, if so try wildcard match, otherwise try substring match wild := strings.ContainsAny(filter, "?|*") if wild { match := wildcard.Match(filter, tag) if match { return true } } else if tag == filter { return true } } } return false } func checkFreeleechPercent(announcePercent int, filterPercent string) bool { filters := strings.Split(filterPercent, ",") for _, s := range filters { s = strings.Replace(s, "%", "", -1) s = strings.Trim(s, " ") if strings.Contains(s, "-") { minMax := strings.Split(s, "-") // to int min, err := strconv.ParseInt(minMax[0], 10, 32) if err != nil { return false } max, err := strconv.ParseInt(minMax[1], 10, 32) if err != nil { return false } if min > max { // handle error return false } else { // if announcePercent is greater than min and less than max return true if announcePercent >= int(min) && announcePercent <= int(max) { return true } } } filterPercentInt, err := strconv.ParseInt(s, 10, 32) if err != nil { return false } if int(filterPercentInt) == announcePercent { return true } } return false } func matchHDR(releaseValues []string, filterValues []string) bool { for _, filter := range filterValues { if filter == "" { continue } filter = strings.ToLower(filter) filter = strings.Trim(filter, " ") parts := strings.Split(filter, " ") if len(parts) == 2 { partsMatched := 0 for _, part := range parts { for _, tag := range releaseValues { if tag == "" { continue } tag = strings.ToLower(tag) if tag == part { partsMatched++ } if len(parts) == partsMatched { return true } } } } else { for _, tag := range releaseValues { if tag == "" { continue } tag = strings.ToLower(tag) if tag == filter { return true } } } } return false }