import requests
from nyaapy import utils
from nyaapy import torrent


class Nyaa:

    def __init__(self):
        self.SITE = utils.TorrentSite.NYAASI
        self.URL = "https://nyaa.si"

    def last_uploads(self, number_of_results):
        r = requests.get(self.URL)

        # If anything up with nyaa servers let the user know.
        r.raise_for_status()

        json_data = utils.parse_nyaa(
            request_text=r.text, limit=number_of_results + 1, site=self.SITE
        )

        return torrent.json_to_class(json_data)

    def search(self, keyword, **kwargs):
        base_url = self.URL

        user = kwargs.get("user", None)
        category = kwargs.get("category", 0)
        subcategory = kwargs.get("subcategory", 0)
        filters = kwargs.get("filters", 0)
        page = kwargs.get("page", 0)
        sorting = kwargs.get(
            "sort", "id"
        )  # Sorting by id = sorting by date, this is the default.
        order = kwargs.get("order", "desc")

        user_uri = f"user/{user}" if user else ""

        if page > 0:
            search_uri = "{}/{}?f={}&c={}_{}&q={}&p={}&s={}&o={}".format(
                base_url,
                user_uri,
                filters,
                category,
                subcategory,
                keyword,
                page,
                sorting,
                order,
            )
        else:
            search_uri = "{}/{}?f={}&c={}_{}&q={}&s={}&o={}".format(
                base_url,
                user_uri,
                filters,
                category,
                subcategory,
                keyword,
                sorting,
                order,
            )

        if not user:
            search_uri += "&page=rss"

        http_response = requests.get(search_uri)
        http_response.raise_for_status()

        if user:
            json_data = utils.parse_nyaa(
                request_text=http_response.text, limit=None, site=self.SITE
            )
        else:
            json_data = utils.parse_nyaa_rss(
                request_text=http_response.text, limit=None, site=self.SITE
            )

        # Convert JSON data to a class object
        return torrent.json_to_class(json_data)

    def get(self, view_id):
        r = requests.get(f"{self.URL}/view/{view_id}")
        r.raise_for_status()

        json_data = utils.parse_single(request_text=r.text, site=self.SITE)

        return torrent.json_to_class(json_data)

    def get_user(self, username):
        r = requests.get(f"{self.URL}/user/{username}")
        r.raise_for_status()

        json_data = utils.parse_nyaa(request_text=r.text, limit=None, site=self.SITE)
        return torrent.json_to_class(json_data)