# encoding: utf-8
from __future__ import unicode_literals

from .common import InfoExtractor

from ..compat import (
    compat_urllib_parse_unquote
)

import re
import xml.etree.ElementTree

class TrollvidsIE(InfoExtractor):
    _VALID_URL = r"http://(?:www\.)?trollvids\.com/+video/+(?P<id>[0-9]+)/+(?P<title>[^?&]+)"
    IE_NAME = 'trollvids'

    def _real_extract(self, url):
        match = re.match(self._VALID_URL, url)

        video_id = match.group('id')
        raw_video_title = match.group('title')
        video_title = compat_urllib_parse_unquote(raw_video_title)
        url = "http://trollvids.com/video/%s/%s" % (video_id, raw_video_title)

        info = {
            "id": video_id,
            "title": video_title,
            "webpage_url": url,
            "age_limit": 18
        }

        sdformats = []
        hdformats = []

        tree = self._download_xml("http://trollvids.com/nuevo/player/config.php?v=%s" % video_id, video_id)

        for child in tree:
            tag, val = child.tag, child.text

            if tag == "file":
                sdformats.append({"url": val})
            elif tag == "filehd":
                hdformats.append({"url": val})
            elif tag == "duration":
                info["duration"] = int(float(val))
            elif tag == "image":
                info["thumbnail"] = val
            elif tag == "title":
                info["title"] = val

        info["formats"] = sdformats + hdformats
        return info