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

import math

from haystack import indexes

from search.base_indexes import BaseIndex
from .models import Thread


class ThreadIndex(BaseIndex, indexes.Indexable):
    title = indexes.CharField(model_attr='latest_message__subject_clean')
    description = indexes.CharField(use_template=True)
    latest_description = indexes.CharField(
        model_attr='latest_message__description',
        indexed=False,
    )
    created = indexes.DateTimeField()
    modified = indexes.DateTimeField(
        model_attr='latest_message__modified'
    )
    tag = indexes.CharField(model_attr='mailinglist__name')
    collaborators = indexes.CharField(use_template=True, stored=False)
    mailinglist_url = indexes.CharField(
        model_attr='mailinglist__get_absolute_url',
        indexed=False,
    )
    latest_message_pk = indexes.IntegerField(
        model_attr='latest_message__pk', indexed=False
    )
    score = indexes.IntegerField(model_attr='score')

    def get_model(self):
        return Thread

    def get_updated_field(self):
        return 'latest_message__received_time'

    def prepare_fullname(self, obj):
        return obj.message_set.first().from_address.get_full_name()

    def prepare_fullname_and_username(self, obj):
        from_address = obj.message_set.first().from_address
        if not from_address.user:
            return from_address.get_full_name()

        return u'{}\n{}'.format(
            from_address.get_full_name(),
            from_address.user.username,
        )

    def prepare_author(self, obj):
        first_message = obj.message_set.first()
        return first_message.from_address.get_full_name()

    def prepare_author_url(self, obj):
        first_message = obj.message_set.first()
        return first_message.author_url

    def prepare_modified_by(self, obj):
        modified_by = obj.latest_message.modified_by
        if modified_by:
            return modified_by
        return obj.message_set.first().author

    def prepare_modified_by_url(self, obj):
        modified_by_url = obj.latest_message.modified_by_url
        if modified_by_url:
            return modified_by_url
        return None

    def prepare_created(self, obj):
        return obj.message_set.first().received_time

    def prepare_fullname(self, obj):
        fullname = obj.latest_message.from_address.get_full_name()
        if not fullname:
            fullname = obj.message_set.first().from_address.get_full_name()
        return fullname

    def prepare_icon_name(self, obj):
        return u'envelope'

    def prepare_type(self, obj):
        return u'thread'

    def index_queryset(self, using=None):
        return self.get_model().objects.filter(
            spam=False
        ).exclude(subject_token='')

    def get_boost(self, obj):
        boost = super(ThreadIndex, self).get_boost(obj)

        if obj.score >= 10:
            boost = boost * math.log(obj.score)

        return boost