#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (C) 2010-2016  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# See the file COPYING for details.

import json

from django.core.urlresolvers import reverse
from django.db.models import Q
from django.http import HttpResponse, HttpResponseRedirect, Http404
from django.shortcuts import redirect
from django.utils.translation import ugettext_lazy as _
from django.views.generic import RedirectView

import models
from archaeological_operations.views import site_extra_context
from forms import *
from ishtar_common.utils import put_session_message
from ishtar_common.views import IshtarMixin, LoginRequiredMixin
from ishtar_common.views_item import display_item, get_item, show_item, \
    revert_item
from ishtar_common.wizards import SearchWizard
from wizards import *

show_contextrecord = show_item(
    models.ContextRecord, 'contextrecord',
    extra_dct=site_extra_context
)
revert_contextrecord = revert_item(models.ContextRecord)
display_contextrecord = display_item(models.ContextRecord)


def autocomplete_contextrecord(request):
    if (not request.user.has_perm(
            'archaeological_context_records.view_contextrecord',
            models.ContextRecord)
        and not request.user.has_perm(
            'archaeological_context_records.view_own_contextrecord',
            models.ArchaeologicalSite)):
        return HttpResponse(content_type='text/plain')
    if not request.GET.get('term'):
        return HttpResponse(content_type='text/plain')
    q = request.GET.get('term')
    query = Q()
    if request.GET.get('operation__pk'):
        query = Q(operation__pk=request.GET.get('operation__pk'))
    for q in q.split(' '):
        qt = Q(parcel__section__icontains=q) | \
            Q(parcel__parcel_number__icontains=q) | \
            Q(label__icontains=q)
        query = query & qt
    limit = 15
    items = models.ContextRecord.objects\
        .filter(query).order_by('parcel__section', 'parcel__parcel_number',
                                'label')[:limit]
    data = json.dumps([{'id': item.pk,
                        'value': unicode(item)[:60]}
                       for item in items])
    return HttpResponse(data, content_type='text/plain')


get_contextrecord = get_item(
    models.ContextRecord,
    'get_contextrecord', 'contextrecord')
get_contextrecord_for_ope = get_item(
    models.ContextRecord,
    'get_contextrecord', 'contextrecord',
    own_table_cols=models.ContextRecord.TABLE_COLS_FOR_OPE)

get_contextrecordrelation = get_item(
    models.RecordRelationView, 'get_contextrecordrelation',
    'contextrecordrelation', specific_perms=['view_recordrelation'],
    do_not_deduplicate=True
)

get_contextrecordrelationdetail = get_item(
    models.RecordRelations, 'get_contextrecordrelationdetail',
    'contextrecordrelationdetail', specific_perms=['view_recordrelation'])

record_search_wizard = SearchWizard.as_view([
    ('general-record_search', RecordFormSelection)],
    label=_(u"Context record search"),
    url_name='record_search',)

record_creation_steps = [
    ('selec-record_creation', OperationRecordFormSelection),
    ('general-record_creation', RecordFormGeneral),
    ('datings-record_creation', DatingFormSet),
    ('interpretation-record_creation', RecordFormInterpretation),
    ('relations-record_creation', RecordRelationsFormSet),
    ('final-record_creation', FinalForm)
]

record_creation_wizard = RecordWizard.as_view(
    record_creation_steps,
    label=_(u"New context record"),
    url_name='record_creation',
)

record_modification_steps = [
    ('selec-record_modification', RecordFormSelection),
    ('operation-record_modification', OperationFormSelection),
    ('general-record_modification', RecordFormGeneral),
    ('datings-record_modification', DatingFormSet),
    ('interpretation-record_modification', RecordFormInterpretation),
    ('relations-record_modification', RecordRelationsFormSet),
    ('final-record_modification', FinalForm)
]

record_modification_wizard = RecordModifWizard.as_view(
    record_modification_steps,
    label=_(u"Context record modification"),
    url_name='record_modification',
)


def record_modify(request, pk):
    try:
        record_modification_wizard(request)
    except IndexError:  # no step available
        put_session_message(
            request.session.session_key,
            _(u"You don't have sufficient permissions to do this action."),
            'warning'
        )
        return HttpResponseRedirect("/")

    RecordModifWizard.session_set_value(
        request, 'selec-record_modification', 'pk', pk, reset=True)
    return redirect(reverse('record_modification',
                    kwargs={'step': 'operation-record_modification'}))

record_deletion_wizard = RecordDeletionWizard.as_view([
    ('selec-record_deletion', RecordFormSelection),
    ('final-record_deletion', RecordDeletionForm)],
    label=_(u"Context record deletion"),
    url_name='record_deletion',)


def reset_wizards(request):
    for wizard_class, url_name in (
            (RecordWizard, 'record_creation'),
            (RecordModifWizard, 'record_modification'),
            (RecordDeletionWizard, 'record_deletion')):
        wizard_class.session_reset(request, url_name)


class GenerateRelationImage(IshtarMixin, LoginRequiredMixin, RedirectView):
    def get_redirect_url(self, *args, **kwargs):
        return reverse('display-' + models.ContextRecord.SLUG,
                       args=[self.context_record.pk])

    def get(self, request, *args, **kwargs):
        try:
            self.context_record = models.ContextRecord.objects.get(
                pk=kwargs['pk']
            )
        except models.ContextRecord.DoesNotExist:
            raise Http404()
        self.context_record.generate_relation_image()
        return super(GenerateRelationImage, self).get(request, *args, **kwargs)