# -*- coding: utf-8 -*-

# Copyright(C) 2014 Johann Broudin
#
# This file is part of weboob.
#
# weboob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# weboob is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>.


from weboob.capabilities.radio import CapRadio, Radio
from weboob.capabilities.audiostream import BaseAudioStream
from weboob.tools.capabilities.streaminfo import StreamInfo
from weboob.capabilities.collection import CapCollection
from weboob.tools.backend import Module
from weboob.browser import Browser
from weboob.tools.misc import to_unicode


__all__ = ['VirginRadioModule']


class VirginRadioModule(Module, CapRadio, CapCollection):
    NAME = 'virginradio'
    MAINTAINER = u'Johann Broudin'
    EMAIL = 'Johann.Broudin@6-8.fr'
    VERSION = '1.1'
    DESCRIPTION = u'VirginRadio french radio'
    LICENSE = 'AGPLv3+'
    BROWSER = Browser

    _RADIOS = {
            'officiel': (
                u'Virgin Radio',
                u'Virgin Radio',
                u'http://mp3lg3.scdn.arkena.com/10490/virginradio.mp3',
                64),
            'new': (
                u'Virgin Radio New',
                u'Virgin Radio New',
                u'http://mp3lg3.tdf-cdn.com/9145/lag_103228.mp3',
                64),
            'classics': (
                u'Virgin Radio Classics',
                u'Virgin Radio Classics',
                u'http://mp3lg3.tdf-cdn.com/9146/lag_103325.mp3',
                64),
            'electroshock': (
                u'Virgin Radio Electroshock',
                u'Virgin Radio Electroshock',
                u'http://mp3lg3.tdf-cdn.com/9148/lag_103401.mp3',
                64),
            'hits': (
                u'Virgin Radio Hits',
                u'Virgin Radio Hits',
                u'http://mp3lg3.tdf-cdn.com/9150/lag_103440.mp3',
                64),
            'rock': (
                u'Virgin Radio Rock',
                u'Virgin Radio Rock',
                u'http://mp3lg3.scdn.arkena.com/9151/lag_103523.mp3',
                64)
            }

    def get_stream_info(self, radio, url):
        stream = BaseAudioStream(0)
        current = StreamInfo(0)

        r = self.browser.open(url, stream=True, headers={'Icy-Metadata':1})

        stream.bitrate = int(r.headers['icy-br'].split(',')[0])

        r.raw.read(int(r.headers['icy-metaint']))
        size = ord(r.raw.read(1))
        content = r.raw.read(size*16)
        r.close()

        for s in content.split("\x00")[0].split(";"):
            a = s.split("=")
            if a[0] == "StreamTitle":
                stream.title = to_unicode(a[1].split("'")[1])
                res = a[1].split("'")[1].split(" - ")
                current.who = to_unicode(res[0])
                current.what = to_unicode(res[1])
        stream.format=u'mp3'
        stream.url = url
        return [stream], current

    def get_radio(self, radio):
        if not isinstance(radio, Radio):
            radio = Radio(radio)

        if radio.id not in self._RADIOS:
            return None

        title, description, url, bitrate = self._RADIOS[radio.id]

        radio.title = title
        radio.description = description

        radio.streams, radio.current = self.get_stream_info(radio.id, url)
        return radio

    def iter_resources(self, objs, split_path):
        if Radio in objs:
            self._restrict_level(split_path)

            for id in self._RADIOS.iterkeys():
                yield self.get_radio(id)

    def iter_radios_search(self, pattern):
        for radio in self.iter_resources((Radio, ), []):
            if pattern.lower() in radio.title.lower() or pattern.lower() in radio.description.lower():
                yield radio

    def fill_radio(self, radio, fields):
        if 'current' in fields:
            if not radio.current:
                radio = self.get_radio(radio.id)
        return radio

    OBJECTS = {Radio: fill_radio}