From 2302f32ced5f9ca1959926be326d527c8794e8fa Mon Sep 17 00:00:00 2001
From: Wong Yiu Hang <henrymail2006@ymail.com>
Date: Mon, 1 Feb 2021 16:00:24 +0800
Subject: [PATCH] [Loom] Add new extractor

---
 youtube_dl/extractor/extractors.py |   1 +
 youtube_dl/extractor/loom.py       | 127 +++++++++++++++++++++++++++++
 2 files changed, 128 insertions(+)
 create mode 100644 youtube_dl/extractor/loom.py

diff --git a/youtube_dl/extractor/extractors.py b/youtube_dl/extractor/extractors.py
index ab8d6a5a5..78bc67f09 100644
--- a/youtube_dl/extractor/extractors.py
+++ b/youtube_dl/extractor/extractors.py
@@ -605,6 +605,7 @@ from .livestream import (
 )
 from .lnkgo import LnkGoIE
 from .localnews8 import LocalNews8IE
+from .loom import LoomIE
 from .lovehomeporn import LoveHomePornIE
 from .lrt import LRTIE
 from .lynda import (
diff --git a/youtube_dl/extractor/loom.py b/youtube_dl/extractor/loom.py
new file mode 100644
index 000000000..1a22d7aeb
--- /dev/null
+++ b/youtube_dl/extractor/loom.py
@@ -0,0 +1,127 @@
+from __future__ import unicode_literals
+
+from .common import InfoExtractor
+
+from ..compat import (
+    compat_urllib_parse_urlencode,
+    compat_urllib_request
+)
+from ..utils import (
+    js_to_json,
+    try_get,
+    unified_timestamp,
+    url_or_none
+)
+
+
+class LoomBaseInfoIE(InfoExtractor):
+    _BASE_URL = 'https://www.loom.com/'
+
+    def _extract_video_info_json(self, webpage, video_id):
+        info = self._html_search_regex(
+            r'window.loomSSRVideo = (.+?);',
+            webpage,
+            'info')
+        return self._parse_json(info, 'json', js_to_json)
+
+    def _get_url_by_id_type(self, video_id, type):
+        request = compat_urllib_request.Request(
+            self._BASE_URL + 'api/campaigns/sessions/' + video_id + '/' + type,
+            {})
+        (json, _) = self._download_json_handle(request, video_id)
+        return (url_or_none(json.get('url')), json.get('part_credentials'))
+
+    def _get_m3u8_formats(self, url, video_id, credentials):
+        format_list = self._extract_m3u8_formats(url, video_id)
+        for item in format_list:
+            item['protocol'] = 'm3u8_native'
+            item['url'] += '?' + credentials
+            item['ext'] = 'mp4'
+            item['format_id'] = 'hls-' + str(item.get('height', 0))
+            item['extra_param_to_segment_url'] = credentials
+        return format_list
+
+
+class LoomIE(LoomBaseInfoIE):
+    _VALID_URL = r'https?://(?:www\.)?loom\.com/share/(?P<id>[a-zA-Z0-9]+)'
+    _TESTS = [
+        {
+            'url': 'https://www.loom.com/share/31b41727a5b24dacb6c1417a565b2ebf',
+            'md5': '8b94361aabff2075141dc60bd6d35453',
+            'info_dict': {
+                'id': '31b41727a5b24dacb6c1417a565b2ebf',
+                'ext': 'mp4',
+                'title': 'How to resize your camera bubble',
+                'uploader': 'Allie Hitchcock',
+                'upload_date': '20201007',
+                'timestamp': 1602089241
+            }
+        },
+        {
+            'url': 'https://www.loom.com/share/7e5168ec3b0744cab5e08a340cc7e086',
+            'md5': '47dd14aa1d8054c249b68ca57ad9963f',
+            'info_dict': {
+                'id': '7e5168ec3b0744cab5e08a340cc7e086',
+                'ext': 'mp4',
+                'title': 'How to flip your camera ',
+                'uploader': 'Matthew Flores',
+                'upload_date': '20200423',
+                'timestamp': 1587646164
+            }
+        },
+        {
+            'url': 'https://www.loom.com/share/6670e3eba3c84dc09ada8306c7138075',
+            'md5': 'bfad8181ed49d6252b10dfdeb46c535e',
+            'info_dict': {
+                'id': '6670e3eba3c84dc09ada8306c7138075',
+                'ext': 'mp4',
+                'title': 'How to record your first video on Loom',
+                'uploader': 'Allie Hitchcock',
+                'upload_date': '20201118',
+                'timestamp': 1605729404
+            }
+        }
+    ]
+
+    def _real_extract(self, url):
+        video_id = self._match_id(url)
+        webpage = self._download_webpage(url, video_id)
+
+        info = self._extract_video_info_json(webpage, video_id)
+
+        formats = []
+        for type in ['transcoded-url', 'raw-url']:
+            (url, part_credentials) = self._get_url_by_id_type(video_id, type)
+            ext = self._search_regex(
+                r'\.([a-zA-Z0-9]+)\?',
+                url, 'ext', default=None)
+            if(ext != 'm3u8'):
+                formats.append({
+                    'url': url,
+                    'ext': ext,
+                    'format_id': type,
+                    'width': try_get(info, lambda x: x['video_properties']['width']),
+                    'height': try_get(info, lambda x: x['video_properties']['height'])
+                })
+            else:
+                credentials = compat_urllib_parse_urlencode(part_credentials)
+                m3u8_formats = self._get_m3u8_formats(url, video_id, credentials)
+                for i in range(len(m3u8_formats)):
+                    formats.insert(
+                        (-1, len(formats))[i == len(m3u8_formats) - 1],
+                        m3u8_formats[i])
+
+        return {
+            'id': info.get('id'),
+            'title': info.get('name'),
+            'formats': formats,
+            'thumbnails': [
+                {
+                    'id': key,
+                    'url': url_or_none(self._BASE_URL + value)
+                } for key, value in info.get('thumbnails').items()
+            ],
+            'description': info.get('description'),
+            'uploader': info.get('owner_full_name'),
+            'timestamp': unified_timestamp(info.get('createdAt'))
+        }