# vim: tabstop=4 shiftwidth=4 expandtab
import builtins
import cssutils
import logging
import os
from lxml import etree
from xml.sax.saxutils import escape as xmlescape

cssutils.ser.prefs.lineSeparator = ' '
cssutils.log.setLevel(logging.FATAL)

class SVGTemplate:
    def __init__(self, task):
        self.task = task

    def __enter__(self):
        with builtins.open(os.path.join(self.task.workdir, self.task.infile), 'r') as fp:
            self.svgstr = fp.read()
        return self

    def write(self):
        # open the output-file (named ".gen.svg" in the workdir)
        outfile = os.path.join(self.task.workdir, '.gen.svg')
        with builtins.open(outfile, 'w') as fp:
            # write the generated svg-text into the output-file
            fp.write(self.svgstr)
        return outfile

    def replacetext(self):
        for key in self.task.parameters.keys():
            self.svgstr = self.svgstr.replace(key, xmlescape(str(self.task.parameters[key])))

    def transform(self, frame):
        parser = etree.XMLParser(huge_tree=True)
        svg = etree.fromstring(self.svgstr.encode('utf-8'), parser)
        # apply the replace-pairs to the input text, by finding the specified xml-elements by their id and modify their css-parameter the correct value
        for replaceinfo in frame:
            (id, type, key, value) = replaceinfo
            for el in svg.findall(".//*[@id='" + id.replace("'", "\\'") + "']"):
                if type == 'style':
                    style = cssutils.parseStyle(el.attrib['style'] if 'style' in el.attrib else '')
                    style[key] = str(value)
                    el.attrib['style'] = style.cssText
                elif type == 'attr':
                    el.attrib[key] = str(value)
                elif type == 'text':
                    el.text = str(value)
        # if '$subtitle' in task.parameters and task.parameters['$subtitle'] == '':
        #   child = svg.findall(".//*[@id='subtitle']")[0]
        #   child.getparent().remove(child)
        self.svgstr = etree.tostring(svg, encoding='unicode')

    def __exit__(self, exception_type, exception_value, traceback):
        pass