From efe6eec885ed7fe0a2dc92a9360365215983ded0 Mon Sep 17 00:00:00 2001
From: GeorgeRNG <81434111+GeorgeRNG@users.noreply.github.com>
Date: Sun, 31 Mar 2024 13:33:30 +0100
Subject: [PATCH] It misses args cuz my class doesn't have it, so 100% working
 now it seems

---
 .../dev/dfonline/codeclient/Commands.java     | 19 ++++++++--
 .../codeclient/action/impl/ScanPlot.java      | 36 ++++++++++---------
 .../hypercube/template/Template.java          | 11 ------
 3 files changed, 35 insertions(+), 31 deletions(-)

diff --git a/src/main/java/dev/dfonline/codeclient/Commands.java b/src/main/java/dev/dfonline/codeclient/Commands.java
index 979fdb41..2ea23737 100644
--- a/src/main/java/dev/dfonline/codeclient/Commands.java
+++ b/src/main/java/dev/dfonline/codeclient/Commands.java
@@ -218,18 +218,31 @@ else if(!Files.isDirectory(currentPath)) {
                         }
                     }
 
+                    try {
+                        if(Files.list(currentPath).findFirst().isPresent()) {
+                            Utility.sendMessage("Please use an empty directory to save into.",ChatType.FAIL);
+                            return -1;
+                        }
+                    }
+                    catch (Exception ignored) {
+                        Utility.sendMessage("An error occurred when reading the directory.",ChatType.FAIL);
+                    }
+
                     Utility.sendMessage("Scanning plot. Use /abort to abort.",ChatType.INFO);
-                    var scan = new ArrayList<Template>();
+                    var scan = new ArrayList<ItemStack>();
                     Path finalCurrentPath = currentPath;
                     CodeClient.currentAction = new ScanPlot(() -> {
                         CodeClient.currentAction = new None();
                         Utility.sendMessage("Done!", ChatType.SUCCESS);
-                        for (Template template : scan) {
+                        for (ItemStack item : scan) {
+                            String data = Utility.templateDataItem(item);
+                            var template = Template.parse64(data);
+                            if(template == null) continue;
                             var first = template.blocks.get(0);
                             String name = Objects.requireNonNullElse(first.action != null ? first.action : first.data,"unknown");
                             var filePath = finalCurrentPath.resolve(name + ".dft");
                             try {
-                                Files.write(filePath,template.compress());
+                                Files.write(filePath,Base64.getDecoder().decode(data));
                             }
                             catch (Exception ignored) {
                                 Utility.sendMessage("Couldn't save " + filePath + "...", ChatType.FAIL);
diff --git a/src/main/java/dev/dfonline/codeclient/action/impl/ScanPlot.java b/src/main/java/dev/dfonline/codeclient/action/impl/ScanPlot.java
index b44f3b06..3a72070b 100644
--- a/src/main/java/dev/dfonline/codeclient/action/impl/ScanPlot.java
+++ b/src/main/java/dev/dfonline/codeclient/action/impl/ScanPlot.java
@@ -26,10 +26,9 @@ public class ScanPlot extends Action {
     private List<BlockPos> blocks = null;
     private Action step = null;
     private static final Vec3d goToOffset = new Vec3d(0, 1.5, 0);
-    private Integer progress = 0;
-    private final ArrayList<Template> scanList;
+    private final ArrayList<ItemStack> scanList;
 
-    public ScanPlot(Callback callback, ArrayList<Template> scanList) {
+    public ScanPlot(Callback callback, ArrayList<ItemStack> scanList) {
         super(callback);
         this.scanList = scanList;
         if (!(CodeClient.location instanceof Dev)) {
@@ -61,23 +60,26 @@ public boolean onReceivePacket(Packet<?> packet) {
 //        }
     }
 
+    private void next(int progress) {
+        if(progress >= blocks.size()) {
+            callback();
+            return;
+        }
+        step = new GoTo(blocks.get(progress).toCenterPos().add(goToOffset), () -> {
+            this.step = new pickUpBlock(blocks.get(progress),() -> {
+                CodeClient.LOGGER.info("picked up block");
+                next(progress + 1);
+            });
+            this.step.init();
+        });
+        step.init();
+    }
+
     @Override
     public void onTick() {
         if (step != null) step.onTick();
         else {
-            if(progress >= blocks.size()) {
-                callback();
-                return;
-            }
-            step = new GoTo(blocks.get(progress).toCenterPos().add(goToOffset), () -> {
-                this.step = new pickUpBlock(blocks.get(progress),() -> {
-                    CodeClient.LOGGER.info("picked up block");
-                    step = null;
-                    progress+=1;
-                });
-                this.step.init();
-            });
-            step.init();
+            next(0);
         }
     }
 
@@ -109,7 +111,7 @@ public boolean onReceivePacket(Packet<?> packet) {
                 var data = Utility.templateDataItem(slot.getStack());
                 var template = Template.parse64(data);
                 if (template == null) return false;
-                scanList.add(Template.parse64(data));
+                scanList.add(slot.getStack());
                 net.sendPacket(new CreativeInventoryActionC2SPacket(slot.getSlot(), ItemStack.EMPTY));
                 this.callback();
                 return true;
diff --git a/src/main/java/dev/dfonline/codeclient/hypercube/template/Template.java b/src/main/java/dev/dfonline/codeclient/hypercube/template/Template.java
index 30b77e5e..663b6704 100644
--- a/src/main/java/dev/dfonline/codeclient/hypercube/template/Template.java
+++ b/src/main/java/dev/dfonline/codeclient/hypercube/template/Template.java
@@ -19,10 +19,6 @@ public int getLength() {
         return length;
     }
 
-    public byte[] compress() throws IOException {
-        return compress(CodeClient.gson.toJson(this));
-    }
-
     /**
      * Parse base64+gzip data
      */
@@ -62,11 +58,4 @@ private static byte[] decompress(byte[] compressedData) throws IOException {
             return bos.toByteArray();
         }
     }
-    private static byte[] compress(String uncompressedString) throws IOException {
-        ByteArrayOutputStream byteStream = new ByteArrayOutputStream();
-        try (GZIPOutputStream gzipStream = new GZIPOutputStream(byteStream)) {
-            gzipStream.write(uncompressedString.getBytes());
-        }
-        return byteStream.toByteArray();
-    }
 }