import m3u8
import json


class OriginChannels():

    def __init__(self, fhdhr, origin):
        self.fhdhr = fhdhr
        self.origin = origin

        self.base_api_url = "https://ott-gateway-stirr.sinclairstoryline.com/api/rest/v3"
        self.base_station_url = "https://ott-stationselection.sinclairstoryline.com/stationSelectionByAllStates"

    def get_channels(self):

        channel_list = []

        channel_ids = []

        station_locations = ["national", "abc3340"]

        station_by_state_opn = self.fhdhr.web.session.get(self.base_station_url)
        states_list = station_by_state_opn.json()

        for state_uuid in states_list['page']:
            if state_uuid["pageComponentUuid"].startswith("nearyou-"):
                state_url = state_uuid["content"]

                state_url_req = self.fhdhr.web.session.get(state_url)
                state_url_json = state_url_req.json()
                for item in state_url_json["rss"]["channel"]["pagecomponent"]["component"]:
                    stationitem = item["item"]["media:content"]['sinclair:action_value']
                    if stationitem not in station_locations:
                        station_locations.append(stationitem)

        for station_item in station_locations:

            chan_list_url = "%s/channels/stirr?station=%s" % (self.base_api_url, station_item)
            chan_list_urlopn = self.fhdhr.web.session.get(chan_list_url)
            stirr_chan_list = chan_list_urlopn.json()

            for channel_dict in stirr_chan_list["channel"]:

                if str(channel_dict["id"]) not in channel_ids:

                    chan_item_url = "%s/status/%s" % (self.base_api_url, str(channel_dict["id"]))
                    chan_item_urlopn = self.fhdhr.web.session.get(chan_item_url)
                    try:
                        stirr_chan_item = chan_item_urlopn.json()
                    except json.JSONDecodeError:
                        stirr_chan_item = None

                    if stirr_chan_item:

                        channel_ids.append(str(channel_dict["id"]))

                        try:
                            thumbnail = channel_dict["icon"]["src"].split("?")[0]
                        except TypeError:
                            thumbnail = None

                        clean_station_item = {
                                             "name": stirr_chan_item['rss']["channel"]["title"],
                                             "callsign": channel_dict["display-name"],
                                             "id": str(channel_dict["id"]),
                                             "thumbnail": thumbnail
                                             }
                        channel_list.append(clean_station_item)

        return channel_list

    def get_channel_stream(self, chandict):
        chan_item_url = "%s/status/%s" % (self.base_api_url, str(chandict["origin_id"]))
        chan_item_urlopn = self.fhdhr.web.session.get(chan_item_url)
        stirr_chan_item = chan_item_urlopn.json()
        streamurl = stirr_chan_item['rss']["channel"]["item"]["link"]
        if self.fhdhr.config.dict["origin"]["force_best"]:
            streamurl = self.m3u8_beststream(streamurl)
        return streamurl

    def m3u8_beststream(self, m3u8_url):
        bestStream = None
        videoUrlM3u = m3u8.load(m3u8_url)
        if not videoUrlM3u.is_variant:
            return m3u8_url

        for videoStream in videoUrlM3u.playlists:
            if not bestStream:
                bestStream = videoStream
            elif videoStream.stream_info.bandwidth > bestStream.stream_info.bandwidth:
                bestStream = videoStream

        if not bestStream:
            return bestStream.absolute_uri
        else:
            return m3u8_url


"""
{
'media:title': {
                'content': 'Powered by WBMA'
                },
'media:description': {
                    'content': 'Birmingham'
                    },
'media:thumbnail': [{'width': '548', 'height': '308', 'url': 'https://hummingbird-ott.s3.amazonaws.com/images/stationCards/wbma_al_birmingham_630_548x308.jpg'}, {'width': '375', 'height': '211', 'url': 'https://hummingbird-ott.s3.amazonaws.com/images/stationCards/wbma_al_birmingham_630_375x211.jpg'}],
'sinclair:action': 'storeStation',
'sinclair:logo': {'width': '119', 'height': '18', 'text': 'Powered by WBMA', 'url': ''},
'sinclair:action_value': 'abc3340', 'sinclair:ident': 'abc3340.0.AL', 'sinclair:ad_preroll': '', 'medium': 'document', 'expression': 'nonstop', 'type': 'application/json', 'url': 'https://ott-stationselection.sinclairstoryline.com/stationSelectionByState/', 'duration': '0', 'media:restriction': [{'relationship': 'allow', 'type': 'uri', 'content': ''}]}
"""