m3u8: handle content-type case insensitively

internetarchive · Sep 9, 2024 · 21847ce · 21847ce
1 parent 91c4b85
commit 21847ce
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/internal/pkg/crawl/extractor/m3u8.go b/internal/pkg/crawl/extractor/m3u8.go
@@ -3,14 +3,13 @@ package extractor
 import (
 	"net/http"
 	"net/url"
-	"strings"
 
 	"github.com/grafov/m3u8"
 )
 
 func IsM3U8(resp *http.Response) bool {
-	return strings.Contains(resp.Header.Get("Content-Type"), "application/vnd.apple.mpegurl") ||
-		strings.Contains(resp.Header.Get("Content-Type"), "application/x-mpegURL")
+	return isContentType(resp.Header.Get("Content-Type"), "application/vnd.apple.mpegurl") ||
+		isContentType(resp.Header.Get("Content-Type"), "application/x-mpegURL")
 }
 
 func M3U8(resp *http.Response) (URLs []*url.URL, err error) {

diff --git a/internal/pkg/crawl/extractor/utils.go b/internal/pkg/crawl/extractor/utils.go
@@ -3,8 +3,17 @@ package extractor
 import (
 	"net/url"
 	"sort"
+	"strings"
 )
 
+func isContentType(header, targetContentType string) bool {
+	// Lowercase the header and target content type for case-insensitive comparison
+	header = strings.ToLower(header)
+	targetContentType = strings.ToLower(targetContentType)
+
+	return strings.Contains(header, targetContentType)
+}
+
 // compareURLs compares two slices of *url.URL
 func compareURLs(a, b []*url.URL) bool {
 	if len(a) != len(b) {