Redstoner/Modules
Archived
0
Fork 0
Code Activity

Fixed multiple NPEs due to unchecked data access

Browse Source
This commit is contained in:
Pepich
2017-05-09 10:51:45 +02:00
parent c1ac03fb57
commit c208d1d595
1 changed files with 98 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

138
src/com/redstoner/modules/datamanager/DataManager.java
View File

@@ -5,6 +5,7 @@ import java.io.FilenameFilter;
import java.util.UUID; import java.util.UUID;
import org.bukkit.Bukkit; import org.bukkit.Bukkit;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Player; import org.bukkit.entity.Player;
import org.bukkit.event.EventHandler; import org.bukkit.event.EventHandler;
import org.bukkit.event.Listener; import org.bukkit.event.Listener;
@@ -20,11 +21,11 @@ import com.redstoner.misc.Utils;
import com.redstoner.modules.CoreModule; import com.redstoner.modules.CoreModule;
@AutoRegisterListener @AutoRegisterListener
@Version(major = 3, minor = 0, revision = 0, compatible = 3) @Version(major = 3, minor = 0, revision = 3, compatible = 3)
public final class DataManager implements CoreModule, Listener public final class DataManager implements CoreModule, Listener
{ {
private static final File dataFolder = new File(Main.plugin.getDataFolder(), "data"); private static final File dataFolder = new File(Main.plugin.getDataFolder(), "data");
private static JSONObject data; private static JSONObject data = new JSONObject();
@Override @Override
public void postEnable() public void postEnable()
@@ -42,7 +43,7 @@ public final class DataManager implements CoreModule, Listener
{ {
for (Player p : Bukkit.getOnlinePlayers()) for (Player p : Bukkit.getOnlinePlayers())
{ {
saveAndUnload(p.getUniqueId()); saveAndUnload(p);
} }
} }
@@ -55,7 +56,7 @@ public final class DataManager implements CoreModule, Listener
@EventHandler @EventHandler
public void onPlayerQuit(PlayerQuitEvent event) public void onPlayerQuit(PlayerQuitEvent event)
{ {
saveAndUnload(event.getPlayer().getUniqueId()); saveAndUnload(event.getPlayer());
} }
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
@@ -67,78 +68,116 @@ public final class DataManager implements CoreModule, Listener
data.put(id.toString(), playerData); data.put(id.toString(), playerData);
} }
public static Object getData(UUID id, String key) public static Object getData(CommandSender sender, String key)
{ {
return getData(id, Utils.getCaller(DataManager.class), key); return getData(sender, Utils.getCaller(DataManager.class), key);
} }
public static Object getData(UUID id, String module, String key) public static Object getData(CommandSender sender, String module, String key)
{ {
if (data.containsKey(id.toString())) String id;
return ((JSONObject) ((JSONObject) data.get(id.toString())).get(module)).get(key); if (sender instanceof Player)
id = ((Player) sender).getUniqueId().toString();
else
id = "CONSOLE";
if (data.containsKey(id))
{
JSONObject moduleData = ((JSONObject) ((JSONObject) data.get(id)).get(module));
if (moduleData == null)
return null;
return moduleData.get(key);
}
else else
return loadAndGet(id, module, key); return loadAndGet(id, module, key);
} }
private static Object loadAndGet(UUID id, String module, String key) private static Object loadAndGet(String id, String module, String key)
{ {
JSONObject playerData = JsonManager.getObject(new File(dataFolder, id.toString() + ".json")); JSONObject playerData = JsonManager.getObject(new File(dataFolder, id + ".json"));
if (playerData == null) if (playerData == null)
return null; return null;
return ((JSONObject) playerData.get(module)).get(key); return ((JSONObject) playerData.get(module)).get(key);
} }
public static void setData(UUID id, String key, Object value) public static void setData(CommandSender sender, String key, Object value)
{ {
setData(id, Utils.getCaller(DataManager.class), key, value); setData(sender, Utils.getCaller(DataManager.class), key, value);
} }
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
public static void setData(UUID id, String module, String key, Object value) public static void setData(CommandSender sender, String module, String key, Object value)
{ {
if (data.containsKey(id.toString())) String id;
if (sender instanceof Player)
id = ((Player) sender).getUniqueId().toString();
else
id = "CONSOLE";
if (data.containsKey(id))
{ {
((JSONObject) ((JSONObject) data.get(id.toString())).get(module)).put(key, value); JSONObject moduleData = ((JSONObject) ((JSONObject) data.get(id)).get(module));
save(id); if (moduleData == null)
{
moduleData = new JSONObject();
((JSONObject) data.get(id)).put(module, moduleData);
}
moduleData.put(key, value);
save(sender);
} }
else else
loadAndSet(id, module, key, value); loadAndSet(id, module, key, value);
} }
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
private static void loadAndSet(UUID id, String module, String key, Object value) private static void loadAndSet(String id, String module, String key, Object value)
{ {
File dataFile = new File(dataFolder, id.toString() + ".json"); File dataFile = new File(dataFolder, id + ".json");
JSONObject playerData = JsonManager.getObject(dataFile); JSONObject playerData = JsonManager.getObject(dataFile);
if (playerData == null) if (playerData == null)
playerData = new JSONObject(); playerData = new JSONObject();
((JSONObject) playerData.get(module)).put(key, value); JSONObject moduleData = ((JSONObject) playerData.get(module));
if (moduleData == null)
{
moduleData = new JSONObject();
playerData.put(module, moduleData);
}
moduleData.put(key, value);
JsonManager.save(playerData, dataFile); JsonManager.save(playerData, dataFile);
} }
public static void removeData(UUID id, String key) public static void removeData(CommandSender sender, String key)
{ {
removeData(id, Utils.getCaller(DataManager.class), key); removeData(sender, Utils.getCaller(DataManager.class), key);
} }
public static void removeData(UUID id, String module, String key) public static void removeData(CommandSender sender, String module, String key)
{ {
String id;
if (sender instanceof Player)
id = ((Player) sender).getUniqueId().toString();
else
id = "CONSOLE";
if (data.containsKey(id.toString())) if (data.containsKey(id.toString()))
{ {
((JSONObject) ((JSONObject) data.get(module)).get(id.toString())).remove(key); JSONObject moduleData = ((JSONObject) ((JSONObject) data.get(id)).get(module));
save(id); if (moduleData == null)
return;
moduleData.remove(key);
save(sender);
} }
else else
loadAndRemove(id, module, key); loadAndRemove(id, module, key);
} }
private static void loadAndRemove(UUID id, String module, String key) private static void loadAndRemove(String id, String module, String key)
{ {
File dataFile = new File(dataFolder, id.toString() + ".json"); File dataFile = new File(dataFolder, id + ".json");
JSONObject playerData = JsonManager.getObject(dataFile); JSONObject playerData = JsonManager.getObject(dataFile);
if (playerData == null) if (playerData == null)
return; return;
((JSONObject) playerData.get(module)).remove(key); JSONObject moduleData = ((JSONObject) playerData.get(module));
if (moduleData == null)
return;
moduleData.remove(key);
JsonManager.save(playerData, dataFile); JsonManager.save(playerData, dataFile);
} }
@@ -158,19 +197,19 @@ public final class DataManager implements CoreModule, Listener
} }
})) }))
{ {
migrate(UUID.fromString(s.replace(".json", "")), oldName, newName); migrate(s.replace(".json", ""), oldName, newName);
} }
} }
public static void migrate(UUID id, String oldName) public static void migrate(String id, String oldName)
{ {
migrate(id, oldName, Utils.getCaller(DataManager.class)); migrate(id, oldName, Utils.getCaller(DataManager.class));
} }
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
public static void migrate(UUID id, String oldName, String newName) public static void migrate(String id, String oldName, String newName)
{ {
if (data.containsKey(id.toString())) if (data.containsKey(id))
{ {
data.put(newName, data.get(oldName)); data.put(newName, data.get(oldName));
data.remove(oldName); data.remove(oldName);
@@ -181,29 +220,46 @@ public final class DataManager implements CoreModule, Listener
} }
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
private static void loadAndMigrate(UUID id, String oldName, String newName) private static void loadAndMigrate(String id, String oldName, String newName)
{ {
File dataFile = new File(dataFolder, id.toString() + ".json"); File dataFile = new File(dataFolder, id + ".json");
JSONObject data = JsonManager.getObject(dataFile); JSONObject data = JsonManager.getObject(dataFile);
data.put(newName, data.get(oldName)); data.put(newName, data.get(oldName));
data.remove(oldName); data.remove(oldName);
JsonManager.save(data, dataFile); JsonManager.save(data, dataFile);
} }
public static void save(UUID id) public static void save(CommandSender sender)
{
String id;
if (sender instanceof Player)
id = ((Player) sender).getUniqueId().toString();
else
id = "CONSOLE";
save(id);
}
public static void save(String id)
{ {
Object raw = data.get(id); Object raw = data.get(id);
if (raw == null || ((JSONObject) raw).size() == 0) if (raw == null || ((JSONObject) raw).size() == 0)
return; return;
JsonManager.save((JSONObject) raw, new File(dataFolder, id.toString() + ".json")); JsonManager.save((JSONObject) raw, new File(dataFolder, id + ".json"));
} }
private static void saveAndUnload(UUID id) private static void saveAndUnload(CommandSender sender)
{
String id;
if (sender instanceof Player)
id = ((Player) sender).getUniqueId().toString();
else
id = "CONSOLE";
saveAndUnload(id);
}
private static void saveAndUnload(String id)
{ {
String key = id.toString();
Object raw = data.containsKey(key);
if (!(raw == null || ((JSONObject) raw).size() == 0))
save(id); save(id);
data.remove(key); data.remove(id);
} }
} }