import json
from providers.auto_embed import AutoEmbed
from providers.embed import Embed
from providers.vidsrc import VidSrc
from providers.two_embed import TwoEmbed
from providers.vidzee import Vidzee
from providers.netfree import NetFree

providers = [
    AutoEmbed,
    Embed,
    VidSrc,
    TwoEmbed,
    Vidzee,
    NetFree,
]

def get_streams(tmdb_id, imdb_id, media_type, title, year, season=None, episode=None):
    streams = []
    for provider in providers:
        try:
            encoded_data = provider().get_stream(tmdb_id=tmdb_id, imdb_id=imdb_id, media_type=media_type, title=title, year=year, season=season, episode=episode)
            streams.extend(encoded_data)
        except Exception as e:
            print(f"Error with provider {provider.__name__}: {e}")
    return json.dumps(streams, indent=4)

# if __name__ == "__main__":
#     media_type = "movie"
#     tmdb_id = "822119"
#     imdb_id = "tt14513804"
#     title = "Dhoom Dhaam"
#     year = "2025"
#     season = None
#     episode = None
#     stream_list = get_streams(
#             tmdb_id=tmdb_id,
#             imdb_id=imdb_id,
#             media_type=media_type,
#             title=title,
#             year=year,
#             season=season,
#             episode=episode,
#         )
#     print(stream_list)
    # print("-" * 25)
    # for stream in json.loads(stream_list):
        # print(f"Stream: {stream}")
        # print(f"VideoData(videoSource: '{stream['videoSource']}', videoSourceUrl: '{stream['videoUrl']}', videoSourceHeaders: '{stream['videoHeaders']}',),")