import sys
import traceback
import os
import inkex
import tempfile
from zipfile import ZipFile
from libembroidery import *

from .base import InkstitchExtension
from ..i18n import _
from ..output import write_embroidery_file
from ..stitch_plan import patches_to_stitch_plan
from ..svg import render_stitch_plan, PIXELS_PER_MM
from ..utils.io import save_stdout


class Zip(InkstitchExtension):
    def __init__(self, *args, **kwargs):
        InkstitchExtension.__init__(self)
        self.OptionParser.add_option("-c", "--collapse_len_mm",
                                     action="store", type="float",
                                     dest="collapse_length_mm", default=3.0,
                                     help="max collapse length (mm)")

        # it's kind of obnoxious that I have to do this...
        self.formats = []
        formatList = embFormatList_create()
        curFormat = formatList
        while(curFormat):
            # extension includes the dot, so we'll remove it
            extension = embFormat_extension(curFormat)[1:]
            description = embFormat_description(curFormat)
            writer_state = embFormat_writerState(curFormat)

            if writer_state.strip() and embFormat_type(curFormat) != EMBFORMAT_OBJECTONLY:
                self.OptionParser.add_option('--format-%s' % extension, type="inkbool", dest=extension)
                self.formats.append(extension)
            curFormat = curFormat.next

    def effect(self):
        if not self.get_elements():
            return

        patches = self.elements_to_patches(self.elements)
        stitch_plan = patches_to_stitch_plan(patches, self.options.collapse_length_mm * PIXELS_PER_MM)

        base_file_name = self.get_base_file_name()
        path = tempfile.mkdtemp()

        files = []

        # libembroidery likes to debug log things to stdout.  No way to disable it.
        save_stdout()
        for format in self.formats:
            if getattr(self.options, format):
                output_file = os.path.join(path, "%s.%s" % (base_file_name, format))
                write_embroidery_file(output_file, stitch_plan, self.document.getroot())
                files.append(output_file)

        # I'd love to do restore_stderr() here, but if I do, libembroidery's
        # stuff still prints out and corrupts the zip!  That's because it uses
        # C's buffered stdout, so it hasn't actually written anything to the
        # real standard output yet.

        if not files:
            self.errormsg(_("No embroidery file formats selected."))

        temp_file = tempfile.NamedTemporaryFile(suffix=".zip", delete=False)

        # in windows, failure to close here will keep the file locked
        temp_file.close()

        with ZipFile(temp_file.name, "w") as zip_file:
            for file in files:
                zip_file.write(file, os.path.basename(file))

        # inkscape will read the file contents from stdout and copy
        # to the destination file that the user chose
        with open(temp_file.name) as output_file:
            sys.real_stdout.write(output_file.read())

        os.remove(temp_file.name)
        for file in files:
            os.remove(file)
        os.rmdir(path)

        # don't let inkex output the SVG!
        sys.exit(0)