Merge pull request #22 from Blukzen/feature/13-disabled-items

feature/13: Lab disabled items configuration
brukzen · Apr 19, 2022 · f5399d4 · f5399d4
2 parents 8a0115b + 2870eaf
commit f5399d4
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/main/java/com/blukzen/createlab/CreateLabConfig.java b/src/main/java/com/blukzen/createlab/CreateLabConfig.java
@@ -9,6 +9,7 @@ public class CreateLabConfig {
     public static final ForgeConfigSpec COMMON_SPEC;
 
     public static ForgeConfigSpec.ConfigValue<List<? extends String>> labPortalFrameBlocks;
+    public static ForgeConfigSpec.ConfigValue<List<? extends String>> disabledItems;
     public static ForgeConfigSpec.IntValue tpCommandPermissionLevel;
     public static ForgeConfigSpec.IntValue debugCommandPermissionLevel;
     public static ForgeConfigSpec.IntValue saveCommandPermissionLevel;
@@ -21,6 +22,7 @@ public class CreateLabConfig {
 
     private static void setupConfig(ForgeConfigSpec.Builder builder) {
         labPortalFrameBlocks = builder.comment("Blocks that can be used as a lab portal frame").defineList("lab_portal_frame_blocks", Arrays.asList("minecraft:smooth_quartz", "minecraft:quartz_block"), entry -> true);
+        disabledItems = builder.comment("Blocks and items that are disabled in the laboratory dimension").defineList("disabled_items", Arrays.asList("minecraft:ender_chest"), entry -> true);
 
         builder.push("Command Permission Levels");
         {

diff --git a/src/main/java/com/blukzen/createlab/mixin/ServerPlayerEntityMixin.java b/src/main/java/com/blukzen/createlab/mixin/ServerPlayerEntityMixin.java
@@ -1,15 +1,34 @@
 package com.blukzen.createlab.mixin;
 
+import com.blukzen.createlab.CreateLabConfig;
+import com.blukzen.createlab.dimension.LabDimensions;
 import com.mojang.authlib.GameProfile;
 import net.minecraft.entity.player.PlayerEntity;
 import net.minecraft.entity.player.ServerPlayerEntity;
+import net.minecraft.item.ItemStack;
 import net.minecraft.util.math.BlockPos;
 import net.minecraft.world.World;
 import org.spongepowered.asm.mixin.Mixin;
+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.function.Predicate;
 
 @Mixin(ServerPlayerEntity.class)
 public abstract class ServerPlayerEntityMixin extends PlayerEntity {
+    Predicate<ItemStack> DISABLED_ITEMS = (ItemStack itemStack) -> CreateLabConfig.disabledItems.get().contains(itemStack.getItem().getRegistryName().toString());
+
     public ServerPlayerEntityMixin(World world, BlockPos pos, float yaw, GameProfile profile) {
         super(world, pos, yaw, profile);
     }
+
+    @Inject(method = "tick", at = @At("HEAD"))
+    public void tick(CallbackInfo ci) {
+        if (this.level.dimension().equals(LabDimensions.LABDIM)) {
+            if (DISABLED_ITEMS.test(getMainHandItem())) {
+                getMainHandItem().setCount(0);
+            }
+        }
+    }
 }