1 files changed, 133 insertions, 0 deletions
diff --git a/src/py/javatoolkit/xml/SaxRewriter.py b/src/py/javatoolkit/xml/SaxRewriter.py
new file mode 100644
index 0000000..3e75d99
--- /dev/null
+++ b/src/py/javatoolkit/xml/SaxRewriter.py
@@ -0,0 +1,133 @@
+# -*- coding: UTF-8 -*-
+
+# Copyright 2004-2005 Gentoo Foundation
+# Distributed under the terms of the GNU General Public License v2
+# $Header: $
+
+import os
+import sys
+import StringIO
+
+from xml.sax.saxutils import XMLGenerator
+from xml.sax.saxutils import quoteattr
+
+class SaxRewriter(XMLGenerator):
+    """
+    Using Sax gives us the support for writing back doctypes and all easily
+    and is only marginally slower than expat as it is just a tight layer over it
+    """
+    def __init__(self, **kwds):
+        self.elems = kwds.has_key('elems') and kwds['elems'] or []
+        self.attributes = kwds.has_key('attributes') and kwds['attributes'] or []
+        self.values = kwds.has_key('values') and kwds['values'] or []
+        self.sourceElems = kwds.has_key('sourceElems') and kwds['sourceElems'] or []
+        self.sourceAttributes = kwds.has_key('sourceAttributes') and kwds['sourceAttributes'] or []
+        self.sourceValues = kwds.has_key('sourceValues') and kwds['sourceValues'] or []
+        self.targetElems = kwds.has_key('targetElems') and kwds['targetElems'] or []
+        self.targetAttributes = kwds.has_key('targetAttributes') and kwds['targetAttributes'] or []
+        self.targetValues = kwds.has_key('targetValues') and kwds['targetValues'] or []
+
+        self.deleteElems = kwds.has_key('deleteElems') and kwds['deleteElems'] or []
+        self.deleteAttributes = kwds.has_key('deleteAttributes') and kwds['deleteAttributes'] or []
+
+        self.src_dirs = kwds.has_key('src_dirs') and kwds['src_dirs'] or []
+        self.output_dir = kwds.has_key('output_dir') and kwds['output_dir'] or None
+
+        self.buffer = StringIO.StringIO()
+
+        XMLGenerator.__init__(self, self.buffer, 'UTF-8')
+
+
+    def add_gentoo_javadoc(self, name, attrs):
+        self.p(u'<%s ' % name)
+        for a,v in attrs.items():
+            self.write_attr(a,v)
+
+        self.p(u'>')
+
+        if name == "project":
+            javadoc_str = """
+            <target name=\"gentoojavadoc\" >
+            <mkdir dir=\"""" + self.output_dir + """\" />
+            <javadoc
+            destdir=\"""" + self.output_dir + """\"
+            author="true"
+            version="true"
+            use="true"
+            windowtitle="javadoc">
+            """
+
+            for src_dir in self.src_dirs:
+                javadoc_str += """
+                <fileset dir=\"""" + src_dir + """\"    defaultexcludes="yes">
+                <include name="**/*.java"/>
+                </fileset>
+                """
+
+            javadoc_str += """
+            </javadoc>
+            </target>
+            """
+
+            self.p(u'%s' % javadoc_str)
+
+
+    # write as they are or delete if wanted attributes first
+    # next, add / update
+    def modify_elements(self, name, attrs):
+        self.p(u'<%s ' % name)
+
+        match = ( name in self.elems )
+        matchSource = ( name in self.sourceElems )
+        matchTarget = ( name in self.targetElems )
+        matchDelete = ( name in self.deleteElems )
+
+        for a,v in attrs.items():
+            if not (
+                (match and a in self.attributes)
+                or (matchSource and a in self.sourceAttributes)
+                or (matchTarget and a in self.targetAttributes)
+                or (matchDelete and a in self.deleteAttributes)
+            ):
+                self.write_attr(a,v)
+
+        if matchSource:
+            for i, attr in enumerate(self.sourceAttributes):
+                self.write_attr(attr, self.sourceValues[i])
+
+        if matchTarget:
+            for i, attr in enumerate(self.targetAttributes):
+                self.write_attr(attr, self.targetValues[i])
+
+        if match:
+            for i, attr in enumerate(self.attributes):
+                self.write_attr(attr, self.values[i])
+
+        self.p(u'>')
+
+
+    def char_data(self, data):
+        self.p(escape(data))
+
+
+    def write(self, out_stream):
+        value = self.buffer.getvalue()
+        out_stream.write(value)
+        self.buffer.truncate(0)
+
+
+    def p(self,str):
+        self.buffer.write(str.encode('utf8'))
+
+
+    def write_attr(self,a,v):
+        self.p(u'%s=%s ' % (a,quoteattr(v, {u'©':'&#169;'})))
+
+
+    def process(self, in_stream, callback):
+        self.startElement = callback
+        from xml.sax import parseString
+        parseString(in_stream, self)
+        self.p(u'\n')
+
+# vim: set expandtab tabstop=4 shiftwidth=4 softtabstop=4 nowrap: