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

# Copyright 2025 Mike Fährmann
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.

"""Extractors for https://imhentai.xxx/"""

from .common import GalleryExtractor, Extractor, Message
from .. import text, util

BASE_PATTERN = r"(?:https?://)?(?:www\.)?imhentai\.xxx"


class ImhentaiExtractor(Extractor):
    category = "imhentai"
    root = "https://imhentai.xxx"

    def _pagination(self, url):
        base = self.root + "/gallery/"
        data = {"_extractor": ImhentaiGalleryExtractor}

        while True:
            page = self.request(url).text
            extr = text.extract_from(page)

            while True:
                gallery_id = extr('<a href="/gallery/', '"')
                if not gallery_id:
                    break
                yield Message.Queue, base + gallery_id, data
                extr('<a href="/gallery/', '"')  # skip duplicate GIDs

            href = text.rextract(page, "class='page-link' href='", "'")[0]
            if not href or href == "#":
                return
            url = text.ensure_http_scheme(href)


class ImhentaiGalleryExtractor(ImhentaiExtractor, GalleryExtractor):
    """Extractor for imhentai galleries"""
    pattern = BASE_PATTERN + r"/(?:gallery|view)/(\d+)"
    example = "https://imhentai.xxx/gallery/12345/"

    def __init__(self, match):
        self.gallery_id = match.group(1)
        url = "{}/gallery/{}/".format(self.root, self.gallery_id)
        GalleryExtractor.__init__(self, match, url)

    def metadata(self, page):
        extr = text.extract_from(page)

        data = {
            "gallery_id": text.parse_int(self.gallery_id),
            "title"     : text.unescape(extr("<h1>", "<")),
            "title_alt" : text.unescape(extr('class="subtitle">', "<")),
            "parody"    : self._split(extr(">Parodies:</span>", "</li>")),
            "character" : self._split(extr(">Characters:</span>", "</li>")),
            "tags"      : self._split(extr(">Tags:</span>", "</li>")),
            "artist"    : self._split(extr(">Artists:</span>", "</li>")),
            "group"     : self._split(extr(">Groups:</span>", "</li>")),
            "language"  : self._split(extr(">Languages:</span>", "</li>")),
            "type"      : text.remove_html(extr(">Category:</span>", "<span")),
        }

        if data["language"]:
            data["lang"] = util.language_to_code(data["language"][0])

        return data

    def _split(self, html):
        results = []
        for tag in text.extract_iter(html, ">", "</a>"):
            tag = tag.partition(" <span class='badge'>")[0]
            if "<" in tag:
                tag = text.remove_html(tag)
            results.append(tag)
        return results

    def images(self, _):
        url = "{}/view/{}/1/".format(self.root, self.gallery_id)
        page = self.request(url).text
        data = util.json_loads(text.extr(page, "$.parseJSON('", "'"))
        base = text.extr(page, 'data-src="', '"').rpartition("/")[0] + "/"
        exts = {"j": "jpg", "p": "png", "g": "gif", "w": "webp", "a": "avif"}

        results = []
        for i in map(str, range(1, len(data)+1)):
            ext, width, height = data[i].split(",")
            url = base + i + "." + exts[ext]
            results.append((url, {
                "width" : text.parse_int(width),
                "height": text.parse_int(height),
            }))
        return results


class ImhentaiTagExtractor(ImhentaiExtractor):
    """Extractor for imhentai tag searches"""
    subcategory = "tag"
    pattern = (BASE_PATTERN + r"(/(?:"
               r"artist|category|character|group|language|parody|tag"
               r")/([^/?#]+))")
    example = "https://imhentai.xxx/tag/TAG/"

    def items(self):
        url = self.root + self.groups[0] + "/"
        return self._pagination(url)


class ImhentaiSearchExtractor(ImhentaiExtractor):
    """Extractor for imhentai search results"""
    subcategory = "search"
    pattern = BASE_PATTERN + r"/search/?\?([^#]+)"
    example = "https://imhentai.xxx/search/?key=QUERY"

    def items(self):
        url = self.root + "/search/?" + self.groups[0]
        return self._pagination(url)