plugins.afreecatv: Update for service changes.

Resolves #488.

src/livestreamer/plugins/afreecatv.py

@@ -1,73 +1,58 @@
 import re
 
-from requests.adapters import HTTPAdapter
-
 from livestreamer.plugin import Plugin
-from livestreamer.plugin.api import http
+from livestreamer.plugin.api import http, validate
 from livestreamer.stream import HLSStream
 
 USER_AGENT = (
     "Mozilla/5.0 (iPhone; CPU iPhone OS 6_0 like Mac OS X) AppleWebKit/536.26 "
     "(KHTML, like Gecko) Version/6.0 Mobile/10A5376e Safari/8536.25"
 )
 HEADERS = {"User-Agent": USER_AGENT}
-PLAYLIST_URL = "http://m.afreeca.com/live/stream/a/hls/broad_no/{0}"
-CHANNEL_URL = "http://afbbs.afreeca.com:8080/api/video/get_bj_liveinfo.php"
+PLAYLIST_URL = "http://m.afreeca.com/live/stream/a/hls/broad_no/{broad_no}"
+BROAD_INFO_URL = "https://api.m.afreeca.com/broad/a/getbroadinfo"
 
 _broadcast_re = re.compile(r".+\/(\d+)\.gif")
 _url_re = re.compile("http(s)?://(\w+\.)?afreeca.com/(?P<username>\w+)")
 
-
-class RedirectAdapter(HTTPAdapter):
-    """This adapter redirects connections on the socket level.
-
-    This is done to keep the Host header the same but replace the
-    actual server IP address used for the connection.
-    """
-
-    def __init__(self, *args, **kwargs):
-        self.redirect_source = kwargs.pop("source")
-        self.redirect_target = kwargs.pop("target")
-        HTTPAdapter.__init__(self, *args, **kwargs)
-
-    def get_connection(self, url, proxies=None):
-        url = url.replace(self.redirect_source, self.redirect_target)
-        return HTTPAdapter.get_connection(self, url, proxies=proxies)
+_broadcast_schema = validate.Schema(
+    {
+        "data": validate.any(
+            None,
+            {
+                "broad_no": int,
+            }
+        )
+    },
+    validate.get("data")
+)
 
 
 class AfreecaTV(Plugin):
     @classmethod
     def can_handle_url(self, url):
         return _url_re.match(url)
 
-    def _find_broadcast(self, username):
-        res = http.get(CHANNEL_URL, headers=HEADERS, params=dict(szBjId=username))
-        liveinfo = http.xml(res)
-        thumb = liveinfo.findtext("thumb")
-        if not thumb:
-            return
+    def _get_broadcast(self, username):
+        data = {
+            "szBjId": username
+        }
+        res = http.post(BROAD_INFO_URL, headers=HEADERS, data=data)
 
-        match = _broadcast_re.match(thumb)
-        if match:
-            return match.group(1)
+        return http.json(res, schema=_broadcast_schema)
 
     def _get_streams(self):
         match = _url_re.match(self.url)
-        if not match:
-            return
-
         username = match.group("username")
-        broadcast = self._find_broadcast(username)
+
+        broadcast = self._get_broadcast(username)
         if not broadcast:
             return
 
-        playlist_url = PLAYLIST_URL.format(broadcast)
-        return HLSStream.parse_variant_playlist(self.session, playlist_url,
-                                                headers=HEADERS)
+        playlist_url = PLAYLIST_URL.format(**broadcast)
+        stream = HLSStream(self.session, playlist_url, headers=HEADERS)
 
-# Redirect 123.111.232.16 to 123.111.232.13
-http.mount("http://123.111.232.16", RedirectAdapter(source="123.111.232.16",
-                                                    target="123.111.232.13"))
+        return dict(live=stream)
 
 
 __plugin__ = AfreecaTV