clean up downloader.go

m3u/m3u_test.go

@@ -101,7 +101,7 @@ http://example.com/fox
   defer os.Remove(sqliteDBPath) // Cleanup the database file after the test
 
 	// Test the parseM3UFromURL function with the mock server URL
-	err = parseM3UFromURL(mockServer.URL, 0)
+	err = ParseM3UFromURL(mockServer.URL, 0)
 	if err != nil {
 		t.Errorf("Error parsing M3U from URL: %v", err)
 	}

m3u/parser.go

@@ -10,10 +10,22 @@ import (
 	"m3u-stream-merger/database"
 )
 
-func parseM3UFromURL(m3uURL string, m3uIndex int) (error) {
+func ParseM3UFromURL(m3uURL string, m3uIndex int) (error) {
+	// Set the custom User-Agent header
+	userAgent := "IPTV Smarters/1.0.3 (iPad; iOS 16.6.1; Scale/2.00)"
+
+	// Create a new HTTP client with a custom User-Agent header
+	client := &http.Client{
+		CheckRedirect: func(req *http.Request, via []*http.Request) error {
+			// Follow redirects while preserving the custom User-Agent header
+			req.Header.Set("User-Agent", userAgent)
+			return nil
+		},
+	}
+
 	fmt.Printf("Parsing M3U from URL: %s\n", m3uURL)
 
-  resp, err := http.Get(m3uURL)
+  resp, err := client.Get(m3uURL)
 	if err != nil {
 		return fmt.Errorf("HTTP GET error: %v", err)
 	}

main.go

@@ -38,6 +38,20 @@ func updateSource(ctx context.Context) {
 }
 
 func main() {
+	index := 1
+	for {
+		m3uUrl, m3uExists := os.LookupEnv(fmt.Sprintf("M3U_URL_%d", index))
+		if !m3uExists {
+			break
+		}
+
+    err := m3u.ParseM3UFromURL(m3uUrl, index)
+    if err != nil {
+
+    }
+
+		index++
+	}
 	// Context for graceful shutdown
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()