from django.conf import settings

from django.contrib.auth.decorators import login_required

from django.contrib.auth.models import User, Group

from django.core.cache import cache

from django.core.urlresolvers import reverse

from django.core.paginator import Paginator, InvalidPage, EmptyPage

from django.db.models import Q

from django.forms.models import model_to_dict

from django.forms.util import ErrorList

from django.http import (HttpResponse, HttpResponseRedirect,

    HttpResponseForbidden, HttpResponseServerError)

from ldt.indexation import get_results_with_context, highlight_documents

from django.shortcuts import (render_to_response, get_object_or_404,

    get_list_or_404)

from django.template import RequestContext

from django.template.loader import render_to_string

from django.utils import simplejson

from django.utils.html import escape

from django.utils.translation import ugettext as _, ungettext

from forms import (LdtAddForm, SearchForm, AddProjectForm, CopyProjectForm,

    ContentForm, MediaForm, GroupAddForm)

from guardian.shortcuts import remove_perm, get_objects_for_group, get_objects_for_user

from ldt.ldt_utils.models import Content

from ldt.ldt_utils.utils import boolean_convert, LdtUtils, LdtSearch

from ldt.security.utils import (assign_perm_to_obj, set_forbidden_stream, 

    add_change_attr, get_userlist, get_userlist_model, get_userlist_group)

from ldt.security.cache import get_cached_checker, cached_assign

from lxml.html import fragment_fromstring

from models import Media, Project

from projectserializer import ProjectSerializer

from urllib2 import urlparse

from operator import itemgetter

from itertools import groupby

import base64

import django.core.urlresolvers

import ldt.auth as ldt_auth

import ldt.utils.path as ldt_utils_path

import logging

import lxml.etree

import mimetypes

import os

import urllib2

import subprocess

import re

import datetime

@login_required

def workspace(request):

    # list of contents

    content_list = add_change_attr(request.user, Content.safe_objects.all()) #@UndefinedVariable

    # get list of projects owned by the current user

    project_list = add_change_attr(request.user, Project.safe_objects.filter(owner=request.user)) #@UndefinedVariable

    is_gecko = ((request.META['HTTP_USER_AGENT'].lower().find("firefox")) > -1);

    # render list

    return render_to_response("ldt/ldt_utils/workspace.html",

                              {'contents': content_list, 'projects': project_list,

                               'is_gecko': is_gecko},

                              context_instance=RequestContext(request))

@login_required

def groups(request): 

    # get list of all published projects

    group_list = request.user.groups.exclude(name=settings.PUBLIC_GROUP_NAME)

    is_gecko = ((request.META['HTTP_USER_AGENT'].lower().find("firefox")) > -1);

    # render list

    return render_to_response("ldt/ldt_utils/groups.html",

                              {'groups': group_list,

                              context_instance=RequestContext(request))

@login_required

def published_project(request): 

    # get list of all published projects

    project_list = Project.safe_objects.filter(state=2) #@UndefinedVariable

    # Search form

    form = SearchForm()

    is_gecko = ((request.META['HTTP_USER_AGENT'].lower().find("firefox")) > -1);

    # render list

    return render_to_response("ldt/ldt_utils/published_projects.html",

                              {'projects': project_list, 'form': form,

                               'is_gecko': is_gecko},

                              context_instance=RequestContext(request))

def popup_embed(request):

    json_url = request.GET.get("json_url")

    player_id = request.GET.get("player_id")

    ldt_id = request.GET.get("ldt_id")

    project = Project.safe_objects.get(ldt_id=ldt_id); #@UndefinedVariable

    stream_mode = project.stream_mode

    if stream_mode != "video":

        stream_mode = 'radio'

    player_width = 650

    player_height = 480

    if stream_mode == 'radio':

        player_height = 1

    if not ldt_auth.check_access(request.user, project):

        return HttpResponseForbidden(_("You can not access this project"))

    ps = ProjectSerializer(project, from_contents=False, from_display=True)

    annotations = ps.get_annotations(first_cutting=True)

    rend_dict = {'json_url':json_url, 'player_id':player_id, 'annotations':annotations, 'ldt_id': ldt_id, 'stream_mode': stream_mode, 'player_width': player_width, 'player_height': player_height}

    embed_rendered = dict((typestr,

                           (lambda s:escape(lxml.etree.tostring(fragment_fromstring(render_to_string("ldt/ldt_utils/partial/embed_%s.html" % (s), rend_dict, context_instance=RequestContext(request))), pretty_print=True)))(typestr))

                           for typestr in ('player', 'seo_body', 'seo_meta', 'links'))

    rend_dict['embed_rendered'] = embed_rendered

    return render_to_response("ldt/ldt_utils/embed_popup.html",

                              rend_dict,

                              context_instance=RequestContext(request))

@login_required

def projects_filter(request, filter, is_owner=False, status=0, id_group=0):

    is_owner = boolean_convert(is_owner)

    status = int(status)

    id_group = int(id_group)

    query = Q()

    if is_owner:

        owner = None

        try:

            owner = request.user

        except:

            return HttpResponseServerError("<h1>User not found</h1>")

        query &= Q(owner=owner)

    if status > 0:

        query &= Q(state=status)

    if filter:

        if len(filter) > 0 and filter[0] == '_':

            filter = filter[1:]

        query &= Q(title__icontains=filter)

    is_gecko = ((request.META['HTTP_USER_AGENT'].lower().find("firefox")) > -1);

    show_username = True

    if id_group > 0:    # Search inside a group

        grp = Group.objects.get(id=id_group)

        project_list = get_objects_for_group(grp, 'ldt_utils.view_project').filter(query)

    elif id_group == -1:# Search inside projects shared with a specific user

        project_list = get_objects_for_user(request.user, 'ldt_utils.view_project', use_groups=False).exclude(owner=request.user)

    else:

        project_list = Project.safe_objects.filter(query)

        show_username = False

    project_list = add_change_attr(request.user, project_list)

    if status == 2:

        url_templ = "ldt/ldt_utils/partial/publishedprojectslist.html"

    else:

        url_templ = "ldt/ldt_utils/partial/projectslist.html"

    return render_to_response(url_templ,

                              {'projects': project_list, 'show_username':show_username,

                               'is_gecko': is_gecko, 'group_id':id_group},

                              context_instance=RequestContext(request))

def share_filter(request, filter, use_groups=False):

    use_groups = boolean_convert(use_groups)

    if not filter or len(filter) == 0:

        raise AttributeError("filter should be a string")

    filter = filter[1:]    

    resp = get_userlist(request.user, filter=filter)

    if use_groups:

        groups = Group.objects.filter(name__icontains=filter).exclude(name=settings.PUBLIC_GROUP_NAME)[0:20]

        for g in groups:

            resp.append({'name': g.name, 'id': g.id, 'type': 'group'})

    resp = sorted(resp, key=lambda elem: elem['name'].lower())

    return render_to_response("ldt/ldt_utils/partial/sharewith.html", {'elem_list' : resp}, context_instance=RequestContext(request))

@login_required

def contents_filter(request, filter): 

    if filter and len(filter) > 0 and filter[0] == '_':

        filter = filter[1:]

    if filter:

        content_list = Content.safe_objects.filter(title__icontains=filter) #@UndefinedVariable

    else:

        content_list = Content.safe_objects.all() #@UndefinedVariable

    content_list = add_change_attr(request.user, content_list)

    return render_to_response("ldt/ldt_utils/partial/contentslist.html",

                              {'contents': content_list},

                              context_instance=RequestContext(request))

@login_required

def groups_filter(request, filter):

    if filter and len(filter) > 0 and filter[0] == '_':

        filter = filter[1:]

    if filter:

        group_list = request.user.groups.filter(title__icontains=filter) 

    else:

        group_list = request.user.groups.all()

    group_list = group_list.exclude(name=settings.PUBLIC_GROUP_NAME)

    group_list = sorted(group_list.all(), key=lambda group: group.name.lower())

    return render_to_response("ldt/ldt_utils/partial/groupslist.html",

                              {'groups': group_list},

                              context_instance=RequestContext(request))

def search_form(request): 

    form = SearchForm()

    return render_to_response('ldt/ldt_utils/search_form.html', {'form': form} , context_instance=RequestContext(request))

@login_required

def search_index(request):

    sform = SearchForm(request.POST)

    if sform.is_valid():

        search = sform.cleaned_data["search"]

        queryStr = base64.urlsafe_b64encode(search.encode('utf8'))

        field = request.POST["field"]

        language_code = request.LANGUAGE_CODE[:2]

        if request.POST.has_key("ldt_pres"):

            url = settings.WEB_URL + django.core.urlresolvers.reverse("ldt.ldt_utils.views.search_init", args=[field, queryStr])

            return render_to_response('ldt/ldt_utils/init_ldt_full.html', {'LDT_MEDIA_PREFIX': settings.LDT_MEDIA_PREFIX, 'colorurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/pkg/color.xml', 'i18nurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/pkg/i18n', 'language': language_code, 'baseurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/', 'url': url}, context_instance=RequestContext(request))

        else:

            results = get_results_with_context(field, search)     

            complete_results = []

            proj_list = Project.safe_objects.all()

            results.sort(key=lambda k: k['iri_id'])

            for iri_id, item in groupby(results, itemgetter('iri_id')):                

                try: 

                    content = Content.objects.get(iri_id=iri_id)

                except Content.DoesNotExist:

                    continue

                segments = list(item)

                to_delete = []

                for s in segments:

                    if not s['project_id']:

                        s['project_id'] = '_'

                    else:

                        project = proj_list.filter(ldt_id=s['project_id'])

                        if len(project) == 0:

                            to_delete.append(s)

                for e in to_delete:

                    segments.remove(e)

                if not segments:

                    continue 

                score = sum([seg['score'] for seg in segments])

                desc = content.description if content.description else ''

                complete_results.append({'list' : segments, 'score' : score, 'content_title' : content.title, 'content_id' : content.iri_id, 'content_description' : desc })

            complete_results.sort(key=lambda k: k['score'])

            cache.set('complete_results_%s' % request.user.username, complete_results)

            cache.set('search_%s' % request.user.username, search)

            cache.set('field_%s' % request.user.username, field)

            paginator = Paginator (complete_results, settings.LDT_RESULTS_PER_PAGE)

            page = 1

            try:

                results = paginator.page(page)

            except (EmptyPage, InvalidPage):

                results = paginator.page(paginator.num_pages)

            results.object_list = highlight_documents(results.object_list, search, field)

            return render_to_response('ldt/ldt_utils/search_results.html', {'results': results, 'nb_results' : paginator.count, 'search' : search, 'LDT_MEDIA_PREFIX': settings.LDT_MEDIA_PREFIX, 'colorurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/pkg/color.xml', 'i18nurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/pkg/i18n', 'language': language_code, 'baseurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/'}, context_instance=RequestContext(request))   

    else:

        return HttpResponseRedirect(reverse('ldt.ldt_utils.views.published_project'))

def search_listing(request):

    if not cache.get('complete_results_%s' % request.user.username):

        return HttpResponseRedirect(reverse('ldt.ldt_utils.views.published_project'))

    complete_results = cache.get('complete_results_%s' % request.user.username)

    search = cache.get('search_%s' % request.user.username)

    field = cache.get('field_%s' % request.user.username)

    paginator = Paginator(complete_results, settings.LDT_RESULTS_PER_PAGE)

    language_code = request.LANGUAGE_CODE[:2]

    try:

        page = int(request.GET.get('page', '1'))

    except ValueError:

        page = 1

    try:

        results = paginator.page(page)

    except (EmptyPage, InvalidPage):

        results = paginator.page(paginator.num_pages)

    results.object_list = highlight_documents(results.object_list, search, field)

    return render_to_response('ldt/ldt_utils/search_results.html', {'results': results, 'nb_results' : paginator.count, 'search' : search, 'LDT_MEDIA_PREFIX': settings.LDT_MEDIA_PREFIX, 'colorurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/pkg/color.xml', 'i18nurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/pkg/i18n', 'language': language_code, 'baseurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/'}, context_instance=RequestContext(request))   

def search_index_get(request, field, query):

    language_code = request.LANGUAGE_CODE[:2]

    url = settings.WEB_URL + django.core.urlresolvers.reverse("ldt.ldt_utils.views.search_init", args=[field, query])

    return render_to_response('ldt/ldt_utils/init_ldt.html', {'LDT_MEDIA_PREFIX': settings.LDT_MEDIA_PREFIX, 'colorurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/pkg/color.xml', 'i18nurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/pkg/i18n', 'language': language_code, 'baseurl': settings.LDT_MEDIA_PREFIX + 'swf/ldt/', 'url': url}, context_instance=RequestContext(request))

def search_init(request, field, query):

    ldtgen = LdtUtils()

    doc = ldtgen.generate_init([field, query], 'ldt.ldt_utils.views.search_ldt', 'ldt.ldt_utils.views.search_segments')

    resp = HttpResponse(mimetype="text/xml;charset=utf-8")

    resp.write(lxml.etree.tostring(doc, pretty_print=True, encoding="utf-8")) 

    return resp