org.brunocvcunha.taskerbox.core.TaskerboxXmlReader.java Source code

Introduction

Here is the source code for org.brunocvcunha.taskerbox.core.TaskerboxXmlReader.java

Source

/**
 * Copyright (C) 2015 Bruno Candido Volpato da Cunha (brunocvcunha@gmail.com)
 *
 * 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.brunocvcunha.taskerbox.core;

import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.io.StringWriter;
import java.util.List;

import org.apache.commons.lang3.StringEscapeUtils;
import org.brunocvcunha.taskerbox.Taskerbox;
import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;
import org.dom4j.tree.DefaultAttribute;
import org.dom4j.tree.DefaultElement;

import lombok.extern.log4j.Log4j;

/**
 * Centralized taskerbox*.xml Files Parsing
 *
 * @author Bruno Candido Volpato da Cunha
 *
 */
@Log4j
public class TaskerboxXmlReader {

    private Taskerbox tasker;

    public TaskerboxXmlReader(Taskerbox tasker) {
        super();
        this.tasker = tasker;
    }

    @SuppressWarnings("unchecked")
    public void handleTaskerboxStream(InputStream stream) throws Exception {

        Document dom = new SAXReader().read(stream);

        for (Element xmlChannel : (List<Element>) dom.getRootElement().elements()) {
            try {
                if (xmlChannel.getName().equalsIgnoreCase("defaultProperties")) {
                    handleDefaultPropertiesNode(xmlChannel);
                } else if (xmlChannel.getName().equalsIgnoreCase("systemProperties")) {
                    handleSystemPropertiesNode(xmlChannel);
                } else if (xmlChannel.getName().equalsIgnoreCase("macros")) {
                    handleMacrosNode(xmlChannel);
                } else if (xmlChannel.getName().equalsIgnoreCase("macro")) {
                    handleMacroUsageNode(xmlChannel);
                } else {
                    handleConcreteChannelNode(xmlChannel);
                }
            } catch (Exception e) {
                log.error("Exception Handling node", e);
            }
        }

    }

    private void handleConcreteChannelNode(Element xmlChannel) throws Exception {
        log.info("Found channel: " + xmlChannel.getName() + ": " + xmlChannel.toString());

        StringWriter sw = new StringWriter();
        XMLWriter writer = new XMLWriter(sw);
        writer.write(xmlChannel);

        String macroElement = sw.toString();

        for (String defaultAttr : this.tasker.getDefaultProperties().keySet()) {
            macroElement = macroElement.replace("{" + defaultAttr + "}",
                    this.tasker.getDefaultProperties().get(defaultAttr));
        }
        for (String defaultAttr : this.tasker.getDefaultProperties().keySet()) {
            macroElement = macroElement.replace("default-" + defaultAttr + "=\"\"", defaultAttr + "=\""
                    + StringEscapeUtils.escapeXml(this.tasker.getDefaultProperties().get(defaultAttr)) + "\"");
        }

        log.debug("Creating channel: " + macroElement);
        Element replacedXmlChannel = new SAXReader().read(new StringReader(macroElement)).getRootElement();

        TaskerboxChannel<?> channel = TaskerboxFactory.buildChannel(replacedXmlChannel);
        TaskerboxLauncher.startChannel(channel, this.tasker.getFrame(), this.tasker.getDaemons(),
                this.tasker.getChannels());

    }

    private void handleMacroUsageNode(Element xmlChannel) throws Exception {

        String macroName = xmlChannel.attributeValue("name");
        log.debug("Found macro! " + macroName);

        if (!this.tasker.getMacros().containsKey(macroName)) {
            throw new IllegalArgumentException("Using unexistent macro " + macroName);
        }

        String macroElement = this.tasker.getMacros().get(macroName);

        for (Object attrObj : xmlChannel.attributes()) {
            DefaultAttribute a = (DefaultAttribute) attrObj;
            macroElement = macroElement.replace("{" + a.getName() + "}", a.getValue());
        }

        for (Object attrMacro : this.tasker.getMacroAttrs().get(macroName)) {
            DefaultAttribute a = (DefaultAttribute) attrMacro;
            macroElement = macroElement.replace("{" + a.getName() + "}", a.getValue());
        }

        for (String defaultAttr : this.tasker.getDefaultProperties().keySet()) {
            macroElement = macroElement.replace("{" + defaultAttr + "}",
                    this.tasker.getDefaultProperties().get(defaultAttr));
        }
        for (String defaultAttr : this.tasker.getDefaultProperties().keySet()) {
            macroElement = macroElement.replace("default-" + defaultAttr + "=\"\"", defaultAttr + "=\""
                    + StringEscapeUtils.escapeXml(this.tasker.getDefaultProperties().get(defaultAttr)) + "\"");
        }

        log.debug("Creating Macro channel: " + macroElement);

        Element el = new SAXReader().read(new StringReader(macroElement)).getRootElement();

        TaskerboxChannel<?> channel = TaskerboxFactory.buildChannel(el);
        TaskerboxLauncher.startChannel(channel, this.tasker.getFrame(), this.tasker.getDaemons(),
                this.tasker.getChannels());

    }

    private void handleMacrosNode(Element xmlChannel) throws IOException {

        for (Object attrObj : xmlChannel.elements()) {
            DefaultElement e = (DefaultElement) attrObj;

            StringWriter sw = new StringWriter();
            XMLWriter writer = new XMLWriter(sw);
            writer.write(e.elements());

            if (this.tasker.getMacros().containsKey(e.attributeValue("name"))) {
                throw new RuntimeException("Macro " + e.attributeValue("name") + " already exists in map.");
            }

            this.tasker.getMacros().put(e.attributeValue("name"), sw.toString());
            this.tasker.getMacroAttrs().put(e.attributeValue("name"), e.attributes());
        }

    }

    private void handleSystemPropertiesNode(Element xmlChannel) {
        for (Object attrObj : xmlChannel.elements()) {
            DefaultElement e = (DefaultElement) attrObj;
            System.setProperty(e.attributeValue("name"), e.attributeValue("value"));
            handleDefaultPropertiesNode(xmlChannel);
        }
    }

    private void handleDefaultPropertiesNode(Element xmlChannel) {
        for (Object attrObj : xmlChannel.elements()) {
            DefaultElement e = (DefaultElement) attrObj;

            String propertyValue = e.attributeValue("value");

            for (String defaultAttr : this.tasker.getDefaultProperties().keySet()) {
                propertyValue = propertyValue.replace("{" + defaultAttr + "}",
                        this.tasker.getDefaultProperties().get(defaultAttr));
            }

            this.tasker.getDefaultProperties().put(e.attributeValue("name"), propertyValue);

        }
    }

}