show in slot setting in MapPreview

Author: beanbag44

src/main/java/com/lambda/mixin/render/DrawContextMixin.java

@@ -18,25 +18,73 @@
 package com.lambda.mixin.render;
 
 import com.lambda.module.modules.render.ContainerPreview;
+import com.lambda.module.modules.render.MapPreview;
+import net.minecraft.client.MinecraftClient;
 import net.minecraft.client.font.TextRenderer;
 import net.minecraft.client.gui.DrawContext;
-import net.minecraft.client.gui.tooltip.TooltipComponent;
-import net.minecraft.client.gui.tooltip.TooltipPositioner;
+import net.minecraft.client.render.MapRenderState;
+import net.minecraft.component.DataComponentTypes;
+import net.minecraft.item.FilledMapItem;
 import net.minecraft.item.ItemStack;
+import net.minecraft.item.Items;
 import net.minecraft.item.tooltip.TooltipData;
 import net.minecraft.text.Text;
 import net.minecraft.util.Identifier;
 import org.jetbrains.annotations.Nullable;
+import org.joml.Matrix3x2fStack;
+import org.spongepowered.asm.mixin.Final;
 import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.Unique;
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
 
 import java.util.List;
 import java.util.Optional;
 
+/*
+Map slot rendering code
+Original source: https://github.com/Crec0/map-in-slot
+Copyright (c) 2022 Crec0
+Licensed under MIT License
+ */
 @Mixin(DrawContext.class)
-public class DrawContextMixin {
+public abstract class DrawContextMixin {
+    @Shadow
+    @Final
+    MinecraftClient client;
+
+    @Unique
+    private final MapRenderState mapRenderState = new MapRenderState();
+
+    @Shadow
+    public abstract Matrix3x2fStack getMatrices();
+
+    @Shadow
+    public abstract void drawMap(MapRenderState mapRenderState);
+
+    @Inject(method = "drawStackOverlay(Lnet/minecraft/client/font/TextRenderer;Lnet/minecraft/item/ItemStack;IILjava/lang/String;)V", at = @At(value = "TAIL"))
+    private void injectDrawMap(TextRenderer textRenderer, ItemStack stack, int i, int j, String string, CallbackInfo ci) {
+        if (MapPreview.INSTANCE.isDisabled() || !MapPreview.getShowInSlot()) return;
+
+        if (!stack.isOf(Items.FILLED_MAP)) return;
+
+        var mapId = stack.get(DataComponentTypes.MAP_ID);
+        var savedData = FilledMapItem.getMapState(mapId, client.world);
+
+        if (savedData == null) return;
+
+        this.getMatrices().pushMatrix();
+        this.getMatrices().translate(i, j);
+        this.getMatrices().scale(0.125F, 0.125F);
+
+        client.getMapRenderer().update(mapId, savedData, this.mapRenderState);
+        this.drawMap(this.mapRenderState);
+
+        this.getMatrices().popMatrix();
+    }
+
     @Inject(method = "drawTooltip(Lnet/minecraft/client/font/TextRenderer;Ljava/util/List;Ljava/util/Optional;IILnet/minecraft/util/Identifier;)V", at = @At("HEAD"), cancellable = true)
     private void onDrawTooltip(TextRenderer textRenderer, List<Text> text, Optional<TooltipData> data, int x, int y, @Nullable Identifier texture, CallbackInfo ci) {
         if (!ContainerPreview.INSTANCE.isEnabled()) return;

src/main/kotlin/com/lambda/module/modules/render/MapPreview.kt

@@ -21,10 +21,10 @@ import com.lambda.Lambda.mc
 import com.lambda.module.Module
 import com.lambda.module.tag.ModuleTag
 import net.minecraft.client.font.TextRenderer
+import net.minecraft.client.gl.RenderPipelines
 import net.minecraft.client.gui.DrawContext
 import net.minecraft.client.gui.tooltip.TooltipComponent
 import net.minecraft.client.render.MapRenderState
-import net.minecraft.client.gl.RenderPipelines
 import net.minecraft.component.DataComponentTypes
 import net.minecraft.component.type.MapIdComponent
 import net.minecraft.item.FilledMapItem
@@ -38,6 +38,8 @@ object MapPreview : Module(
     description = "Preview maps in your inventory",
     tag = ModuleTag.RENDER,
 ) {
+    @JvmStatic val showInSlot by setting("Show In Slot", true, "Shows the map in the slot rather than the basic map icon")
+
     private val background = Identifier.ofVanilla("textures/map/map_background.png")
 
     class MapComponent(val stack: ItemStack) : TooltipData, TooltipComponent {