import re
from typing import Any, override

import papis.config
import papis.document
import papis.format
import papis.logging

logger = papis.logging.get_logger(__name__)

OPTIONS_SECTION = "plugins.bbt"
DEFAULT_OPTIONS = {
    OPTIONS_SECTION: {
        "default-formatter": "python",
        "title-words": 3,
        "title-chars": -1,
    }
}
papis.config.register_default_settings(DEFAULT_OPTIONS)


class BBTFormatter(papis.format.Formatter):
    """Provides zotero better-bibtex-like keys."""

    @override
    def format(
        self,
        fmt: str,
        doc: papis.document.DocumentLike,
        doc_key: str = "",
        additional: dict[str, Any] | None = None,
        default: str | None = None,
    ) -> str:
        if not fmt.startswith("bbt:"):
            fallback_formatter = papis.config.getstring(
                "default-formatter", OPTIONS_SECTION
            )
            return self.format_with_fallback(
                fallback_formatter, fmt, doc, doc_key, additional, default
            )

        return self.format_with_bbt(doc)

    def format_with_fallback(
        self,
        fallback_formatter: str,
        fmt: str,
        doc: papis.document.DocumentLike,
        doc_key: str,
        additional: dict[str, Any] | None = None,
        default: str | None = None,
    ) -> str:
        _saved = papis.format.FORMATTER
        papis.format.FORMATTER = None
        fallback_formatted: str = papis.format.get_formatter(fallback_formatter).format(
            fmt, doc, doc_key, additional, default
        )
        papis.format.FORMATTER = _saved
        return fallback_formatted

    def format_with_bbt(self, doc: papis.document.DocumentLike) -> str:
        author_unfmt = (
            doc["author_list"][0]["family"]
            if "author_list" in doc
            else doc["author"].split(maxsplit=1)[0]
            if "author" in doc
            else "UNKNOWN"
        )
        author = re.sub("[^a-z]+", "", author_unfmt.lower())
        year = self.get_year(
            doc["year"] if "year" in doc else doc["date"] if "date" in doc else ""
        )
        title = self.get_title(doc["title"] if "title" in doc else "NO TITLE")
        return f"{author}{year}{title}"

    def get_year(self, date: str) -> str:
        """Returns year string according to set year display options.

        Returns either the full 4-digit year or a shortened 2-digit
        version depending on the plugin year options."""
        date_match = re.search(r"\d{4}", date)
        if not date_match:
            return "0000"
        date_str = date_match[0]
        if papis.config.getboolean("full-year", OPTIONS_SECTION):
            return date_str
        return date_str[-2:]

    def get_title(self, title: str) -> str:
        """Returns cleaned and shortened title.

        Removes skip-words, cleans any punctuation and spaces and trims
        the title length to that set in plugin length options."""
        title = re.sub("[^0-9a-z ]+", "", title.lower())
        title_words = list(
            map(
                str.capitalize,
                filter(lambda word: word and word not in SKIP_WORDS, title.split()),
            )
        )
        wlen = papis.config.getint("title-words", OPTIONS_SECTION)
        clen = papis.config.getint("title-chars", OPTIONS_SECTION)
        wlen = None if wlen == -1 else wlen
        clen = None if clen == -1 else clen
        title = "".join(title_words[:wlen])[:clen]
        return title


SKIP_WORDS = set(
    [
        "about",
        "above",
        "across",
        "afore",
        "after",
        "against",
        "al",
        "along",
        "alongside",
        "amid",
        "amidst",
        "among",
        "amongst",
        "anenst",
        "apropos",
        "apud",
        "around",
        "as",
        "aside",
        "astride",
        "at",
        "athwart",
        "atop",
        "barring",
        "before",
        "behind",
        "below",
        "beneath",
        "beside",
        "besides",
        "between",
        "beyond",
        "but",
        "by",
        "circa",
        "despite",
        "down",
        "during",
        "et",
        "except",
        "for",
        "forenenst",
        "from",
        "given",
        "in",
        "inside",
        "into",
        "lest",
        "like",
        "modulo",
        "near",
        "next",
        "notwithstanding",
        "of",
        "off",
        "on",
        "onto",
        "out",
        "over",
        "per",
        "plus",
        "pro",
        "qua",
        "sans",
        "since",
        "than",
        "through",
        " thru",
        "throughout",
        "thruout",
        "till",
        "to",
        "toward",
        "towards",
        "under",
        "underneath",
        "until",
        "unto",
        "up",
        "upon",
        "versus",
        "vs.",
        "v.",
        "vs",
        "v",
        "via",
        "vis-à-vis",
        "with",
        "within",
        "without",
        "according to",
        "ahead of",
        "apart from",
        "as for",
        "as of",
        "as per",
        "as regards",
        "aside from",
        "back to",
        "because of",
        "close to",
        "due to",
        "except for",
        "far from",
        "inside of",
        "instead of",
        "near to",
        "next to",
        "on to",
        "out from",
        "out of",
        "outside of",
        "prior to",
        "pursuant to",
        "rather than",
        "regardless of",
        "such as",
        "that of",
        "up to",
        "where as",
        "or",
        "yet",
        "so",
        "for",
        "and",
        "nor",
        "a",
        "an",
        "the",
        "de",
        "d'",
        "von",
        "van",
        "c",
        "ca",
    ]
)