com.technophobia.substeps.model.Configuration.java Source code

Introduction

Here is the source code for com.technophobia.substeps.model.Configuration.java

Source

/*
 *  Copyright Technophobia Ltd 2012
 *
 *   This file is part of Substeps.
 *
 *    Substeps is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU Lesser General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    Substeps is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU Lesser General Public License for more details.
 *
 *    You should have received a copy of the GNU Lesser General Public License
 *    along with Substeps.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.technophobia.substeps.model;

import com.technophobia.substeps.model.exception.SubstepsConfigurationException;
import org.apache.commons.configuration.CombinedConfiguration;
import org.apache.commons.configuration.ConfigurationException;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.apache.commons.configuration.tree.OverrideCombiner;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.URL;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

/**
 * @author ian
 */
public enum Configuration {

    INSTANCE;

    private static final Logger logger = LoggerFactory.getLogger(Configuration.class);

    private final CombinedConfiguration combinedConfig = new CombinedConfiguration(new OverrideCombiner());

    private Configuration() {
        initialise();
    }

    private void initialise() {

        final String resourceBundleName = resourceBundleName();

        final URL customPropsUrl = Configuration.class.getResource(resourceBundleName);

        if (customPropsUrl != null) {

            try {
                final PropertiesConfiguration customProps = new PropertiesConfiguration(customPropsUrl);
                combinedConfig.addConfiguration(customProps, "customProps");

            } catch (final ConfigurationException e) {
                logger.error("error loading custom properties", e);

            }
        }
    }

    /**
     * Implementors of substep libraries should call this with default
     * properties for their library
     *
     * @param url  to a properties file containing default values
     * @param name to name of the properties file that is being added
     */
    public void addDefaultProperties(final URL url, final String name) {

        if (url != null) {
            try {

                final PropertiesConfiguration defaultProps = new PropertiesConfiguration(url);

                combinedConfig.addConfiguration(defaultProps, name);
            } catch (final ConfigurationException e) {
                logger.error("error loading default properties", e);
                throw new SubstepsConfigurationException(e);
            }
        }
    }

    public void addDefaultProperty(final String key, final Object value) {
        combinedConfig.addProperty(key, value);
    }

    public String getConfigurationInfo() {

        final List<String> configurationNameList = combinedConfig.getConfigurationNameList();

        final StringBuilder buf = new StringBuilder();

        for (final String configurationName : configurationNameList) {

            buf.append("In config: ").append(configurationName).append("\n");

            final org.apache.commons.configuration.Configuration cfg = combinedConfig
                    .getConfiguration(configurationName);

            final Iterator<String> keys = cfg.getKeys();

            while (keys.hasNext()) {
                final String key = keys.next();

                final String val = cfg.getString(key);

                buf.append("key: ").append(key).append("\tval: [").append(val).append("]\n");
            }

            buf.append("\n");
        }
        return buf.toString();
    }

    private static String resourceBundleName() {
        return "/" + System.getProperty("environment", "localhost") + ".properties";
    }

    public String getString(final String key) {
        return combinedConfig.getString(key);
    }

    public int getInt(final String key) {
        return combinedConfig.getInt(key);
    }

    public long getLong(final String key) {
        return combinedConfig.getLong(key);
    }

    public boolean getBoolean(final String key) {
        return combinedConfig.getBoolean(key);
    }
}