io.milton.http.report.AbstractMultiGetReport Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of milton-server-ent Show documentation
Milton Enterprise: Supports DAV level 2 and above, including Caldav and Carddav. Available on AGPL or commercial licenses
The newest version!
/*
 * Copyright 2012 McEvoy Software Ltd.
 *
 */

package io.milton.http.report;

import io.milton.common.Utils;
import io.milton.http.HttpManager;
import io.milton.http.ResourceFactory;
import io.milton.http.exceptions.BadRequestException;
import io.milton.http.exceptions.NotAuthorizedException;
import io.milton.http.webdav.*;
import io.milton.resource.PropFindableResource;
import io.milton.resource.Resource;
import org.jdom2.Document;
import org.jdom2.Element;
import org.jdom2.Namespace;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.xml.namespace.QName;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * @author brad
 */
public abstract class AbstractMultiGetReport implements QualifiedReport {

    private static final Logger log = LoggerFactory.getLogger(AbstractMultiGetReport.class);
    protected final ResourceFactory resourceFactory;
    protected final PropFindPropertyBuilder propertyBuilder;
    protected final PropFindXmlGenerator xmlGenerator;
    protected final Namespace NS_DAV = Namespace.getNamespace(WebDavProtocol.NS_DAV.getPrefix(), WebDavProtocol.NS_DAV.getName());

    public AbstractMultiGetReport(ResourceFactory resourceFactory, PropFindPropertyBuilder propertyBuilder, PropFindXmlGenerator xmlGenerator) {
        if (resourceFactory == null) {
            throw new NullPointerException("resourceFactory is null");
        }
        this.resourceFactory = resourceFactory;
        this.propertyBuilder = propertyBuilder;
        this.xmlGenerator = xmlGenerator;
    }

    @Override
    public String process(String host, String path, Resource calendar, Document doc) throws NotAuthorizedException, BadRequestException {
        log.debug("process report: " + getName());
        // The requested properties
        Set props = getProps(doc);
        // The requested resources
        List hrefs = getHrefs(doc);

        PropertiesRequest parseResult = PropertiesRequest.toProperties(props);

        // Generate the response
        List respProps = new ArrayList<>();

        for (String href : hrefs) {
            if(!href.startsWith("/")) {
                href = Utils.suffixSlash(path) + href;
            }
            String decodedHref = HttpManager.decodeUrl(href);
            Resource r = resourceFactory.getResource(host, decodedHref);
            if (r != null) {
                if (r instanceof PropFindableResource) {
                    PropFindableResource pfr = (PropFindableResource) r;
                    try {
                        respProps.addAll(propertyBuilder.buildProperties(pfr, 0, parseResult, href));
                    } catch (URISyntaxException ex) {
                        throw new RuntimeException("There was an unencoded url requested: " + href, ex);
                    }
                } else {
                    log.warn("requested href is for a non PropFindableResource: " + r.getClass() + " - " + href);
                }
            } else {
                log.warn("requested href not found: " + href + " on resource factory: " + resourceFactory.getClass());
            }
        }

        String xml = xmlGenerator.generate(respProps);
        if (log.isTraceEnabled()) {
            log.trace(getName() + ": response");
            log.trace(xml);
        }
        return xml;
    }

    private List getHrefs(Document doc) {
        List list = new ArrayList<>();
        for (Object o : doc.getRootElement().getChildren()) {
            if (o instanceof Element) {
                Element el = (Element) o;
                if (el.getName().equals("href")) {
                    String href = el.getText();
                    list.add(href);
                    if (log.isTraceEnabled()) {
                        log.trace(" href: " + href);
                    }
                }
            }
        }
        return list;
    }

    private Set getProps(Document doc) {
        Element elProp = doc.getRootElement().getChild("prop", NS_DAV);
        if (elProp == null) {
            throw new RuntimeException("No prop element");
        }

        Set set = new HashSet<>();
        for (Object o : elProp.getChildren()) {
            if (o instanceof Element) {
                Element el = (Element) o;
                String local = el.getName();
                String ns = el.getNamespaceURI();
                set.add(new QName(ns, local, el.getNamespacePrefix()));
            }
        }
        return set;
    }
}