io.sitoolkit.wt.gui.infra.config.PropertyManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sit-wt-app Show documentation
The newest version!
package io.sitoolkit.wt.gui.infra.config;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Properties;
import io.sitoolkit.wt.infra.log.SitLogger;
import io.sitoolkit.wt.infra.log.SitLoggerFactory;
import io.sitoolkit.wt.util.infra.util.StrUtils;

public class PropertyManager {

  private static final String FILE_NAME = "sit-wt-app.properties";

  private static final String SEPARATOR = ",";

  private static final String BASE_URL = "baseUrl";

  private static final String BASE_URL_LIMIT = "baseUrlLimit";

  private static final String CSV_CHARSET = "script.file.csv.charset";

  private static final String CSV_BOM = "script.file.csv.bom";

  private static final SitLogger LOG = SitLoggerFactory.getLogger(PropertyManager.class);

  private static final PropertyManager pm = new PropertyManager();

  private Properties prop = new Properties();

  private File baseDir;

  private List baseUrls = new ArrayList<>();

  private PropertyManager() {}

  public static PropertyManager get() {
    return pm;
  }

  public void load(File baseDir) {

    this.baseDir = baseDir;

    File propertyFile = new File(baseDir, FILE_NAME);

    if (!propertyFile.exists()) {
      LOG.debug("app.noPropertyFile", baseDir.getAbsolutePath());
      return;
    }

    try (FileInputStream fis = new FileInputStream(propertyFile)) {

      prop.load(fis);
      LOG.info("app.loadProperties", prop);

      List savedBaseUrls = Arrays.asList(getProp(BASE_URL).split(SEPARATOR));
      savedBaseUrls = savedBaseUrls.subList(0, Math.min(savedBaseUrls.size(), getBaseUrlLimit()));
      baseUrls.addAll(savedBaseUrls);

    } catch (IOException e) {

      LOG.warn("app.loadPropertiesFailed", e);

    }
  }

  public void save() {

    if (baseDir == null) {
      return;
    }

    try (FileOutputStream fos = new FileOutputStream(new File(baseDir, FILE_NAME), false)) {

      setProp(BASE_URL, StrUtils.join(baseUrls));
      setProp(CSV_CHARSET, getCsvCharset().name());
      setProp(CSV_BOM, String.valueOf(getCsvHasBOM()));

      prop.store(fos, "SI-Toolkit for Web Testing");
      LOG.info("app.saveProperties", prop);

    } catch (IOException e) {

      LOG.warn("app.savePropertiesFailed", e);

    }

  }

  public void setBaseUrls(List baseUrls) {
    this.baseUrls = baseUrls;
  }

  public List getBaseUrls() {
    return baseUrls;
  }

  public int getBaseUrlLimit() {
    return Integer.parseInt(getProp(BASE_URL_LIMIT, "5"));
  }

  public Charset getCsvCharset() {
    return Charset.forName(getProp(CSV_CHARSET, "UTF-8"));
  }

  public boolean getCsvHasBOM() {
    return Boolean.valueOf(getProp(CSV_BOM, "true"));
  }

  private String getProp(String key) {
    return getProp(key, "");
  }

  private String getProp(String key, String defaultValue) {
    return prop.getProperty(key, defaultValue);
  }

  private void setProp(String key, String value) {
    prop.setProperty(key, value);
  }

}