Improve ACR, introduce migration from JCR.
[lgpl/argeo-commons.git] / org.argeo.cms / src / org / argeo / cms / acr / xml / DomContent.java
index bfea129b11110a7561a9c3b38a1828851b2e58c1..30ecd8e82b649be7251473c204062937f5a3af8c 100644 (file)
@@ -1,8 +1,10 @@
 package org.argeo.cms.acr.xml;
 
 import java.nio.CharBuffer;
+import java.util.ArrayList;
 import java.util.HashSet;
 import java.util.Iterator;
+import java.util.List;
 import java.util.Optional;
 import java.util.Set;
 import java.util.concurrent.CompletableFuture;
@@ -10,24 +12,31 @@ import java.util.concurrent.CompletableFuture;
 import javax.xml.XMLConstants;
 import javax.xml.namespace.NamespaceContext;
 import javax.xml.namespace.QName;
+import javax.xml.transform.Source;
+import javax.xml.transform.Transformer;
+import javax.xml.transform.TransformerException;
+import javax.xml.transform.dom.DOMResult;
+import javax.xml.transform.dom.DOMSource;
 
 import org.argeo.api.acr.Content;
 import org.argeo.api.acr.ContentName;
-import org.argeo.api.acr.spi.AbstractContent;
 import org.argeo.api.acr.spi.ProvidedContent;
 import org.argeo.api.acr.spi.ProvidedSession;
+import org.argeo.cms.acr.AbstractContent;
 import org.argeo.cms.acr.ContentUtils;
 import org.w3c.dom.Attr;
+import org.w3c.dom.DOMException;
 import org.w3c.dom.Document;
+import org.w3c.dom.DocumentFragment;
 import org.w3c.dom.Element;
 import org.w3c.dom.NamedNodeMap;
 import org.w3c.dom.Node;
 import org.w3c.dom.NodeList;
 import org.w3c.dom.Text;
 
+/** Content persisted as a DOM element. */
 public class DomContent extends AbstractContent implements ProvidedContent {
 
-       private final ProvidedSession session;
        private final DomContentProvider provider;
        private final Element element;
 
@@ -35,7 +44,7 @@ public class DomContent extends AbstractContent implements ProvidedContent {
        private Boolean hasText = null;
 
        public DomContent(ProvidedSession session, DomContentProvider contentProvider, Element element) {
-               this.session = session;
+               super(session);
                this.provider = contentProvider;
                this.element = element;
        }
@@ -49,7 +58,7 @@ public class DomContent extends AbstractContent implements ProvidedContent {
                if (element.getParentNode() == null) {// root
                        String mountPath = provider.getMountPath();
                        if (mountPath != null) {
-                               Content mountPoint = session.getMountPoint(mountPath);
+                               Content mountPoint = getSession().getMountPoint(mountPath);
                                return mountPoint.getName();
                        }
                }
@@ -128,12 +137,31 @@ public class DomContent extends AbstractContent implements ProvidedContent {
                Object previous = get(key);
                String namespaceUriOrNull = XMLConstants.NULL_NS_URI.equals(key.getNamespaceURI()) ? null
                                : key.getNamespaceURI();
+               String prefixToUse = registerPrefixIfNeeded(key);
                element.setAttributeNS(namespaceUriOrNull,
-                               namespaceUriOrNull == null ? key.getLocalPart() : key.getPrefix() + ":" + key.getLocalPart(),
+                               namespaceUriOrNull == null ? key.getLocalPart() : prefixToUse + ":" + key.getLocalPart(),
                                value.toString());
                return previous;
        }
 
+       protected String registerPrefixIfNeeded(QName name) {
+               String namespaceUriOrNull = XMLConstants.NULL_NS_URI.equals(name.getNamespaceURI()) ? null
+                               : name.getNamespaceURI();
+               String prefixToUse;
+               if (namespaceUriOrNull != null) {
+                       String registeredPrefix = provider.getPrefix(namespaceUriOrNull);
+                       if (registeredPrefix != null) {
+                               prefixToUse = registeredPrefix;
+                       } else {
+                               provider.registerPrefix(name.getPrefix(), namespaceUriOrNull);
+                               prefixToUse = name.getPrefix();
+                       }
+               } else {
+                       prefixToUse = null;
+               }
+               return prefixToUse;
+       }
+
        @Override
        public boolean hasText() {
 //             return element instanceof Text;
@@ -178,7 +206,7 @@ public class DomContent extends AbstractContent implements ProvidedContent {
        @Override
        public Iterator<Content> iterator() {
                NodeList nodeList = element.getChildNodes();
-               return new ElementIterator(this, session, provider, nodeList);
+               return new ElementIterator(this, getSession(), provider, nodeList);
        }
 
        @Override
@@ -189,7 +217,7 @@ public class DomContent extends AbstractContent implements ProvidedContent {
                        if (mountPath == null)
                                return null;
                        String[] parent = ContentUtils.getParentPath(mountPath);
-                       return session.get(parent[0]);
+                       return getSession().get(parent[0]);
                }
                if (parentNode instanceof Document)
                        return null;
@@ -204,8 +232,9 @@ public class DomContent extends AbstractContent implements ProvidedContent {
                Document document = this.element.getOwnerDocument();
                String namespaceUriOrNull = XMLConstants.NULL_NS_URI.equals(name.getNamespaceURI()) ? null
                                : name.getNamespaceURI();
+               String prefixToUse = registerPrefixIfNeeded(name);
                Element child = document.createElementNS(namespaceUriOrNull,
-                               namespaceUriOrNull == null ? name.getLocalPart() : name.getPrefix() + ":" + name.getLocalPart());
+                               namespaceUriOrNull == null ? name.getLocalPart() : prefixToUse + ":" + name.getLocalPart());
                element.appendChild(child);
                return new DomContent(this, child);
        }
@@ -226,28 +255,79 @@ public class DomContent extends AbstractContent implements ProvidedContent {
 
        }
 
+       @SuppressWarnings("unchecked")
        @Override
        public <A> A adapt(Class<A> clss) throws IllegalArgumentException {
                if (CharBuffer.class.isAssignableFrom(clss)) {
                        String textContent = element.getTextContent();
                        CharBuffer buf = CharBuffer.wrap(textContent);
                        return (A) buf;
+               } else if (Source.class.isAssignableFrom(clss)) {
+                       DOMSource source = new DOMSource(element);
+                       return (A) source;
                }
                return super.adapt(clss);
        }
 
+       @SuppressWarnings("unchecked")
        public <A> CompletableFuture<A> write(Class<A> clss) {
                if (String.class.isAssignableFrom(clss)) {
                        CompletableFuture<String> res = new CompletableFuture<>();
                        res.thenAccept((s) -> {
-                               session.notifyModification(this);
+                               getSession().notifyModification(this);
                                element.setTextContent(s);
                        });
                        return (CompletableFuture<A>) res;
+               } else if (Source.class.isAssignableFrom(clss)) {
+                       CompletableFuture<Source> res = new CompletableFuture<>();
+                       res.thenAccept((source) -> {
+                               try {
+                                       Transformer transformer = provider.getTransformerFactory().newTransformer();
+                                       DocumentFragment documentFragment = element.getOwnerDocument().createDocumentFragment();
+                                       DOMResult result = new DOMResult(documentFragment);
+                                       transformer.transform(source, result);
+                                       // Node parentNode = element.getParentNode();
+                                       Element resultElement = (Element) documentFragment.getFirstChild();
+                                       QName resultName = toQName(resultElement);
+                                       if (!resultName.equals(getName()))
+                                               throw new IllegalArgumentException(resultName + "+ is not compatible with " + getName());
+
+                                       // attributes
+                                       NamedNodeMap attrs = resultElement.getAttributes();
+                                       for (int i = 0; i < attrs.getLength(); i++) {
+                                               Attr attr2 = (Attr) element.getOwnerDocument().importNode(attrs.item(i), true);
+                                               element.getAttributes().setNamedItem(attr2);
+                                       }
+
+                                       // Move all the children
+                                       while (element.hasChildNodes()) {
+                                               element.removeChild(element.getFirstChild());
+                                       }
+                                       while (resultElement.hasChildNodes()) {
+                                               element.appendChild(resultElement.getFirstChild());
+                                       }
+//                                     parentNode.replaceChild(resultNode, element);
+//                                     element = (Element)resultNode;
+                                       
+                               } catch (DOMException | TransformerException e) {
+                                       throw new RuntimeException("Cannot write to element", e);
+                               }
+                       });
+                       return (CompletableFuture<A>) res;
                }
                return super.write(clss);
        }
 
+       /*
+        * TYPING
+        */
+       @Override
+       public List<QName> getContentClasses() {
+               List<QName> res = new ArrayList<>();
+               res.add(getName());
+               return res;
+       }
+
        /*
         * MOUNT MANAGEMENT
         */
@@ -259,10 +339,7 @@ public class DomContent extends AbstractContent implements ProvidedContent {
                return new DomContent(this, childElement);
        }
 
-       public ProvidedSession getSession() {
-               return session;
-       }
-
+       @Override
        public DomContentProvider getProvider() {
                return provider;
        }