me.mayo.telnetkek.config.ConfigLoader.java Source code

Introduction

Here is the source code for me.mayo.telnetkek.config.ConfigLoader.java

Source

/* 
 * Copyright (C) 2012-2017 Steven Lawson
 *
 * This file is part of FreedomTelnetClient.
 *
 * FreedomTelnetClient 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 3 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 <http://www.gnu.org/licenses/>.
 */
package me.mayo.telnetkek.config;

import java.io.*;
import java.util.*;
import java.util.logging.Level;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.*;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import me.mayo.telnetkek.button.FavoriteButtonEntry;
import me.mayo.telnetkek.player.PlayerCommandEntry;
import me.mayo.telnetkek.ServerEntry;
import me.mayo.telnetkek.TelnetKek;
import org.apache.commons.io.FileUtils;
import org.w3c.dom.*;
import org.xml.sax.SAXException;

public class ConfigLoader {

    private static final String SETTINGS_FILE = "ftc_settings.xml";

    private final ServerEntry.ServerEntryList servers = new ServerEntry.ServerEntryList();
    private final PlayerCommandEntry.PlayerCommandEntryList playerCommands = new PlayerCommandEntry.PlayerCommandEntryList();
    private final FavoriteButtonEntry.FavoriteButtonEntryList favoriteButtons = new FavoriteButtonEntry.FavoriteButtonEntryList();

    public ConfigLoader() {
    }

    public boolean load(boolean verbose) {
        File settings = new File(SETTINGS_FILE);

        if (!settings.exists()) {
            if (extractFileFromJar("/" + SETTINGS_FILE, SETTINGS_FILE)) {
                if (verbose) {
                    System.out.println("Copied default " + SETTINGS_FILE + ".");
                }
            }
        }

        if (settings.exists()) {
            boolean loadError = loadXML(settings);

            generateXML(settings);

            if (verbose) {
                if (loadError) {
                    System.out.println("Settings loaded with errors.");
                } else {
                    System.out.println("Settings loaded.");
                }
            }

            return true;
        } else if (verbose) {
            System.out.println("Can't load " + SETTINGS_FILE + ".");
        }

        return false;
    }

    public boolean save() {
        return generateXML(new File(SETTINGS_FILE));
    }

    public Collection<PlayerCommandEntry> getCommands() {
        return this.playerCommands.getList();
    }

    public Collection<ServerEntry> getServers() {
        return this.servers.getList();
    }

    public Collection<FavoriteButtonEntry> getFavoriteButtons() {
        return favoriteButtons.getList();
    }

    private boolean generateXML(final File file) {
        try {
            final Document doc = DocumentBuilderFactory.newInstance().newDocumentBuilder().newDocument();

            final Element rootElement = doc.createElement("configuration");
            doc.appendChild(rootElement);

            rootElement.appendChild(this.servers.listToXML(doc));
            rootElement.appendChild(this.playerCommands.listToXML(doc));
            rootElement.appendChild(this.favoriteButtons.listToXML(doc));

            final Transformer transformer = TransformerFactory.newInstance().newTransformer();
            transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
            transformer.setOutputProperty(OutputKeys.INDENT, "yes");

            transformer.transform(new DOMSource(doc), new StreamResult(file));

            return true;
        } catch (IllegalArgumentException | ParserConfigurationException | TransformerException | DOMException ex) {
            TelnetKek.LOGGER.log(Level.SEVERE, null, ex);
        }

        return false;
    }

    private boolean loadXML(final File file) {
        boolean hadErrors = false;

        try {
            final Document doc = DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(file);
            doc.getDocumentElement().normalize();

            if (!this.servers.listFromXML(doc)) {
                System.out.println("Error loading servers.");
                hadErrors = true;
            }

            if (!this.playerCommands.listFromXML(doc)) {
                System.out.println("Error loading playerCommands.");
                hadErrors = true;
            }

            if (!this.favoriteButtons.listFromXML(doc)) {
                System.out.println("Error favorite buttons.");
                hadErrors = true;
            }
        } catch (IOException | ParserConfigurationException | SAXException ex) {
            hadErrors = true;

            TelnetKek.LOGGER.log(Level.SEVERE, null, ex);
        }

        return hadErrors;
    }

    private static boolean extractFileFromJar(final String resourceName, final String fileName) {
        final InputStream resource = ConfigLoader.class.getResourceAsStream(resourceName);
        if (resource != null) {
            final File destination = new File(fileName);
            try {
                FileUtils.copyInputStreamToFile(resource, destination);
                return true;
            } catch (IOException ex) {
                TelnetKek.LOGGER.log(Level.SEVERE, null, ex);
            }
        }

        return false;
    }
}