import re
import uuid

from streamlink.plugin import Plugin, pluginmatcher
from streamlink.plugin.api import validate
from streamlink.stream import HLSStream

API_HLS = "https://chaturbate.com/get_edge_hls_url_ajax/"

_url_re = re.compile(r"https?://(\w+\.)?chaturbate\.com/(?P<username>\w+)")

_post_schema = validate.Schema(
    {
        "url": str,
        "room_status": str,
        "success": int
    }
)

@pluginmatcher(_url_re)
class chaturbate(Plugin):
    def _get_streams(self):
        match = _url_re.match(self.url)
        username = match.group("username")

        CSRFToken = str(uuid.uuid4().hex.upper()[0:32])

        headers = {
            "Content-Type": "application/x-www-form-urlencoded",
            "X-CSRFToken": CSRFToken,
            "X-Requested-With": "XMLHttpRequest",
            "Referer": self.url,
        }

        cookies = {
            "csrftoken": CSRFToken,
        }

        post_data = "room_slug={0}&bandwidth=high".format(username)

        res = self.session.http.post(API_HLS, headers=headers, cookies=cookies, data=post_data)
        data = self.session.http.json(res, schema=_post_schema)

        self.logger.info("Stream status: {0}".format(data["room_status"]))

        if data["success"] and data["room_status"] == "public" and data["url"]:
            updated_url = data["url"]
            updated_url = updated_url.replace("live-hls", "live-c-fhls")
            updated_url = updated_url.replace("playlist.m3u8", "playlist_sfm4s.m3u8")

            self.logger.info(f"Stream URL: {updated_url}")

            for s in HLSStream.parse_variant_playlist(self.session, updated_url).items():
                yield s


__plugin__ = chaturbate