fix(processor): don't match sample in torrents

nuxencs · Nov 8, 2024 · cf5704a · cf5704a
1 parent 65be43b
commit cf5704a
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/internal/http/processor.go b/internal/http/processor.go
@@ -275,7 +275,7 @@ func (p *processor) processSeasonPack() (domain.StatusCode, error) {
 			var fileName string
 			var size int64
 			for _, f := range *torrentFiles {
-				if filepath.Ext(f.Name) != ".mkv" {
+				if !release.IsValidEpisodeFile(f.Name) {
 					continue
 				}
 

diff --git a/internal/release/release.go b/internal/release/release.go
@@ -152,3 +152,19 @@ func PercentOfTotalEpisodes(totalEps int, foundEps int) float32 {
 
 	return float32(foundEps) / float32(totalEps)
 }
+
+func IsValidEpisodeFile(torrentFileName string) bool {
+	torrentFileRls := rls.ParseString(filepath.Base(torrentFileName))
+
+	// ignore non video files
+	if torrentFileRls.Ext != "mkv" {
+		return false
+	}
+
+	// ignore sample files
+	if rls.MustNormalize(torrentFileRls.Group) == "sample" {
+		return false
+	}
+
+	return true
+}