From bc3dd777a21dbf4d6522c46de12064f2254dfa2b Mon Sep 17 00:00:00 2001
From: Mathieu Baudier <mbaudier@argeo.org>
Date: Wed, 28 Apr 2010 16:24:31 +0000
Subject: [PATCH] Introduce depth in JCR serializer

git-svn-id: https://svn.argeo.org/commons/trunk@3516 4cfe0d0a-d680-48aa-b62c-e0a02a3f76cc
---
 .../org.argeo.server.jackrabbit/pom.xml       |  5 ++
 .../jcr/mvc/JcrXmlServerSerializer.java       | 72 +++++++++++++++++--
 2 files changed, 70 insertions(+), 7 deletions(-)

diff --git a/server/runtime/org.argeo.server.jackrabbit/pom.xml b/server/runtime/org.argeo.server.jackrabbit/pom.xml
index d0b857192..cf67d176f 100644
--- a/server/runtime/org.argeo.server.jackrabbit/pom.xml
+++ b/server/runtime/org.argeo.server.jackrabbit/pom.xml
@@ -95,6 +95,11 @@
 			<artifactId>org.springframework.security.core</artifactId>
 		</dependency>
 
+		<dependency>
+			<groupId>org.springframework.ws</groupId>
+			<artifactId>org.springframework.xml</artifactId>
+		</dependency>
+
 		<!-- Logging -->
 		<dependency>
 			<groupId>org.slf4j</groupId>
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..0ae91e22e 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,60 @@ public class JcrXmlServerSerializer implements ServerSerializer {
 		String noRecurseStr = request.getParameter("noRecurse");
 		boolean noRecurse = noRecurseStr != null && noRecurseStr.equals("true");
 
+		String depthStr = request.getParameter("depth");
 		Node node = (Node) obj;
-		response.setContentType("text/xml;charset=" + contentTypeCharset);
 		try {
-			node.getSession().exportDocumentView(node.getPath(),
-					response.getOutputStream(), true, noRecurse);
+			if (depthStr == null) {
+				response.setContentType("text/xml;charset="
+						+ contentTypeCharset);
+				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) {
-- 
2.30.2