import re

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

_url_re = re.compile(r"https?://(\w+\.)?streamate\.com/cam/(?P<username>[a-zA-Z0-9_-]+)")

class Streamate(Plugin):
    @classmethod
    def can_handle_url(cls, url):
        return _url_re.match(url)

    def _get_streams(self):
        match = _url_re.match(self.url)
        username = match.group("username")
        content_json = "https://manifest-server.naiadsystems.com/live/s:{0}.json".format(username)
        headers = {
            "Content-Type": "application/x-www-form-urlencoded",
            "X-Requested-With": "XMLHttpRequest",
            "Referer": self.url,
        }

        res = self.session.http.get(content_json, headers=headers, raise_for_status=False)
        if res.status_code == 404:
            self.logger.error("Model is offline or nickname is incorrect or non exist")
            return
        elif res.status_code == 403:
            self.logger.error("Model is in private or access denied")
            return
        else:
            data = self.session.http.json(res)

        server = "https://manifest-server.naiadsystems.com/live/{0}.m3u8".format(data["formats"]["webrtc"]["origin"]["publicKey"])

        self.logger.debug("Key: {0}".format(data["formats"]["webrtc"]["origin"]["publicKey"]))

        if server:
            for s in HLSStream.parse_variant_playlist(self.session,server,headers={'Referer': self.url}).items():
                yield s


__plugin__ = Streamate

@echo off

set /p name=nickname: 

set quality=best
set dir=D:\Records
set seconds=30

:loop
cls
echo www.streamate.com
echo %name%
echo ====================

for /f "delims=" %%a in ('curl -ksS -w %%{size_download} https://manifest-server.naiadsystems.com/live/s:%name%.json -o nul') do set size=%%a
if %size% equ 0 (title OFF & echo Offline & timeout %seconds% & goto loop)

title Record - www.streamate.com - %name%

for /f "delims=" %%a in ('curl -ksS https://manifest-server.naiadsystems.com/live/s:%name%.json ^| jq -r .self') do set self=%%a
for /f "tokens=1 delims=?" %%a in ("%self%") do set link=%%a

set URL=%link:~0,-5%.m3u8

streamlink --hls-segment-threads 3 %URL% %quality% -o "%dir%\Streamate_%name%_{time}.ts"

timeout %seconds%
goto loop