add additional sources (#3)

parser_test.go

@@ -262,6 +262,7 @@ func TestParser_Parse(t *testing.T) {
 				Group:       "cookiemonster",
 				Title:       "stephen colbert",
 				Date:        "2019-02-03",
+				Source:      "web",
 				ContentType: TV,
 			},
 		},
@@ -360,6 +361,7 @@ func TestParser_Parse(t *testing.T) {
 				Resolution:  "4k",
 				Codec:       "x265",
 				Group:       "NAHOM",
+				Source:      "bdremux",
 				Languages:   []string{"ita", "eng"},
 				Hdr:         true,
 				Season:      -1,

source.go

@@ -12,7 +12,7 @@ var (
 )
 
 func init() {
-	sourceGeneral = regexp.MustCompile(`(?i)\b(?:(?:HD-?)?CAM|HD-?Rip|HDTV|BRRip|BDRip|DVDRip|DVDscr|(?:HD-?)?TVRip|TC|PPVRip|R5|VHSSCR|Bluray|WEB-?DL|WEB-?Rip|(?:DL|WEB|BD|BR)MUX)\b`)
+	sourceGeneral = regexp.MustCompile(`(?i)\b(?:(?:HD-?)?CAM|HD-?Rip|HDTV|BRRip|BDRip|DVDRip|DVDscr|(?:HD-?)?TVRip|TC|PPVRip|R5|VHSSCR|Blu-?ray|WEB-?DL|WEB-?Rip|(?:DL|WEB|BD|BR|BDRE|RE)MUX)|WEB\b`)
 	sourceTelesync = regexp.MustCompile("(?i)\b(?:HD-?)?T(?:ELE)?S(?:YNC)?\b")
 	sourceDvd = regexp.MustCompile("(?i)(DVD)(?:R[0-9])?")
 }

source_test.go

@@ -18,6 +18,26 @@ func TestParser_GetSource(t *testing.T) {
 			name: "Pitch.Perfect.2.2015.REPACK.1080p.BluRay.DTS.x264-VietHD.mkv",
 			want: "bluray",
 		},
+		{
+			desc: "bdremux extracted",
+			name: "Pitch.Perfect.2.2015.REPACK.1080p.BDRemux.DTS.x264-VietHD.mkv",
+			want: "bdremux",
+		},
+		{
+			desc: "remux extracted",
+			name: "Pitch.Perfect.2.2015.REPACK.1080p.REMUX.DTS.x264-VietHD.mkv",
+			want: "remux",
+		},
+		{
+			desc: "blu-ray extracted",
+			name: "Pitch.Perfect.2.2015.REPACK.1080p.Blu-Ray.DTS.x264-VietHD.mkv",
+			want: "blu-ray",
+		},
+		{
+			desc: "extract web",
+			name: "Stephen.Colbert.2023.02.15.Jim.Gaffigan.1080p.WEB.H264-JEBAITED.mkv",
+			want: "web",
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {