# VERSION: 1.2
# AUTHORS: LightDestory (https://github.com/LightDestory) and Snake

import re
from time import sleep
from helpers import retrieve_url
from novaprinter import prettyPrinter


class kickasstorrents(object):
    url = 'https://katcr.to/'
    name = 'Kickasstorrents'

    supported_categories = {'all': '', 'movies': 'movies', 'tv': 'tv', 'music': 'music', 'games': 'games', 'anime': 'anime', 'software': 'apps'}
    class HTMLParser:

        def __init__(self, url):
            self.url = url
            self.noTorrents = False

        def feed(self, html):
            self.noTorrents = False
            torrents = self.__findTorrents(html)
            if len(torrents) == 0:
                self.noTorrents = True
                return

        def __findTorrents(self, html):
            # Find all TR nodes with class odd or even
            trs = re.findall(r'<tr class=\"(?:odd|even)\"\s*>.*?</tr>', html, re.DOTALL)
            for tr in trs:
                url_titles = re.search(r'<div class="torrentname">.*?<a href="([^"]+)"\s+class="cellMainLink">\s*(.*?)\s*</a>.*?<td[^>]*>\s*([\d\.]+\s*(?:TB|GB|MB|KB))\s*</td>.*?<td class="green center">\s*(\d+)\s*</td>.*?<td class="red lasttd center">\s*(\d+)\s*</td>', tr, re.DOTALL)
                if url_titles:
                    detail_link = '{0}{1}'.format(self.url, url_titles.group(1))
                    download_link = self.__retrieve_download_link(detail_link)
                    data = {
                        'link': download_link,
                        'name': url_titles.group(2),
                        'size': url_titles.group(3).replace(",", ""),
                        'seeds': url_titles.group(4).replace(",", ""),
                        'leech': url_titles.group(5).replace(",", ""),
                        'engine_url': self.url,
                        'desc_link': detail_link,
                    }
                    prettyPrinter(data)
                    sleep(1)
            return trs

        def __retrieve_download_link(self, detail_link):
            torrent_page = retrieve_url(detail_link)
            magnet_match = re.search(r'\"(magnet:.*?)\"', torrent_page)
            if magnet_match and magnet_match.groups():
                return str(magnet_match.groups()[0])
            else:
                return "NotFound"

    def search(self, what, cat='all'):
        parser = self.HTMLParser(self.url)
        category = "" if cat == "all" else 'category/{0}/'.format(self.supported_categories[cat])
        counter: int = 0
        while True:
            url = '{0}search/{1}/{2}{3}/'.format(self.url, what, category, counter)
            # Some replacements to format the html source
            html = retrieve_url(url)
            html = re.sub("<strong[^>]*>|</strong>", "", html)
            parser.feed(html)
            if parser.noTorrents:
                print("no torrents")
                break
            counter += 1