from django.conf import settings

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

from django.db import models

from django.db import transaction

from django.utils.translation import ugettext_lazy as _

from guardian.shortcuts import assign, remove_perm, get_perms

from ldt.core.models import Document

from ldt.security import (get_current_user_or_admin, set_current_user, 

    get_current_user)

from ldt.security.manager import SafeManager

from ldt.security.models import SafeModel

from sorl.thumbnail import ImageField

from tagging.models import Tag

from utils import (create_ldt, copy_ldt, create_empty_iri, update_iri, 

    generate_uuid)

from ldt.utils import generate_hash 

import datetime

import lxml.etree

import mimetypes

import os.path

import re

import tagging.fields

import uuid

import logging

from shutil import rmtree, move

from django.core.files.storage import default_storage

#from ldt.core.models import Document, Owner

class Author(SafeModel):

    handle = models.CharField(max_length=255, unique=True, blank=True, null=True)

    email = models.EmailField(unique=False, blank=True, null=True)

    firstname = models.CharField(max_length=512, blank=True, null=True)

    lastname = models.CharField(max_length=512, blank=True, null=True)

    def __unicode__(self):

        return unicode(self.id) + " - " + self.handle + ", " + self.email + ", " + self.firstname + " " + self.lastname

    class Meta:

        permissions = (

                       ('view_author', 'Can view author'),

                       )

class Media(SafeModel):

    external_id = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('media.external_id'))

    external_permalink = models.URLField(max_length=1024, verify_exists=False, null=True, blank=True, verbose_name=_('media.external_permalink'))

    external_publication_url = models.URLField(max_length=1024, verify_exists=True, null=True, blank=True, verbose_name=_('media.external_publication_url'))

    external_src_url = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('media.external_src_url'))

    creation_date = models.DateTimeField(auto_now_add=True, verbose_name=_('media.creation_date'))

    media_creation_date = models.DateTimeField(null=True, blank=True, verbose_name=_('media.media_creation_date'))

    update_date = models.DateTimeField(auto_now=True, verbose_name=_('media.update_date'))

    videopath = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('media.videopath'))

    duration = models.IntegerField(null=True, blank=True, verbose_name=_('media.duration'))

    creator = models.ForeignKey(User, blank=True, null=True, verbose_name=_('media.creator'))

    description = models.TextField(null=True, blank=True, verbose_name=_('description'))

    title = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('title'))

    src = models.CharField(max_length=1024, verbose_name=_('media.src'))

    #TODO: use a fixed length char field

    src_hash = models.CharField(max_length=128, unique=True, verbose_name=_('media.src_hash'), blank=True)

    mimetype_field = models.CharField(max_length=512, null=True, blank=True, verbose_name=_('media.mimetype'))

    class Meta:

        permissions = (

                       ('view_media', 'Can view media'),

                       )

    def mimetype(): #@NoSelf

        def fget(self):

            if self.mimetype_field :

                return self.mimetype_field

            elif self.src:

                return mimetypes.guess_type(self.src.rstrip())[0]

            else:

                return None

        def fset(self, value):

            self.mimetype_field = value

        return locals()

    mimetype = property(**mimetype())

    def stream_src(): #@NoSelf

        def fget(self):

            res_src = self.src.rstrip()

            if self.videopath and self.videopath.startswith("rtmp://") and "mp3:" not in res_src and "mp4:" not in res_src:

                extension = res_src.split(".")[-1]

                res_src = {

                    'flv': lambda s: s,

                    'mp3': lambda s: "%s:%s" % ("mp3", res_src[:-4]),

                    'mp4': lambda s: "%s:%s" % ("mp4", res_src),

                    'f4v': lambda s: "%s:%s" % ("mp4", res_src),

                    'mov': lambda s: "%s:%s" % ("mp4", res_src),

                }.get(extension, lambda s:s)(res_src)

            return res_src

        return locals()

    stream_src = property(**stream_src())

    def is_public(): #@NoSelf

        def fget(self):

            if self.pk:

                everyone = Group.objects.get(name=settings.PUBLIC_GROUP_NAME)

                if 'view_media' in get_perms(everyone, self):

                    return True

            return False

        def fset(self, value):

            if self.pk:

                everyone = Group.objects.get(name=settings.PUBLIC_GROUP_NAME)

                if value:

                    assign('ldt_utils.view_media', everyone, self.media_obj)               

                else:

                    remove_perm('ldt_utils.view_media', everyone, self.media_obj)

        return locals()

    is_public = property(**is_public())

    def save(self, *args, **kwargs):

        self.src_hash = generate_hash(self.src)

        super(Media, self).save(*args, **kwargs)

        for content in self.content_set.all():

            content.sync_iri_file()

    def __unicode__(self):

        strings = []

        if self.title:

            strings.append(unicode(self.title))

        else:

            strings.append(unicode(self.src))

        if self.external_id:

            strings.append(unicode(self.external_id))

        return "|".join(strings)

class ContentManager(SafeManager):

    def __init__(self):

        super(ContentManager, self).__init__(check_perm=False)

    def get_by_natural_key(self, iri_id):

        return self.get(iri_id=iri_id)

class Content(SafeModel):

    objects = ContentManager()

    iri_id = models.CharField(max_length=255, unique=True, default=generate_uuid, verbose_name=_('content.iri_id'))

    iriurl = models.CharField(max_length=1024, verbose_name=_('content.iriurl'))

    creation_date = models.DateTimeField(auto_now_add=True, verbose_name=_('content.creation_date'))

    update_date = models.DateTimeField(auto_now=True, verbose_name=_('content.update_date'))

    title = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('content.title'))

    description = models.TextField(null=True, blank=True, verbose_name=_('content.description'))

    authors = models.ManyToManyField(Author, blank=True, verbose_name=_('content.authors'))

    duration = models.IntegerField(null=True, blank=True, verbose_name=_('content.duration'))

    content_creation_date = models.DateTimeField(null=True, blank=True, verbose_name=_('content.content_creation_date'))

    tags = tagging.fields.TagField(max_length=2048, null=True, blank=True)

    media_obj = models.ForeignKey('Media', blank=True, null=True)

    image = ImageField(upload_to="thumbnails/contents/", default=settings.DEFAULT_CONTENT_ICON, max_length=200)

    front_project = models.ForeignKey('Project', null=True, blank=True)

    class Meta:

        ordering = ["title"]

        permissions = (

                       ('view_content', 'Can view content'),

                       )

    def __init__(self, *args, **kwargs):

        super(Content, self).__init__(*args, **kwargs)

        if not hasattr(Content, 'pol_positive'):

            self.__add_polemic_attributes()

    def delete(self):

        super(Content, self).delete()

        iri_file_path = self.iri_file_path()

        thumbnail = os.path.join(settings.MEDIA_ROOT, unicode(self.image))

        if os.path.exists(iri_file_path):

            dir = os.path.dirname(iri_file_path)

            temp = os.path.join(os.path.join(os.path.dirname(dir), "temp"), self.iri_id)             

            try:

                move(dir, temp)

            except Exception, e:

                raise e

        if os.path.exists(thumbnail):

            if os.path.basename(thumbnail) != os.path.basename(settings.DEFAULT_CONTENT_ICON):

                temp_thumbnail = os.path.join(os.path.dirname(thumbnail), "temp")

                try:    

                    if not os.path.exists(temp_thumbnail):

                        os.makedirs(temp_thumbnail)

                    move(thumbnail, os.path.join(temp_thumbnail, os.path.basename(thumbnail)))            

                except Exception, e:

                    raise e

    #del .iri, and .png from temp directory

    def commit(self):

        iri_file_path=self.iri_file_path()

        dir = os.path.dirname(iri_file_path)

        temp = os.path.join(os.path.join(os.path.dirname(dir), "temp"), self.iri_id)

        thumbnail = os.path.join(settings.MEDIA_ROOT, unicode(self.image))

        temp_thumbnail = os.path.join(os.path.dirname(thumbnail), "temp")

        if os.path.exists(temp):

            default_storage.delete(os.path.join(temp, os.path.basename(self.iriurl)))

            os.rmdir(temp)

        if os.path.exists(temp_thumbnail):

            default_storage.delete(os.path.join(temp_thumbnail, os.path.basename(thumbnail)))

            os.rmdir(temp_thumbnail)

    #move .iri, and .png to there original directory

    def rollback(self):

        iri_file_path=self.iri_file_path()

        dir = os.path.dirname(iri_file_path)

        temp = os.path.join(os.path.join(os.path.dirname(dir), "temp"), self.iri_id)

        thumbnail = os.path.join(settings.MEDIA_ROOT, unicode(self.image))

        temp_thumbnail = os.path.join(os.path.dirname(thumbnail), "temp")

        if os.path.exists(temp):

            move(temp, dir)

            os.rmdir(os.path.dirname(temp))

        if os.path.exists(temp_thumbnail):

            move(os.path.join(temp_thumbnail, os.path.basename(thumbnail)), os.path.dirname(thumbnail))

            os.rmdir(temp_thumbnail)

    def natural_key(self):

        return self.iri_id

    # added for import

    def get_by_natural_key(self, iri_id):

        return self.get(iri_id=iri_id)

    def get_duration(self):

        if self.duration is None:

            doc = lxml.etree.parse(self.iri_file_path())

            res = doc.xpath("/iri/body/medias/media[@id='video']/video")

            if len(res) > 0:

                try:

                    self.duration = int(res[0].get(u'dur', 0) or 0)

                except:

                    self.duration = 0

            else:

                self.duration = 0

            self.save()

        return self.duration    

    def mimetype(): #@NoSelf

        def fget(self):

            if self.media_obj:

                return self.media_obj.mimetype

            else:

                return None

        return locals()

    mimetype = property(**mimetype())

    def sync_iri_file(self):

        # create iri file if needed

        created = False

        try:

            iri_file_path = self.iri_file_path()

            if not os.path.exists(iri_file_path):

                dir = os.path.dirname(iri_file_path)

                if not os.path.exists(dir):

                    os.makedirs(dir)

                created = True

                file = open(iri_file_path, "w")

                create_empty_iri(file, self, "IRI")

            else:

                created = False

                update_iri(iri_file_path, self, "IRI")

        except Exception, e:

            if created:

                if os.path.exists(iri_file_path):

                    os.remove(iri_file_path)

            raise e

    #TODO: better manage the change in .iri name and error scenario (save in temp file + rename

    def save(self, *args, **kwargs):

        create_front_project = False

        if not self.pk:

        super(Content, self).save(*args, **kwargs)

        if create_front_project: 

            # We need a primary key for self in create_project, so

            # save() has to be called first

            self.create_front_project()

            assign('ldt_utils.change_content', get_current_user(), self)

    def __unicode__(self):

        return str(self.id) + ": " + self.iri_id

    def iri_url(self, web_url=settings.WEB_URL):

        if 'http' in self.iriurl or 'https' in self.iriurl:

            return self.iriurl

        else:

            return unicode(web_url) + unicode(settings.MEDIA_URL) + u"ldt/" + unicode(self.iriurl)

    def iri_file_path(self):

        return os.path.join(os.path.join(os.path.join(settings.MEDIA_ROOT, "ldt"), self.iri_id), os.path.basename(self.iriurl))

    def iri_url_template(self):

        return "${web_url}${media_url}ldt/" + unicode(self.iri_id) + "/" + os.path.basename(self.iriurl)

    def __get_empty_media(self):

        if settings.EMPTY_MEDIA_EXTERNALID:

            empty_media = Media.objects.get(external_id=settings.EMPTY_MEDIA_EXTERNALID)

            return empty_media

        else:

            return None

    def stream_src(): #@NoSelf

        def fget(self):

            if self.media_obj is not None:

                return self.media_obj.stream_src

            else:

                empty_media = self.__get_empty_media()

                if empty_media:

                    return empty_media.stream_src

                else:

                    return ""

        return locals()

    stream_src = property(**stream_src())