# -*- coding: utf-8 -*-
# --------------------------------------------------------
# Conector UpVID By Alfa development Group
# --------------------------------------------------------

import re
import urllib
from core import httptools
from core import scrapertools
from platformcode import logger

import sys, os
import re, base64
from lib.aadecode import decode as aadecode


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

    data = httptools.downloadpage(page_url)

    if data.code == 404:
        return False, "[upvid] El archivo no existe o  ha sido borrado"

    return True, ""


def get_video_url(page_url, premium = False, user = "", password = "", video_password = ""):
    logger.info("url=" + page_url)

    video_urls = []
    headers = {'referer': page_url}

    for i in range(0, 3):
        data = httptools.downloadpage(page_url, headers=headers).data
        data = re.sub(r'"|\n|\r|\t|&nbsp;|<br>|\s{2,}', "", data)
        if '<input type=hidden' in data:
            break
        else:
            page_url = scrapertools.find_single_match(data, "iframe src=(.*?) scrolling")


    # logger.debug(data)
    # decodificar función para obtener función y clave
    # ------------------------------------------------
    code = re.findall('<script>\s*ﾟωﾟ(.*?)</script>', data, flags=re.DOTALL)[0]
    text_decode = aadecode(code)
    funcion, clave = re.findall("func\.innerHTML = (\w*)\('([^']*)', ", text_decode, flags=re.DOTALL)[0]

    # decodificar javascript en campos html hidden
    # --------------------------------------------
    oculto = re.findall('<input type=hidden value=([^ ]+) id=func', data, flags=re.DOTALL)[0]
    funciones = resuelve(clave, base64.b64decode(oculto))

    oculto = re.findall('<input type=hidden value=([^ ]+) id=code', data, flags=re.DOTALL)[0]
    codigo = resuelve(clave, base64.b64decode(oculto))

    url, type = scrapertools.find_single_match(funciones, "setAttribute\('src', '(.*?)'\);\s.*?type', 'video/(.*?)'")

    video_urls.append(['upvid [%s]' % type ,url])

    return video_urls


def resuelve(r, o):
    a = '';
    n = 0
    e = range(256)
    for f in range(256):
        n = (n + e[f] + ord(r[(f % len(r))])) % 256
        t = e[f];
        e[f] = e[n];
        e[n] = t
    f = 0;
    n = 0
    for h in range(len(o)):
        f = (f + 1) % 256
        n = (n + e[f]) % 256
        t = e[f];
        e[f] = e[n];
        e[n] = t
        a += chr(ord(o[h]) ^ e[(e[f] + e[n]) % 256])
    return a