import m3u8


class OriginChannels():

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

        self.channel_json_url = "https://www.toonamiaftermath.com/tatv.json"
        self.base_api = 'https://api.toonamiaftermath.com'
        self.stream_url_base = "%s/streamUrl?&useHttps=true&channelName=" % self.base_api

    def get_channels(self):

        channel_list = []
        channels_json = self.fhdhr.web.session.get(self.channel_json_url).json()
        for channel_dict in channels_json:

            clean_station_item = {
                                 "name": channel_dict["title"],
                                 "callsign": channel_dict["id"].title(),
                                 "id": channel_dict["id"],
                                 }
            channel_list.append(clean_station_item)

        movie_channel = {
                         "name": "Toonami Aftermath's Random Movies",
                         "callsign": "Movies",
                         "id": "movies",
                         }
        channel_list.append(movie_channel)

        snickelodeon_est_channel = {
                                     "name": "Snickelodeon (Eastern Time)",
                                     "callsign": "SNick-EST",
                                     "id": "snick-est",
                                     }
        channel_list.append(snickelodeon_est_channel)
        snickelodeon_pst_channel = {
                                     "name": "Snickelodeon (Pacific Time)",
                                     "callsign": "SNick-PST",
                                     "id": "snick-pst",
                                     }
        channel_list.append(snickelodeon_pst_channel)

        mtv97_est_channel = {
                             "name": "MTV97 (Eastern Time)",
                             "callsign": "MTV97-EST",
                             "id": "mtv97",
                             }
        channel_list.append(mtv97_est_channel)
        mtv97_pst_channel = {
                             "name": "MTV97 (Pacific Time)",
                             "callsign": "MTV97-PST",
                             "id": "mtv97-PSToff",
                             }
        channel_list.append(mtv97_pst_channel)

        return channel_list

    def get_channel_stream(self, chandict):
        if "pst" in chandict["origin_id"]:
            streamurl = self.stream_url_base + chandict["origin_id"].replace("pst", "est")
            streamurl += "&streamDelay=180"
        elif "-PSToff" in chandict["origin_id"]:
            streamurl = self.stream_url_base + chandict["origin_id"].replace("-PSToff", "")
            streamurl += "&streamDelay=180"
        else:
            streamurl = self.stream_url_base + chandict["origin_id"]
        streamurl = self.fhdhr.web.session.get(streamurl).text
        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