From cd55c6ccd7b9cd0c48d475330c40f382eb0bc625 Mon Sep 17 00:00:00 2001
From: sh!zeeg <shizeeque@gmail.com>
Date: Wed, 4 Jan 2017 01:51:08 +0300
Subject: [PATCH] [beam:live] Add extractor

---
 youtube_dl/extractor/beampro.py    | 82 ++++++++++++++++++++++++++++++
 youtube_dl/extractor/extractors.py |  1 +
 2 files changed, 83 insertions(+)
 create mode 100644 youtube_dl/extractor/beampro.py

diff --git a/youtube_dl/extractor/beampro.py b/youtube_dl/extractor/beampro.py
new file mode 100644
index 000000000..dc0a2b4af
--- /dev/null
+++ b/youtube_dl/extractor/beampro.py
@@ -0,0 +1,82 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+from .common import InfoExtractor
+from ..utils import (
+    ExtractorError,
+    clean_html,
+    compat_str,
+    int_or_none,
+    parse_iso8601,
+    try_get,
+)
+
+
+class BeamProLiveIE(InfoExtractor):
+    IE_NAME = 'Beam:live'
+    _VALID_URL = r'https?://(?:\w+.)?beam.pro/(?P<id>[^?]+)$'
+    _API_CHANNEL = 'https://beam.pro/api/v1/channels/{0}'
+    _API_MANIFEST = 'https://beam.pro/api/v1/channels/{0}/manifest.m3u8'
+    _RATINGS = {'family': 0, 'teen': 13, '18+': 18}
+
+    _TEST = {
+        'url': 'http://www.beam.pro/niterhayven',
+        'info_dict': {
+            'id': '261562',
+            'ext': 'mp4',
+            'uploader': 'niterhayven',
+            'timestamp': 1483477281,
+            'age_limit': 18,
+            'title': 'Introducing The Witcher 3 //  The Grind Starts Now!',
+            'thumbnail': r're:https://.*\.jpg$',
+            'upload_date': '20170103',
+            'uploader_id': 373396,
+            'description': 'md5:0b161ac080f15fe05d18a07adb44a74d',
+            'is_live': True,
+        },
+        'skip': 'niterhayven is offline',
+        'params': {
+            'skip_download': True,
+        },
+    }
+
+    def _real_extract(self, url):
+        channel_id = self._match_id(url)
+        chan_data = self._download_json(self._API_CHANNEL.format(channel_id), channel_id)
+
+        if not chan_data.get('online'):
+            raise ExtractorError('{0} is offline'.format(channel_id), expected=True)
+
+        formats = self._extract_m3u8_formats(
+            self._API_MANIFEST.format(chan_data.get('id')), channel_id, ext='mp4')
+
+        self._sort_formats(formats)
+        info = {}
+        info['formats'] = formats
+        if chan_data:
+            info.update(self._extract_info(chan_data))
+        if not info.get('title'):
+            info['title'] = self._live_title(channel_id)
+        if not info.get('id'):  # barely possible but just in case
+            info['id'] = compat_str(abs(hash(channel_id)) % (10 ** 8))
+
+        return info
+
+    def _extract_info(self, info):
+        thumbnail = try_get(info, lambda x: x['thumbnail']['url'], compat_str)
+        username = try_get(info, lambda x: x['user']['url'], compat_str)
+        video_id = compat_str(info['id']) if info.get('id') else None
+        rating = info.get('audience')
+
+        return {
+            'id': video_id,
+            'title': info.get('name'),
+            'description': clean_html(info.get('description')),
+            'age_limit': self._RATINGS[rating] if rating in self._RATINGS else None,
+            'is_live': True if info.get('online') else False,
+            'timestamp': parse_iso8601(info.get('updatedAt')),
+            'uploader': info.get('token') or username,
+            'uploader_id': int_or_none(info.get('userId')),
+            'view_count': int_or_none(info.get('viewersTotal')),
+            'thumbnail': thumbnail,
+        }
diff --git a/youtube_dl/extractor/extractors.py b/youtube_dl/extractor/extractors.py
index 5ba8efb0e..9d0610d21 100644
--- a/youtube_dl/extractor/extractors.py
+++ b/youtube_dl/extractor/extractors.py
@@ -88,6 +88,7 @@
     BBCCoUkPlaylistIE,
     BBCIE,
 )
+from .beampro import BeamProLiveIE
 from .beeg import BeegIE
 from .behindkink import BehindKinkIE
 from .bellmedia import BellMediaIE