from django import forms

from django.core.urlresolvers import reverse

from django.http import HttpResponse, HttpResponseRedirect, Http404

from django.shortcuts import render_to_response

from django.template.loader import render_to_string

from django.template import RequestContext

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

from django.contrib.auth.models import User

from cm.converters.pandoc_converters import pandoc_convert, do_tidy

from cm.models import Text, TextVersion, Attachment, Comment

from cm.security import get_viewable_comments

import mimetypes

import simplejson

from cm.cm_settings import USE_ABI

EXPORT2_INFOS = {

# key -> { mimetype, extension}

's5' :   {},

'pdf' :  {'mimetype': 'application/pdf', 'extension':'pdf'},

'markdown' :  {'mimetype': 'text/plain', 'extension':'mkd'},

'odt' :  {'mimetype': 'application/vnd.oasis.opendocument.text', 'extension':'odt'},

'doc' :  {'mimetype': 'application/msword', 'extension':'doc'},

'docx' :  {'mimetype': 'application/vnd.openxmlformats-officedocument.wordprocessingml.document', 'extension':'docx'},

'latex' :{'mimetype': 'text/x-tex', 'extension':'tex'},

'html' :{'mimetype': 'text/html', 'extension':'html'},

'epub' :{'mimetype': 'application/epub+zip', 'extension':'epub'},

'raw' : {'mimetype': 'text/plain', 'extension':'txt'},

'xml' : {'mimetype': 'text/xml', 'extension':'xml'},

}

def content_export2(request, content, title, content_format, format, use_pandoc, download_response):

    # TODO : formats must be imported from converters

#    import pdb;pdb.set_trace()

    if format == 'raw' :

        export_content = content

    elif content_format == 'html' and format == 'html':

        export_content = HTML_HEADER % content

    elif content_format == 'markdown' and format == 'markdown':

        export_content = content

    else:

        if use_pandoc :

          # markdown2pdf is buggy => convert to HTML and use abiword to export in PDF

            html_content = pandoc_convert(content, content_format, 'html', full=True)

            from cm.converters.abi_converters import AbiFileConverter

            converter = AbiFileConverter()

            export_content = converter.convert_from_html(fix_content, format)

          else:

            export_content = pandoc_convert(content, content_format, format, full=True)

        else :

            fix_content = content

            if content_format == 'html':

                if USE_ABI:

                  from cm.converters.abi_converters import AbiFileConverter

                  converter = AbiFileConverter()

                else:

                  from cm.converters.oo_converters import combine_css_body                

                  fix_content = combine_css_body(content, '')

            if USE_ABI:

              from cm.converters.abi_converters import AbiFileConverter

              converter = AbiFileConverter()

              # replaces images url by their actual path

              from django.conf import settings

              site_url = settings.SITE_URL

              import re

              attach_re = r'/attach/(?P<attach_key>\w*)/'

              attach_str = r'%s/attach/%s/'

              for match in re.findall(attach_re, fix_content):

                link = attach_str %(site_url, match)

                attach = Attachment.objects.get(key=match)

                fix_content = fix_content.replace(link, attach.data.path)

              export_content = converter.convert_from_html(fix_content, format)

            else:

              from cm.converters.oo_converters import convert_html as oo_convert                

              export_content = oo_convert(fix_content, format)

    export_infos = EXPORT2_INFOS[format]

    if download_response:

        return _response_download(export_content, title, export_infos['mimetype'], export_infos['extension']) ;

    else:

        return _response_write(export_content)    

def _response_download(content, title, mimetype, extension):

    response = HttpResponse(mimetype=mimetype)

    file_title = title + '.' + extension

    from email.header import Header

    encoded_name = str(Header(file_title.encode('utf8'), charset='utf8', maxlinelen=500))

    # TODO: find a way to include long (more than 76 chars) into header

    encoded_name = encoded_name.replace('\n','')

    response['Content-Disposition'] = 'attachment; filename=%s' % encoded_name

    response.write(content)

    return response        

def _response_write(content):

    response = HttpResponse()

    response.write(content)

    return response

def xml_export(request, text_version, whichcomments):

  # Text version infos

  # Comments

  comments = [] # whichcomments=="none"

  if whichcomments == "filtered" or whichcomments == "all":

    _comments = text_version.comment_set.all()

    if whichcomments == "filtered" :

      filteredIds = []

      if request.method == 'POST' :

        ll = request.POST.get('filteredIds',[]).split(",")

        filteredIds = [ int(l) for l in ll if l]

      _comments = text_version.comment_set.filter(id__in=filteredIds)

    comments = get_viewable_comments(request, _comments, text_version, order_by=('start_wrapper','start_offset','end_wrapper','end_offset'))

    # Add user name/email if missing comment name/email

    for comment in comments:

      users = User.objects.filter(id=comment.user_id)

      if not(comment.name):

        comment.name = users[0].username

      if not(comment.email):

        comment.email = users[0].email

    template_dict['comments'] = comments

  # Author

  users = User.objects.filter(id=text_version.user_id)

  if text_version.name:

    template_dict['name'] = text_version.name

  else:

    template_dict['name'] = users[0].username

  if text_version.email:

    template_dict['email'] = text_version.email

  else:

    template_dict['email'] = users[0].email

  # Renders template

  export_content = render_to_string('site/export.xml', template_dict, context_instance=RequestContext(request))

  # Returns HTTP response

  export_infos = EXPORT2_INFOS['xml']

  return _response_download(export_content, text_version.title, export_infos['mimetype'], export_infos['extension']) ;