/*
* Copyright (C) 2009 eXo Platform SAS.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.exoplatform.services.jcr.webdav.command.dasl;
import org.exoplatform.services.jcr.impl.Constants;
import org.exoplatform.services.jcr.webdav.resource.CollectionResource;
import org.exoplatform.services.jcr.webdav.resource.FileResource;
import org.exoplatform.services.jcr.webdav.resource.Resource;
import org.exoplatform.services.jcr.webdav.resource.ResourceUtil;
import org.exoplatform.services.jcr.webdav.resource.VersionedCollectionResource;
import org.exoplatform.services.jcr.webdav.resource.VersionedFileResource;
import org.exoplatform.services.jcr.webdav.util.TextUtil;
import org.exoplatform.services.jcr.webdav.xml.PropertyWriteUtil;
import org.exoplatform.services.jcr.webdav.xml.PropstatGroupedRepresentation;
import org.exoplatform.services.jcr.webdav.xml.WebDavNamespaceContext;
import org.exoplatform.services.log.ExoLogger;
import org.exoplatform.services.log.Log;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URI;
import java.util.HashSet;
import java.util.Set;
import javax.jcr.Node;
import javax.jcr.NodeIterator;
import javax.jcr.query.QueryResult;
import javax.ws.rs.core.StreamingOutput;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLOutputFactory;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
/**
* Created by The eXo Platform SAS.
* @author Vitaly Guly - gavrikvetal@gmail.com
*
* @version $Id: $
*/
public class SearchResultResponseEntity implements StreamingOutput
{
/**
* logger.
*/
private static final Log LOG = ExoLogger.getLogger("exo.jcr.component.webdav.SearchResultResponseEntity");
/**
* Namespace conetext.
*/
private final WebDavNamespaceContext nsContext;
/**
* Query result.
*/
private QueryResult queryResult;
/**
* base uri.
*/
private String baseURI;
/**
* The list of properties.
*/
private static Set<QName> properties = new HashSet<QName>();
static
{
properties.add(new QName("DAV:", "displayname"));
properties.add(new QName("DAV:", "resourcetype"));
properties.add(new QName("DAV:", "creationdate"));
properties.add(new QName("DAV:", "getlastmodified"));
properties.add(new QName("DAV:", "getcontentlength"));
properties.add(new QName("DAV:", "getcontenttype"));
}
/**
* Constructor.
*
* @param queryResult Result of request
* @param nsContext namespace context
* @param baseURI base uri
*/
public SearchResultResponseEntity(QueryResult queryResult, final WebDavNamespaceContext nsContext, String baseURI)
{
this.queryResult = queryResult;
this.nsContext = nsContext;
this.baseURI = baseURI;
}
/**
* {@inheritDoc}
*/
public void write(OutputStream outStream) throws IOException
{
try
{
XMLStreamWriter xmlStreamWriter =
XMLOutputFactory.newInstance().createXMLStreamWriter(outStream, Constants.DEFAULT_ENCODING);
xmlStreamWriter.setNamespaceContext(nsContext);
xmlStreamWriter.writeStartDocument();
xmlStreamWriter.writeStartElement("D", "multistatus", "DAV:");
xmlStreamWriter.writeNamespace("D", "DAV:");
xmlStreamWriter.writeAttribute("xmlns:b", "urn:uuid:c2f41010-65b3-11d1-a29f-00aa00c14882/");
NodeIterator nodeIter = queryResult.getNodes();
Set<URI> resultNodes = new HashSet<URI>();
while (nodeIter.hasNext())
{
Node nextNode = nodeIter.nextNode();
if (nextNode.isNodeType("nt:resource"))
{
if (nextNode.getParent().isNodeType("nt:file"))
{
nextNode = nextNode.getParent();
}
else
{
// skipping
continue;
}
}
URI uri = new URI(TextUtil.escape(baseURI + nextNode.getPath(), '%', true));
// if URI is new, and wasn't previously being written, then write it
if (resultNodes.add(uri))
{
Resource resource;
if (ResourceUtil.isVersioned(nextNode))
{
if (ResourceUtil.isFile(nextNode))
{
resource = new VersionedFileResource(uri, nextNode, nsContext);
}
else
{
resource = new VersionedCollectionResource(uri, nextNode, nsContext);
}
}
else
{
if (ResourceUtil.isFile(nextNode))
{
resource = new FileResource(uri, nextNode, nsContext);
}
else
{
resource = new CollectionResource(uri, nextNode, nsContext);
}
}
xmlStreamWriter.writeStartElement("DAV:", "response");
xmlStreamWriter.writeStartElement("DAV:", "href");
xmlStreamWriter.writeCharacters(resource.getIdentifier().toASCIIString());
xmlStreamWriter.writeEndElement();
PropstatGroupedRepresentation propstat = new PropstatGroupedRepresentation(resource, properties, false);
PropertyWriteUtil.writePropStats(xmlStreamWriter, propstat.getPropStats());
xmlStreamWriter.writeEndElement();
}
}
// for better GC
resultNodes.clear();
resultNodes = null;
// D:multistatus
xmlStreamWriter.writeEndElement();
xmlStreamWriter.writeEndDocument();
}
catch (XMLStreamException exc)
{
throw new IOException(exc.getMessage(), exc);
}
catch (Exception exc)
{
LOG.error(exc.getMessage(), exc);
throw new IOException(exc.getMessage(), exc);
}
}
}