info.magnolia.cms.util.WebXmlUtil.java Source code

Java tutorial

Introduction

Here is the source code for info.magnolia.cms.util.WebXmlUtil.java

Source

/**
 * This file Copyright (c) 2003-2012 Magnolia International
 * Ltd.  (http://www.magnolia-cms.com). All rights reserved.
 *
 *
 * This file is dual-licensed under both the Magnolia
 * Network Agreement and the GNU General Public License.
 * You may elect to use one or the other of these licenses.
 *
 * This file is distributed in the hope that it will be
 * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the
 * implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT.
 * Redistribution, except as permitted by whichever of the GPL
 * or MNA you select, is prohibited.
 *
 * 1. For the GPL license (GPL), you can redistribute and/or
 * modify this file under the terms of the GNU General
 * Public License, Version 3, as published by the Free Software
 * Foundation.  You should have received a copy of the GNU
 * General Public License, Version 3 along with this program;
 * if not, write to the Free Software Foundation, Inc., 51
 * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * 2. For the Magnolia Network Agreement (MNA), this file
 * and the accompanying materials are made available under the
 * terms of the MNA which accompanies this distribution, and
 * is available at http://www.magnolia-cms.com/mna.html
 *
 * Any modifications to this file must keep this entire header
 * intact.
 *
 */
package info.magnolia.cms.util;

import info.magnolia.cms.core.Path;
import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.collections.Transformer;
import org.jdom.Document;
import org.jdom.Element;
import org.jdom.JDOMException;
import org.jdom.input.SAXBuilder;
import org.jdom.xpath.XPath;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;

/**
 * Contains utility methods to register or check for the existence of elements in web.xml.
 *
 * @author gjoseph
 * @version $Revision: $ ($Author: $)
 */
public class WebXmlUtil {
    private final Document doc;

    public WebXmlUtil() {
        final File source = new File(Path.getAppRootDir() + "/WEB-INF/web.xml");
        if (!source.exists()) {
            throw new IllegalStateException("Failed to locate web.xml : " + source.getAbsolutePath());
        }
        final SAXBuilder builder = new SAXBuilder();
        try {
            doc = builder.build(source);
        } catch (JDOMException e) {
            throw new RuntimeException(e); // TODO
        } catch (IOException e) {
            throw new RuntimeException(e); // TODO
        }
    }

    /**
     * Test-friendly constructor.
     */
    public WebXmlUtil(InputStream inputStream) {
        final SAXBuilder builder = new SAXBuilder();
        try {
            doc = builder.build(inputStream);
        } catch (JDOMException e) {
            throw new RuntimeException(e); // TODO
        } catch (IOException e) {
            throw new RuntimeException(e); // TODO
        }
    }

    public boolean isServletOrMappingRegistered(String servletName) {
        return isServletRegistered(servletName) || isServletMappingRegistered(servletName);
    }

    public boolean isServletRegistered(String servletName) {
        return xpathMatches("/webxml:web-app/webxml:servlet[webxml:servlet-name='" + servletName + "']");
    }

    public boolean isServletMappingRegistered(String servletName) {
        return xpathMatches("/webxml:web-app/webxml:servlet-mapping[webxml:servlet-name='" + servletName + "']");
    }

    public boolean isServletMappingRegistered(String servletName, String urlPattern) {
        final String xpathExpr = "/webxml:web-app/webxml:servlet-mapping[webxml:servlet-name='" + servletName
                + "' and webxml:url-pattern='" + urlPattern + "']";
        return xpathMatches(xpathExpr);
    }

    public Collection getServletMappings(String servletName) {
        final String servletMappingXPathExpr = "/webxml:web-app/webxml:servlet-mapping[webxml:servlet-name='"
                + servletName + "']/webxml:url-pattern";
        final List servletMappings = getElementsFromXPath(servletMappingXPathExpr);

        return CollectionUtils.collect(servletMappings, new Transformer() {
            @Override
            public Object transform(Object input) {
                final Element servletMapping = (Element) input;
                return servletMapping.getText();
            }
        });
    }

    public boolean isFilterRegistered(String filterClass) {
        return getFilterElement(filterClass) != null;
    }

    /**
     * @deprecated since 3.6.4, use checkFilterDispatchersConfiguration()
     */
    public boolean areFilterDispatchersConfiguredProperly(String filterClass, List mandatoryDispatchers,
            List optionalDispatchers) {
        return checkFilterDispatchersConfiguration(filterClass, mandatoryDispatchers, optionalDispatchers) >= 0;
    }

    /**
     * Returns:
     * +1 if all mandatory dispatchers are present and no additional unsupported dispatcher is present, or this filter class isn't registered.
     * 0  if all mandatory dispatchers are present but additional unsupported dispatchers are present.
     * -1  if not all mandatory dispatchers are present.
     */
    public int checkFilterDispatchersConfiguration(String filterClass, List mandatoryDispatchers,
            List optionalDispatchers) {
        final Element filterEl = getFilterElement(filterClass);
        if (filterEl != null) {
            final String filterName = filterEl.getTextNormalize();
            final String filterMappingXPathExpr = "/webxml:web-app/webxml:filter-mapping[webxml:filter-name='"
                    + filterName + "']/webxml:dispatcher";
            final List dispatchersEl = getElementsFromXPath(filterMappingXPathExpr);
            final List registeredDispatchers = new ArrayList();
            final Iterator it = dispatchersEl.iterator();
            while (it.hasNext()) {
                final Element dispatcherEl = (Element) it.next();
                registeredDispatchers.add(dispatcherEl.getTextNormalize());
            }
            registeredDispatchers.removeAll(optionalDispatchers);
            if (CollectionUtils.isEqualCollection(mandatoryDispatchers, registeredDispatchers)) {
                return 1;
            } else if (CollectionUtils.isSubCollection(mandatoryDispatchers, registeredDispatchers)) {
                return 0;
            } else {
                return -1;
            }

        }
        return 1;
    }

    public boolean isListenerRegistered(String deprecatedListenerClass) {
        final String xpathExpr = "/webxml:web-app/webxml:listener[webxml:listener-class='" + deprecatedListenerClass
                + "']";
        return xpathMatches(xpathExpr);
    }

    private Element getFilterElement(String filterClass) {
        final String filterXPathExpr = "/webxml:web-app/webxml:filter[webxml:filter-class='" + filterClass
                + "']/webxml:filter-name";

        return getElementFromXPath(filterXPathExpr);
    }

    private List getElementsFromXPath(String xpathExpr) {
        try {
            final XPath xpath = XPath.newInstance(xpathExpr);
            // must add the namespace and use it: there is no default namespace elsewise
            xpath.addNamespace("webxml", doc.getRootElement().getNamespace().getURI());
            return xpath.selectNodes(doc);
        } catch (JDOMException e) {
            throw new RuntimeException(e); // TODO
        }
    }

    private Element getElementFromXPath(String xpathExpr) {
        final List list = getElementsFromXPath(xpathExpr);
        return (Element) (list.size() > 0 ? list.get(0) : null);
    }

    private boolean xpathMatches(String xpathExpr) {
        return getElementsFromXPath(xpathExpr).size() > 0;
    }
}