FileConfigurationProvider.java

/*
 * Copyright (C) 2012 Google Inc.
 *
 * 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 interactivespaces.launcher.bootstrap;

import interactivespaces.system.core.configuration.ConfigurationProvider;

import org.apache.commons.logging.Log;

import java.io.File;
import java.io.FileInputStream;
import java.io.FilenameFilter;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Properties;

/**
 * A configuration provider from a file.
 *
 * @author Keith M. Hughes
 */
public class FileConfigurationProvider implements ConfigurationProvider {

  /**
   * Extensions on config files.
   */
  private static final String CONFIGURATION_FILES_EXTENSION = ".conf";

  /**
   * The base install folder.
   */
  private File baseInstallFolder;

  /**
   * The initial configuration folder.
   */
  private File configFolder;

  /**
   * The current configuration.
   */
  private Map<String, String> currentConfiguration;

  /**
   * The logger to use.
   */
  private Log log;

  /**
   * Construct a new provider.
   *
   * @param baseInstallFolder
   *          base install folder for the Interactive Spaces Container.
   * @param configFolder
   *          the configuration folder for this component
   * @param log
   *          the logger to use during
   */
  public FileConfigurationProvider(File baseInstallFolder, File configFolder, Log log) {
    this.baseInstallFolder = baseInstallFolder;
    this.configFolder = configFolder;
    this.log = log;
    currentConfiguration = new HashMap<String, String>();
  }

  @Override
  public Map<String, String> getInitialConfiguration() {
    return currentConfiguration;
  }

  /**
   * Load all conf files in the configuration folder.
   */
  public void load() {
    // Look in the specified bundle directory to create a list
    // of all JAR files to install.
    File[] files = configFolder.listFiles(new FilenameFilter() {
      @Override
      public boolean accept(File dir, String name) {
        return name.toLowerCase().endsWith(CONFIGURATION_FILES_EXTENSION);
      }
    });
    if (files == null || files.length == 0) {
      log.error(String.format("Couldn't load config files from %s\n", configFolder.getAbsolutePath()));
    }

    for (File file : files) {
      Properties props = new Properties();
      try {
        props.load(new FileInputStream(file));
        for (Entry<Object, Object> p : props.entrySet()) {
          currentConfiguration.put((String) p.getKey(), (String) p.getValue());
        }
      } catch (IOException e) {
        log.error(String.format("Couldn't load config file %s\n", file));
      }
    }
  }

  @Override
  public File getConfigFolder() {
    return configFolder;
  }

  /**
   * Add in a new configuration value into the configuration.
   *
   * @param configuration
   *          the configuration name
   * @param value
   *          the value of the configuration
   */
  public void put(String configuration, String value) {
    currentConfiguration.put(configuration, value);
  }
}