import requests
from flask import Flask, Response
from flask_cors import CORS
import threading

app = Flask(__name__)
CORS(app)  # Enable CORS for all routes

# List of M3U file URLs
URLS = [
    "https://iptv-org.github.io/iptv/index.country.m3u",
    "https://forge.fsky.io/frost/repo/raw/branch/main/tv.m3u",
    "https://raw.githubusercontent.com/Free-TV/IPTV/refs/heads/master/playlist.m3u8"
]

# Cached M3U content and a lock for thread safety
combined_m3u = None
lock = threading.Lock()

def fetch_m3u(url):
    """Fetch M3U content from a given URL with error handling."""
    try:
        response = requests.get(url, timeout=5)  # Set timeout to prevent long waits
        response.raise_for_status()  # Raise an error for bad responses
        return response.text
    except requests.RequestException as e:
        print(f"Error fetching {url}: {e}")
        return ""

def combine_m3u(urls):
    """Combine M3U content from multiple URLs and remove duplicates."""
    unique_streams = {}
    
    for url in urls:
        m3u_content = fetch_m3u(url)
        lines = m3u_content.splitlines()
        current_entry = None
        
        for line in lines:
            if line.startswith('#EXTINF:'):
                current_entry = line
            elif line and current_entry:
                stream_link = line.strip()
                if stream_link not in unique_streams:
                    unique_streams[stream_link] = f"{current_entry}\n{stream_link}"
                current_entry = None  # Reset after adding
            
    return '#EXTM3U\n' + '\n'.join(unique_streams.values())

@app.route('/all.m3u')
def serve_m3u():
    """Serve the cached combined M3U file, generating it if necessary."""
    global combined_m3u
    with lock:
        if combined_m3u is None:  # Generate if not already done
            combined_m3u = combine_m3u(URLS)
    return Response(combined_m3u, mimetype='application/vnd.apple.mpegurl')

@app.route('/generate')
def generate_m3u():
    """Force regeneration of the M3U file from the sources."""
    global combined_m3u
    with lock:
        combined_m3u = combine_m3u(URLS)  # Regenerate the M3U content
    return Response("M3U playlist regenerated!", mimetype='text/plain')

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000, threaded=True)