--- /dev/null
+# coding: utf-8
+from __future__ import unicode_literals
+
+import re
+import base64
+import calendar
+import datetime
+
+from .common import InfoExtractor
+from ..utils import (
+ ExtractorError,
+ update_url_query,
+ clean_html,
+ unified_timestamp,
+)
+from ..compat import compat_urllib_parse
+
+
+class RadikoBaseIE(InfoExtractor):
+ _FULL_KEY = None
+
+ def _auth_client(self):
+ auth_cache = self._downloader.cache.load('radiko', 'auth_data')
+ if auth_cache:
+ return auth_cache
+
+ _, auth1_handle = self._download_webpage_handle(
+ 'https://radiko.jp/v2/api/auth1', None, 'Downloading authentication page',
+ headers={
+ 'x-radiko-app': 'pc_html5',
+ 'x-radiko-app-version': '0.0.1',
+ 'x-radiko-device': 'pc',
+ 'x-radiko-user': 'dummy_user',
+ })
+ auth1_header = auth1_handle.info()
+
+ auth_token = auth1_header['X-Radiko-AuthToken']
+ kl = int(auth1_header['X-Radiko-KeyLength'])
+ ko = int(auth1_header['X-Radiko-KeyOffset'])
+ raw_partial_key = self._extract_full_key()[ko:ko + kl]
+ partial_key = base64.b64encode(raw_partial_key).decode()
+
+ area_id = self._download_webpage(
+ 'https://radiko.jp/v2/api/auth2', None, 'Authenticating',
+ headers={
+ 'x-radiko-device': 'pc',
+ 'x-radiko-user': 'dummy_user',
+ 'x-radiko-authtoken': auth_token,
+ 'x-radiko-partialkey': partial_key,
+ }).split(',')[0]
+
+ auth_data = (auth_token, area_id)
+ self._downloader.cache.store('radiko', 'auth_data', auth_data)
+ return auth_data
+
+ def _extract_full_key(self):
+ if self._FULL_KEY:
+ return self._FULL_KEY
+
+ jscode = self._download_webpage(
+ 'https://radiko.jp/apps/js/playerCommon.js', None,
+ note='Downloading player js code')
+ full_key = self._search_regex(
+ (r"RadikoJSPlayer\([^,]*,\s*(['\"])pc_html5\1,\s*(['\"])(?P<fullkey>[0-9a-f]+)\2,\s*{"),
+ jscode, 'full key', fatal=False, group='fullkey')
+
+ if full_key:
+ full_key = full_key.encode()
+ else: # use full key ever known
+ full_key = b'bcd151073c03b352e1ef2fd66c32209da9ca0afa'
+
+ self._FULL_KEY = full_key
+ return full_key
+
+ def _find_program(self, video_id, station, cursor):
+ station_program = self._download_xml(
+ 'https://radiko.jp/v3/program/station/weekly/%s.xml' % station, video_id,
+ note='Downloading radio program for %s station' % station)
+
+ prog = None
+ for p in station_program.findall('.//prog'):
+ ft_str, to_str = p.attrib['ft'], p.attrib['to']
+ ft = unified_timestamp(ft_str, False)
+ to = unified_timestamp(to_str, False)
+ if ft <= cursor and cursor < to:
+ prog = p
+ break
+ if not prog:
+ raise ExtractorError('Cannot identify radio program to download!')
+ assert ft, to
+ return prog, station_program, ft, ft_str, to_str
+
+ def _extract_formats(self, video_id, station, is_onair, ft, cursor, auth_token, area_id, query):
+ m3u8_playlist_data = self._download_xml(
+ 'https://radiko.jp/v3/station/stream/pc_html5/%s.xml' % station, video_id,
+ note='Downloading m3u8 information')
+ m3u8_urls = m3u8_playlist_data.findall('.//url')
+
+ formats = []
+ found = set()
+ for url_tag in m3u8_urls:
+ pcu = url_tag.find('playlist_create_url')
+ url_attrib = url_tag.attrib
+ playlist_url = update_url_query(pcu.text, {
+ 'station_id': station,
+ **query,
+ 'l': '15',
+ 'lsid': '77d0678df93a1034659c14d6fc89f018',
+ 'type': 'b',
+ })
+ if playlist_url in found:
+ continue
+ else:
+ found.add(playlist_url)
+
+ time_to_skip = None if is_onair else cursor - ft
+
+ subformats = self._extract_m3u8_formats(
+ playlist_url, video_id, ext='m4a',
+ live=True, fatal=False, m3u8_id=None,
+ headers={
+ 'X-Radiko-AreaId': area_id,
+ 'X-Radiko-AuthToken': auth_token,
+ })
+ for sf in subformats:
+ domain = sf['format_id'] = compat_urllib_parse.urlparse(sf['url']).netloc
+ if re.match(r'^[cf]-radiko\.smartstream\.ne\.jp$', domain):
+ # Prioritize live radio vs playback based on extractor
+ sf['preference'] = 100 if is_onair else -100
+ if not is_onair and url_attrib['timefree'] == '1' and time_to_skip:
+ sf['_ffmpeg_args'] = ['-ss', time_to_skip]
+ formats.extend(subformats)
+
+ self._sort_formats(formats)
+ return formats
+
+
+class RadikoIE(RadikoBaseIE):
+ _VALID_URL = r'https?://(?:www\.)?radiko\.jp/#!/ts/(?P<station>[A-Z0-9-]+)/(?P<id>\d+)'
+
+ _TESTS = [{
+ # QRR (文化放送) station provides <desc>
+ 'url': 'https://radiko.jp/#!/ts/QRR/20210425101300',
+ 'only_matching': True,
+ }, {
+ # FMT (TOKYO FM) station does not provide <desc>
+ 'url': 'https://radiko.jp/#!/ts/FMT/20210810150000',
+ 'only_matching': True,
+ }, {
+ 'url': 'https://radiko.jp/#!/ts/JOAK-FM/20210509090000',
+ 'only_matching': True,
+ }]
+
+ def _real_extract(self, url):
+ station, video_id = self._match_valid_url(url).groups()
+ vid_int = unified_timestamp(video_id, False)
+
+ auth_token, area_id = self._auth_client()
+
+ prog, station_program, ft, radio_begin, radio_end = self._find_program(video_id, station, vid_int)
+
+ title = prog.find('title').text
+ description = clean_html(prog.find('info').text)
+ station_name = station_program.find('.//name').text
+
+ formats = self._extract_formats(
+ video_id=video_id, station=station, is_onair=False,
+ ft=ft, cursor=vid_int, auth_token=auth_token, area_id=area_id,
+ query={
+ 'start_at': radio_begin,
+ 'ft': radio_begin,
+ 'end_at': radio_end,
+ 'to': radio_end,
+ 'seek': video_id,
+ })
+
+ return {
+ 'id': video_id,
+ 'title': title,
+ 'description': description,
+ 'uploader': station_name,
+ 'uploader_id': station,
+ 'timestamp': vid_int,
+ 'formats': formats,
+ 'is_live': True,
+ }
+
+
+class RadikoRadioIE(RadikoBaseIE):
+ _VALID_URL = r'https?://(?:www\.)?radiko\.jp/#!/live/(?P<id>[A-Z0-9-]+)'
+
+ _TESTS = [{
+ # QRR (文化放送) station provides <desc>
+ 'url': 'https://radiko.jp/#!/live/QRR',
+ 'only_matching': True,
+ }, {
+ # FMT (TOKYO FM) station does not provide <desc>
+ 'url': 'https://radiko.jp/#!/live/FMT',
+ 'only_matching': True,
+ }, {
+ 'url': 'https://radiko.jp/#!/live/JOAK-FM',
+ 'only_matching': True,
+ }]
+
+ def _real_extract(self, url):
+ station = self._match_id(url)
+ self.report_warning('Downloader will not stop at the end of the program! Press Ctrl+C to stop')
+
+ auth_token, area_id = self._auth_client()
+ # get current time in JST (GMT+9:00 w/o DST)
+ vid_now = datetime.datetime.now(datetime.timezone(datetime.timedelta(hours=9)))
+ vid_now = calendar.timegm(vid_now.timetuple())
+
+ prog, station_program, ft, _, _ = self._find_program(station, station, vid_now)
+
+ title = prog.find('title').text
+ description = clean_html(prog.find('info').text)
+ station_name = station_program.find('.//name').text
+
+ formats = self._extract_formats(
+ video_id=station, station=station, is_onair=True,
+ ft=ft, cursor=vid_now, auth_token=auth_token, area_id=area_id,
+ query={})
+
+ return {
+ 'id': station,
+ 'title': title,
+ 'description': description,
+ 'uploader': station_name,
+ 'uploader_id': station,
+ 'timestamp': ft,
+ 'formats': formats,
+ 'is_live': True,
+ }