Social Media Girls

import re

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


class Bigo(Plugin):
    _url_re = re.compile(r"https?://(?:www\.)?bigo\.tv/([^/]+)$")
    _api_url = "https://ta.bigo.tv/official_website/studio/getInternalStudioInfo?siteId={0}"

    _video_info_schema = validate.Schema({
        "code": 0,
        "msg": "success",
        "data": {
            validate.optional("videoSrc"): validate.any(None, "", validate.url()),
            validate.optional("hls_src"): validate.any(None, "", validate.url()),
            validate.optional("cdn_src"): object
        }
    })

    @classmethod
    def can_handle_url(cls, url):
        return cls._url_re.match(url) is not None

    def _get_streams(self):
        match = self._url_re.match(self.url)
        res = self.session.http.post(
            self._api_url.format(match.group(1)),
            allow_redirects=True,
            headers={"User-Agent": useragents.IPHONE_6}
        )
        data = self.session.http.json(res, schema=self._video_info_schema)

        # 🔥 แก้ตรงนี้
        videourl = (
            data["data"].get("hls_src")
            or data["data"].get("videoSrc")
            or (data["data"].get("cdn_src")[0] if data["data"].get("cdn_src") else None)
        )

        if videourl:
            yield "live", HLSStream(self.session, videourl)


__plugin__ = Bigo