org.atmosphere.util.AtmosphereConfigReader Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2008-2024 Async-IO.org
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

package org.atmosphere.util;

import org.atmosphere.config.ApplicationConfiguration;
import org.atmosphere.config.AtmosphereHandlerConfig;
import org.atmosphere.config.AtmosphereHandlerProperty;
import org.atmosphere.config.FrameworkConfiguration;
import org.atmosphere.cpr.AtmosphereConfig;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;

import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;

/**
 * Descriptor for an Atmosphere configuraton file.
 *
 * @author Jerome Dochez (for the version shipped in GlassFish v3).
 * @author Jeanfrancois Arcand
 * @author Sebastien Dionne : [email protected]
 */
public class AtmosphereConfigReader {

    private static final AtmosphereConfigReader instance = new AtmosphereConfigReader();
    private static final Logger logger = LoggerFactory.getLogger(AtmosphereConfigReader.class);

    private AtmosphereConfigReader() {
    }

    public AtmosphereConfig parse(AtmosphereConfig config, String filename) throws FileNotFoundException {

        DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
        try {
            return parse(config, factory.newDocumentBuilder().parse(filename));
        } catch (SAXException | IOException | ParserConfigurationException e) {
            logger.error(e.getMessage(), e);
            throw new RuntimeException(e);
        }

    }

    public AtmosphereConfig parse(AtmosphereConfig config, InputStream stream) throws FileNotFoundException {

        DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
        try {
            return parse(config, factory.newDocumentBuilder().parse(stream));
        } catch (SAXException | IOException | ParserConfigurationException e) {
            logger.error(e.getMessage(), e);
            throw new RuntimeException(e);
        }

    }

    /**
     * Parse the atmosphere-handlers element.
     *
     * @param document
     */
    private AtmosphereConfig parse(AtmosphereConfig config, Document document) {

        Element element = document.getDocumentElement();
        NodeList atmosphereHandlers = element.getElementsByTagName("atmosphere-handler");
        for (int i = 0; i < atmosphereHandlers.getLength(); i++) {
            AtmosphereHandlerConfig atmoHandler = new AtmosphereHandlerConfig();

            Node root = atmosphereHandlers.item(i);

            // parse Attributes
            for (int j = 0; j < root.getAttributes().getLength(); j++) {

                Node attribute = root.getAttributes().item(j);

                switch (attribute.getNodeName()) {
                    case "support-session":
                        atmoHandler.setSupportSession(attribute.getNodeValue());
                        break;
                    case "context-root":
                        atmoHandler.setContextRoot(attribute.getNodeValue());
                        break;
                    case "class-name":
                        atmoHandler.setClassName(attribute.getNodeValue());
                        break;
                    case "broadcaster":
                        atmoHandler.setBroadcaster(attribute.getNodeValue());
                        break;
                    case "broadcasterCache":
                        atmoHandler.setBroadcasterCache(attribute.getNodeValue());
                        break;
                    case "broadcastFilterClasses": {
                        String[] values = attribute.getNodeValue().split(",");
                        for (String value : values) {
                            atmoHandler.getBroadcastFilterClasses().add(value);
                        }
                        break;
                    }
                    case "comet-support":
                        atmoHandler.setCometSupport(attribute.getNodeValue());
                        break;
                    case "interceptorClasses": {
                        String[] values = attribute.getNodeValue().split(",");
                        for (String value : values) {
                            atmoHandler.getAtmosphereInterceptorClasses().add(value);
                        }
                        break;
                    }
                }
            }

            NodeList list = root.getChildNodes();

            for (int j = 0; j < list.getLength(); j++) {
                Node n = list.item(j);
                switch (n.getNodeName()) {
                    case "property": {
                        String param = n.getAttributes().getNamedItem("name").getNodeValue();
                        String value = n.getAttributes().getNamedItem("value").getNodeValue();

                        atmoHandler.getProperties().add(new AtmosphereHandlerProperty(param, value));
                        break;
                    }
                    case "applicationConfig": {

                        String param = null;
                        String value = null;
                        for (int k = 0; k < n.getChildNodes().getLength(); k++) {

                            Node n2 = n.getChildNodes().item(k);

                            if (n2.getNodeName().equals("param-name")) {
                                param = n2.getFirstChild().getNodeValue();
                            } else if (n2.getNodeName().equals("param-value")) {
                                value = n2.getFirstChild().getNodeValue();
                            }

                        }

                        if (param != null) {
                            atmoHandler.getApplicationConfig().add(new ApplicationConfiguration(param, value));
                        }

                        break;
                    }
                    case "frameworkConfig": {
                        String param = null;
                        String value = null;
                        for (int k = 0; k < n.getChildNodes().getLength(); k++) {

                            Node n2 = n.getChildNodes().item(k);

                            if (n2.getNodeName().equals("param-name")) {
                                param = n2.getFirstChild().getNodeValue();
                            } else if (n2.getNodeName().equals("param-value")) {
                                value = n2.getFirstChild().getNodeValue();
                            }
                        }

                        if (param != null) {
                            atmoHandler.getFrameworkConfig().add(
                                    new FrameworkConfiguration(param, value));
                        }

                        break;
                    }
                }

            }

            config.getAtmosphereHandlerConfig().add(atmoHandler);
        }

        return config;

    }

    public static AtmosphereConfigReader getInstance() {
        return instance;
    }
}