Initial code dump.

1 files changed, 99 insertions, 0 deletions

diff --git a/src/main/java/org/chrisoft/trashyaddon/commands/MapTallyCommand.java b/src/main/java/org/chrisoft/trashyaddon/commands/MapTallyCommand.java
new file mode 100644
index 0000000..cce0687
--- /dev/null
+++ b/src/main/java/org/chrisoft/trashyaddon/commands/MapTallyCommand.java
@@ -0,0 +1,99 @@
+package org.chrisoft.trashyaddon.commands;
+
+import com.mojang.brigadier.builder.LiteralArgumentBuilder;
+import meteordevelopment.meteorclient.MeteorClient;
+import meteordevelopment.meteorclient.commands.Command;
+import meteordevelopment.meteorclient.utils.player.ChatUtils;
+import meteordevelopment.orbit.EventPriority;
+import meteordevelopment.orbit.listeners.IListener;
+import meteordevelopment.meteorclient.events.packets.InventoryEvent;
+import net.minecraft.command.CommandSource;
+import net.minecraft.client.MinecraftClient;
+import net.minecraft.item.FilledMapItem;
+import net.minecraft.item.ItemStack;
+import net.minecraft.network.packet.s2c.play.InventoryS2CPacket;
+import net.minecraft.util.Util;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.nio.ByteBuffer;
+import java.nio.ByteOrder;
+import java.util.HashSet;
+import java.util.List;
+import java.util.zip.GZIPOutputStream;
+
+import static com.mojang.brigadier.Command.SINGLE_SUCCESS;
+
+public class MapTallyCommand extends Command {
+    IListener invListener;
+    HashSet<Integer> ids = new HashSet<>();
+    private final MinecraftClient mc = MinecraftClient.getInstance();
+    public MapTallyCommand() {
+        super("mt", "Tally maps inside containers opened after execution of this command.");
+        invListener = null;
+    }
+
+    @Override
+    public void build(LiteralArgumentBuilder<CommandSource> builder) {
+        builder.then(literal("stop").executes(context -> {
+            if (invListener != null) {
+                info(ids.size() + " distinct map(s) total found.");
+                try {
+                    ByteBuffer buf = ByteBuffer.allocate(4 * ids.size());
+                    buf.order(ByteOrder.LITTLE_ENDIAN);
+                    for (int id : ids)
+                        buf.putInt(id);
+                    File dirpath = new File(MeteorClient.FOLDER, "mapman");
+                    if (!dirpath.isDirectory())
+                        dirpath.mkdir();
+                    File dumppath = new File(dirpath, "mt-" + Util.getFormattedCurrentTime() + ".gz");
+                    FileOutputStream fo = new FileOutputStream(dumppath);
+                    GZIPOutputStream output = new GZIPOutputStream(fo);
+                    output.write(buf.array());
+                    output.close();
+                    fo.close();
+                    info("Tally saved to " + dumppath.toString());
+                }
+                catch (Exception e) {
+                    error("Cannot write output:" + e.getMessage());
+                    return 0;
+                }
+                ids.clear();
+                MeteorClient.EVENT_BUS.unsubscribe(invListener);
+                invListener = null;
+            } else error("Map tally not running.");
+            return SINGLE_SUCCESS;
+        }));
+        builder.executes(context -> {
+            if (invListener == null) {
+                ids.clear();
+                invListener = new IListener() {
+                    @Override
+                    public void call(Object target) {
+                        InventoryEvent e = (InventoryEvent)target;
+                        List<ItemStack> stacks = e.packet.getContents();
+                        if (stacks.size() == 46 || stacks.size() <= 36) {
+                            // player inventory / 3x3 crafting table or invalid
+                            return;
+                        }
+                        int nmaps = 0;
+                        for (ItemStack s : stacks.subList(0, stacks.size() - 36)) {
+                            if (s.getItem() instanceof FilledMapItem) {
+                                ids.add(FilledMapItem.getMapId(s));
+                                ++nmaps;
+                            }
+                        }
+                        info("Found " + nmaps + " map(s) in that container.");
+                    }
+                    @Override public Class<?> getTarget() { return InventoryEvent.class; }
+                    @Override public int getPriority() { return EventPriority.MEDIUM; }
+                    @Override public boolean isStatic() { return false; }
+                };
+                MeteorClient.EVENT_BUS.subscribe(invListener);
+                info("Open the containers that contain maps you wish to tally...");
+            }
+            else error("Already tallying!");
+            return SINGLE_SUCCESS;
+        });
+    }
+}