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

import re

from core import httptools
from core import scrapertools
from core import tmdb
from core.item import Item
from lib import jsunpack
from platformcode import config, logger

host = 'http://pelisencasa.net'

tgenero = {"Comedia": "https://s7.postimg.org/ne9g9zgwb/comedia.png",
           "Suspense": "https://s13.postimg.org/wmw6vl1cn/suspenso.png",
           "Drama": "https://s16.postimg.org/94sia332d/drama.png",
           "Acción": "https://s3.postimg.org/y6o9puflv/accion.png",
           "Aventura": "https://s10.postimg.org/6su40czih/aventura.png",
           "Romance": "https://s15.postimg.org/fb5j8cl63/romance.png",
           "Animación": "https://s13.postimg.org/5on877l87/animacion.png",
           "Ciencia ficción": "https://s9.postimg.org/diu70s7j3/cienciaficcion.png",
           "Terror": "https://s7.postimg.org/yi0gij3gb/terror.png",
           "Documental": "https://s16.postimg.org/7xjj4bmol/documental.png",
           "Música": "https://s29.postimg.org/bbxmdh9c7/musical.png",
           "Western": "https://s23.postimg.org/lzyfbjzhn/western.png",
           "Fantasía": "https://s13.postimg.org/65ylohgvb/fantasia.png",
           "Guerra": "https://s23.postimg.org/71itp9hcr/belica.png",
           "Misterio": "https://s1.postimg.org/w7fdgf2vj/misterio.png",
           "Crimen": "https://s4.postimg.org/6z27zhirx/crimen.png",
           "Historia": "https://s15.postimg.org/fmc050h1n/historia.png",
           "Familia": "https://s7.postimg.org/6s7vdhqrf/familiar.png"}

tletras = {'#': 'https://s32.postimg.org/drojt686d/image.png',
           'a': 'https://s32.postimg.org/llp5ekfz9/image.png',
           'b': 'https://s32.postimg.org/y1qgm1yp1/image.png',
           'c': 'https://s32.postimg.org/vlon87gmd/image.png',
           'd': 'https://s32.postimg.org/3zlvnix9h/image.png',
           'e': 'https://s32.postimg.org/bgv32qmsl/image.png',
           'f': 'https://s32.postimg.org/y6u7vq605/image.png',
           'g': 'https://s32.postimg.org/9237ib6jp/image.png',
           'h': 'https://s32.postimg.org/812yt6pk5/image.png',
           'i': 'https://s32.postimg.org/6nbbxvqat/image.png',
           'j': 'https://s32.postimg.org/axpztgvdx/image.png',
           'k': 'https://s32.postimg.org/976yrzdut/image.png',
           'l': 'https://s32.postimg.org/fmal2e9yd/image.png',
           'm': 'https://s32.postimg.org/m19lz2go5/image.png',
           'n': 'https://s32.postimg.org/b2ycgvs2t/image.png',
           'o': 'https://s32.postimg.org/c6igsucpx/image.png',
           'p': 'https://s32.postimg.org/jnro82291/image.png',
           'q': 'https://s32.postimg.org/ve5lpfv1h/image.png',
           'r': 'https://s32.postimg.org/nmovqvqw5/image.png',
           's': 'https://s32.postimg.org/zd2t89jol/image.png',
           't': 'https://s32.postimg.org/wk9lo8jc5/image.png',
           'u': 'https://s32.postimg.org/w8s5bh2w5/image.png',
           'v': 'https://s32.postimg.org/e7dlrey91/image.png',
           'w': 'https://s32.postimg.org/fnp49k15x/image.png',
           'x': 'https://s32.postimg.org/dkep1w1d1/image.png',
           'y': 'https://s32.postimg.org/um7j3zg85/image.png',
           'z': 'https://s32.postimg.org/jb4vfm9d1/image.png'}


def mainlist(item):
    logger.info()

    itemlist = []

    itemlist.append(item.clone(title="Todas", action="lista", thumbnail='https://s18.postimg.org/fwvaeo6qh/todas.png',
                               fanart='https://s18.postimg.org/fwvaeo6qh/todas.png', url=host))

    itemlist.append(
        item.clone(title="Generos", action="seccion", thumbnail='https://s3.postimg.org/5s9jg2wtf/generos.png',
                   fanart='https://s3.postimg.org/5s9jg2wtf/generos.png', url=host, extra='generos'))

    itemlist.append(
        item.clone(title="Alfabetico", action="seccion", thumbnail='https://s17.postimg.org/fwi1y99en/a-z.png',
                   fanart='https://s17.postimg.org/fwi1y99en/a-z.png', url=host, extra='letras'))

    itemlist.append(item.clone(title="Buscar", action="search", url=host + '/?s=',
                               thumbnail='https://s30.postimg.org/pei7txpa9/buscar.png',
                               fanart='https://s30.postimg.org/pei7txpa9/buscar.png'))

    return itemlist


def lista(item):
    logger.info()

    itemlist = []
    data = httptools.downloadpage(item.url).data
    if item.extra != 'letras':
        patron = '<li class="TPostMv">.*?<a href="(.*?)"><div class="Image">.*?src="(.*?)\?resize=.*?".*?class="Title">(.*?)<\/h2>.*?'
        patron += '<span class="Year">(.*?)<\/span>.*?<span class="Qlty">(.*?)<\/span><\/p><div class="Description"><p>(.*?)<\/p>'
    else:
        patron = '<td class="MvTbImg"> <a href="(.*?)".*?src="(.*?)\?resize=.*?".*?<strong>(.*?)<\/strong> <\/a><\/td><td>(.*?)<\/td><td>.*?'
        patron += 'class="Qlty">(.*?)<\/span><\/p><\/td><td>(.*?)<\/td><td>'
    matches = re.compile(patron, re.DOTALL).findall(data)

    for scrapedurl, scrapedthumbnail, scrapedtitle, scrapedyear, calidad, scrapedplot in matches:
        url = scrapedurl
        thumbnail = scrapedthumbnail
        plot = scrapedplot
        contentTitle = scrapedtitle
        title = contentTitle + ' (' + calidad + ')'
        year = scrapedyear
        fanart = ''
        itemlist.append(
            Item(channel=item.channel, action='findvideos', title=title, url=url, thumbnail=thumbnail, plot=plot,
                 fanart=fanart, contentTitle=contentTitle, infoLabels={'year': year}))
    tmdb.set_infoLabels_itemlist(itemlist, seekTmdb=True)
    # Paginacion

    if itemlist != []:
        actual_page_url = item.url
        next_page = scrapertools.find_single_match(data, '<a class="nextpostslink" rel="next" href="(.*?)">')
        if next_page != '':
            itemlist.append(Item(channel=item.channel, action="lista", title='Siguiente >>>', url=next_page,
                                 thumbnail='https://s16.postimg.org/9okdu7hhx/siguiente.png'))
    return itemlist


def seccion(item):
    logger.info()
    itemlist = []
    data = httptools.downloadpage(item.url).data
    if item.extra == 'generos':
        patron = 'menu-item-object-category menu-item-.*?"><a href="(.*?)">(.*?)<\/a><\/li>'
    else:
        patron = '<li><a href="(.*?\/letter\/.*?)">(.*?)<\/a><\/li>'

    matches = re.compile(patron, re.DOTALL).findall(data)

    for scrapedurl, scrapedtitle in matches:
        thumbnail = ''
        if item.extra == 'generos' and scrapedtitle in tgenero:
            thumbnail = tgenero[scrapedtitle]
        elif scrapedtitle.lower() in tletras:
            thumbnail = tletras[scrapedtitle.lower()]
        fanart = ''
        title = scrapedtitle
        url = scrapedurl

        itemlist.append(
            Item(channel=item.channel, action="lista", title=title, fulltitle=item.title, url=url, thumbnail=thumbnail,
                 fanart=fanart, extra=item.extra))
    return itemlist


def findvideos(item):
    logger.info()
    itemlist = []
    data = httptools.downloadpage(item.url).data
    patron = 'class="Num">.*?<\/span>.*?href="(.*?)" class="Button STPb">.*?<\/a>.*?<span>(.*?)<\/span><\/td><td><span>(.*?)<\/span><\/td><td><span>.*?<\/span>'
    matches = re.compile(patron, re.DOTALL).findall(data)
    infoLabels = item.infoLabels
    for scrapedurl, servidor, idioma in matches:
        new_item = (item.clone(url=scrapedurl, servidor=servidor, idioma=idioma, infoLabels=infoLabels))
        itemlist += get_video_urls(new_item)

    if config.get_videolibrary_support() and len(itemlist) > 0 and item.extra != 'findvideos':
        itemlist.append(
            Item(channel=item.channel, title='[COLOR yellow]Añadir esta pelicula a la videoteca[/COLOR]', url=item.url,
                 action="add_pelicula_to_library", extra="findvideos", contentTitle=item.contentTitle))

    itemlist.insert(len(itemlist) - 1, item.clone(channel='trailertools', action='buscartrailer',
                                                  title='[COLOR orange]Trailer en Youtube[/COLOR]'))

    return itemlist


def get_video_urls(item):
    logger.info()
    itemlist = []
    data = httptools.downloadpage(item.url).data
    data = scrapertools.find_single_match(data, '<script type="text\/javascript">(.*?)<\/script>')
    data = jsunpack.unpack(data)
    patron = '"file":"(.*?)","label":"(.*?)","type":"video.*?"}'
    subtitle = scrapertools.find_single_match(data, 'tracks:\[{"file":"(.*?)","label":".*?","kind":"captions"}')
    matches = re.compile(patron, re.DOTALL).findall(data)
    for url, calidad in matches:
        if item.servidor == 'PELISENCASA':
            item.servidor = 'Directo'
        title = item.contentTitle + ' (' + item.idioma + ')' + ' (' + calidad + ')' + ' (' + item.servidor + ')'
        itemlist.append(item.clone(title=title, url=url, calidad=calidad, action='play', subtitle=subtitle))

    return itemlist


def search(item, texto):
    logger.info()
    texto = texto.replace(" ", "+")
    item.url = item.url + texto

    if texto != '':
        return lista(item)
    else:
        return []


def newest(categoria):
    logger.info()
    itemlist = []
    item = Item()
    try:
        if categoria == 'peliculas':
            item.url = host
        elif categoria == 'infantiles':
            item.url = host + '/category/animacion/'
        itemlist = lista(item)
        if itemlist[-1].title == 'Siguiente >>>':
            itemlist.pop()
    except:
        import sys
        for line in sys.exc_info():
            logger.error("{0}".format(line))
        return []

    return itemlist