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

# 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://bato.to and aliases (v3x only)"""

from .common import ChapterExtractor, MangaExtractor
from .. import text, exception
import re

BASE_PATTERN = r"(?:https?://)?(?:bato\.to|dto\.to|batotoo\.com|wto\.to)"
MANGA_PATTERN = r"/title/\d+(?:-[0-9a-z]+)*/?"
CHAPTER_PATTERN = r"/\d+(?:-vol_\d+)?-ch_\d+\.?\d*/?"


class BatoBase():
    """Base class for bato v3x extractors"""
    category = "bato"
    root = "https://bato.to"


class BatoChapterExtractor(BatoBase, ChapterExtractor):
    """Extractor for manga chapters from bato.to"""
    pattern = BASE_PATTERN + "(" + MANGA_PATTERN + CHAPTER_PATTERN + ")"
    # There are three possible patterns for a chapter
    example = "https://bato.to/title/12345-manga-name-with-spaces/54212-ch_1.5"
    example2 = \
        "https://bato.to/title/12345-manga-name-with-spaces/54212-vol1-ch_1.5"
    example3 = "https://bato.to/title/12345/54212"
    # v2x, not supported
    example4 = "https://bato.to/chapter/54212"

    def __init__(self, match):
        self.path = match.group(1)
        ChapterExtractor.__init__(self, match, self.root + self.path)

    def metadata(self, page):
        info, _ = text.extract(
            page, "<title>", r" - Read Free Manga Online at Bato.To</title>"
        )
        info = info.encode('latin-1').decode('utf-8').replace("\n", "")

        match = re.match(
            r"(.+) - "
            r"(?:Volume *(\d+) )?"
            r"Chapter *([\d\.]+)", info)
        manga, volume, chapter = match.groups() if match else ("", "", info)
        chapter, sep, minor = chapter.partition(".")
        title_container = text.extr(page, f'<a href="{self.path}"', "</a>")
        title = text.extr(title_container, "<!-- -->", "</span>")

        return {
            "manga"        : text.unescape(manga),
            "title"        : text.unescape(title),
            "author"       : "",
            "volume"       : text.parse_int(volume),
            "chapter"      : text.parse_int(chapter),
            "chapter_minor": sep + minor,
        }

    def images(self, page):
        images_container = text.extr(page, 'pageOpts', ':[0,0]}"')
        images_container = text.unescape(images_container)
        return [
            (url, None)
            for url in text.extract_iter(images_container, r"\"", r"\"")
        ]


class BatoMangaExtractor(BatoBase, MangaExtractor):
    """Extractor for manga from bato.to"""
    reverse = False
    chapterclass = BatoChapterExtractor
    pattern = BASE_PATTERN + "(" + MANGA_PATTERN + "$" + ")"
    # There are two possible patterns for a manga
    example = "https://bato.to/title/12345-manga-name-with-spaces/"
    example2 = "https://bato.to/title/12345/"
    # v2x, not supported
    example3 = "https://bato.to/series/12345/manga-name-with-space"

    def chapters(self, page):
        data = {}
        num_chapters, _ = text.extract(page, ">Chapters<", "</div>")
        num_chapters, _ = text.extract(num_chapters, r"<!-- -->", r"<!-- -->")
        num_chapters = text.parse_int(num_chapters)
        if num_chapters == 0:
            raise exception.NotFoundError("chapter")

        manga, _ = text.extract(
            page, "<title>", r" - Read Free Manga Online at Bato.To</title>"
        )
        manga = manga.encode('latin-1').decode('utf-8').replace("\n", "")
        data["manga"] = manga

        results = []
        for chapter_num in range(num_chapters):
            chapter, _ = text.extract(
                page, f'<div data-hk="0-0-{chapter_num}-0"', r"</time><!--/-->"
            )
            chapter += r"</time><!--/-->"  # so we can match the date
            url, pos = text.extract(chapter, '<a href="', '"')

            chapter_no = re.search(r"-ch_([\d\.]+)", url).group(1)
            chapter_major, sep, chapter_minor = chapter_no.partition(".")
            title, _ = text.extract(
                chapter, f'<span data-hk="0-0-{chapter_num}-1"', "</span>"
            )
            title, _ = text.extract(title, r"<!--#-->", r"<!--/-->")
            if title is None or title == "" or title == "<!--/-->":
                title, _ = text.extract(chapter, ">", "</a>", pos)

            date, _ = text.extract(chapter, "<time", "</time>")
            date, _ = text.extract(date, 'time="', '"')

            data["date"] = date
            data["title"] = title
            data["chapter"] = text.parse_int(chapter_major)
            data["chapter_minor"] = sep + chapter_minor

            if url.startswith("/"):
                url = self.root + url
            results.append((url, data.copy()))
        return results