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

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

* chore(processor): add tests for IsValidEpisodeFile

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
 				}
 

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
+}

internal/release/release_test.go

@@ -191,3 +191,55 @@ func Test_MatchEpToSeasonPackEp(t *testing.T) {
 		})
 	}
 }
+
+func Test_IsValidEpisodeFile(t *testing.T) {
+	type args struct {
+		torrentFileName string
+	}
+	tests := []struct {
+		name string
+		args args
+		want bool
+	}{
+		{
+			name: "sample_with_dash",
+			args: args{
+				torrentFileName: "test.release.s06e03.dutch.1080p.web.h264-rlsgrp-sample.mkv",
+			},
+			want: false,
+		},
+		{
+			name: "sample_with_dot",
+			args: args{
+				torrentFileName: "test.release.s06e03.dutch.1080p.web.h264-rlsgrp.sample.mkv",
+			},
+			want: false,
+		},
+		{
+			name: "wrong_ext",
+			args: args{
+				torrentFileName: "test.release.s06e03.dutch.1080p.web.h264-rlsgrp.nfo",
+			},
+			want: false,
+		},
+		{
+			name: "wrong_ext_and_sample",
+			args: args{
+				torrentFileName: "test.release.s06e03.dutch.1080p.web.h264-rlsgrp.sample.nfo",
+			},
+			want: false,
+		},
+		{
+			name: "valid_release",
+			args: args{
+				torrentFileName: "test.release.s06e03.dutch.1080p.web.h264-rlsgrp.mkv",
+			},
+			want: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			assert.Equalf(t, tt.want, IsValidEpisodeFile(tt.args.torrentFileName), "IsValidEpisodeFile(%v)", tt.args.torrentFileName)
+		})
+	}
+}