# -*- coding: utf-8 -*-

from core import httptools
from core import scrapertools
from platformcode import config, logger


def test_video_exists(page_url):
    logger.info("(page_url='%s')" % page_url)

    data = httptools.downloadpage(page_url, cookies=False).data
    if 'File you are looking for is not found.' in data:
        return False, config.get_localized_string(70449) % "Onlystream"

    return True, ""


def get_video_url(page_url, premium=False, user="", password="", video_password=""):
    logger.info("url=" + page_url)
    video_urls = []
    data = httptools.downloadpage(page_url).data
    # logger.info(data)
    block = scrapertools.find_single_match(data, r'sources: \[([^\]]+)\]')
    sources = scrapertools.find_multiple_matches(block, r'file:\s*"([^"]+)"(?:,label:\s*"([^"]+)")?')
    if not sources:
        sources = scrapertools.find_multiple_matches(data, r'src:\s*"([^"]+)",\s*type:\s*"[^"]+",[^,]+,\s*label:\s*"([^"]+)"')
    for url, quality in sources:
        quality = 'auto' if not quality else quality
        video_urls.append(['.' + url.split('.')[-1] + ' [' + quality + '] [Onlystream]', url])
    video_urls.sort(key=lambda x: x[0].split()[1])
    return video_urls