camsoda.py

import random
import re
import json

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

_url_re = re.compile(r"http(s)?://(www\.)?camsoda\.com/(?P<username>[^\"\']+)")

_api_user_schema = validate.Schema(
    {
        "status": validate.any(int, validate.text),
        validate.optional("user"): validate.Schema({
            "chat": validate.Schema ({
                    "status": validate.any(int, validate.text)
            })
        })
    }
)

_api_video_schema = validate.Schema(
    {
        "token": validate.text,
        "edge_servers": [validate.text],
        "stream_name": validate.text
    }
)


class Camsoda(Plugin):
    API_URL_VIDEO = "https://www.camsoda.com/api/v1/video/vtoken/{0}?username=guest_{1}"
    HLS_URL_VIDEO_EDGE = "https://{server}/{stream_name}_v1/index.m3u8?token={token}"
    HLS_URL_VIDEO = "https://{server}/mp4:{stream_name}_aac/playlist.m3u8?token={token}"
    headers = {
        "User-Agent": useragents.FIREFOX
    }

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

    def _stream_status(self, data_user):

        invalid_username = data_user["status"] is False
        if invalid_username:
            self.logger.info("No validate username found for {0}".format(self.url))
            return

        is_online = data_user["status"] is True and data_user["user"]["chat"]["status"] == "online"
        if is_online is False:
            self.logger.info("Stream is currently offline or private")
            return

        return True

    def _get_api_video(self, username):
        res = http.get(self.API_URL_VIDEO.format(username, str(random.randint(1000, 99999))), headers=self.headers, verify=False)
        data_video = http.json(res, schema=_api_video_schema)
        return data_video

    def _get_streams(self):
        match = _url_re.match(self.url)
        username = match.group("username")
        username = username.replace("/", "")

        data_video = self._get_api_video(username)

        if data_video["edge_servers"]:
            hls_url = self.HLS_URL_VIDEO.format(
                server=data_video["edge_servers"][0],
                stream_name=data_video["stream_name"],
                token=data_video["token"]
            )
            if "edge" in data_video["edge_servers"][0]:
                self.session.http.verify = False
                hls_url = self.HLS_URL_VIDEO_EDGE.format(
                    server=data_video["edge_servers"][0],
                    stream_name=data_video["stream_name"],
                    token=data_video["token"]
                )

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


__plugin__ = Camsoda