# coding: utf-8
from __future__ import unicode_literals

import re

from .common import InfoExtractor
from ..compat import compat_xpath
from ..utils import (
    determine_ext,
    ExtractorError,
    int_or_none,
    url_or_none,
    urlencode_postdata,
    xpath_text,
)


class AfreecaTVIE(InfoExtractor):
    IE_NAME = "afreecatv"
    IE_DESC = "afreecatv.com"
    _VALID_URL = r"""(?x)
                    https?://
                        (?:
                            (?:(?:live|afbbs|www)\.)?afreeca(?:tv)?\.com(?::\d+)?
                            (?:
                                /app/(?:index|read_ucc_bbs)\.cgi|
                                /player/[Pp]layer\.(?:swf|html)
                            )\?.*?\bnTitleNo=|
                            vod\.afreecatv\.com/PLAYER/STATION/
                        )
                        (?P<id>\d+)
                    """
    _NETRC_MACHINE = "afreecatv"
    _TESTS = [
        {
            "url": "http://live.afreecatv.com:8079/app/index.cgi?szType=read_ucc_bbs&szBjId=dailyapril&nStationNo=16711924&nBbsNo=18605867&nTitleNo=36164052&szSkin=",
            "md5": "f72c89fe7ecc14c1b5ce506c4996046e",
            "info_dict": {
                "id": "36164052",
                "ext": "mp4",
                "title": "데일리 에이프릴 요정들의 시상식!",
                "thumbnail": "re:^https?://(?:video|st)img.afreecatv.com/.*$",
                "uploader": "dailyapril",
                "uploader_id": "dailyapril",
                "upload_date": "20160503",
            },
            "skip": "Video is gone",
        },
        {
            "url": "http://afbbs.afreecatv.com:8080/app/read_ucc_bbs.cgi?nStationNo=16711924&nTitleNo=36153164&szBjId=dailyapril&nBbsNo=18605867",
            "info_dict": {
                "id": "36153164",
                "title": "BJ유트루와 함께하는 '팅커벨 메이크업!'",
                "thumbnail": "re:^https?://(?:video|st)img.afreecatv.com/.*$",
                "uploader": "dailyapril",
                "uploader_id": "dailyapril",
            },
            "playlist_count": 2,
            "playlist": [
                {
                    "md5": "d8b7c174568da61d774ef0203159bf97",
                    "info_dict": {
                        "id": "36153164_1",
                        "ext": "mp4",
                        "title": "BJ유트루와 함께하는 '팅커벨 메이크업!'",
                        "upload_date": "20160502",
                    },
                },
                {
                    "md5": "58f2ce7f6044e34439ab2d50612ab02b",
                    "info_dict": {
                        "id": "36153164_2",
                        "ext": "mp4",
                        "title": "BJ유트루와 함께하는 '팅커벨 메이크업!'",
                        "upload_date": "20160502",
                    },
                },
            ],
            "skip": "Video is gone",
        },
        {
            "url": "http://vod.afreecatv.com/PLAYER/STATION/18650793",
            "info_dict": {
                "id": "18650793",
                "ext": "mp4",
                "title": "오늘은 다르다! 쏘님의 우월한 위아래~ 댄스리액션!",
                "thumbnail": r"re:^https?://.*\.jpg$",
                "uploader": "윈아디",
                "uploader_id": "badkids",
                "duration": 107,
            },
            "params": {
                "skip_download": True,
            },
        },
        {
            "url": "http://vod.afreecatv.com/PLAYER/STATION/10481652",
            "info_dict": {
                "id": "10481652",
                "title": "BJ유트루와 함께하는 '팅커벨 메이크업!'",
                "thumbnail": "re:^https?://(?:video|st)img.afreecatv.com/.*$",
                "uploader": "dailyapril",
                "uploader_id": "dailyapril",
                "duration": 6492,
            },
            "playlist_count": 2,
            "playlist": [
                {
                    "md5": "d8b7c174568da61d774ef0203159bf97",
                    "info_dict": {
                        "id": "20160502_c4c62b9d_174361386_1",
                        "ext": "mp4",
                        "title": "BJ유트루와 함께하는 '팅커벨 메이크업!' (part 1)",
                        "thumbnail": "re:^https?://(?:video|st)img.afreecatv.com/.*$",
                        "uploader": "dailyapril",
                        "uploader_id": "dailyapril",
                        "upload_date": "20160502",
                        "duration": 3601,
                    },
                },
                {
                    "md5": "58f2ce7f6044e34439ab2d50612ab02b",
                    "info_dict": {
                        "id": "20160502_39e739bb_174361386_2",
                        "ext": "mp4",
                        "title": "BJ유트루와 함께하는 '팅커벨 메이크업!' (part 2)",
                        "thumbnail": "re:^https?://(?:video|st)img.afreecatv.com/.*$",
                        "uploader": "dailyapril",
                        "uploader_id": "dailyapril",
                        "upload_date": "20160502",
                        "duration": 2891,
                    },
                },
            ],
            "params": {
                "skip_download": True,
            },
        },
        {
            # non standard key
            "url": "http://vod.afreecatv.com/PLAYER/STATION/20515605",
            "info_dict": {
                "id": "20170411_BE689A0E_190960999_1_2_h",
                "ext": "mp4",
                "title": "혼자사는여자집",
                "thumbnail": "re:^https?://(?:video|st)img.afreecatv.com/.*$",
                "uploader": "♥이슬이",
                "uploader_id": "dasl8121",
                "upload_date": "20170411",
                "duration": 213,
            },
            "params": {
                "skip_download": True,
            },
        },
        {
            # PARTIAL_ADULT
            "url": "http://vod.afreecatv.com/PLAYER/STATION/32028439",
            "info_dict": {
                "id": "20180327_27901457_202289533_1",
                "ext": "mp4",
                "title": "[생]빨개요♥ (part 1)",
                "thumbnail": "re:^https?://(?:video|st)img.afreecatv.com/.*$",
                "uploader": "[SA]서아",
                "uploader_id": "bjdyrksu",
                "upload_date": "20180327",
                "duration": 3601,
            },
            "params": {
                "skip_download": True,
            },
            "expected_warnings": ["adult content"],
        },
        {
            "url": "http://www.afreecatv.com/player/Player.swf?szType=szBjId=djleegoon&nStationNo=11273158&nBbsNo=13161095&nTitleNo=36327652",
            "only_matching": True,
        },
        {
            "url": "http://vod.afreecatv.com/PLAYER/STATION/15055030",
            "only_matching": True,
        },
    ]

    @staticmethod
    def parse_video_key(key):
        video_key = {}
        m = re.match(r"^(?P<upload_date>\d{8})_\w+_(?P<part>\d+)$", key)
        if m:
            video_key["upload_date"] = m.group("upload_date")
            video_key["part"] = int(m.group("part"))
        return video_key

    def _real_initialize(self):
        self._login()

    def _login(self):
        username, password = self._get_login_info()
        if username is None:
            return

        login_form = {
            "szWork": "login",
            "szType": "json",
            "szUid": username,
            "szPassword": password,
            "isSaveId": "false",
            "szScriptVar": "oLoginRet",
            "szAction": "",
        }

        response = self._download_json(
            "https://login.afreecatv.com/app/LoginAction.php",
            None,
            "Logging in",
            data=urlencode_postdata(login_form),
        )

        _ERRORS = {
            -4: "Your account has been suspended due to a violation of our terms and policies.",
            -5: "https://member.afreecatv.com/app/user_delete_progress.php",
            -6: "https://login.afreecatv.com/membership/changeMember.php",
            -8: "Hello! AfreecaTV here.\nThe username you have entered belongs to \n an account that requires a legal guardian's consent. \nIf you wish to use our services without restriction, \nplease make sure to go through the necessary verification process.",
            -9: "https://member.afreecatv.com/app/pop_login_block.php",
            -11: "https://login.afreecatv.com/afreeca/second_login.php",
            -12: "https://member.afreecatv.com/app/user_security.php",
            0: "The username does not exist or you have entered the wrong password.",
            -1: "The username does not exist or you have entered the wrong password.",
            -3: "You have entered your username/password incorrectly.",
            -7: "You cannot use your Global AfreecaTV account to access Korean AfreecaTV.",
            -10: "Sorry for the inconvenience. \nYour account has been blocked due to an unauthorized access. \nPlease contact our Help Center for assistance.",
            -32008: "You have failed to log in. Please contact our Help Center.",
        }

        result = int_or_none(response.get("RESULT"))
        if result != 1:
            error = _ERRORS.get(result, "You have failed to log in.")
            raise ExtractorError(
                "Unable to login: %s said: %s" % (self.IE_NAME, error), expected=True
            )

    def _real_extract(self, url):
        video_id = self._match_id(url)

        webpage = self._download_webpage(url, video_id)

        if re.search(r'alert\(["\']This video has been deleted', webpage):
            raise ExtractorError("Video %s has been deleted" % video_id, expected=True)

        station_id = self._search_regex(r"nStationNo\s*=\s*(\d+)", webpage, "station")
        bbs_id = self._search_regex(r"nBbsNo\s*=\s*(\d+)", webpage, "bbs")
        video_id = self._search_regex(
            r"nTitleNo\s*=\s*(\d+)", webpage, "title", default=video_id
        )

        partial_view = False
        for _ in range(2):
            query = {
                "nTitleNo": video_id,
                "nStationNo": station_id,
                "nBbsNo": bbs_id,
            }
            if partial_view:
                query["partialView"] = "SKIP_ADULT"
            video_xml = self._download_xml(
                "http://afbbs.afreecatv.com:8080/api/video/get_video_info.php",
                video_id,
                "Downloading video info XML%s"
                % (" (skipping adult)" if partial_view else ""),
                video_id,
                headers={
                    "Referer": url,
                },
                query=query,
            )

            flag = xpath_text(video_xml, "./track/flag", "flag", default=None)
            if flag and flag == "SUCCEED":
                break
            if flag == "PARTIAL_ADULT":
                self._downloader.report_warning(
                    "In accordance with local laws and regulations, underage users are restricted from watching adult content. "
                    "Only content suitable for all ages will be downloaded. "
                    "Provide account credentials if you wish to download restricted content."
                )
                partial_view = True
                continue
            elif flag == "ADULT":
                error = "Only users older than 19 are able to watch this video. Provide account credentials to download this content."
            else:
                error = flag
            raise ExtractorError("%s said: %s" % (self.IE_NAME, error), expected=True)
        else:
            raise ExtractorError("Unable to download video info")

        video_element = video_xml.findall(compat_xpath("./track/video"))[-1]
        if video_element is None or video_element.text is None:
            raise ExtractorError("Video %s does not exist" % video_id, expected=True)

        video_url = video_element.text.strip()

        title = xpath_text(video_xml, "./track/title", "title", fatal=True)

        uploader = xpath_text(video_xml, "./track/nickname", "uploader")
        uploader_id = xpath_text(video_xml, "./track/bj_id", "uploader id")
        duration = int_or_none(xpath_text(video_xml, "./track/duration", "duration"))
        thumbnail = xpath_text(video_xml, "./track/titleImage", "thumbnail")

        common_entry = {
            "uploader": uploader,
            "uploader_id": uploader_id,
            "thumbnail": thumbnail,
        }

        info = common_entry.copy()
        info.update(
            {
                "id": video_id,
                "title": title,
                "duration": duration,
            }
        )

        if not video_url:
            entries = []
            file_elements = video_element.findall(compat_xpath("./file"))
            one = len(file_elements) == 1
            for file_num, file_element in enumerate(file_elements, start=1):
                file_url = url_or_none(file_element.text)
                if not file_url:
                    continue
                key = file_element.get("key", "")
                upload_date = self._search_regex(
                    r"^(\d{8})_", key, "upload date", default=None
                )
                file_duration = int_or_none(file_element.get("duration"))
                format_id = key if key else "%s_%s" % (video_id, file_num)
                if determine_ext(file_url) == "m3u8":
                    formats = self._extract_m3u8_formats(
                        file_url,
                        video_id,
                        "mp4",
                        entry_protocol="m3u8_native",
                        m3u8_id="hls",
                        note="Downloading part %d m3u8 information" % file_num,
                    )
                else:
                    formats = [
                        {
                            "url": file_url,
                            "format_id": "http",
                        }
                    ]
                if not formats:
                    continue
                self._sort_formats(formats)
                file_info = common_entry.copy()
                file_info.update(
                    {
                        "id": format_id,
                        "title": title if one else "%s (part %d)" % (title, file_num),
                        "upload_date": upload_date,
                        "duration": file_duration,
                        "formats": formats,
                    }
                )
                entries.append(file_info)
            entries_info = info.copy()
            entries_info.update(
                {
                    "_type": "multi_video",
                    "entries": entries,
                }
            )
            return entries_info

        info = {
            "id": video_id,
            "title": title,
            "uploader": uploader,
            "uploader_id": uploader_id,
            "duration": duration,
            "thumbnail": thumbnail,
        }

        if determine_ext(video_url) == "m3u8":
            info["formats"] = self._extract_m3u8_formats(
                video_url, video_id, "mp4", entry_protocol="m3u8_native", m3u8_id="hls"
            )
        else:
            app, playpath = video_url.split("mp4:")
            info.update(
                {
                    "url": app,
                    "ext": "flv",
                    "play_path": "mp4:" + playpath,
                    "rtmp_live": True,  # downloading won't end without this
                }
            )

        return info
