import time
import mimetypes
import os
import re
import stat
import urllib
try:
    from cStringIO import StringIO
except ImportError:
    from StringIO import StringIO

try:
    # python 2.6
    import json
except:
    # python 2.4 with simplejson
    import simplejson as json

from django.template import Context, RequestContext, loader
from django.http import Http404, HttpResponse, HttpResponseRedirect, HttpResponseNotModified
from django.core.urlresolvers import reverse
from django.template.loader import get_template
from django.contrib.auth.decorators import login_required
from django.core.exceptions import ObjectDoesNotExist, SuspiciousOperation
from django.utils.http import http_date                 
from django.conf import settings
from django.views.static import was_modified_since
from django.core.files.uploadedfile import SimpleUploadedFile
from django.core.context_processors import csrf
from django.utils.translation import ugettext as _

from twistranet.twistapp.models import *
from twistranet.twistapp.forms.resource_forms import ResourceForm, ResourceBrowserForm
from twistranet.twistapp.lib.decorators import require_access
from twistranet.twistapp.lib.log import log
from twistranet.twistorage.storage import Twistorage
from twistranet.twistapp.lib import utils
from twistranet.core.views import *

def serve(request, path, document_root = None, show_indexes = False, nocache = False):
    """
    Adapted from django.views.static to handle the creation/modification date of the resource's publisher
    instead of only the file's value.
    
    Serve static files below a given point in the directory structure.

    To use, put a URL pattern such as::

        (r'^(?P<path>.*)$', 'django.views.static.serve', {'document_root' : '/path/to/my/files/'})

    in your URLconf. You must provide the ``document_root`` param. You may
    also set ``show_indexes`` to ``True`` if you'd like to serve a basic index
    of the directory.  This index view will use the template hardcoded below,
    but if you'd like to override it, you can create a template called
    ``static/directory_index.html``.
    """

    # Clean up given path to only allow serving files below document_root.
    path = urllib.unquote(path).lstrip('/')
    newpath = ''
    for part in path.split('/'):
        if not part:
            # Strip empty path components.
            continue
        drive, part = os.path.splitdrive(part)
        head, part = os.path.split(part)
        if part in (os.curdir, os.pardir):
            # Strip '.' and '..' in path.
            continue
        newpath = os.path.join(newpath, part).replace('\\', '/')
    if newpath and path != newpath:
        return HttpResponseRedirect(newpath)
    fullpath = os.path.join(document_root, newpath)
    if os.path.isdir(fullpath):
        # if show_indexes:
        #     return directory_index(newpath, fullpath)
        raise Http404("Directory indexes are not allowed here.")
    if not os.path.exists(fullpath):
        raise Http404('"%s" does not exist' % fullpath)
    
    # Respect the If-Modified-Since header.
    statobj = os.stat(fullpath)
    mimetype = mimetypes.guess_type(fullpath)[0] or 'application/octet-stream'
    # XXX TODO: Handle this correctly!!! Disabled to avoid using file mod. time instead of obj mod. time
    if not nocache:
        if not was_modified_since(
            request.META.get(
                'HTTP_IF_MODIFIED_SINCE'),
                    statobj[stat.ST_MTIME], statobj[stat.ST_SIZE]
        ):
            return HttpResponseNotModified(mimetype=mimetype)
            
    # XXX This is subsubsuboptimal!
    contents = open(fullpath, 'rb').read()
    response = HttpResponse(contents, mimetype=mimetype)
    response["Last-Modified"] = http_date(statobj[stat.ST_MTIME])
    response["Content-Length"] = len(contents)
    return response

def _getResourceResponse(request, resource, last_modified = None, force_download = False):
    """
    Return the proper HTTP stream for a resource object
    XXX TODO: Handle the last_modified parameter
    If force_download is True, then we return attachment instead of inline.
    """
    # Determinate the appropriate rendering scheme: file or URL
    if resource.resource_url:
        raise NotImplementedError("We yet have to implement resource URLs")
    
    # Resource file: get storage and check if path exists
    elif resource.resource_file:
        path = resource.resource_file.name      # XXX TODO: Handle subdirectories somehow here
        storage = Twistorage()
        if not storage.exists(path):
            raise Http404

    # Neither a file nor a URL? Then it's probably invalid.
    # Maybe we should implement an "alias" type of resource?
    else:
        raise ValueError("Invalid resource: %s" % resource)

    # Return the underlying file, adapt the Last-Modified header as necessary
    response = serve(request, path, document_root = storage.location, show_indexes = False, nocache = True)
    response["Content-Type"] = resource.mimetype
    content_disposition = force_download and "attachment" or "inline"
    response["Content-Disposition"] = "%s; filename=\"%s\"" % (content_disposition, urllib.quote(resource.filename.encode("ascii", "ignore")))
    return response
    
@require_access
def resource_cache(request, cache_path):
    """
    Return a resource by its cache path.
    """
    return serve(request, cache_path, os.path.join(settings.MEDIA_ROOT, "cache"))

@require_access
def resource_by_id(request, resource_id):
    """
    Return a resource by id
    """
    resource = Resource.objects.get(id = resource_id)
    return _getResourceResponse(request, resource)

@require_access
def resource_by_slug(request, slug):
    """
    XXX TODO: Make this more efficient?
    """
    resource = Resource.objects.get(slug = slug)
    return _getResourceResponse(request, resource)

@require_access
def download_by_id(request, resource_id):
    """
    Return a resource by id
    """
    resource = Resource.objects.get(id = resource_id)
    return _getResourceResponse(request, resource, force_download = True)

@require_access
def download_by_slug(request, slug):
    """
    XXX TODO: Make this more efficient?
    """
    resource = Resource.objects.get(slug = slug)
    return _getResourceResponse(request, resource, force_download = True)

@require_access
def resource_by_slug_or_id(request, slug_or_id):
    """
    Return a resource by its alias or by its id if not found
    """
    try:
        resource = Resource.objects.get(slug = slug_or_id)
    except ObjectDoesNotExist:
        resource = Resource.objects.get(id = slug_or_id)
    return _getResourceResponse(request, resource)
    
@require_access
def resource_by_account(request, account_id, property):
    """
    Fetch a resource by an account attribute.
    """
    account = Account.objects.get(id = account_id)
    resource = getattr(account, property, None)
    if isinstance(resource, Resource):
        return _getResourceResponse(request, resource)
    else:
        # XXX should raise 404 if pty not found
        raise ValueError("Invalid property")
        
@require_access
def resource_by_content(request, content_id, property):
    """
    Fetch a resource by an content attribute.
    """
    content = Content.objects.get(id = content_id)
    resource = getattr(content, property, None)
    if isinstance(resource, Resource):
        return _getResourceResponse(request, resource)
    else:
        # XXX should raise 404 if pty not found
        raise ValueError("Invalid property")

@require_access
def edit_resource(request, resource_id = None):
    """
    Edit the given resource or create a new one if necessary
    """
    # Get basic information
    account = request.user.get_profile()
    if resource_id is not None:
        resource = Resource.objects.get(id = resource_id)
        # XXX TODO: Implement security here.
        # if not resource.can_view:
        #     raise NotImplementedError("Should implement a permission denied exception here")
        # if not resource.can_edit:
        #     raise NotImplementedError("Should redirect to the regular view? or raise a permission denied exception here.")
        # form_entry = form_registry.getFormEntry(content.content_type)
    else:
        # XXX TODO: Check some kind of "can_create_content" permission?
        resource = None
        # form_entry = form_registry.getFormEntry(content_type)

    # Process form
    if request.method == 'POST': # If the form has been submitted...
        if resource:
            form = ResourceForm(request.POST, request.FILES, instance = resource)
        else:
            form = ResourceForm(request.POST, request.FILES)

        if form.is_valid():
            return HttpResponseRedirect(reverse('twistranet.views.resource_by_id', args = (resource.id,)))
    else:
        if resource:
            form = ResourceForm(instance = resource.object)
        else:
            form = ResourceForm()

    # Template hapiness
    t = loader.get_template('resource/edit.html')
    c = RequestContext(
        request,
        {
            "account": account,
            "resource": resource,
            "form": form,
        },
        )
    return HttpResponse(t.render(c))

@require_access
def create_resource(request):
    """
    """
    return edit_resource(request)


class ResourceBrowser(BaseView):
    """
    A view used to browse and upload resources
    Used by wysiwyg editors
    Based on resource field
    """
    name = "resource_browser"
    template_variables = BaseView.template_variables + [
        "account",
        "actions",
        "allow_browser_selection",
        "form",
    ]
    template = 'resource/resource_browser_form.html'
    title = _("Resource browser")
    
    def prepare_view(self,):
        self.account = self.auth
        self.actions = None
        self.allow_browser_selection = int(self.request.GET.get('allow_browser_selection', 0) or 0)
        self.form = ResourceBrowserForm()


###############################
# Resource Quick Upload views #
###############################


# JS String used inline by resource_quickupload template
# XXX TODO JMG : some cleaning (put template in fileupload.js, remove unused methods ...)
UPLOAD_JS = """       
    var fillTitles = %(ul_fill_titles)s;
    var auto = %(ul_auto_upload)s;
    var uploadparams = {};
    if (typeof getActivePublisher!='undefined') {
        uploadparams['publisher_id'] = getActivePublisher();
    }
    uploadparams['csrfmiddlewaretoken'] = '%(ul_csrf_token)s';
    addUploadFields_%(ul_id)s = function(file, id) {
        var uploader = xhr_%(ul_id)s;
        TwistranetQuickUpload.addUploadFields(uploader, uploader._element, file, id, fillTitles);
    }
    sendDataAndUpload_%(ul_id)s = function() {
        var uploader = xhr_%(ul_id)s;
        TwistranetQuickUpload.sendDataAndUpload(uploader, uploader._element, '%(typeupload)s');
    }    
    clearQueue_%(ul_id)s = function() {
        var uploader = xhr_%(ul_id)s;
        TwistranetQuickUpload.clearQueue(uploader, uploader._element);    
    }
    onUploadComplete_%(ul_id)s = function(id, fileName, responseJSON) {       
        var uploader = xhr_%(ul_id)s;
        TwistranetQuickUpload.onUploadComplete(uploader, uploader._element, id, fileName, responseJSON);
    } 
    createUploader_%(ul_id)s= function(){    
        xhr_%(ul_id)s = new qq.FileUploader({
            element: jQuery('#%(ul_id)s')[0],
            action: '%(home_url)sresource_quickupload_file/',
            params: uploadparams,
            autoUpload: auto,
            onAfterSelect: addUploadFields_%(ul_id)s,
            onComplete: onUploadComplete_%(ul_id)s,
            allowedExtensions: %(ul_extensions_list)s,
            sizeLimit: %(ul_xhr_size_limit)s,
            simUploadLimit: %(ul_sim_upload_limit)s,
            template: '<div class="qq-uploader">' +
                      '<div class="qq-upload-drop-area"><span>%(ul_draganddrop_text)s</span></div>' +
                      '<div class="qq-upload-button">%(ul_button_text)s</div>' +
                      '<ul class="qq-upload-list"></ul>' + 
                      '</div>',
            fileTemplate: '<li>' +
                    '<a class="qq-upload-cancel" href="#">&nbsp;</a>' +
                    '<div class="qq-upload-infos"><span class="qq-upload-file"></span>' +
                    '<span class="qq-upload-spinner"></span>' +
                    '<span class="qq-upload-failed-text">%(ul_msg_failed)s</span></div>' +
                    '<div class="qq-upload-size"></div>' +
                '</li>',                      
            messages: {
                serverError: "%(ul_error_server)s",
                typeError: "%(ul_error_bad_ext)s {file}. %(ul_error_onlyallowed)s {extensions}.",
                sizeError: "%(ul_error_file_large)s {file}, %(ul_error_maxsize_is)s {sizeLimit}.",
                emptyError: "%(ul_error_empty_file)s {file}, %(ul_error_try_again_wo)s",
                unexpectedError: "%(ul_error_unexpected)s"
            }            
        });           
    }
    jQuery(document).ready(createUploader_%(ul_id)s); 
"""

def _content_types_infos (media_type):
    """
    return content types infos depending on media_type
    media_type could be 'image', 'video', 'audio' or any
    extensions string like '*.doc;*.pdf'
    return a tuple (string for flashupload, list for html5 uploader, upload label)
    """
    ext = '*.*;'
    extlist = []
    msg = u'Choose files to upload'
    if media_type == 'image' :
        ext = '*.jpg;*.jpeg;*.gif;*.png;'
        msg = u'Choose images to upload'
    elif media_type == 'video' :
        ext = '*.flv;*.avi;*.wmv;*.mpg;'
        msg = u'Choose video files to upload'
    elif media_type == 'audio' :
        ext = '*.mp3;*.wav;*.ogg;*.mp4;*.wma;*.aif;'
        msg = u'Choose audio files to upload'
    elif media_type == 'flash' :
        ext = '*.swf;'
        msg = u'Choose flash files to upload'
    elif media_type and media_type != 'file':
        # you can also pass a list of extensions in media_type request var
        # with this syntax '*.aaa;*.bbb;'
        ext = media_type 
        msg = u'Choose file for upload : ' + ext 
    try :
        extlist = [f.split('.')[1].strip() for f in ext.split(';') if f.strip()]
    except :
        ext = '*.*;'
        extlist = []
    if extlist==['*'] :
        extlist = []
    return ( ext, extlist, _(msg))


# This view is rendering html with inline javascript and is called in ajax
# XXX TODO (JMG), not urgent : call it with a simple include
@require_access
def resource_quickupload(request):
    media_type = str(request.GET.get('media_type',u'file'))
    content_types_infos = _content_types_infos(media_type)
    qu_settings = dict(
        typeupload             = 'File',
        home_url               =  reverse("twistranet_home"),
        ul_id                  = 'tnuploader', # improve it to get multiple uploader in a same page, change it also in 'resource_quickupload.html'
        ul_extensions_list     = str(content_types_infos[1]), #could be "['jpg,'png','gif']"
        ul_fill_titles         = settings.QUICKUPLOAD_FILL_TITLES and 'true' or 'false',
        ul_auto_upload         = settings.QUICKUPLOAD_AUTO_UPLOAD and 'true' or 'false',
        ul_xhr_size_limit      = settings.QUICKUPLOAD_SIZE_LIMIT and str(settings.QUICKUPLOAD_SIZE_LIMIT*1024) or '0',
        ul_sim_upload_limit    = str(settings.QUICKUPLOAD_SIM_UPLOAD_LIMIT),
        ul_button_text         = _(u'Browse'),
        ul_draganddrop_text    = _(u'Drag and drop files to upload'),
        ul_msg_all_sucess      = _( u'All files uploaded with success.'),
        ul_msg_some_sucess     = _( u' files uploaded with success, '),
        ul_msg_some_errors     = _( u" uploads return an error."),
        ul_msg_failed          = _( u"Failed"),
        ul_csrf_token          = csrf(request)['csrf_token'],
        ul_error_try_again_wo  = _( u"please select files again without it."),
        ul_error_try_again     = _( u"please try again."),
        ul_error_empty_file    = _( u"This file is empty:"),
        ul_error_file_large    = _( u"This file is too large:"),
        ul_error_maxsize_is    = _( u"maximum file size is:"),
        ul_error_bad_ext       = _( u"This file has invalid extension:"),
        ul_error_onlyallowed   = _( u"Only allowed:"),
        ul_error_server        = _( u"Server error, please contact support and/or try again."),
        ul_error_unexpected    = _( u"Unexpected error, please use the 'Browse' button."),
    )
    qu_script = UPLOAD_JS % qu_settings
    c = Context({ 'qu_script': qu_script, }) 
    t = loader.get_template('resource/resource_quickupload.html')
    return HttpResponse(t.render(c))

@require_access
def resource_quickupload_file(request):
    """
    json view used by quikupload script
    when uploading a file
    return success/error + file infos (url/preview/title ...)
    """               
    msg = {}
    if request.environ.get('HTTP_X_REQUESTED_WITH') == 'XMLHttpRequest':
        file_name = urllib.unquote(request.environ['HTTP_X_FILE_NAME'])
        title = request.GET.get('title', '')
        upload_with = "XHR"        
        try:
            # the solution for sync ajax file upload
            file_data = SimpleUploadedFile(file_name, request.raw_post_data)
        except:
            log.debug("XHR Upload of %s has been aborted" %file_name)
            # not really useful here since the upload block
            # is removed by "cancel" action, but
            # could be useful if someone change the js behavior
            msg = {u'error': u'emptyError'}
    else:
        # MSIE fallback behavior (classic upload with iframe)
        file_data = request.FILES.get("qqfile", None)
        filename = getattr(file_data,'name', '')
        file_name = filename.split("\\")[-1]
        title = request.POST.get('title', '')
        upload_with = "CLASSIC FORM POST"
        # we must test the file size in this case (because there is no size client test with msie file field)
        if not utils._check_file_size(file_data):
            log.debug("The file %s is too big, quick iframe upload rejected" % file_name) 
            msg = {u'error': u'sizeError'}

    if file_data and not msg:
        publisher_id = request.GET.get('publisher_id', request.POST.get('publisher_id', ''))
        try:                publisher_id = int(publisher_id)
        except ValueError:  publisher_id = None
        content_type = mimetypes.guess_type(file_name)[0] or 'application/octet-stream'

        try:
            # Create the resource itself
            resource = Resource.objects.create(
                resource_file = file_data,
                title = title,
                publisher_id = publisher_id and int(publisher_id),
                filename = file_name,
            )
            is_image = resource.is_image
            type = is_image and 'image' or 'file'
            # Generate the preview thumbnails
            thumbnails = resource.thumbnails
            msg = {
                'success':       True,
                'value':         resource.id,
                'url':           resource.get_absolute_url(),
                'thumbnail_url': is_image and thumbnails['medium'].url or thumbnails['big_icon'].url,
                'mini_url':      is_image and thumbnails['summary_preview'].url or thumbnails['big_icon'].url,
                # only used by image size selection (wysiwyg browser)
                'summary_url':   is_image and thumbnails['summary'].url or '',
                'preview_url':   is_image and thumbnails['preview'].url or '',
                'legend':        title and title or file_name, 
                'scope':         publisher_id,
                'type':          type
            }
        except:
            # TODO : improve error messages with Unauthorized error
            log.exception("Unexpected error while trying to upload a file.")
            msg = {u'error': u'unexpectedError'}
    else:
        msg = {u'error': u'serverError'}

    return HttpResponse( json.dumps(msg),
                         mimetype='text/html')

###############################
# Resource Browser json views #
###############################

@require_access
def resource_by_publisher_json(request, publisher_id):
    """
    return all resources for a publisher id
    as json dict (for images dict contains all thumb urls) 
    """
    # minimal security check
    publisher_id = int(publisher_id)
    account = Twistable.objects.getCurrentAccount(request)
    selectable_accounts_ids = [a.id for a in Resource.objects.selectable_accounts(account)] # XXX Suboptimal, should be 1! query
    request_account = Account.objects.get(id = publisher_id)
    if not publisher_id == account.id:
        if int(publisher_id) not in selectable_accounts_ids:
            raise SuspiciousOperation("Attempted access to '%s' denied." % request_account.slug)
    
    selection = request.GET.get('selection','')  or 0
    media_type = str(request.GET.get('media_type',u'file'))
    # XXX TODO (JMG) : use haystack for search and batch
    batch_size = 30
    files = Resource.objects.filter(publisher=request_account)
    results = []
    i = 0
    for file_ in files:
        is_image = file_.is_image
        type_ = is_image and 'image' or 'file'
        if media_type=='file' or (media_type=='image' and type_=='image'):
            if len(results) <= batch_size:
                is_selected = file_.id == (int(selection) or 0)
                thumbnails = file_.thumbnails
                result = {
                        "url":              file_.get_absolute_url(),
                        "thumbnail_url":    is_image and thumbnails['medium'].url or thumbnails['big_icon'].url,
                        "mini_url":         is_image and thumbnails['summary_preview'].url or thumbnails['big_icon'].url,
                        # only used by image size selection (wysiwyg browser)
                        "summary_url":      is_image and thumbnails['summary'].url or '',
                        "preview_url":      is_image and thumbnails['preview'].url or '',
                        "id":               file_.id,
                        "title":            file_.title,
                        "selected":         is_selected and ' checked="checked"' or '',
                        "type":             type_,
                        }
                results.append(result)
    data = {}
    # XXX TODO : change it for batch
    data['page'] = 1
    data['results'] = results
    # used only by Wysiwyg browser window
    data['sizes'] = ((_('Small size'), '100*100'), (_('Medium size'), '500*500'), (_('Full size'), 'full') )
    return HttpResponse( json.dumps(data),
                         mimetype='text/plain')
