implement #86

moodle_dl/moodle_connector/results_handler.py

@@ -82,7 +82,7 @@ def _get_files_in_modules(self, section_name: str, section_modules: []) -> [File
             if module_modname in ['moodecvideo', 'page']:
                 module_modname = 'index_mod-' + module_modname
 
-            if module_modname in ['kalvidres']:
+            if module_modname in ['kalvidres', 'helixmedia']:
                 module_modname = 'cookie_mod-' + module_modname
                 files += self._handle_cookie_mod(section_name, module_name, module_modname, module_id, module_url)
 

moodle_dl/ydl_extractors/extractors.py

@@ -2,13 +2,15 @@
 
 from moodle_dl.ydl_extractors.zoomus import ZoomUSIE
 from moodle_dl.ydl_extractors.opencast import OpencastIE, OpencastPlaylistIE
+from moodle_dl.ydl_extractors.helixmedia import Helixmedia
 
 
 def add_additional_extractors(ydl: YoutubeDL):
     additional_extractors = [
         OpencastIE(ydl),
         OpencastPlaylistIE(ydl),
         ZoomUSIE(ydl),
+        Helixmedia(ydl),
     ]
 
     for extractor in additional_extractors:

moodle_dl/ydl_extractors/helixmedia.py

@@ -0,0 +1,122 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+
+import re
+import json
+
+from youtube_dl.compat import (
+    compat_parse_qs,
+    compat_urllib_parse_urlparse,
+)
+
+from youtube_dl.extractor.common import InfoExtractor
+from youtube_dl.utils import (
+    ExtractorError,
+    urlencode_postdata,
+    js_to_json,
+    determine_ext,
+)
+
+
+class Helixmedia(InfoExtractor):
+    IE_NAME = 'helixmedia'
+    _VALID_URL = r'(?P<scheme>https?://)(?P<host>[^/]+)(?P<path>.*)?/mod/helixmedia/view.php\?.*?id=(?P<id>\d+)'
+
+    # _TEST = {'url': 'http://localhost/moodle/mod/helixmedia/view.php?id=3'}
+
+    def _real_extract(self, url):
+        mobj = re.match(self._VALID_URL, url)
+        scheme = mobj.group('scheme')
+        host = mobj.group('host')
+        path = mobj.group('path')
+        video_id = mobj.group('id')
+
+        launch_url = scheme + host + path + '/mod/helixmedia/launch.php?type=1&id=' + video_id
+
+        # webpage = self._download_webpage(url, video_id)
+        launch_webpage = self._download_webpage(launch_url, video_id)
+        launch_inputs = self._hidden_inputs(launch_webpage)
+
+        action_url = 'https://na1upload.medialibrary.com/Lti/Launch'
+        submit_page, start_urlh = self._download_webpage_handle(
+            action_url, video_id, 'Launch Helix Media', data=urlencode_postdata(launch_inputs)
+        )
+
+        if 'UploadSessionId' not in start_urlh.geturl():
+            raise ExtractorError('Unable to launch helixmedia video', expected=True)
+
+        qs = compat_parse_qs(compat_urllib_parse_urlparse(start_urlh.geturl()).query)
+        UploadSessionId = qs.get('UploadSessionId', [None])[0]
+
+        default_video_url = (
+            'https://na1upload.medialibrary.com/Lti/HomeSplit?mobile=N&fullWidth=940&fullHeight=906&UploadSessionId='
+        )
+        video_webpage = self._download_webpage(default_video_url + UploadSessionId, video_id)
+
+        video_model = json.loads(js_to_json(self._search_regex(r'var model = ([^;]+);', video_webpage, 'video model')))
+
+        video_title = video_model.get('VideoTitle', None)
+        video_description = video_model.get('VideoDescription', '')
+        video_id = str(video_model.get('VideoId', video_id))
+        download_url = video_model.get('DownloadUrl', None)
+        video_json = json.loads(video_model.get('PlayScreenVm', {}).get('VodPlayerModel', {}).get('PlayerJson', '{}'))
+
+        thumbnail_list = video_json.get('tracks', [])
+        thumbnail = None
+        if len(thumbnail_list) >= 1:
+            thumbnail = thumbnail_list[0].get('file', None)
+            if thumbnail is not None:
+                thumbnail = thumbnail.replace('vtt', 'jpg')
+
+        sources_list = video_json.get('sources', [])
+
+        formats = []
+        for track in sources_list:
+            href = track['file']
+            ext = determine_ext(href, None)
+
+            if ext == 'mpd':
+                # DASH
+                formats.extend(self._extract_mpd_formats(href, video_id, mpd_id='dash', fatal=False))
+            elif ext == 'm3u8':
+                # HLS
+                formats.extend(
+                    self._extract_m3u8_formats(href, video_id, m3u8_id='hls', entry_protocol='m3u8_native', fatal=False)
+                )
+            elif ext == 'f4m':
+                # HDS
+                formats.extend(self._extract_f4m_formats(href, video_id, f4m_id='hds', fatal=False))
+            elif ext == 'smil':
+                formats.extend(self._extract_smil_formats(href, video_id, fatal=False))
+            elif ext == 'mp4':
+                track_obj = {
+                    'url': href,
+                    'ext': ext,
+                }
+                formats.append(track_obj)
+
+        if download_url is not None:
+            track_obj_direct = {
+                'url': download_url,
+                'ext': 'mp4',
+            }
+            formats.append(track_obj_direct)
+
+        self._sort_formats(formats)
+
+        result_obj = {'formats': formats}
+
+        if video_id is not None:
+            result_obj.update({'id': video_id})
+
+        if video_title is not None:
+            result_obj.update({'title': video_title})
+
+        if video_description is not None:
+            result_obj.update({'creator': video_description})
+
+        if thumbnail is not None:
+            result_obj.update({'thumbnail': thumbnail})
+
+        return result_obj