# VERSION: 1.0
# AUTHORS: LightDestory (https://github.com/LightDestory)

import re
import urllib.parse
from helpers import retrieve_url
from novaprinter import prettyPrinter


class nitro(object):
    url = 'https://nitro.to/'
    name = 'Nitro.to'
    max_pages = 10

    class HTMLParser:

        def __init__(self, url):
            self.url = url
            self.pageResSize = 0

        def feed(self, html):
            self.pageResSize = 0
            torrents = self.__findTorrents(html)
            resultSize = len(torrents)
            if resultSize == 0:
                return
            else:
                self.pageResSize = resultSize
                count = 0
            for torrent in range(resultSize):
                count = count + 1
                data = {
                    'link': torrents[torrent][0],
                    'name': torrents[torrent][1],
                    'size': torrents[torrent][2],
                    'seeds': torrents[torrent][3],
                    'leech': torrents[torrent][4],
                    'engine_url': self.url,
                    'desc_link': urllib.parse.unquote(torrents[torrent][0])
                }
                prettyPrinter(data)

        def __findTorrents(self, html):
            torrents = []
            trs = re.findall(
                r'<tr>.*?td.+?center\">\d\d/\d\d/\d\d\d\d.+?</tr>', html)
            for tr in trs:
                url_titles = re.search(
                    r'.+?center\" class=\"lista\".+?href=\"/(.+?)#.+?title=\"(.+?)\".+?lista\">([0-9\,\.]+ (TB|GB|MB|KB)).+?class=\"(?:red|green)\">([0-9,]+).+?class=\"(?:red|green)\">([0-9,]+)',
                    tr)
                if url_titles:
                    torrent_data = [
                        urllib.parse.quote('{0}{1}'.format(self.url, url_titles.group(1))),
                        url_titles.group(2).replace("Przeglądasz detale:", ""),
                        url_titles.group(3).replace(",", ""),
                        url_titles.group(5).replace(",", ""),
                        url_titles.group(6).replace(",", "")
                    ]
                    torrents.append(torrent_data)
            return torrents

    def download_torrent(self, info):
        torrent_id = info.split("_")[-1]
        magnet_raw = retrieve_url(f"https://nitro.to/download_magnet.php?id={torrent_id}")
        magnet_match = re.search(r'\"(magnet:.*?)&dn=\"', magnet_raw)
        if magnet_match and magnet_match.groups():
            print('{0} {1}'.format(magnet_match.groups()[0], info))
        else:
            raise Exception('Error, please fill a bug report!')

    def search(self, what, cat='all'):
        parser = self.HTMLParser(self.url)
        for currPage in range(0, self.max_pages):
            url = '{0}tags.php?search={1}&per=50&by=desc&order=data&page={2}&where=1'.format(self.url, what, currPage)
            # Some replacements to format the html source
            html = retrieve_url(url).replace("	", "").replace("\n", "").replace("\r", "")
            parser.feed(html)
            if parser.pageResSize <= 0:
                break