X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=server%2Fruntime%2Forg.argeo.server.jackrabbit%2Fsrc%2Fmain%2Fjava%2Forg%2Fargeo%2Fserver%2Fjcr%2Fmvc%2FJcrXmlServerSerializer.java;h=54cc5b7b5a986fc08cd54075f81dc12c12ea991f;hb=946518ac41ee428e6cfd27b8d8b7e92b6d38470c;hp=01eab4b34fe0fea17a1c29209a66dff04f031871;hpb=6f9b81862316741292bf782395f90f2c7e57bfc7;p=lgpl%2Fargeo-commons.git

diff --git a/server/runtime/org.argeo.server.jackrabbit/src/main/java/org/argeo/server/jcr/mvc/JcrXmlServerSerializer.java b/server/runtime/org.argeo.server.jackrabbit/src/main/java/org/argeo/server/jcr/mvc/JcrXmlServerSerializer.java
index 01eab4b34..54cc5b7b5 100644
--- a/server/runtime/org.argeo.server.jackrabbit/src/main/java/org/argeo/server/jcr/mvc/JcrXmlServerSerializer.java
+++ b/server/runtime/org.argeo.server.jackrabbit/src/main/java/org/argeo/server/jcr/mvc/JcrXmlServerSerializer.java
@@ -1,19 +1,32 @@
 package org.argeo.server.jcr.mvc;
 
-import java.io.OutputStream;
-import java.util.Set;
-
 import javax.jcr.Node;
+import javax.jcr.NodeIterator;
+import javax.jcr.RepositoryException;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import javax.xml.parsers.DocumentBuilderFactory;
+import javax.xml.transform.Transformer;
+import javax.xml.transform.TransformerFactory;
+import javax.xml.transform.dom.DOMSource;
+import javax.xml.transform.stream.StreamResult;
 
 import org.argeo.ArgeoException;
 import org.argeo.server.ServerSerializer;
-import org.xml.sax.helpers.DefaultHandler;
+import org.springframework.xml.dom.DomContentHandler;
+import org.w3c.dom.Document;
+import org.w3c.dom.Element;
+import org.w3c.dom.NodeList;
+import org.xml.sax.SAXException;
 
 public class JcrXmlServerSerializer implements ServerSerializer {
 	private String contentTypeCharset = "UTF-8";
 
+	private final DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory
+			.newInstance();
+	private final TransformerFactory transformerFactory = TransformerFactory
+			.newInstance();
+
 	public void serialize(Object obj, HttpServletRequest request,
 			HttpServletResponse response) {
 		if (!(obj instanceof Node))
@@ -22,15 +35,75 @@ public class JcrXmlServerSerializer implements ServerSerializer {
 		String noRecurseStr = request.getParameter("noRecurse");
 		boolean noRecurse = noRecurseStr != null && noRecurseStr.equals("true");
 
+		String depthStr = request.getParameter("depth");
+		String downloadStr = request.getParameter("download");
+
 		Node node = (Node) obj;
-		response.setContentType("text/xml;charset=" + contentTypeCharset);
+
 		try {
-			node.getSession().exportDocumentView(node.getPath(),
-					response.getOutputStream(), true, noRecurse);
+			String contentType = "text/xml;charset=" + contentTypeCharset;
+			// download case
+			if (downloadStr != null && downloadStr.equals("true")) {
+				String fileName = node.getName().replace(':', '_') + ".xml";
+				contentType = contentType + ";name=\"" + fileName + "\"";
+				response.setHeader("Content-Disposition",
+						"attachment; filename=\"" + fileName + "\"");
+				response.setHeader("Expires", "0");
+				response
+						.setHeader("Cache-Control", "no-cache, must-revalidate");
+				response.setHeader("Pragma", "no-cache");
+			}
+
+			response.setContentType(contentType);
+			if (depthStr == null) {
+				node.getSession().exportDocumentView(node.getPath(),
+						response.getOutputStream(), true, noRecurse);
+			} else {
+				int depth = Integer.parseInt(depthStr);
+				Document document = documentBuilderFactory.newDocumentBuilder()
+						.newDocument();
+				serializeLevelToDom(node, document, 0, depth);
+				Transformer transformer = transformerFactory.newTransformer();
+				transformer.transform(new DOMSource(document),
+						new StreamResult(response.getOutputStream()));
+			}
 		} catch (Exception e) {
 			throw new ArgeoException("Cannot serialize " + node, e);
 		}
+	}
+
+	protected void serializeLevelToDom(Node currentJcrNode,
+			org.w3c.dom.Node currentDomNode, int currentDepth, int targetDepth)
+			throws RepositoryException, SAXException {
+		DomContentHandler domContentHandler = new DomContentHandler(
+				currentDomNode);
+		currentJcrNode.getSession().exportDocumentView(
+				currentJcrNode.getPath(), domContentHandler, true, true);
 
+		if (currentDepth == targetDepth)
+			return;
+
+		// TODO: filter
+		NodeIterator nit = currentJcrNode.getNodes();
+		while (nit.hasNext()) {
+			Node nextJcrNode = nit.nextNode();
+			org.w3c.dom.Node nextDomNode;
+			if (currentDomNode instanceof Document)
+				nextDomNode = ((Document) currentDomNode).getDocumentElement();
+			else {
+				String name = currentJcrNode.getName();
+				NodeList nodeList = ((Element) currentDomNode)
+						.getElementsByTagName(name);
+				if (nodeList.getLength() < 1)
+					throw new ArgeoException("No elment named " + name
+							+ " under " + currentDomNode);
+				// we know it is the last one added
+				nextDomNode = nodeList.item(nodeList.getLength() - 1);
+			}
+			// recursive call
+			serializeLevelToDom(nextJcrNode, nextDomNode, currentDepth + 1,
+					targetDepth);
+		}
 	}
 
 	public void setContentTypeCharset(String contentTypeCharset) {