Fix Namespace Prefixes in Marshaled Messages (msg: instead of ns#:)

- Implement own NamespacePrefixMapper that maps prefixes depending on
  whether a msg or a mzs object is being marshaled.
- Namespaces are not hardcoded but extracted from jaxb's generated
  sources (somewhat indirect, can be considered a hack, but there's no
  public API to extract namespaces)

3 files changed, 106 insertions, 10 deletions

diff --git a/src/main/java/at/gv/egiz/moazs/Marshaller.java b/src/main/java/at/gv/egiz/moazs/Marshaller.java
index c126f03..7e2ab17 100644
--- a/src/main/java/at/gv/egiz/moazs/Marshaller.java
+++ b/src/main/java/at/gv/egiz/moazs/Marshaller.java
@@ -1,8 +1,6 @@
 package at.gv.egiz.moazs;
 
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.oxm.jaxb.Jaxb2Marshaller;
-import org.springframework.stereotype.Component;
 
 import javax.xml.transform.Result;
 import javax.xml.transform.stream.StreamResult;
@@ -14,11 +12,13 @@ import java.io.StringWriter;
  * @author xerx593
  * Source: https://stackoverflow.com/questions/44676532/how-to-use-spring-to-marshal-and-unmarshal-xml
  */
-@Component
 public class Marshaller {
 
-    @Autowired
-    private Jaxb2Marshaller marshaller;
+    private final Jaxb2Marshaller marshaller;
+
+    public Marshaller(Jaxb2Marshaller marshaller) {
+        this.marshaller = marshaller;
+    }
 
     public <T> String marshallXml(final T obj) {
         StringWriter sw = new StringWriter();
diff --git a/src/main/java/at/gv/egiz/moazs/MoaZSPrefixMapper.java b/src/main/java/at/gv/egiz/moazs/MoaZSPrefixMapper.java
new file mode 100644
index 0000000..ef85989
--- /dev/null
+++ b/src/main/java/at/gv/egiz/moazs/MoaZSPrefixMapper.java
@@ -0,0 +1,54 @@
+package at.gv.egiz.moazs;
+
+import com.sun.xml.bind.marshaller.NamespacePrefixMapper;
+import org.springframework.stereotype.Component;
+
+import java.util.HashMap;
+import java.util.Map;
+
+@Component
+public class MoaZSPrefixMapper extends NamespacePrefixMapper {
+
+    private final Map<String, String> map = new HashMap<>();
+
+    public MoaZSPrefixMapper() {
+        map.put(getMsgNamespaceUri(),  "msg");
+        map.put(getMsgPNamespaceUri(), "msgp");
+        map.put(getMzsNamespaceUri(),  "mzs");
+        map.put(getMzsPNamespaceUri(), "mzsp");
+        map.put(getDsigNamespaceUri(), "dsig");
+    }
+
+    public MoaZSPrefixMapper(boolean isMzs) {
+        this();
+        map.put((isMzs) ? getMzsPNamespaceUri() : getMsgPNamespaceUri(), "p");
+        map.put((isMzs) ? getMzsNamespaceUri()  : getMsgNamespaceUri(),  "");
+    }
+
+    @Override
+    public String getPreferredPrefix(String namespaceUri, String suggestion, boolean requirePrefix) {
+        return map.getOrDefault(namespaceUri, suggestion);
+    }
+
+    private String getMsgNamespaceUri() {
+        return new at.gv.zustellung.msg.xsd.ObjectFactory().createDeliveryRequest(null).getName().getNamespaceURI();
+    }
+
+    private String getMsgPNamespaceUri() {
+        return new at.gv.zustellung.msg.xsd.persondata.ObjectFactory().createPerson(null).getName().getNamespaceURI();
+    }
+
+    private String getMzsNamespaceUri() {
+        return new at.gv.zustellung.app2mzs.xsd.ObjectFactory().createDeliveryRequest(null).getName().getNamespaceURI();
+    }
+
+    private String getMzsPNamespaceUri() {
+        return new at.gv.zustellung.app2mzs.xsd.persondata.ObjectFactory().createAbstractPersonData(null).getName().getNamespaceURI();
+    }
+
+    private String getDsigNamespaceUri() {
+        return new  org.w3._2000._09.xmldsig_.ObjectFactory().createCanonicalizationMethod(null).getName().getNamespaceURI();
+    }
+
+
+}
diff --git a/src/main/java/at/gv/egiz/moazs/config/MarshallerConfig.java b/src/main/java/at/gv/egiz/moazs/config/MarshallerConfig.java
index e6e3cc1..6fdd128 100644
--- a/src/main/java/at/gv/egiz/moazs/config/MarshallerConfig.java
+++ b/src/main/java/at/gv/egiz/moazs/config/MarshallerConfig.java
@@ -1,5 +1,9 @@
 package at.gv.egiz.moazs.config;
 
+import at.gv.egiz.moazs.Marshaller;
+import at.gv.egiz.moazs.MoaZSPrefixMapper;
+import com.sun.xml.bind.marshaller.NamespacePrefixMapper;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.oxm.jaxb.Jaxb2Marshaller;
@@ -10,19 +14,57 @@ import java.util.HashMap;
 public class MarshallerConfig {
 
     @Bean
-    public Jaxb2Marshaller jaxb2Marshaller() {
+    public NamespacePrefixMapper msgMapper() {
+        return new MoaZSPrefixMapper(false);
+    }
+
+    @Bean
+    public NamespacePrefixMapper mzsMapper() {
+        return new MoaZSPrefixMapper(true);
+    }
+
+    @Bean
+    public Jaxb2Marshaller mzsJaxbMarshaller() {
+        Jaxb2Marshaller marshaller = new Jaxb2Marshaller();
+
+        marshaller.setClassesToBeBound(
+            at.gv.zustellung.app2mzs.xsd.ObjectFactory.class,
+            at.gv.zustellung.app2mzs.xsd.persondata.ObjectFactory.class,
+            at.gv.zustellung.msg.xsd.ObjectFactory.class);
+
+        marshaller.setMarshallerProperties(new HashMap<String, Object>() {{
+            put(javax.xml.bind.Marshaller.JAXB_FORMATTED_OUTPUT, true);
+            put("com.sun.xml.bind.namespacePrefixMapper", mzsMapper());
+        }});
+
+        return marshaller;
+    }
+
+    @Bean
+    public Jaxb2Marshaller msgJaxbMarshaller() {
         Jaxb2Marshaller marshaller = new Jaxb2Marshaller();
+
         marshaller.setClassesToBeBound(
-                at.gv.zustellung.msg.xsd.ObjectFactory.class,
-                at.gv.zustellung.msg.xsd.persondata.ObjectFactory.class,
-                at.gv.zustellung.app2mzs.xsd.ObjectFactory.class,
-                at.gv.zustellung.app2mzs.xsd.persondata.ObjectFactory.class);
+            at.gv.zustellung.msg.xsd.ObjectFactory.class,
+            at.gv.zustellung.msg.xsd.persondata.ObjectFactory.class);
 
         marshaller.setMarshallerProperties(new HashMap<String, Object>() {{
             put(javax.xml.bind.Marshaller.JAXB_FORMATTED_OUTPUT, true);
+            put("com.sun.xml.bind.namespacePrefixMapper", msgMapper());
         }});
 
         return marshaller;
     }
 
+    @Bean
+    public Marshaller mzsMarshaller() {
+        return new Marshaller(mzsJaxbMarshaller());
+    }
+
+    @Bean
+    public Marshaller msgMarshaller() {
+        return new Marshaller(msgJaxbMarshaller());
+    }
+
+
 }