From 4bdbe0efee1bc349d320d9009bd51dcbe42360f6 Mon Sep 17 00:00:00 2001
From: luiz-otavio <luizfarias2006@hotmail.com>
Date: Sun, 27 Feb 2022 06:57:37 -0300
Subject: [PATCH] refactor: ensure almost compression types to load/write each
 `NBTTagCompound`

---
 .../controllers/PlayerDataController.java     | 106 ++++++++++++++++--
 1 file changed, 94 insertions(+), 12 deletions(-)

diff --git a/src/main/java/noppes/npcs/controllers/PlayerDataController.java b/src/main/java/noppes/npcs/controllers/PlayerDataController.java
index 4c72212d7..c50a9ff44 100644
--- a/src/main/java/noppes/npcs/controllers/PlayerDataController.java
+++ b/src/main/java/noppes/npcs/controllers/PlayerDataController.java
@@ -1,21 +1,24 @@
 package noppes.npcs.controllers;
 
-import java.io.EOFException;
-import java.io.File;
-import java.io.FileInputStream;
-import java.io.FileOutputStream;
-import java.io.IOException;
+import java.io.*;
+import java.nio.file.Files;
+import java.nio.file.StandardOpenOption;
 import java.util.HashMap;
 import java.util.Map;
 
+import me.luizotavio.compressor.ZstdCompressor;
 import net.minecraft.entity.player.EntityPlayer;
 import net.minecraft.nbt.CompressedStreamTools;
+import net.minecraft.nbt.NBTSizeTracker;
 import net.minecraft.nbt.NBTTagCompound;
 import net.minecraft.server.MinecraftServer;
 import noppes.npcs.CustomNpcs;
 import noppes.npcs.LogWriter;
+import noppes.npcs.util.JsonException;
 import noppes.npcs.util.NBTJsonUtil;
 
+import static me.luizotavio.compressor.executor.IOExecutor.IO_EXECUTOR;
+
 public class PlayerDataController {		
 	public static PlayerDataController instance;
 	
@@ -74,10 +77,9 @@ public class PlayerDataController {
 			filename = "noplayername";
 		filename += ".json";
 		try {
-	        File file = new File(saveDir, filename);
-	        if(file.exists()){
-		        return NBTJsonUtil.LoadFile(file);
-	        }
+			return ensureCompressor(
+				new File(saveDir, filename)
+			);
 		} catch (Exception e) {
 			LogWriter.error("Error loading: " + filename, e);
 		}
@@ -87,12 +89,23 @@ public class PlayerDataController {
 	
 	public void savePlayerData(PlayerData data){
 		NBTTagCompound compound = data.getNBT();
-		String filename = data.uuid + ".json";
+		String filename;
+
+		if (CustomNpcs.CompressorType.equalsIgnoreCase("lz4")) {
+			filename = data.uuid + ".lz4";
+		} else if (CustomNpcs.CompressorType.equalsIgnoreCase("json")) {
+			filename = data.uuid + ".json";
+		} else {
+			filename = data.uuid + ".dat";
+		}
+
 		try {
 			File saveDir = getSaveDir();
             File file = new File(saveDir, filename+"_new");
             File file1 = new File(saveDir, filename);
-            NBTJsonUtil.SaveFile(file, compound);
+
+			writeCompressor(file, compound);
+
             if(file1.exists()){
                 file1.delete();
             }
@@ -164,7 +177,8 @@ public class PlayerDataController {
         	if(file.isDirectory() || !file.getName().endsWith(".json"))
         		continue;
 			try {
-				NBTTagCompound compound = NBTJsonUtil.LoadFile(file);
+				NBTTagCompound compound = ensureCompressor(file);
+
 	        	if(compound.hasKey("PlayerName")){
 	        		map.put(compound.getString("PlayerName"), compound);
 	        	}
@@ -178,4 +192,72 @@ public class PlayerDataController {
 	public boolean hasMail(EntityPlayer player) {
 		return getPlayerData(player).mailData.hasMail();
 	}
+
+	private void writeCompressor(File file, NBTTagCompound compound) {
+		if (compound == null) {
+			throw new IllegalArgumentException("Compound cannot be null!");
+		}
+
+		if (CustomNpcs.CompressorType.equalsIgnoreCase("json")) {
+			IO_EXECUTOR.submit(() -> {
+				try {
+					NBTJsonUtil.SaveFile(file, compound);
+				} catch (IOException | JsonException e) {
+					e.printStackTrace();
+				}
+			});
+		} else if (CustomNpcs.CompressorType.equalsIgnoreCase("zstd")) {
+			IO_EXECUTOR.submit(() -> {
+				try {
+					byte[] compressed = ZstdCompressor.writeCompound(compound);
+
+					Files.write(
+						file.toPath(),
+						compressed,
+						StandardOpenOption.CREATE,
+						StandardOpenOption.TRUNCATE_EXISTING,
+						StandardOpenOption.WRITE
+					);
+				} catch (Exception e) {
+					e.printStackTrace();
+				}
+			});
+		}
+	}
+
+	private NBTTagCompound ensureCompressor(File file) throws JsonException, IOException {
+		if (!file.exists()) {
+			return new NBTTagCompound();
+		}
+
+		if (CustomNpcs.CompressorType.equalsIgnoreCase("json")) {
+			if (file.getName().endsWith(".zstd")) {
+				byte[] readCompressed = ZstdCompressor.readCompressed(
+					new RandomAccessFile(file, "r")
+				);
+
+				try (DataInputStream dataInputStream = new DataInputStream(new ByteArrayInputStream(readCompressed))) {
+					return CompressedStreamTools.func_152456_a(dataInputStream, NBTSizeTracker.field_152451_a);
+				}
+			}
+
+			return NBTJsonUtil.LoadFile(file);
+		}
+
+		if (CustomNpcs.CompressorType.equalsIgnoreCase("lz4")) {
+			if (file.getName().endsWith(".json")) {
+				return NBTJsonUtil.LoadFile(file);
+			}
+
+			byte[] readCompressed = ZstdCompressor.readCompressed(
+				new RandomAccessFile(file, "r")
+			);
+
+			try (DataInputStream dataInputStream = new DataInputStream(new ByteArrayInputStream(readCompressed))) {
+				return CompressedStreamTools.func_152456_a(dataInputStream, NBTSizeTracker.field_152451_a);
+			}
+		}
+
+		throw new IOException("Unknown compressor type: " + CustomNpcs.CompressorType);
+	}
 }