TARDISLanguageUpdater.java

/*
 * Copyright (C) 2016 eccentric_nz
 *
 * 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 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.eccentric_nz.TARDIS.files;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import me.eccentric_nz.TARDIS.TARDIS;
import org.bukkit.ChatColor;
import org.bukkit.configuration.file.FileConfiguration;
import org.bukkit.configuration.file.YamlConfiguration;

/**
 * Updates the configured language file with any new message strings. It will
 * then be up to the server administrator to update the translation.
 *
 * @author eccentric_nz
 */
public class TARDISLanguageUpdater {

    private final TARDIS plugin;

    public TARDISLanguageUpdater(TARDIS plugin) {
        this.plugin = plugin;
    }

    public void update() {
        // get currently configured language
        String lang = plugin.getConfig().getString("preferences.language");
        if (lang.equals("en")) {
            return;
        }
        // get the English language config
        String en_path = plugin.getDataFolder() + File.separator + "language" + File.separator + "en.yml";
        File en_file = new File(en_path);
        FileConfiguration en = YamlConfiguration.loadConfiguration(en_file);
        // loop through the keys and set values that are missing
        int i = 0;
        for (String key : en.getKeys(false)) {
            if (!plugin.getLanguage().contains(key)) {
                plugin.getLanguage().set(key, en.getString(key));
                i++;
            } else if (key.equals("ARG_COLOUR")
                    || key.equals("BIND_NOT_VALID")
                    || key.equals("CONDENSE_REQUIRE")
                    || key.equals("MATERIAL_NOT_VALID")
                    || key.equals("REQUEST_TRAVEL")
                    || key.equals("RESCUE_REQUEST")
                    || key.equals("SAVE_NAME_NOT_VALID")
                    || key.equals("TARDIS_LOCS_INFO")
                    || key.equals("UPDATE_INTERFACE")
                    || key.equals("UPDATE_LOCATION")
                    || key.equals("UPGRADE_PERCENT_EXPLAIN")) {
                plugin.getLanguage().set(key, en.getString(key));
            }
        }
        try {
            // save the config
            String lang_path = plugin.getDataFolder() + File.separator + "language" + File.separator + lang + ".yml";
            plugin.getLanguage().save(new File(lang_path));
        } catch (IOException ex) {
            plugin.debug("Could not save language config file after adding entries! " + ex.getMessage());
        }
        // loop through the keys and remove values that are no longer in en.yml
        int j = 0;
        for (String key : plugin.getLanguage().getKeys(false)) {
            if (!en.contains(key)) {
                plugin.getLanguage().set(key, null);
                j++;
            }
        }
        // resave file
        String lang_path = plugin.getDataFolder() + File.separator + "language" + File.separator + lang + ".yml";
        try {
            // save the config
            plugin.getLanguage().save(new File(lang_path));
        } catch (IOException ex) {
            plugin.debug("Could not save language config file after removing entries! " + ex.getMessage());
        }
        if (i > 0) {
            plugin.getConsole().sendMessage(plugin.getPluginName() + "Added " + ChatColor.AQUA + i + ChatColor.RESET + " new messages to " + lang + ".yml");
            // sort language file alphabetically if additions were made
            try {
                FileReader fileReader = new FileReader(lang_path);
                BufferedReader bufferedReader = new BufferedReader(fileReader);
                String inputLine;
                List<String> lineList = new ArrayList<String>();
                while ((inputLine = bufferedReader.readLine()) != null) {
                    lineList.add(inputLine);
                }
                fileReader.close();

                Collections.sort(lineList);

                FileWriter fileWriter = new FileWriter(lang_path);
                PrintWriter out = new PrintWriter(fileWriter, false);
                for (String outputLine : lineList) {
                    out.println(outputLine);
                }
                out.flush();
                out.close();
                fileWriter.close();
            } catch (IOException ex) {
                plugin.debug("Could not alphabetize language config file! " + ex.getMessage());
            }
        }
        if (j > 0) {
            plugin.getConsole().sendMessage(plugin.getPluginName() + "Removed " + ChatColor.AQUA + j + ChatColor.RESET + " redundant messages from " + lang + ".yml");
        }
    }
}