com.softwaremagico.tm.file.configurator.PropertiesSourceFile Maven / Gradle / Ivy

Go to download
package com.softwaremagico.tm.file.configurator;

/*-
 * #%L
 * Think Machine (Rules)
 * %%
 * Copyright (C) 2017 - 2019 Softwaremagico
 * %%
 * This software is designed by Jorge Hortelano Otero. Jorge Hortelano Otero
 *  Valencia (Spain).
 *
 * This program 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.
 *
 * This program 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.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; If not, see .
 * #L%
 */

import com.softwaremagico.tm.exceptions.PropertyNotStoredException;
import com.softwaremagico.tm.exceptions.ResurceNotFoundException;
import com.softwaremagico.tm.file.watcher.FileWatcher;
import com.softwaremagico.tm.file.watcher.FileWatcher.FileModifiedListener;
import com.softwaremagico.tm.log.ConfigurationLog;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.nio.file.NoSuchFileException;
import java.nio.file.Path;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Properties;
import java.util.Set;
import java.util.TreeMap;

public class PropertiesSourceFile extends SourceFile implements IPropertiesSource {
    private Set fileModifiedListeners;
    private FileWatcher fileWatcher;

    public PropertiesSourceFile(String fileName) {
        super(fileName);
        fileModifiedListeners = new HashSet<>();
    }

    public PropertiesSourceFile(String filePath, String fileName) {
        super(filePath, fileName);
        fileModifiedListeners = new HashSet<>();
    }

    public void addFileModifiedListeners(FileModifiedListener fileModifiedListener) {
        fileModifiedListeners.add(fileModifiedListener);
    }

    @Override
    public void setFilePath(String filePath) {
        super.setFilePath(filePath);
    }

    public void storeInFile(Map propertiesValues) throws IOException, PropertyNotStoredException {
        final Properties properties = new Properties();
        // Sort properties and add it.
        for (final Entry entry : new TreeMap(propertiesValues).entrySet()) {
            properties.setProperty(entry.getKey(), entry.getValue());
        }
        PropertiesFile.store(properties, getFilePath(), getFileName());
        ConfigurationLog.debug(this.getClass().getName(),
                "Storing '" + properties + "' at properties file '" + getFilePath() + File.separator + getFileName() + "'.");
    }

    @Override
    public Properties loadFile() {
        try {
            if (getFilePath() == null) {
                return PropertiesFile.load(getFileName());
            } else {
                return PropertiesFile.load(getFilePath(), getFileName());
            }
        } catch (FileNotFoundException e) {
            ConfigurationLog.debug(this.getClass().getName(), e.getMessage());
        } catch (IOException e) {
            ConfigurationLog.errorMessage(this.getClass().getName(), e);
        } catch (NullPointerException e) {
            ConfigurationLog.info(this.getClass().getName(), e.getMessage());
        }
        return null;
    }

    private String getDirectoryToWatch() throws ResurceNotFoundException {
        try {
            return (getFilePath() != null ? getFilePath() : this.getClass().getClassLoader().getResource(".").getPath());
        } catch (NullPointerException e) {
            throw new ResurceNotFoundException("No modules found.");
        }
    }

    @SuppressWarnings("unused")
    private void setWatcher() {
        final Set checkedFiles = new HashSet<>(Arrays.asList(new String[]{getFileName()}));

        try {
            fileWatcher = new FileWatcher(getDirectoryToWatch(), checkedFiles);
            fileWatcher.addFileModifiedListener(new FileModifiedListener() {

                @Override
                public void changeDetected(Path pathToFile) {
                    // Pass the listener to current listeners.
                    for (final FileModifiedListener fileModifiedListener : fileModifiedListeners) {
                        fileModifiedListener.changeDetected(pathToFile);
                    }
                }
            });
        } catch (NoSuchFileException | NullPointerException e) {
            try {
                ConfigurationLog.warning(this.getClass().getName(), "Directory '" + getDirectoryToWatch() + "' to watch not found!");
            } catch (ResurceNotFoundException e1) {
                ConfigurationLog.warning(this.getClass().getName(), "Modules not found!");
            }
        } catch (IOException e) {
            ConfigurationLog.errorMessage(this.getClass().getName(), e);
        } catch (ResurceNotFoundException e) {
            ConfigurationLog.warning(this.getClass().getName(), "Modules not found!");
        } catch (Exception e) {
            ConfigurationLog.warning(this.getClass().getName(), "Watcher not enabled!");
        }
    }

    public void stopFileWatcher() {
        if (fileWatcher != null) {
            fileWatcher.closeFileWatcher();
        }
    }
}