Add /ckit preview <name> (#232)

* Add /ckit preview <name>

* Optimise imports

* Implement requested changes

Author: xpple

src/main/java/net/earthcomputer/clientcommands/command/KitCommand.java

@@ -1,5 +1,6 @@
 package net.earthcomputer.clientcommands.command;
 
+import com.mojang.blaze3d.systems.RenderSystem;
 import com.mojang.brigadier.CommandDispatcher;
 import com.mojang.brigadier.arguments.StringArgumentType;
 import com.mojang.brigadier.exceptions.CommandSyntaxException;
@@ -12,14 +13,19 @@
 import net.fabricmc.loader.api.FabricLoader;
 import net.minecraft.SharedConstants;
 import net.minecraft.client.MinecraftClient;
+import net.minecraft.client.gui.screen.ingame.AbstractInventoryScreen;
+import net.minecraft.client.util.math.MatrixStack;
 import net.minecraft.command.CommandSource;
 import net.minecraft.datafixer.TypeReferences;
 import net.minecraft.entity.player.PlayerInventory;
 import net.minecraft.item.ItemStack;
 import net.minecraft.nbt.*;
+import net.minecraft.screen.PlayerScreenHandler;
 import net.minecraft.screen.slot.Slot;
 import net.minecraft.server.command.ServerCommandSource;
+import net.minecraft.text.LiteralText;
 import net.minecraft.text.TranslatableText;
+import net.minecraft.util.Formatting;
 import net.minecraft.util.Util;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
@@ -36,7 +42,7 @@
 
 public class KitCommand {
 
-    private static final Logger logger = LogManager.getLogger("clientcommands");
+    private static final Logger LOGGER = LogManager.getLogger("clientcommands");
 
     private static final SimpleCommandExceptionType SAVE_FAILED_EXCEPTION = new SimpleCommandExceptionType(new TranslatableText("commands.ckit.saveFile.failed"));
 
@@ -55,7 +61,7 @@ public class KitCommand {
         try {
             loadFile();
         } catch (IOException e) {
-            logger.info("Could not load kits file, hence /ckit will not work!");
+            LOGGER.info("Could not load kits file, hence /ckit will not work!");
         }
     }
 
@@ -82,7 +88,11 @@ public static void register(CommandDispatcher<ServerCommandSource> dispatcher) {
                                         .executes(ctx -> load(ctx.getSource(), StringArgumentType.getString(ctx, "name"), true)))
                                 .executes(ctx -> load(ctx.getSource(), StringArgumentType.getString(ctx, "name"), false))))
                 .then(literal("list")
-                        .executes(ctx -> list(ctx.getSource()))));
+                        .executes(ctx -> list(ctx.getSource())))
+                .then(literal("preview")
+                        .then(argument("name", StringArgumentType.string())
+                                .suggests((ctx, builder) -> CommandSource.suggestMatching(kits.keySet(), builder))
+                                .executes(ctx -> preview(ctx.getSource(), StringArgumentType.getString(ctx, "name"))))));
     }
 
     private static int create(ServerCommandSource source, String name) throws CommandSyntaxException {
@@ -151,6 +161,24 @@ private static int list(ServerCommandSource source) {
         return kits.size();
     }
 
+    private static int preview(ServerCommandSource source, String name) throws CommandSyntaxException {
+        ListTag kit = kits.get(name);
+        if (kit == null) {
+            throw NOT_FOUND_EXCEPTION.create(name);
+        }
+
+        PlayerInventory tempInv = new PlayerInventory(client.player);
+        tempInv.deserialize(kit);
+        /*
+            After executing a command, the current screen will be closed (the chat hud).
+            And if you open a new screen in a command, that new screen will be closed
+            instantly along with the chat hud. Slightly delaying the opening of the
+            screen fixes this issue.
+         */
+        client.send(() -> client.openScreen(new PreviewScreen(new PlayerScreenHandler(tempInv, true, client.player), tempInv, name)));
+        return 0;
+    }
+
     private static void saveFile() throws CommandSyntaxException {
         try {
             CompoundTag rootTag = new CompoundTag();
@@ -192,3 +220,30 @@ private static void loadFile() throws IOException {
         }
     }
 }
+
+class PreviewScreen extends AbstractInventoryScreen<PlayerScreenHandler> {
+
+    public PreviewScreen(PlayerScreenHandler playerScreenHandler, PlayerInventory inventory, String name) {
+        super(playerScreenHandler, inventory, new LiteralText(name).styled(style -> style.withColor(Formatting.RED)));
+        this.passEvents = true;
+        this.titleX = 80;
+    }
+
+    protected void drawForeground(MatrixStack matrices, int mouseX, int mouseY) {
+        this.textRenderer.draw(matrices, this.title, (float) this.titleX, (float) this.titleY, 0x404040);
+    }
+
+    public void render(MatrixStack matrices, int mouseX, int mouseY, float delta) {
+        this.drawStatusEffects = false;
+        this.renderBackground(matrices);
+        super.render(matrices, mouseX, mouseY, delta);
+
+        this.drawMouseoverTooltip(matrices, mouseX, mouseY);
+    }
+
+    protected void drawBackground(MatrixStack matrices, float delta, int mouseX, int mouseY) {
+        RenderSystem.color4f(1.0F, 1.0F, 1.0F, 1.0F);
+        this.client.getTextureManager().bindTexture(BACKGROUND_TEXTURE);
+        this.drawTexture(matrices, this.x, this.y, 0, 0, this.backgroundWidth, this.backgroundHeight);
+    }
+}