import datetime
import re

from .common import InfoExtractor
from ..utils import (
    ExtractorError,
    float_or_none,
    join_nonempty,
    qualities,
    smuggle_url,
    str_or_none,
    strip_or_none,
    traverse_obj,
    update_url_query,
)


class TVerIE(InfoExtractor):
    _VALID_URL = r'https?://(?:www\.)?tver\.jp/(?:(?P<type>lp|corner|series|episodes?|feature)/)+(?P<id>[a-zA-Z0-9]+)'
    _TESTS = [
        {
            'skip': 'videos are only available for 7 days',
            'url': 'https://tver.jp/episodes/ep83nf3w4p',
            'info_dict': {
                'title': '家事ヤロウ!!! 売り場席巻のチーズSP＆財前直見×森泉親子の脱東京暮らし密着！',
                'description': 'md5:dc2c06b6acc23f1e7c730c513737719b',
                'series': '家事ヤロウ!!!',
                'episode': '売り場席巻のチーズSP＆財前直見×森泉親子の脱東京暮らし密着！',
                'alt_title': '売り場席巻のチーズSP＆財前直見×森泉親子の脱東京暮らし密着！',
                'channel': 'テレビ朝日',
                'id': 'ep83nf3w4p',
                'ext': 'mp4',
            },
        },
        {
            'url': 'https://tver.jp/corner/f0103888',
            'only_matching': True,
        },
        {
            'url': 'https://tver.jp/lp/f0033031',
            'only_matching': True,
        },
        {
            'url': 'https://tver.jp/series/srtxft431v',
            'info_dict': {
                'id': 'srtxft431v',
                'title': '名探偵コナン',
            },
            'playlist': [
                {
                    'md5': '779ffd97493ed59b0a6277ea726b389e',
                    'info_dict': {
                        'id': 'ref:conan-1137-241005',
                        'ext': 'mp4',
                        'title': '名探偵コナン #1137「行列店、味変の秘密」',
                        'uploader_id': '5330942432001',
                        'tags': [],
                        'channel': '読売テレビ',
                        'series': '名探偵コナン',
                        'description': 'md5:601fccc1d2430d942a2c8068c4b33eb5',
                        'episode': '#1137「行列店、味変の秘密」',
                        'duration': 1469.077,
                        'timestamp': 1728030405,
                        'upload_date': '20241004',
                        'alt_title': '名探偵コナン #1137「行列店、味変の秘密」 読売テレビ 10月5日(土)放送分',
                        'thumbnail': r're:https://.+\.jpg',
                    },
                },
            ],
        },
        {
            'url': 'https://tver.jp/series/sru35hwdd2',
            'info_dict': {
                'id': 'sru35hwdd2',
                'title': '神回だけ見せます！',
            },
            'playlist_count': 11,
        },
        {
            'url': 'https://tver.jp/series/srkq2shp9d',
            'only_matching': True,
        },
    ]

    BRIGHTCOVE_URL_TEMPLATE = 'http://players.brightcove.net/%s/default_default/index.html?videoId=%s'

    STREAKS_URL_TEMPLATE = 'https://playback.api.streaks.jp/v1/projects/%s/medias/%s'

    _HEADERS = {
        'x-tver-platform-type': 'web',
        'origin': 'https://tver.jp/',
        'referer': 'https://tver.jp/',
    }
    _PLATFORM_QUERY = {}

    def _real_initialize(self):
        session_info = self._download_json(
            'https://platform-api.tver.jp/v2/api/platform_users/browser/create',
            None,
            'Creating session',
            data=b'device_type=pc',
        )
        self._PLATFORM_QUERY = traverse_obj(
            session_info,
            (
                'result',
                {
                    'platform_uid': 'platform_uid',
                    'platform_token': 'platform_token',
                },
            ),
        )

    def _call_platform_api(self, path, video_id, note=None, fatal=True, query=None):
        return self._download_json(
            f'https://platform-api.tver.jp/service/api/{path}',
            video_id,
            note,
            fatal=fatal,
            headers=self._HEADERS,
            query={
                **self._PLATFORM_QUERY,
                **(query or {}),
            },
        )

    def _yield_episode_ids_for_series(self, series_id):
        seasons_info = self._download_json(
            f'https://service-api.tver.jp/api/v1/callSeriesSeasons/{series_id}',
            series_id,
            'Downloading seasons info',
            headers=self._HEADERS,
        )
        for season_id in traverse_obj(
            seasons_info,
            ('result', 'contents', lambda _, v: v['type'] == 'season', 'content', 'id', {str}),
        ):
            episodes_info = self._call_platform_api(
                f'v1/callSeasonEpisodes/{season_id}',
                series_id,
                f'Downloading season {season_id} episodes info',
            )
            yield from traverse_obj(
                episodes_info,
                (
                    'result',
                    'contents',
                    lambda _, v: v['type'] == 'episode',
                    'content',
                    'id',
                    {str},
                ),
            )

    def _real_extract(self, url):
        video_id, video_type = self._match_valid_url(url).group('id', 'type')

        if video_type == 'series':
            series_info = self._call_platform_api(f'v2/callSeries/{video_id}', video_id, 'Downloading series info')
            return self.playlist_from_matches(
                self._yield_episode_ids_for_series(video_id),
                video_id,
                traverse_obj(series_info, ('result', 'content', 'content', 'title', {str})),
                ie=TVerIE,
                getter=lambda x: f'https://tver.jp/episodes/{x}',
            )

        if video_type != 'episodes':
            webpage = self._download_webpage(url, video_id, note='Resolving to new URL')
            video_id = self._match_id(
                self._search_regex(
                    (
                        r'canonical"\s*href="(https?://tver\.jp/[^"]+)"',
                        r'&link=(https?://tver\.jp/[^?&]+)[?&]',
                    ),
                    webpage,
                    'url regex',
                ),
            )

        episode_info = self._call_platform_api(
            f'v1/callEpisode/{video_id}',
            video_id,
            'Downloading episode info',
            fatal=False,
            query={
                'require_data': 'mylist,later[epefy106ur],good[epefy106ur],resume[epefy106ur]',
            },
        )
        episode_content = traverse_obj(episode_info, ('result', 'episode', 'content')) or {}

        version = traverse_obj(episode_content, ('version', {str_or_none}), default='5')

        video_info = self._download_json(
            f'https://statics.tver.jp/content/episode/{video_id}.json',
            video_id,
            'Downloading video info',
            query={'v': version},
            headers={'Referer': 'https://tver.jp/'},
        )

        episode = strip_or_none(episode_content.get('title'))
        series = str_or_none(episode_content.get('seriesTitle'))
        title = join_nonempty(series, episode, delim=' ') or str_or_none(video_info.get('title'))
        provider = str_or_none(episode_content.get('productionProviderName'))
        onair_label = str_or_none(episode_content.get('broadcastDateLabel'))

        thumbnails = [
            {
                'id': quality,
                'url': update_url_query(
                    f'https://statics.tver.jp/images/content/thumbnail/episode/{quality}/{video_id}.jpg',
                    {'v': version},
                ),
                'width': width,
                'height': height,
            }
            for quality, width, height in [
                ('small', 480, 270),
                ('medium', 640, 360),
                ('large', 960, 540),
                ('xlarge', 1280, 720),
            ]
        ]

        data = {
            'title': title,
            'series': series,
            'episode': episode,
            # an another title which is considered "full title" for some viewers
            'alt_title': join_nonempty(title, provider, onair_label, delim=' '),
            'channel': provider,
            'description': str_or_none(video_info.get('description')),
            'thumbnails': thumbnails,
        }

        ts = traverse_obj(video_info, ('viewStatus', 'startAt', {int}), default=None)
        if ts:
            data['timestamp'] = ts

        episode_number = traverse_obj(video_info, ('no', {str_or_none}), default=None)
        if episode_number:
            data['episode_number'] = int(episode_number)

        if onair_label:
            data.update(self._format_broadcast_date(onair_label))

        backend = self._configuration_arg('backend', ['streaks'])[0]

        if backend not in ('brightcove', 'streaks'):
            raise ExtractorError(f'Invalid backend value: {backend}', expected=True)

        if backend == 'brightcove':
            data = self._brightcove_backend(data, video_info)
        else:
            data = self._streaks_backend(data, video_info, video_id)

        return data

    def _format_broadcast_date(self, onair_label):
        """
        Extracts the broadcast date from the onair label

        Truth to be said, we cannot be sure or guarantee that the broadcast date is correct
        as TVer doesn't really have consistent date format for the broadcast date.
        At best we can only assume the following:
        - If there is only year, this mean the broadcast is old.
        - If there is only month and day, this mean the broadcast is recent within the current year or the previous year.

        :param onair_label: The onair label string
        :return: A dictionary containing the formatted broadcast date or an empty dictionary if the date is not found

        """
        if not onair_label:
            return {}

        match = re.search(
            pattern=r'(?:(?P<year>\d{4})年)|(?:(?P<month>\d{1,2})\D(?P<day>\d{1,2})\D)',
            string=onair_label,
        )

        if not match:
            return {}

        data = {}

        broadcast_date = match.groupdict()

        if broadcast_date.get('year'):
            data['release_year'] = int(broadcast_date['year'])

        if broadcast_date.get('day') and broadcast_date.get('month'):
            if 'release_year' in data:
                year = data['release_year']
            else:
                year = datetime.datetime.now().year
                dt = datetime.datetime.strptime(
                    f"{year}-{broadcast_date['month']}-{broadcast_date['day']}",
                    '%Y-%m-%d',
                )
                # if the date in the future, it means the broadcast date is in the previous year
                # ref: https://github.com/yt-dlp/yt-dlp/pull/12282#issuecomment-2678132806
                if dt > datetime.datetime.now():
                    year -= 1

            data['release_timestamp'] = datetime.datetime(
                year=year,
                month=int(broadcast_date['month']),
                day=int(broadcast_date['day']),
            ).timestamp()

        return data

    def _brightcove_backend(self, result, video_info):
        self.write_debug('Using Brightcove backend')

        p_id = video_info['video']['accountID']
        r_id = traverse_obj(video_info, ('video', ('videoRefID', 'videoID')), get_all=False)

        if not r_id:
            raise ExtractorError('Failed to extract reference ID for Brightcove')

        if not r_id.isdigit():
            r_id = f'ref:{r_id}'

        result.update(
            {
                '_type': 'url_transparent',
                'url': smuggle_url(
                    self.BRIGHTCOVE_URL_TEMPLATE % (p_id, r_id),
                    {'geo_countries': ['JP']},
                ),
                'ie_key': 'BrightcoveNew',
            },
        )

        return result

    def _streaks_backend(self, result, video_info, video_id):
        self.write_debug('Using streaks.jp backend')

        ref_id = traverse_obj(video_info, ('streaks', 'videoRefID'), get_all=False)
        project_id = traverse_obj(video_info, ('streaks', 'projectID'), get_all=False)

        if not ref_id:
            raise ExtractorError('Failed to extract reference ID for streaks.jp stream info')

        if not project_id:
            raise ExtractorError('Failed to extract project ID for streaks.jp stream info')

        if not ref_id.startswith('ref:'):
            ref_id = f'ref:{ref_id}'

        url = self.STREAKS_URL_TEMPLATE % (project_id, ref_id)
        self.write_debug(f'Streaks URL: {url}')

        json_info = self._download_json(
            url,
            video_id,
            'Downloading streaks.jp streams video info',
            headers={
                'origin': 'https://tver.jp/',
                'referer': 'https://tver.jp/',
                **self.geo_verification_headers(),
            },
        )

        sources = traverse_obj(json_info, ('sources'), default=[])

        formats = []
        subtitles = {}

        for item in sources:
            m3u8_url = traverse_obj(item, ('src'), default=None)
            if not m3u8_url:
                continue

            item_formats, item_subtitles = self._extract_m3u8_formats_and_subtitles(
                m3u8_url,
                video_id,
                'mp4',
                m3u8_id='hls',
                headers={'origin': 'https://tver.jp/', 'referer': 'https://tver.jp/'},
                note='Downloading streaks.jp m3u8 information',
            )

            build_qualities = []

            for f in item_formats:
                # hls-ts_AUDIO-1_1-pro_105ba7d8d0f4452bb452acf09466b2a5
                if mobj := re.search(r'hls-ts_AUDIO-(.+?)-', f['format_id']):
                    build_qualities.append(mobj.group(1))

            if len(build_qualities) > 0:
                # it seems best quality is 0_X. see ref for more info
                # ref: https://github.com/yt-dlp/yt-dlp/issues/12643#issuecomment-2745263178
                build_qualities.sort(key=lambda x: int(x.split('_')[0]), reverse=True)
                quality = qualities(tuple(build_qualities))

                for f in item_formats:
                    if mobj := re.search(r'hls-ts_AUDIO-(.+?)-', f['format_id']):
                        f['quality'] = quality(mobj.group(1))

            if len(item_formats) > 0:
                formats.extend(item_formats)

            if len(item_subtitles) > 0:
                subtitles.update(item_subtitles)

        if len(formats) < 1:
            raise ExtractorError('Failed to extract any m3u8 streams from streaks.jp video info')

        result.update(
            {
                'id': video_id,
                'formats': formats,
                'subtitles': subtitles,
            },
        )

        duration = float_or_none(json_info.get('duration'), 1000)
        if duration:
            result['duration'] = duration

        return result