import os
from re import compile
import warnings

from django.conf import settings
from django.contrib.sites.models import Site
from django.template.context import RequestContext
from django.template.response import HttpResponse
from django.views.generic import TemplateView

from wkhtmltopdf.utils import template_to_temp_file, wkhtmltopdf


class PDFResponse(HttpResponse):
    def __init__(self, content, *args, **kwargs):
        filename = kwargs.pop('filename', None)
        super(PDFResponse, self).__init__(content, 'application/pdf', *args, **kwargs)
        if filename:
            header_content = 'attachment; filename={0}'.format(filename)
            self.__setitem__('Content-Disposition', header_content)


class PdfResponse(PDFResponse):
    def __init__(self, content, filename):
        warnings.warn('PdfResponse is deprecated in favour of PDFResponse. It will be removed in version 1.',
                      PendingDeprecationWarning, 2)
        super(PdfResponse, self).__init__(content, filename)


class PDFTemplateView(TemplateView):
    filename = 'rendered_pdf.pdf'
    footer_template = None
    header_template = None
    orientation = 'portrait'
    margin_bottom = 0
    margin_left = 0
    margin_right = 0
    margin_top = 0
    response = PDFResponse
    _tmp_files = None

    def __init__(self, *args, **kwargs):
        self._tmp_files = []
        super(PDFTemplateView, self).__init__(*args, **kwargs)

    def get(self, request, context_instance=None, *args, **kwargs):
        if request.GET.get('as', '') == 'html':
            return super(PDFTemplateView, self).get(request, *args, **kwargs)

        if context_instance:
            self.context_instance = context_instance
        else:
            self.context_instance = RequestContext(request, self.get_context_data(**kwargs))

        page_path = template_to_temp_file(self.get_template_names(), self.get_context_data(), self.context_instance)
        pdf_kwargs = self.get_pdf_kwargs()
        output = wkhtmltopdf(page_path, **pdf_kwargs)
        if self._tmp_files:
            map(os.remove, self._tmp_files)
        return self.response(output, filename=self.get_filename())

    def get_filename(self):
        return self.filename

    def get_pdf_kwargs(self):
        kwargs = {
            'margin_bottom': self.margin_bottom,
            'margin_left': self.margin_left,
            'margin_right': self.margin_right,
            'margin_top': self.margin_top,
            'orientation': self.orientation,
        }
        if self.header_template:
            kwargs['header_html'] = template_to_temp_file(self.header_template, self.get_context_data(), self.context_instance)
            self._tmp_files.append(kwargs['header_html'])
        if self.footer_template:
            kwargs['footer_html'] = template_to_temp_file(self.footer_template, self.get_context_data(), self.context_instance)
            self._tmp_files.append(kwargs['footer_html'])
        return kwargs

    def get_context_data(self, **kwargs):
        context = super(PDFTemplateView, self).get_context_data(**kwargs)

        match_full_url = compile(r'^https?://')
        if not match_full_url.match(settings.STATIC_URL):
            context['STATIC_URL'] = 'http://' + Site.objects.get_current().domain + settings.STATIC_URL
        if not match_full_url.match(settings.MEDIA_URL):
            context['MEDIA_URL'] = 'http://' + Site.objects.get_current().domain + settings.MEDIA_URL

        return context


class PdfTemplateView(PDFTemplateView): #TODO: Remove this in v1.0
    def __init__(self, *args, **kwargs):
        warnings.warn('PdfTemplateView is deprecated in favour of PDFTemplateView. It will be removed in version 1.',
                      PendingDeprecationWarning, 2)
        super(PdfTemplateView, self).__init__(*args, **kwargs)