import time

from django.conf.urls import include, url
from wagtail.contrib.modeladmin.options import ModelAdmin, modeladmin_register
from wagtail.contrib.modeladmin.views import IndexView
from wagtail.wagtailcore import hooks

from personalisation import admin_urls
from personalisation.forms import (
    ReferralRuleForm, TimeRuleForm, VisitCountRuleForm)
from personalisation.models import Segment


@hooks.register('register_admin_urls')
def register_admin_urls():
    """Adds the administration urls for the personalisation apps."""
    return [
        url(r'^personalisation/', include(
            admin_urls,
            app_name='personalisation',
            namespace='personalisation')),
    ]


class SegmentIndexView(IndexView):
    def get_context_data(self, **kwargs):
        context = {
            'additional_forms': (TimeRuleForm, ReferralRuleForm, VisitCountRuleForm),
        }
        context.update(kwargs)
        return super(SegmentIndexView, self).get_context_data(**context)


class SegmentModelAdmin(ModelAdmin):
    """The base model for the Segments administration interface."""
    model = Segment
    menu_icon = 'group'
    add_to_settings_menu = False
    list_display = ('status', 'name', 'create_date', 'edit_date')
    index_view_extra_css = ['personalisation/segment/index.css']
    form_view_extra_css = ['personalisation/segment/form.css']
    inspect_view_enabled = True

modeladmin_register(SegmentModelAdmin)


@hooks.register('before_serve_page')
def set_visit_count(page, request, serve_args, serve_kwargs):
    """Update the users visit count before each page visit."""
    # Update the segment visit count
    for seg in request.session['segments']:
        segment = Segment.objects.get(pk=seg['id'])
        segment.visit_count = segment.visit_count + 1
        segment.save()

    # Update the page visit count
    def create_new_counter(page):
        """Create a new counter dict and place it in session storage."""
        countdict = {
            "slug": page.slug,
            "id": page.pk,
            "path": page.url,
            "count": 1,
        }
        request.session['visit_count'].append(countdict)

    if len(request.session['visit_count']) > 0:
        for index, counter in enumerate(request.session['visit_count']):
            if counter['id'] == page.pk:
                # Counter already exisits. Increase the count value by 1.
                newcount = counter['count'] + 1
                request.session['visit_count'][index]['count'] = newcount
                request.session.modified = True
            else:
                # Counter doesn't exist. Create a new counter with count value 1.
                create_new_counter(page)
    else:
        # No counters exist. Create a new counter with count value 1.
        create_new_counter(page)