import poppler

import pdfformfiller.models
import pdb

class PopplerController(object):
    def __init__(self, pdf=""):
        doc_path = self._get_pdf_path(pdf)
        self.doc = self._load_poppler_doc(doc_path)


    def _get_pdf_path(self, pdf):
        import os
        from django.conf import settings

        search = [
            os.path.abspath(pdf),
            os.path.abspath(os.path.join(getattr(settings, 'STATIC_FORMS_ROOT', ''), pdf)),
        ]

        for path in search:
            if os.path.isfile(path):
                return path

        raise Exception("Path not found in %s" % search)

    def _load_poppler_doc(self, pdf_path):
        return poppler.document_new_from_file('file://' + pdf_path, password=None)


    def generate_page_images(self, pages=[], clear_page=True):
        import cairo
        import StringIO

        page = self.doc.get_page(pages[0] if pages else 0)

        width, height = page.get_size()
        surface = cairo.ImageSurface(cairo.FORMAT_RGB24, int(width*2), int(height*2))
        context = cairo.Context(surface)
        context.scale(2, 2)

        for page_num in pages or xrange(self.doc.get_n_pages()):
            page = self.doc.get_page(page_num)

            if clear_page:
                self.clear_page(page)

            context.set_source_rgb(1.0, 1.0, 1.0)
            context.rectangle(0.0, 0.0, width*2, height*2)
            context.fill()

            page.render(context)
            fd = StringIO.StringIO()
            surface.write_to_png(fd)
            fd.seek(0)
            yield fd


    def clear_page(self, page):
        for field_mapping in page.get_form_field_mapping():
            field = field_mapping.field
            field_type = field.get_field_type().value_nick

            if field_type == 'text':
                field.text_set_text('')
            elif field_type == 'button':
                field.button_set_state(0)


    def get_page_fields(self, page_num):
        page = self.doc.get_page(page_num)
        for field in page.get_form_field_mapping():
            area = field.area
            data = field.field
            field_type = data.get_field_type()

            if field_type.value_nick == 'text':
                value = data.text_get_text()
            elif field_type.value_nick == 'button':
                value = data.button_get_state()

            yield {
                'name': data.get_name() or data.get_mapping_name() or data.get_partial_name(),
                'value': value,
                'type': field_type.value_nick,
                'area': {
                    'pos_x': area.x1,
                    'pos_y': area.y1,
                    'width': area.x2 - area.x1,
                    'height': area.y2 - area.y1,
                }
            }


class PDFFormFillerPostSave(PopplerController):
    def __init__(self, document):
        pass