331 lines
12 KiB
Java
331 lines
12 KiB
Java
package com.pahimar.ee3.util;
|
|
|
|
import com.google.common.io.Files;
|
|
import com.google.gson.stream.JsonReader;
|
|
import com.google.gson.stream.JsonWriter;
|
|
import com.pahimar.ee3.api.exchange.EnergyValue;
|
|
import com.pahimar.ee3.exchange.EnergyValueRegistry;
|
|
import com.pahimar.ee3.exchange.EnergyValueStackMapping;
|
|
import com.pahimar.ee3.exchange.WrappedStack;
|
|
import com.pahimar.ee3.knowledge.TransmutationKnowledge;
|
|
import com.pahimar.ee3.reference.Reference;
|
|
import cpw.mods.fml.common.FMLCommonHandler;
|
|
import cpw.mods.fml.common.Loader;
|
|
import cpw.mods.fml.common.ModContainer;
|
|
import net.minecraft.nbt.CompressedStreamTools;
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import org.apache.commons.codec.digest.DigestUtils;
|
|
|
|
import java.io.*;
|
|
import java.util.*;
|
|
|
|
public class SerializationHelper
|
|
{
|
|
private static File instanceDataDirectory;
|
|
private static File instancePlayerDataDirectory;
|
|
|
|
/**
|
|
* Returns a File reference to the mod specific directory in the data directory
|
|
*
|
|
* @return
|
|
*/
|
|
public static File getInstanceDataDirectory()
|
|
{
|
|
return instanceDataDirectory;
|
|
}
|
|
|
|
/**
|
|
* Returns a File reference to the mod specific directory in the playerdata directory
|
|
*
|
|
* @return
|
|
*/
|
|
public static File getInstancePlayerDataDirectory()
|
|
{
|
|
return instancePlayerDataDirectory;
|
|
}
|
|
|
|
/**
|
|
* Creates (if one does not exist already) and initializes a mod specific File reference inside of the current world's playerdata directory
|
|
*/
|
|
public static void initModDataDirectories()
|
|
{
|
|
instanceDataDirectory = new File(FMLCommonHandler.instance().getMinecraftServerInstance().getEntityWorld().getSaveHandler().getWorldDirectory(), "data" + File.separator + Reference.LOWERCASE_MOD_ID);
|
|
instanceDataDirectory.mkdirs();
|
|
|
|
instancePlayerDataDirectory = new File(FMLCommonHandler.instance().getMinecraftServerInstance().getEntityWorld().getSaveHandler().getWorldDirectory(), "playerdata" + File.separator + Reference.LOWERCASE_MOD_ID);
|
|
instancePlayerDataDirectory.mkdirs();
|
|
}
|
|
|
|
public static String getModListMD5()
|
|
{
|
|
List<String> modList = new ArrayList<String>();
|
|
|
|
for (ModContainer modContainer : Loader.instance().getModList())
|
|
{
|
|
modList.add("[" + modContainer.getModId() + "-" + modContainer.getName() + "-" + modContainer.getVersion() + "]");
|
|
}
|
|
|
|
Collections.sort(modList);
|
|
|
|
StringBuilder modListString = new StringBuilder();
|
|
for (String modEntry : modList)
|
|
{
|
|
modListString.append(modEntry);
|
|
}
|
|
|
|
return DigestUtils.md5Hex(modListString.toString());
|
|
}
|
|
|
|
public static NBTTagCompound readNBTFromFile(File nbtEncodedFile)
|
|
{
|
|
if (nbtEncodedFile.exists() && nbtEncodedFile.isFile())
|
|
{
|
|
try
|
|
{
|
|
return CompressedStreamTools.readCompressed(new FileInputStream(nbtEncodedFile));
|
|
}
|
|
catch (IOException e)
|
|
{
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public static void writeNBTToFile(File directory, String fileName, INBTTaggable nbtTaggable)
|
|
{
|
|
writeNBTToFile(directory, fileName, nbtTaggable, false);
|
|
}
|
|
|
|
public static void writeNBTToFile(File directory, String fileName, INBTTaggable nbtTaggable, boolean verboseLogging)
|
|
{
|
|
if (directory != null && fileName != null && nbtTaggable != null)
|
|
{
|
|
if (!directory.exists())
|
|
{
|
|
directory.mkdirs();
|
|
}
|
|
|
|
NBTTagCompound nbtTagCompound = new NBTTagCompound();
|
|
nbtTaggable.writeToNBT(nbtTagCompound);
|
|
|
|
try
|
|
{
|
|
File file1 = new File(directory, fileName + ".tmp");
|
|
File file2 = new File(directory, fileName);
|
|
CompressedStreamTools.writeCompressed(nbtTagCompound, new FileOutputStream(file1));
|
|
|
|
if (file2.exists())
|
|
{
|
|
file2.delete();
|
|
}
|
|
|
|
file1.renameTo(file2);
|
|
|
|
if (verboseLogging)
|
|
{
|
|
LogHelper.info("Successfully saved {} to file: {}", nbtTaggable.getTagLabel(), file2.getAbsolutePath());
|
|
}
|
|
}
|
|
catch (Exception exception)
|
|
{
|
|
LogHelper.warn("Failed to save {} to file: {}{}{}", nbtTaggable.getTagLabel(), directory.getAbsolutePath(), File.separator, fileName);
|
|
exception.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
|
|
public static TransmutationKnowledge readTransmutationKnowledgeFromFile(File directory, String fileName)
|
|
{
|
|
if (!directory.exists())
|
|
{
|
|
directory.mkdirs();
|
|
}
|
|
|
|
return TransmutationKnowledge.readFromFile(new File(directory, fileName));
|
|
}
|
|
|
|
public static void writeTransmutationKnowledgeToFile(File directory, String fileName, TransmutationKnowledge transmutationKnowledge)
|
|
{
|
|
writeTransmutationKnowledgeToFile(directory, fileName, transmutationKnowledge, false);
|
|
}
|
|
|
|
public static void writeTransmutationKnowledgeToFile(File directory, String fileName, TransmutationKnowledge transmutationKnowledge, boolean verboseLogging)
|
|
{
|
|
if (directory != null && fileName != null)
|
|
{
|
|
if (!directory.exists())
|
|
{
|
|
directory.mkdirs();
|
|
}
|
|
|
|
if (transmutationKnowledge == null)
|
|
{
|
|
transmutationKnowledge = new TransmutationKnowledge();
|
|
}
|
|
|
|
try
|
|
{
|
|
File file1 = new File(directory, fileName + ".tmp");
|
|
File file2 = new File(directory, fileName);
|
|
TransmutationKnowledge.writeToFile(file1, transmutationKnowledge);
|
|
|
|
if (file2.exists())
|
|
{
|
|
file2.delete();
|
|
}
|
|
|
|
file1.renameTo(file2);
|
|
|
|
if (verboseLogging)
|
|
{
|
|
LogHelper.info("Successfully saved TransmutationKnowledge to file: {}", file2.getAbsolutePath());
|
|
}
|
|
}
|
|
catch (Exception exception)
|
|
{
|
|
exception.printStackTrace();
|
|
LogHelper.error("Failed to save TransmutationKnowledge to file: {}{}", directory.getAbsolutePath(), fileName);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static Map<WrappedStack, EnergyValue> readEnergyValueStackMapFromJsonFile(String fileName)
|
|
{
|
|
File energyValuesDataDirectory = new File(FMLCommonHandler.instance().getMinecraftServerInstance().getEntityWorld().getSaveHandler().getWorldDirectory(), "data" + File.separator + Reference.LOWERCASE_MOD_ID + File.separator + "energyvalues");
|
|
return readEnergyValueStackMapFromJsonFile(new File(energyValuesDataDirectory, fileName));
|
|
}
|
|
|
|
public static Map<WrappedStack, EnergyValue> readEnergyValueStackMapFromJsonFile(File jsonFile)
|
|
{
|
|
Map<WrappedStack, EnergyValue> energyValueStackMap = new TreeMap<WrappedStack, EnergyValue>();
|
|
JsonReader jsonReader;
|
|
|
|
try
|
|
{
|
|
jsonReader = new JsonReader(new FileReader(jsonFile));
|
|
jsonReader.beginArray();
|
|
while (jsonReader.hasNext())
|
|
{
|
|
EnergyValueStackMapping energyValueStackMapping = EnergyValueStackMapping.jsonSerializer.fromJson(jsonReader, EnergyValueStackMapping.class);
|
|
if (energyValueStackMapping != null)
|
|
{
|
|
energyValueStackMap.put(energyValueStackMapping.wrappedStack, energyValueStackMapping.energyValue);
|
|
}
|
|
}
|
|
jsonReader.endArray();
|
|
jsonReader.close();
|
|
}
|
|
catch (FileNotFoundException ignored)
|
|
{
|
|
}
|
|
catch (IOException e)
|
|
{
|
|
e.printStackTrace();
|
|
}
|
|
|
|
return energyValueStackMap;
|
|
}
|
|
|
|
public static void writeEnergyValueStackMapToJsonFile(String fileName, Map<WrappedStack, EnergyValue> energyValueMap)
|
|
{
|
|
File energyValuesDataDirectory = new File(FMLCommonHandler.instance().getMinecraftServerInstance().getEntityWorld().getSaveHandler().getWorldDirectory(), "data" + File.separator + Reference.LOWERCASE_MOD_ID + File.separator + "energyvalues");
|
|
writeEnergyValueStackMapToJsonFile(new File(energyValuesDataDirectory, fileName), energyValueMap);
|
|
}
|
|
|
|
public static void writeEnergyValueStackMapToJsonFile(File jsonFile, Map<WrappedStack, EnergyValue> energyValueMap)
|
|
{
|
|
JsonWriter jsonWriter;
|
|
|
|
try
|
|
{
|
|
jsonWriter = new JsonWriter(new FileWriter(jsonFile));
|
|
jsonWriter.setIndent(" ");
|
|
jsonWriter.beginArray();
|
|
for (WrappedStack wrappedStack : energyValueMap.keySet())
|
|
{
|
|
if (wrappedStack != null && wrappedStack.getWrappedObject() != null)
|
|
{
|
|
EnergyValueStackMapping.jsonSerializer.toJson(new EnergyValueStackMapping(wrappedStack, energyValueMap.get(wrappedStack)), EnergyValueStackMapping.class, jsonWriter);
|
|
}
|
|
}
|
|
|
|
jsonWriter.endArray();
|
|
jsonWriter.close();
|
|
}
|
|
catch (IOException e)
|
|
{
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static void compressEnergyValueStackMapToFile(String fileName, Map<WrappedStack, EnergyValue> energyValueMap)
|
|
{
|
|
File energyValuesDataDirectory = new File(FMLCommonHandler.instance().getMinecraftServerInstance().getEntityWorld().getSaveHandler().getWorldDirectory(), "data" + File.separator + Reference.LOWERCASE_MOD_ID + File.separator + "energyvalues");
|
|
compressEnergyValueStackMapToFile(new File(energyValuesDataDirectory, fileName), energyValueMap);
|
|
}
|
|
|
|
public static void compressEnergyValueStackMapToFile(File file, Map<WrappedStack, EnergyValue> energyValueMap)
|
|
{
|
|
try
|
|
{
|
|
byte[] energyValueRegistryArray = CompressionHelper.compressStringToByteArray(EnergyValueRegistry.getInstance().toJson());
|
|
FileOutputStream fos = new FileOutputStream(file);
|
|
fos.write(energyValueRegistryArray);
|
|
fos.close();
|
|
}
|
|
catch (UnsupportedEncodingException e)
|
|
{
|
|
e.printStackTrace();
|
|
}
|
|
catch (FileNotFoundException e)
|
|
{
|
|
e.printStackTrace();
|
|
}
|
|
catch (IOException e)
|
|
{
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static Map<WrappedStack, EnergyValue> decompressEnergyValueStackMapFromFile(String fileName)
|
|
{
|
|
File energyValuesDataDirectory = new File(FMLCommonHandler.instance().getMinecraftServerInstance().getEntityWorld().getSaveHandler().getWorldDirectory(), "data" + File.separator + Reference.LOWERCASE_MOD_ID + File.separator + "energyvalues");
|
|
return decompressEnergyValueStackMapFromFile(new File(energyValuesDataDirectory, fileName));
|
|
}
|
|
|
|
public static Map<WrappedStack, EnergyValue> decompressEnergyValueStackMapFromFile(File file)
|
|
{
|
|
Map<WrappedStack, EnergyValue> energyValueStackMap = new TreeMap<WrappedStack, EnergyValue>();
|
|
|
|
try
|
|
{
|
|
String jsonEnergyValueStackMap = CompressionHelper.decompressStringFromByteArray(Files.toByteArray(file));
|
|
JsonReader jsonReader = new JsonReader(new StringReader(jsonEnergyValueStackMap));
|
|
jsonReader.beginArray();
|
|
while (jsonReader.hasNext())
|
|
{
|
|
EnergyValueStackMapping energyValueStackMapping = EnergyValueStackMapping.jsonSerializer.fromJson(jsonReader, EnergyValueStackMapping.class);
|
|
if (energyValueStackMapping != null)
|
|
{
|
|
energyValueStackMap.put(energyValueStackMapping.wrappedStack, energyValueStackMapping.energyValue);
|
|
}
|
|
}
|
|
jsonReader.endArray();
|
|
jsonReader.close();
|
|
}
|
|
catch (FileNotFoundException e)
|
|
{
|
|
e.printStackTrace();
|
|
}
|
|
catch (IOException e)
|
|
{
|
|
e.printStackTrace();
|
|
}
|
|
|
|
|
|
return energyValueStackMap;
|
|
}
|
|
}
|