import re
from .common import InfoExtractor
from ..utils import (
    determine_ext,
    extract_attributes,
    int_or_none,
    traverse_obj,
    unified_strdate,
)


class FranceCultureIE(InfoExtractor):
    _VALID_URL = r'https?://(?:www\.)?franceculture\.fr/emissions/(?:[^/]+/)*(?P<id>[^/?#&]+)'
    _TESTS = [{
        # playlist
        'url': 'https://www.franceculture.fr/emissions/serie/hasta-dente',
        'playlist_count': 12,
        'info_dict': {
            'id': 'hasta-dente',
            'title': 'Hasta Dente',
            'description': 'md5:57479af50648d14e9bb649e6b1f8f911',
            'thumbnail': r're:^https?://.*\.jpg$',
            'upload_date': '20201024',
        },
        'playlist': [{
            'info_dict': {
                'id': '3c1c2e55-41a0-11e5-9fe0-005056a87c89',
                'ext': 'mp3',
                'title': 'Jeudi, vous avez dit bizarre ?',
                'description': 'md5:47cf1e00cc21c86b0210279996a812c6',
                'duration': 604,
                'upload_date': '20201024',
                'thumbnail': r're:^https?://.*\.jpg$',
                'timestamp': 1603576680
            },
        },
        ],
    }, {
        'url': 'https://www.franceculture.fr/emissions/carnet-nomade/rendez-vous-au-pays-des-geeks',
        'info_dict': {
            'id': 'rendez-vous-au-pays-des-geeks',
            'display_id': 'rendez-vous-au-pays-des-geeks',
            'ext': 'mp3',
            'title': 'Rendez-vous au pays des geeks',
            'thumbnail': r're:^https?://.*\.jpg$',
            'upload_date': '20140301',
            'vcodec': 'none',
            'duration': 3569,
        },
    }, {
        # no thumbnail
        'url': 'https://www.franceculture.fr/emissions/la-recherche-montre-en-main/la-recherche-montre-en-main-du-mercredi-10-octobre-2018',
        'only_matching': True,
    }]

    def _real_extract(self, url):
        display_id = self._match_id(url)
        webpage = self._download_webpage(url, display_id)

        info = {
            'id': display_id,
            'title': self._html_search_regex(
                r'(?s)<h1[^>]*itemprop="[^"]*name[^"]*"[^>]*>(.+?)</h1>',
                webpage, 'title', default=self._og_search_title(webpage)),
            'description': self._html_search_regex(
                r'(?s)<div[^>]+class="excerpt"[^>]*>(.*?)</div>', webpage, 'description', default=None),
            'thumbnail': self._og_search_thumbnail(webpage),
            'uploader': self._html_search_regex(
                r'(?s)<span class="author">(.*?)</span>', webpage, 'uploader', default=None),
            'upload_date': unified_strdate(self._html_search_regex(
                r'(?s)class="teaser-text-date".*?(\d{2}/\d{2}/\d{4})', webpage, 'date', default=None)),
        }

        playlist_data = self._search_regex(
            r'''(?sx)
                <section[^>]+data-xiti-place="[^"]*?liste_episodes[^"?]*?"[^>]*>
                (.*?)
                </section>
            ''',
            webpage, 'playlist data', fatal=False, default=None)

        if playlist_data:
            entries = []
            for item, item_description in re.findall(
                    r'(?s)(<button[^<]*class="[^"]*replay-button[^>]*>).*?<p[^>]*class="[^"]*teaser-text-chapo[^>]*>(.*?)</p>',
                    playlist_data):

                item_attributes = extract_attributes(item)
                entries.append({
                    'id': item_attributes.get('data-emission-uuid'),
                    'url': item_attributes.get('data-url'),
                    'title': item_attributes.get('data-diffusion-title'),
                    'duration': int_or_none(traverse_obj(item_attributes, 'data-duration-seconds', 'data-duration-seconds')),
                    'description': item_description,
                    'timestamp': int_or_none(item_attributes.get('data-start-time')),
                    'thumbnail': info['thumbnail'],
                    'uploader': info['uploader'],
                })

            return {
                '_type': 'playlist',
                'entries': entries,
                **info
            }

        video_data = extract_attributes(self._search_regex(
            r'''(?sx)
                (?:
                    </h1>|
                    <div[^>]+class="[^"]*?(?:title-zone-diffusion|heading-zone-(?:wrapper|player-button))[^"]*?"[^>]*>
                ).*?
                (<button[^>]+data-(?:url|asset-source)="[^"]+"[^>]+>)
            ''',
            webpage, 'video data'))
        video_url = traverse_obj(video_data, 'data-url', 'data-asset-source')
        ext = determine_ext(video_url.lower())

        return {
            'display_id': display_id,
            'url': video_url,
            'ext': ext,
            'vcodec': 'none' if ext == 'mp3' else None,
            'duration': int_or_none(video_data.get('data-duration')),
            **info
        }