info.novatec.testit.livingdoc.agent.server.AgentConfiguration.java Source code

Introduction

Here is the source code for info.novatec.testit.livingdoc.agent.server.AgentConfiguration.java

Source

/**
 * Copyright (c) 2008 Pyxis Technologies inc.
 * <p>
 * This is free software; you can redistribute it and/or modify it under the
 * terms of the GNU General Public License as published by the Free Software
 * Foundation; either version 2 of the License, or (at your option) any later
 * version.
 * <p>
 * This software 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 General Public License for more
 * details.
 * <p>
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 51
 * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA, or see the FSF site:
 * http://www.fsf.org.
 */
package info.novatec.testit.livingdoc.agent.server;

import org.apache.commons.lang3.*;
import org.apache.logging.log4j.*;

import java.io.*;
import java.util.*;

public class AgentConfiguration {

    private final static int DEFAULT_PORT = 56000;
    private final static String DEFAULT_CONFIG_FILE = "remoteagent.properties";

    private final static String AGENT_KEY_NAME = "livingdoc.remoteagent.";
    private final static String PORT_PROPERTY_NAME = AGENT_KEY_NAME + "port";
    private final static String SECURED_PROPERTY_NAME = AGENT_KEY_NAME + "secured";
    private final static String KEYSTORE_PROPERTY_NAME = AGENT_KEY_NAME + "keystore.file";
    private final static String KEYSTORE_PASSWORD_PROPERTY_NAME = AGENT_KEY_NAME + "keystore.password";

    private static final Logger log = LogManager.getLogger(AgentConfiguration.class);

    private static int port = 56000;
    private static boolean secured = false;
    private static String keyStore = null;
    private static String keyStorePassword = null;

    public AgentConfiguration(String... args) throws IOException {
        ComandLineHelper commandLineHelper = new ComandLineHelper(args);

        Properties properties = loadProperties(commandLineHelper.getConfig());

        port = getPort(properties, commandLineHelper.getPort(DEFAULT_PORT));
        secured = isSecured(properties, commandLineHelper.isSecured());
        keyStore = getKeyStore(properties, commandLineHelper.getKeyStore());
        keyStorePassword = getKeyStorePassword(properties);
    }

    public static int getPort() {
        return port;
    }

    public static boolean isSecured() {
        return secured;
    }

    public static String getKeyStore() throws IOException {
        if (!isSecured()) {
            return null;
        }

        if (keyStore == null) {
            throw new IllegalArgumentException("You must specify keystore file location");
        }

        final File keyStoreFile = new File(keyStore);

        if (!keyStoreFile.exists()) {
            throw new FileNotFoundException(
                    String.format("KeyStore file '%s' does not exists", keyStoreFile.getAbsolutePath()));
        }

        return keyStore;
    }

    public static String getKeyStorePassword() {
        if (!isSecured()) {
            return null;
        }

        return keyStorePassword;
    }

    public void setKeyStorePassword(String pass) {
        keyStorePassword = pass;
    }

    private Properties loadProperties(String config) throws IOException {
        Properties properties = new Properties();

        if (config == null) {
            File configFile = new File(DEFAULT_CONFIG_FILE);

            if (configFile.exists()) {
                loadProperties(properties, configFile);
            } else {
                log.info("Default configuration file not found.  Skip.");
            }
        } else {
            File configFile = new File(config);

            if (!configFile.exists()) {
                throw new FileNotFoundException(
                        String.format("Configuration file '%s' does not exist", configFile.getAbsolutePath()));
            }

            loadProperties(properties, configFile);
        }

        return properties;
    }

    private void loadProperties(Properties properties, File filename) throws IOException {
        log.info(String.format("Reading configuration file '%s'...", filename.getAbsolutePath()));

        FileInputStream fis = new FileInputStream(filename);
        properties.load(fis);
        fis.close();
    }

    private int getPort(Properties properties, int defaultValue) {
        return Integer.parseInt(properties.getProperty(PORT_PROPERTY_NAME, String.valueOf(defaultValue)));
    }

    private boolean isSecured(Properties properties, boolean defaultValue) {
        return Boolean.parseBoolean(properties.getProperty(SECURED_PROPERTY_NAME, String.valueOf(defaultValue)));
    }

    private String getKeyStore(Properties properties, String defaultValue) {
        return StringUtils.stripToNull(properties.getProperty(KEYSTORE_PROPERTY_NAME, defaultValue));
    }

    private String getKeyStorePassword(Properties properties) {
        return StringUtils.stripToNull(properties.getProperty(KEYSTORE_PASSWORD_PROPERTY_NAME, null));
    }
}