from .common import InfoExtractor
from ..utils import float_or_none, smuggle_url, unified_timestamp, unsmuggle_url
from ..utils.traversal import traverse_obj


class AniGamerIE(InfoExtractor):
    _VALID_URL = r'https?://ani\.gamer\.com\.tw/animeVideo\.php\?sn=(?P<id>\d+)'

    RATING_TO_AGE_LIMIT = {
        1: 0,
        2: 6,
        3: 12,
        4: 15,
        # Seems like there's no age limit for '5'
        6: 18,
    }

    def _real_extract(self, url):
        url, unsmuggled_data = unsmuggle_url(url, {})
        video_id = self._match_id(url)
        device_id = self._download_json(
            'https://ani.gamer.com.tw/ajax/getdeviceid.php', video_id,
            'Downloading device ID', 'Failed to download device ID',
            headers=self.geo_verification_headers())['deviceid']
        metadata = {}
        format_id = '0'
        if api_result := self._download_json(
                'https://api.gamer.com.tw/anime/v1/video.php', video_id,
                'Downloading video info', 'Failed to download video info',
                query={'videoSn': video_id}).get('data'):

            metadata.update(traverse_obj(api_result, ('anime', {
                'description': 'content',
                'thumbnail': 'cover',
                'tags': 'tags',
                'creators': ('director', {lambda x: [x]}),
                'title': 'title',
            })))
            playlist_id = traverse_obj(api_result, ('video', 'animeSn')) or ''
            if self._yes_playlist(playlist_id, video_id) and unsmuggled_data.get('extract_playlist') is not False:
                return self.playlist_result(
                    (self.url_result(
                        smuggle_url(f'https://ani.gamer.com.tw/animeVideo.php?sn={ep["videoSn"]}', {
                            'extract_playlist': False,
                        }),
                        AniGamerIE,
                        video_id=ep['videoSn'], thumbnail=ep['cover']) for ep in traverse_obj(
                            api_result,
                            # This (the first ellipsis) extracts episodes of all languages,
                            # maybe just extract episodes of the current language?
                            ('anime', 'episodes', ..., ...))),
                    playlist_id=playlist_id, **metadata)

            # video-specific metadata, extract after returning the playlist result
            metadata.update(traverse_obj(api_result, ('video', {
                'thumbnail': 'cover',
                'title': 'title',
                'timestamp': ('upTime', {unified_timestamp}),
                'duration': ('duration', {float_or_none}, {lambda x: x * 60}),
                'age_limit': ('rating', {lambda x: self.RATING_TO_AGE_LIMIT.get(x)}),
            })))
            format_id = traverse_obj(api_result, ('video', 'quality'), default=format_id)

        m3u8_info = self._download_json('https://ani.gamer.com.tw/ajax/m3u8.php', video_id, query={
            'sn': video_id,
            'device': device_id,
        }, headers=self.geo_verification_headers(), expected_status=400)

        error_code = traverse_obj(m3u8_info, ('error', 'code'))
        if error_code == 1011:
            self.raise_geo_restricted()
        # TODO: handle more error codes
        src = m3u8_info['src']
        return {
            **metadata,
            'id': video_id,
            'formats': [{
                'format_id': format_id,
                'url': src,
                'ext': 'mp4',
            }],
        }