001    /**
002     * Copyright (c) 2000-2012 Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.webdav.methods;
016    
017    import com.liferay.portal.kernel.log.Log;
018    import com.liferay.portal.kernel.log.LogFactoryUtil;
019    import com.liferay.portal.kernel.util.FileUtil;
020    import com.liferay.portal.kernel.util.StringPool;
021    import com.liferay.portal.kernel.util.Validator;
022    import com.liferay.portal.kernel.webdav.WebDAVException;
023    import com.liferay.portal.kernel.webdav.WebDAVRequest;
024    import com.liferay.portal.kernel.webdav.WebDAVUtil;
025    import com.liferay.portal.kernel.xml.Document;
026    import com.liferay.portal.kernel.xml.Element;
027    import com.liferay.portal.kernel.xml.Namespace;
028    import com.liferay.portal.kernel.xml.QName;
029    import com.liferay.portal.kernel.xml.SAXReaderUtil;
030    import com.liferay.portal.webdav.InvalidRequestException;
031    import com.liferay.util.xml.XMLFormatter;
032    
033    import java.util.HashSet;
034    import java.util.Iterator;
035    import java.util.Set;
036    
037    import javax.servlet.http.HttpServletRequest;
038    import javax.servlet.http.HttpServletResponse;
039    
040    /**
041     * @author Brian Wing Shun Chan
042     * @author Alexander Chow
043     */
044    public class PropfindMethodImpl extends BasePropMethodImpl implements Method {
045    
046            public int process(WebDAVRequest webDavRequest) throws WebDAVException {
047                    try {
048                            Set<QName> props = getProps(webDavRequest);
049    
050                            return writeResponseXML(webDavRequest, props);
051                    }
052                    catch (InvalidRequestException ire) {
053                            return HttpServletResponse.SC_BAD_REQUEST;
054                    }
055                    catch (Exception e) {
056                            throw new WebDAVException(e);
057                    }
058            }
059    
060            protected Set<QName> generateProps(Set<QName> props) {
061                    props.add(DISPLAYNAME);
062                    props.add(RESOURCETYPE);
063                    props.add(GETCONTENTTYPE);
064                    props.add(GETCONTENTLENGTH);
065                    props.add(GETLASTMODIFIED);
066                    props.add(LOCKDISCOVERY);
067    
068                    // RFC 3253 Currently Unsupported
069    
070                    //props.add(new Tuple("checked-in", WebDAVUtil.DAV_URI));
071                    //props.add(new Tuple("checked-out", WebDAVUtil.DAV_URI));
072                    //props.add(new Tuple("version-name", WebDAVUtil.DAV_URI));
073    
074                    return props;
075            }
076    
077            protected Set<QName> getProps(WebDAVRequest webDavRequest)
078                    throws InvalidRequestException {
079    
080                    try {
081                            Set<QName> props = new HashSet<QName>();
082    
083                            HttpServletRequest request = webDavRequest.getHttpServletRequest();
084    
085                            String xml = new String(
086                                    FileUtil.getBytes(request.getInputStream()));
087    
088                            if (Validator.isNull(xml)) {
089    
090                                    // Windows XP does not generate an xml request so the PROPFIND
091                                    // must be generated manually. See LEP-4920.
092    
093                                    return generateProps(props);
094                            }
095    
096                            if (_log.isDebugEnabled()) {
097                                    _log.debug(
098                                            "Request XML: \n" +
099                                                    XMLFormatter.toString(xml, StringPool.FOUR_SPACES));
100                            }
101    
102                            Document doc = SAXReaderUtil.read(xml);
103    
104                            Element root = doc.getRootElement();
105    
106                            if (root.element(ALLPROP.getName()) != null) {
107    
108                                    // Generate props if <allprop> tag is used. See LEP-6162.
109    
110                                    return generateProps(props);
111                            }
112    
113                            Element prop = root.element("prop");
114    
115                            Iterator<Element> itr = prop.elements().iterator();
116    
117                            while (itr.hasNext()) {
118                                    Element el = itr.next();
119    
120                                    String prefix = el.getNamespacePrefix();
121                                    String uri = el.getNamespaceURI();
122    
123                                    Namespace namespace = WebDAVUtil.createNamespace(prefix, uri);
124    
125                                    props.add(SAXReaderUtil.createQName(el.getName(), namespace));
126                            }
127    
128                            return props;
129                    }
130                    catch (Exception e) {
131                            throw new InvalidRequestException(e);
132                    }
133            }
134    
135            private static Log _log = LogFactoryUtil.getLog(PropfindMethodImpl.class);
136    
137    }