from django_elasticsearch_dsl import fields
from django_elasticsearch_dsl.registries import registry

from blog.models import Post
from core.elasticsearch import COMMON_ANALYSIS, ActiveOnlyMixin, add_multilang_fields
from core.elasticsearch.documents import BaseDocument


class PostDocument(ActiveOnlyMixin, BaseDocument):
    title = fields.TextField(
        attr="title",
        analyzer="standard",
        fields={
            "raw": fields.KeywordField(ignore_above=256),
            "ngram": fields.TextField(analyzer="name_ngram", search_analyzer="icu_query"),
            "phonetic": fields.TextField(analyzer="name_phonetic"),
        },
    )

    class Index:
        name = "posts"
        settings = {
            "number_of_shards": 1,
            "number_of_replicas": 0,
            "analysis": COMMON_ANALYSIS,
            "index": {"max_ngram_diff": 18},
        }

    class Django:
        model = Post
        fields = ["uuid"]

    def prepare_title(self, instance):
        return getattr(instance, "title", "") or ""


add_multilang_fields(PostDocument)
registry.register_document(PostDocument)