from bot.entries.daijirin2.phrase_entry import PhraseEntry

from bot.yomichan.terms.terminator import Terminator
from bot.yomichan.glossary.daijirin2 import make_glossary
from bot.yomichan.grammar import sudachi_rules, tags_to_rules


class Daijirin2Terminator(Terminator):
    def _definition_tags(self, entry):
        return ""

    def _inflection_rules(self, entry, expression):
        if isinstance(entry, PhraseEntry):
            return sudachi_rules(expression)
        pos_tags = entry.get_part_of_speech_tags()
        if len(pos_tags) > 0:
            rules = tags_to_rules(expression, pos_tags,
                                  self._inflection_categories)
        else:
            rules = sudachi_rules(expression)
        return rules

    def _glossary(self, entry):
        if entry.entry_id in self._glossary_cache:
            return self._glossary_cache[entry.entry_id]
        glossary = make_glossary(entry, self._image_dir)
        self._glossary_cache[entry.entry_id] = glossary
        return glossary

    def _sequence(self, entry):
        return entry.entry_id[0] * 100000 + entry.entry_id[1]

    def _term_tags(self, entry):
        return ""

    def _link_glossary_parameters(self, entry):
        return [
            [entry.children, "子"],
            [entry.phrases, "句"],
        ]

    def _subentry_lists(self, entry):
        return [
            entry.children,
            entry.phrases,
        ]