from __future__ import annotations
import collections
import glob
import os
from collections.abc import Iterator
from typing import Any

from . import core
from . import resources

__copyright__ = """
2012 Peter Norvig (http://norvig.com/spell-correct.html)
2013-2026 David Aguilar <davvid@gmail.com>
"""


class GlobalState:
    ALPHABET = 'abcdefghijklmnopqrstuvwxyz'
    LETTERS = set(ALPHABET)

    @classmethod
    def train(
        cls, features: Any, model: dict[str, int], all_train_words: set[str]
    ) -> dict[str, int]:
        """Add words to the model"""
        for word in features:
            if word not in all_train_words:
                all_train_words.add(word)
                model[word] += 1
                for letter in word:
                    cls.LETTERS.add(letter)
        return model

    @classmethod
    def update(cls) -> None:
        """Update the alphabet after all dictionaries have been loaded"""
        cls.ALPHABET = ''.join(sorted(cls.LETTERS))


def edits1(word: str) -> set[str]:
    splits = [(word[:i], word[i:]) for i in range(len(word) + 1)]
    deletes = [a + b[1:] for a, b in splits if b]
    transposes = [a + b[1] + b[0] + b[2:] for a, b in splits if len(b) > 1]
    replaces = [a + c + b[1:] for a, b in splits for c in GlobalState.ALPHABET if b]
    inserts = [a + c + b for a, b in splits for c in GlobalState.ALPHABET]
    return set(deletes + transposes + replaces + inserts)


def known_edits2(word: str, words: Any) -> set[str]:
    return {e2 for e1 in edits1(word) for e2 in edits1(e1) if e2 in words}


def known(word: Any, words: Any) -> set[str]:
    return {w for w in word if w in words}


def suggest(word: str, words: Any) -> set[str] | list[str]:
    candidates = (
        known([word], words)
        or known(edits1(word), words)
        or known_edits2(word, words)
        or [word]
    )
    return candidates


def correct(word: str, words: Any) -> Any:
    candidates = suggest(word, words)
    return max(candidates, key=words.get)


class NorvigSpellCheck:
    def __init__(
        self,
        words: str = 'dict/words',
        propernames: str = 'dict/propernames',
    ) -> None:
        data_dirs = resources.xdg_data_dirs()
        self.dictwords = resources.find_first(words, data_dirs)
        self.propernames = resources.find_first(propernames, data_dirs)
        self.words = collections.defaultdict(lambda: 1)
        self.extra_words = set()
        self.extra_dictionaries = set()
        self.initialized = False
        self.all_words = set()
        self.aspell_enabled = False
        self.aspell_langs = set()
        self.aspell_ok = False

    def add_dictionaries(self, dictionaries: list[Any]) -> None:
        """Add additional dictionaries to the spellcheck engine"""
        self.extra_dictionaries.update(dictionaries)

    def init(self) -> None:
        if self.initialized:
            return
        self.initialized = True

        all_train_words = set()
        if self.aspell_enabled:
            GlobalState.train(self.read_aspell_words(), self.words, all_train_words)
        if not self.aspell_ok:
            GlobalState.train(self.read(), self.words, all_train_words)
        GlobalState.train(self.extra_words, self.words, all_train_words)

        GlobalState.update()

    def set_aspell_enabled(self, enabled: bool) -> None:
        """Enable aspell support"""
        self.aspell_enabled = enabled

    def set_aspell_langs(self, langs: list[Any]) -> None:
        """Set the aspell languages to query"""
        self.aspell_langs = set(langs)

    def add_word(self, word: str) -> None:
        self.extra_words.add(word)

    def suggest(self, word: str) -> list[str] | set[str]:
        self.init()
        return suggest(word, self.words)

    def check(self, word: str) -> bool:
        self.init()
        word = word.replace('.', '')
        return word in self.words or word.lower() in self.words

    def read(self, use_common_files: bool = True) -> Iterator[str]:
        """Read dictionary words"""
        paths = []
        words = self.dictwords
        propernames = self.propernames

        if use_common_files and words and os.path.exists(words):
            paths.append(words)

        if use_common_files and propernames and os.path.exists(propernames):
            paths.append(propernames)

        for path in self.extra_dictionaries:
            paths.append(path)

        all_words = self.all_words
        for path in paths:
            is_dic_file = path.endswith('.dic')
            try:
                with open(path, encoding='utf-8', errors='ignore') as words_file:
                    # Ignore the first word count line in *.dic files.
                    if is_dic_file:
                        words_file.readline()
                    for line in words_file:
                        word = line.strip().split('/', 1)[0]
                        if word not in all_words:
                            all_words.add(word)
                            yield word
            except OSError:
                pass

    def read_aspell_words(self) -> Iterator[Any]:
        """Read words from aspell"""
        # First, determine the languages to query.
        # Use "aspell dicts" and filter out any strings that are longer than 2
        # characters. This should leave *just* the main language names.
        if self.aspell_langs:
            aspell_langs = self.aspell_langs
        else:
            aspell_langs = _get_default_aspell_langs()  # type: ignore[assignment]

        ok = False
        all_words = self.all_words
        for lang in aspell_langs:
            cmd = ['aspell', 'dump', 'master', f'--lang={lang}']
            status, out, _ = core.run_command(cmd)
            if status == 0:
                for line in out.splitlines():
                    # Strip "/A" "/LR", "/H" and other suffixes from the
                    # output produced by "aspell dump master --lang=ru".
                    line = line.strip().split('/', 1)[0]
                    if not line:
                        continue
                    ok = True
                    if line not in all_words:
                        all_words.add(line)
                        yield line

        # Read extra dictionaries configured using `cola.dictionary`.
        self.aspell_ok = ok
        if ok:
            yield from self.read(use_common_files=False)


def _get_default_aspell_langs() -> list[str]:
    cmd = ['aspell', 'dicts']
    status, out, _ = core.run_command(cmd)
    if status != 0:
        return []
    return [line for line in out.splitlines() if len(line) == 2]


def get_available_dictionaries() -> list[str]:
    """Query available dictionary files from hunspell"""
    dictionaries = []
    hunspell_cmd = core.find_executable('hunspell')
    if hunspell_cmd:
        # Transform "/usr/bin/hunspell" into "/usr".
        hunspell_prefix = os.path.dirname(os.path.dirname(hunspell_cmd))
        # Create a "/usr/share/hunspell/*.dic" glob pattern.
        hunspell_pattern = os.path.join(hunspell_prefix, 'share', 'hunspell', '*.dic')
        dictionaries.extend(
            path for path in glob.glob(hunspell_pattern) if not os.path.islink(path)
        )
    return dictionaries
