[jwplatform] add support for playlist extraction and relative urls and improve audio...
[youtube-dl] / youtube_dl / extractor / jwplatform.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5
6 from .common import InfoExtractor
7 from ..compat import compat_urlparse
8 from ..utils import (
9     determine_ext,
10     float_or_none,
11     int_or_none,
12     mimetype2ext,
13 )
14
15
16 class JWPlatformBaseIE(InfoExtractor):
17     @staticmethod
18     def _find_jwplayer_data(webpage):
19         # TODO: Merge this with JWPlayer-related codes in generic.py
20
21         mobj = re.search(
22             'jwplayer\((?P<quote>[\'"])[^\'" ]+(?P=quote)\)\.setup\((?P<options>[^)]+)\)',
23             webpage)
24         if mobj:
25             return mobj.group('options')
26
27     def _extract_jwplayer_data(self, webpage, video_id, *args, **kwargs):
28         jwplayer_data = self._parse_json(
29             self._find_jwplayer_data(webpage), video_id)
30         return self._parse_jwplayer_data(
31             jwplayer_data, video_id, *args, **kwargs)
32
33     def _parse_jwplayer_data(self, jwplayer_data, video_id, require_title=True, m3u8_id=None, rtmp_params=None, base_url=None):
34         # JWPlayer backward compatibility: flattened playlists
35         # https://github.com/jwplayer/jwplayer/blob/v7.4.3/src/js/api/config.js#L81-L96
36         if 'playlist' not in jwplayer_data:
37             jwplayer_data = {'playlist': [jwplayer_data]}
38
39         entries = []
40         for video_data in jwplayer_data['playlist']:
41             # JWPlayer backward compatibility: flattened sources
42             # https://github.com/jwplayer/jwplayer/blob/v7.4.3/src/js/playlist/item.js#L29-L35
43             if 'sources' not in video_data:
44                 video_data['sources'] = [video_data]
45
46             formats = []
47             for source in video_data['sources']:
48                 source_url = self._proto_relative_url(source['file'])
49                 if base_url:
50                     source_url = compat_urlparse.urljoin(base_url, source_url)
51                 source_type = source.get('type') or ''
52                 ext = mimetype2ext(source_type) or determine_ext(source_url)
53                 if source_type == 'hls' or ext == 'm3u8':
54                     formats.extend(self._extract_m3u8_formats(
55                         source_url, video_id, 'mp4', 'm3u8_native', m3u8_id=m3u8_id, fatal=False))
56                 # https://github.com/jwplayer/jwplayer/blob/master/src/js/providers/default.js#L67
57                 elif source_type.startswith('audio') or ext in ('oga', 'aac', 'mp3', 'mpeg', 'vorbis'):
58                     formats.append({
59                         'url': source_url,
60                         'vcodec': 'none',
61                         'ext': ext,
62                     })
63                 else:
64                     a_format = {
65                         'url': source_url,
66                         'width': int_or_none(source.get('width')),
67                         'height': int_or_none(source.get('height')),
68                         'ext': ext,
69                     }
70                     if source_url.startswith('rtmp'):
71                         a_format['ext'] = 'flv',
72
73                         # See com/longtailvideo/jwplayer/media/RTMPMediaProvider.as
74                         # of jwplayer.flash.swf
75                         rtmp_url_parts = re.split(
76                             r'((?:mp4|mp3|flv):)', source_url, 1)
77                         if len(rtmp_url_parts) == 3:
78                             rtmp_url, prefix, play_path = rtmp_url_parts
79                             a_format.update({
80                                 'url': rtmp_url,
81                                 'play_path': prefix + play_path,
82                             })
83                         if rtmp_params:
84                             a_format.update(rtmp_params)
85                     formats.append(a_format)
86             self._sort_formats(formats)
87
88             subtitles = {}
89             tracks = video_data.get('tracks')
90             if tracks and isinstance(tracks, list):
91                 for track in tracks:
92                     if track.get('file') and track.get('kind') == 'captions':
93                         subtitles.setdefault(track.get('label') or 'en', []).append({
94                             'url': self._proto_relative_url(track['file'])
95                         })
96
97             entries.append({
98                 'id': video_id,
99                 'title': video_data['title'] if require_title else video_data.get('title'),
100                 'description': video_data.get('description'),
101                 'thumbnail': self._proto_relative_url(video_data.get('image')),
102                 'timestamp': int_or_none(video_data.get('pubdate')),
103                 'duration': float_or_none(jwplayer_data.get('duration')),
104                 'subtitles': subtitles,
105                 'formats': formats,
106             })
107         if len(entries) == 1:
108             return entries[0]
109         else:
110             return self.playlist_result(entries)
111
112
113 class JWPlatformIE(JWPlatformBaseIE):
114     _VALID_URL = r'(?:https?://content\.jwplatform\.com/(?:feeds|players|jw6)/|jwplatform:)(?P<id>[a-zA-Z0-9]{8})'
115     _TEST = {
116         'url': 'http://content.jwplatform.com/players/nPripu9l-ALJ3XQCI.js',
117         'md5': 'fa8899fa601eb7c83a64e9d568bdf325',
118         'info_dict': {
119             'id': 'nPripu9l',
120             'ext': 'mov',
121             'title': 'Big Buck Bunny Trailer',
122             'description': 'Big Buck Bunny is a short animated film by the Blender Institute. It is made using free and open source software.',
123             'upload_date': '20081127',
124             'timestamp': 1227796140,
125         }
126     }
127
128     @staticmethod
129     def _extract_url(webpage):
130         mobj = re.search(
131             r'<script[^>]+?src=["\'](?P<url>(?:https?:)?//content.jwplatform.com/players/[a-zA-Z0-9]{8})',
132             webpage)
133         if mobj:
134             return mobj.group('url')
135
136     def _real_extract(self, url):
137         video_id = self._match_id(url)
138         json_data = self._download_json('http://content.jwplatform.com/feeds/%s.json' % video_id, video_id)
139         return self._parse_jwplayer_data(json_data, video_id)