Merge pull request #104 from sonroyaalmerol/m3u-support

Consider success if a playlist file reaches EOF

stream_handler.go

@@ -101,7 +101,7 @@ func proxyStream(m3uIndex int, resp *http.Response, r *http.Request, w http.Resp
 		if err != nil {
 			if err == io.EOF {
 				log.Printf("Stream ended (EOF reached): %s\n", r.RemoteAddr)
-				statusChan <- 1
+				statusChan <- 2
 				return
 			}
 			log.Printf("Error reading stream: %s\n", err.Error())
@@ -201,7 +201,10 @@ func streamHandler(w http.ResponseWriter, r *http.Request, db *database.Instance
 			streamExitCode := <-exitStatus
 			log.Printf("Exit code %d received from %s\n", streamExitCode, selectedUrl)
 
-			if streamExitCode == 1 {
+			if streamExitCode == 2 && utils.IsPlaylistFile(selectedUrl) {
+				log.Printf("Successfully proxied playlist (M3U) file: %s\n", r.RemoteAddr)
+				cancel()
+			} else if streamExitCode == 1 || streamExitCode == 2 {
 				// Retry on server-side connection errors
 				log.Printf("Retrying other servers...\n")
 			} else {

utils/url.go

@@ -1,6 +1,9 @@
 package utils
 
-import "encoding/base64"
+import (
+	"encoding/base64"
+	"strings"
+)
 
 func GetStreamUrl(slug string) string {
 	return base64.URLEncoding.EncodeToString([]byte(slug))
@@ -13,3 +16,9 @@ func GetStreamSlugFromUrl(streamUID string) string {
 	}
 	return string(decoded)
 }
+
+func IsPlaylistFile(url string) bool {
+	urlClean := strings.TrimSpace(strings.ToLower(url))
+
+	return strings.HasSuffix(urlClean, ".m3u") || strings.HasSuffix(urlClean, ".m3u8")
+}