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

from haystack import indexes
from django.db.models import Count

from proxy.models import Revision, Ticket, Wiki
from .models import User


class UserIndex(indexes.SearchIndex, indexes.Indexable):
    # common fields
    text = indexes.CharField(document=True, use_template=True, stored=False)
    url = indexes.CharField(model_attr='get_absolute_url', indexed=False)
    title = indexes.CharField(model_attr='get_full_name')
    description = indexes.CharField(null=True)
    type = indexes.CharField()
    icon_name = indexes.CharField()

    # extra fields
    username = indexes.CharField(model_attr='username', stored=False)
    name = indexes.CharField(model_attr='get_full_name')
    email = indexes.CharField(model_attr='email', stored=False)
    institution = indexes.CharField(model_attr='institution', null=True)
    role = indexes.CharField(model_attr='role', null=True)
    google_talk = indexes.CharField(model_attr='google_talk', null=True,
                                    stored=False)
    webpage = indexes.CharField(model_attr='webpage', null=True, stored=False)
    message_count = indexes.IntegerField(stored=False)
    changeset_count = indexes.IntegerField(stored=False)
    ticket_count = indexes.IntegerField(stored=False)
    wiki_count = indexes.IntegerField(stored=False)
    contribution_count = indexes.IntegerField(stored=False)

    def get_model(self):
        return User

    def get_updated_field(self):
        return 'date_joined'

    def prepare(self, obj):
        prepared_data = super(UserIndex, self).prepare(obj)

        message_count = self.prepared_data['message_count']
        changeset_count = self.prepared_data['changeset_count']
        ticket_count = self.prepared_data['ticket_count']
        wiki_count = self.prepared_data['wiki_count']

        prepared_data['contribution_count'] = sum((
            message_count,
            changeset_count,
            ticket_count,
            wiki_count
        ))

        return prepared_data

    def prepare_description(self, obj):
        return u'{}\n{}\n{}\n{}'.format(
            obj.institution, obj.role, obj.username, obj.get_full_name()
        )

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

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

    def prepare_message_count(self, obj):
        return obj.emails.aggregate(Count('message'))['message__count']

    def prepare_changeset_count(self, obj):
        return Revision.objects.filter(author=obj.username).count()

    def prepare_ticket_count(self, obj):
        return Ticket.objects.filter(author=obj.username).count()

    def prepare_wiki_count(self, obj):
        return Wiki.objects.filter(author=obj.username).count()

    def index_queryset(self, using=None):
        return self.get_model().objects.filter(is_active=True)