1 files changed, 15 insertions, 9 deletions

diff --git a/src/main/java/net/tylermurphy/hideAndSeek/configuration/Localization.java b/src/main/java/net/tylermurphy/hideAndSeek/configuration/Localization.java
index 177b5bf..9980a4e 100644
--- a/src/main/java/net/tylermurphy/hideAndSeek/configuration/Localization.java
+++ b/src/main/java/net/tylermurphy/hideAndSeek/configuration/Localization.java
@@ -1,6 +1,9 @@
 package net.tylermurphy.hideAndSeek.configuration;
 
 import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -13,25 +16,28 @@ public class Localization {
 
 	public static final Map<String,LocalizationString> LOCAL = new HashMap<String,LocalizationString>();
 	
-	static YamlConfiguration config;
+	private static ConfigManager manager;
 	
-	public static boolean init() {
-		Main.plugin.saveResource("localization.yml", false);
-		String path = Main.data.getAbsolutePath()+File.separator + "localization.yml";
-		config = YamlConfiguration.loadConfiguration(new File(path));
-		for(String key : config.getConfigurationSection("Localization").getKeys(false)) {
+	public static boolean loadLocalization() {
+
+		manager = new ConfigManager("localization.yml");
+		manager.saveConfig();
+
+		for(String key : manager.getConfigurationSection("Localization").getKeys(false)) {
 			LOCAL.put(
 					key, 
-					new LocalizationString( ChatColor.translateAlternateColorCodes('&', config.getString("Localization."+key) ) )
+					new LocalizationString( ChatColor.translateAlternateColorCodes('&', manager.getString("Localization."+key) ) )
 					);
 		}
+
 		return true;
 	}
 	
 	public static LocalizationString message(String key) {
 		LocalizationString temp = LOCAL.get(key);
-		if(temp == null)
-			return new LocalizationString(key+" missing from localization.yml");
+		if(temp == null) {
+			return new LocalizationString(ChatColor.RED + "" + ChatColor.ITALIC + key + "is not found in localization.yml. This is a plugin issue, please report it.");
+		}
 		return new LocalizationString(temp.toString());
 	}
 }