diff --git a/utils/files.go b/utils/files.go index d3fa9f7f..a3a07148 100644 --- a/utils/files.go +++ b/utils/files.go @@ -23,6 +23,6 @@ func IsImageFile(filePath string) bool { } func IsPlaylist(filePath string) bool { - extension := filepath.Ext(filePath) - return strings.ToLower(extension) == ".m3u" + extension := strings.ToLower(filepath.Ext(filePath)) + return extension == ".m3u" || extension == ".m3u8" } diff --git a/utils/files_test.go b/utils/files_test.go index 8f5cf5bb..97f54b7a 100644 --- a/utils/files_test.go +++ b/utils/files_test.go @@ -49,6 +49,10 @@ var _ = Describe("Files", func() { Expect(IsPlaylist(filepath.Join("path", "to", "test.m3u"))).To(BeTrue()) }) + It("returns true for a M3U8 file", func() { + Expect(IsPlaylist(filepath.Join("path", "to", "test.m3u8"))).To(BeTrue()) + }) + It("returns false for a non-playlist file", func() { Expect(IsPlaylist("testm3u")).To(BeFalse()) })