aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/livestreamer/plugins/ownedtv.py
blob: 5279777bf5be362b68f586bb12e7a9f75ed5e288 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#!/usr/bin/env python3

from livestreamer.compat import urllib
from livestreamer.plugins import Plugin, PluginError, register_plugin
from livestreamer.stream import RTMPStream
from livestreamer.utils import urlget

import xml.dom.minidom, re

class RelativeRedirectHandler(urllib.HTTPRedirectHandler):
    def http_error_302(self, req, fp, code, msg, headers):
        if "location" in headers and headers["location"][0] == "/":
            absurl = ("{scheme}://{host}{path}").format(
                      scheme=req.get_type(), host=req.get_host(),
                      path=headers["location"])
            del headers["location"]
            headers["location"] = absurl

        return urllib.HTTPRedirectHandler.http_error_301(
               self, req, fp, code, msg, headers)

urlopener = urllib.build_opener(RelativeRedirectHandler)


class OwnedTV(Plugin):
    ConfigURL = "http://www.own3d.tv/livecfg/{0}"
    CDN = {
        "cdn1": "rtmp://fml.2010.edgecastcdn.net/202010",
        "cdn2": "rtmp://owned.fc.llnwd.net:1935/owned",
        "cdn3": "http://hwcdn.net/u4k2r7c4/fls",
    }

    @classmethod
    def can_handle_url(self, url):
        return "own3d.tv" in url

    def _get_channel_id(self, url):
        data = urlget(url, opener=urlopener)
        match = re.search(b'flashvars.config = "livecfg/(\d+)', data)
        if match:
            return int(match.group(1))
            
        match = re.search(b"document.location.hash='/live/(\d+)'", data)
        if match:
            return int(match.group(1))

        match = re.search(b"xajax_load_live_config\((\d+),", data)
        if match:
            return int(match.group(1))

    def _get_streams(self):
        channelid = self._get_channel_id(self.url)
        streams = {}

        if channelid:
            data = urlget(self.ConfigURL.format(channelid))

            try:
                dom = xml.dom.minidom.parseString(data)
            except Exception as err:
                raise PluginError(("Unable to parse config XML: {0})").format(err))

            channels = dom.getElementsByTagName("channels")[0]
            clip = channels.getElementsByTagName("clip")[0]

            for item in clip.getElementsByTagName("item"):
                base = item.getAttribute("base")
                if not base:
                    continue

                if base[0] == "$":
                    ref = re.match("\${(.+)}", base).group(1)
                    base = self.CDN[ref]

                for streamel in item.getElementsByTagName("stream"):
                    name = streamel.getAttribute("label").lower().replace(" ", "_")
                    playpath = streamel.getAttribute("name")

                    if not name in streams:
                        streams[name] = RTMPStream({
                            "rtmp": ("{0}/{1}").format(base, playpath),
                            "live": 1
                        })

        return streams

register_plugin("own3dtv", OwnedTV)