Implement #207: allow mapping custom prefixes for namespace URIs

Author: cowtowncoder

src/main/java/tools/jackson/dataformat/xml/ser/NamespaceBinding.java

@@ -0,0 +1,40 @@
+package tools.jackson.dataformat.xml.ser;
+
+import javax.xml.stream.XMLStreamException;
+
+import org.codehaus.stax2.XMLStreamWriter2;
+
+import tools.jackson.dataformat.xml.util.ArgUtil;
+
+/**
+ * Entity representing binding from prefix to namespace URI, used
+ * to determine prefix to use for given namespace URI (and dynamically
+ * adding necessary declarations)
+ *
+ * @since 3.2
+ */
+public class NamespaceBinding
+    implements XmlGeneratorWritable
+{
+    private final String _prefix;
+
+    private final String _namespaceURI;
+
+    public NamespaceBinding(String prefix, String namespaceURI) {
+        _prefix = ArgUtil.emptyToNull(prefix);
+        _namespaceURI = ArgUtil.nonEmptyNonNull("namespaceURI", namespaceURI);
+    }
+
+    @Override
+    public void write(ToXmlGenerator xmlGen, XMLStreamWriter2 sw) throws XMLStreamException {
+        if (_prefix == null) {
+            sw.setDefaultNamespace(_namespaceURI);
+        } else {
+            sw.setPrefix(_prefix, _namespaceURI);
+        }
+    }
+
+    public boolean isDefaultNamespace() {
+        return _prefix == null;
+    }
+}

src/main/java/tools/jackson/dataformat/xml/ser/PrologDirective.java

@@ -1,27 +1,14 @@
 package tools.jackson.dataformat.xml.ser;
 
-import javax.xml.stream.XMLStreamException;
-
-import org.codehaus.stax2.XMLStreamWriter2;
-
 /**
- * Common API for XML nodes -- Comments, DTDs, Processing Instructions -- to
+ * Base API for XML nodes -- Comments, DTDs, Processing Instructions -- to
  * be written in Document Prolog (before actual XML Document (root node),
  * after optional XML Declaration).
  *
  * @since 3.2
  */
 public interface PrologDirective
+    extends XmlGeneratorWritable
 {
-    /**
-     * Method to call to actually write out the directive using given
-     * {@link XMLStreamWriter2}. {@link ToXmlGenerator} is only passed
-     * in case access to configuration was needed.
-     *
-     * @param xmlGen Generator that called this method: MUST NOT call
-     *   its output methods, only to be used for configuration access
-     *   @param sw Writer to use for actual output of XML event
-     */
-    public void write(ToXmlGenerator xmlGen, XMLStreamWriter2 sw)
-        throws XMLStreamException;
+    // All we need is so far in XmlGeneratorWritable
 }

src/main/java/tools/jackson/dataformat/xml/ser/ToXmlGenerator.java

@@ -99,6 +99,13 @@ public class ToXmlGenerator
     /**********************************************************************
      */
 
+    /**
+     * XML directives (DTD, Comments, PIs) to write, if any.
+     *
+     * @since 3.2
+     */
+    protected List<NamespaceBinding> _namespaceBindings;
+
     /**
      * XML directives (DTD, Comments, PIs) to write, if any.
      *
@@ -228,16 +235,19 @@ public ToXmlGenerator(ObjectWriteContext writeCtxt, IOContext ioCtxt,
 
     /**
      * Method called by {@link XmlGeneratorInitializer} to inject
-     * necessary configuration.
+     * necessary configuration like Prolog Directives and namespace
+     * bindings.
      *
      * @since 3.2
      */
-    public void initProlog(boolean lfBetweenPrologDirectives,
-            List<PrologDirective> directives)
+    public void initDocument(boolean lfBetweenPrologDirectives,
+            List<PrologDirective> directives,
+            List<NamespaceBinding> nsBindings)
     {
         if (_initialized) { // sanity check
             _reportError("Internal error: cannot call `initConfig()` after generator already initialized");
         }
+        _namespaceBindings = nsBindings;
         _lfBetweenPrologDirectives = lfBetweenPrologDirectives;
         _prologDirectives = directives;
     }
@@ -285,7 +295,11 @@ public void initGenerator() throws JacksonException
                     }
                 }
             }
-
+            if (_namespaceBindings != null) {
+                for (NamespaceBinding ns : _namespaceBindings) {
+                    ns.write(this, _xmlWriter);
+                }
+            }
         } catch (XMLStreamException e) {
             StaxUtil.throwAsWriteException(e, this);
         }

src/main/java/tools/jackson/dataformat/xml/ser/XmlGeneratorInitializer.java

@@ -2,6 +2,7 @@
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 
 import tools.jackson.core.JacksonException;
 import tools.jackson.core.JsonGenerator;
@@ -31,16 +32,26 @@
 public class XmlGeneratorInitializer
     implements GeneratorInitializer
 {
+    /**
+     * Prolog Directives to pass for generator to write.
+     */
     protected List<PrologDirective> _directives;
 
+    /**
+     * Namespace bindings (prefix to URL) to register with
+     * generator.
+     */
+    protected List<NamespaceBinding> _namespaceBindings;
+
     protected boolean _addLfBetweenPrologDirectives = true;
 
     protected boolean _hasDTD;
 
     @Override
     public void initialize(SerializationConfig config, JsonGenerator g) throws JacksonException {
         if (g instanceof ToXmlGenerator xg) {
-            xg.initProlog(_addLfBetweenPrologDirectives, _directives);
+            xg.initDocument(_addLfBetweenPrologDirectives, _directives,
+                    _namespaceBindings);
         }
     }
 
@@ -124,6 +135,39 @@ public XmlGeneratorInitializer addPI(String target, String data) {
         return _add(new PrologPI(target, data));
     }
 
+    /**
+     * Method for specifying namespace URI to preferentially bind to the
+     * "default namespace" (one used when element has no prefix).
+     * This will guide underlying generator to add necessary
+     * declarations when actually writing elements with matching
+     * namespace URI.
+     *
+     * @param namespaceURI URI of the default namespace
+     *
+     * @return This initializer for call chaining
+     */
+    public XmlGeneratorInitializer addDefaultNamespace(String namespaceURI) {
+        return addNamespace(null, namespaceURI);
+    }
+
+    /**
+     * Method for adding a mapping (binding) between given prefix and matching
+     * namespace URI. This will guide underlying generator to add necessary
+     * declarations when actually writing namespaced attributes and elements.
+     *
+     * @param prefix Prefix to use for namespace
+     * @param namespaceURI URI of the namespace
+     *
+     * @return This initializer for call chaining
+     */
+    public XmlGeneratorInitializer addNamespace(String prefix, String namespaceURI) {
+        if (_namespaceBindings == null) {
+            _namespaceBindings = new ArrayList<>();
+        }
+        _namespaceBindings.add(new NamespaceBinding(prefix, namespaceURI));
+        return this;
+    }
+
     protected XmlGeneratorInitializer _add(PrologDirective d) {
         if (_directives == null) {
             _directives = new ArrayList<>();

src/main/java/tools/jackson/dataformat/xml/ser/XmlGeneratorWritable.java

@@ -0,0 +1,25 @@
+package tools.jackson.dataformat.xml.ser;
+
+import javax.xml.stream.XMLStreamException;
+
+import org.codehaus.stax2.XMLStreamWriter2;
+
+/**
+ * Common API for XML entities to write via {@link ToXmlGenerator}.
+ *
+ * @since 3.2
+ */
+public interface XmlGeneratorWritable
+{
+    /**
+     * Method to call to actually write out the entity using given
+     * {@link XMLStreamWriter2}. {@link ToXmlGenerator} is only passed
+     * in case access to configuration was needed.
+     *
+     * @param xmlGen Generator that called this method: MUST NOT call
+     *   its output methods, only to be used for configuration access
+     * @param sw Writer to use for actual output of XML event
+     */
+    public void write(ToXmlGenerator xmlGen, XMLStreamWriter2 sw)
+        throws XMLStreamException;
+}