Allow shrinking data component to create new machine rooms

Author: robotgryphon

neoforge-main/src/main/java/dev/compactmods/machines/machine/block/UnboundCompactMachineBlock.java

@@ -9,6 +9,8 @@
 import dev.compactmods.machines.api.shrinking.PSDTags;
 import dev.compactmods.machines.machine.Machines;
 import dev.compactmods.machines.room.RoomHelper;
+import dev.compactmods.machines.shrinking.PersonalShrinkingDevice;
+import dev.compactmods.machines.shrinking.Shrinking;
 import net.minecraft.core.BlockPos;
 import net.minecraft.server.MinecraftServer;
 import net.minecraft.server.level.ServerLevel;
@@ -64,7 +66,7 @@ protected ItemInteractionResult useItemOn(ItemStack stack, BlockState state, Lev
 		}
 
 		MinecraftServer server = level.getServer();
-		if (stack.is(PSDTags.ITEM) && player instanceof ServerPlayer sp) {
+		if ((stack.is(PSDTags.ITEM) || stack.has(Shrinking.DataComponents.SHRINKING_CONFIG)) && player instanceof ServerPlayer sp) {
 			level.getBlockEntity(pos, Machines.BlockEntities.UNBOUND_MACHINE.get()).ifPresent(unboundEntity -> {
 
 				RoomTemplate template = RoomTemplateHelper.getTemplate(level, unboundEntity.templateId());
@@ -84,7 +86,8 @@ protected ItemInteractionResult useItemOn(ItemStack stack, BlockState state, Lev
 							ent.setData(Machines.Attachments.MACHINE_COLOR, color);
 
 							try {
-								RoomHelper.teleportPlayerIntoRoom(server, sp, newRoom, RoomEntryPoint.playerEnteringMachine(player));
+								RoomHelper.teleportPlayerIntoRoom(server, sp, newRoom, RoomEntryPoint.playerEnteringMachine(player))
+										.thenAccept(res -> PersonalShrinkingDevice.handleSuccessfulAtomicShift(stack, sp, PersonalShrinkingDevice.config(stack)));
 							} catch (MissingDimensionException e) {
 								throw new RuntimeException(e);
 							}

neoforge-main/src/main/java/dev/compactmods/machines/shrinking/PersonalShrinkingDevice.java

@@ -76,14 +76,18 @@ public InteractionResultHolder<ItemStack> use(Level world, Player player, Intera
     public static void handleSuccessfulAtomicShift(ItemStack stack, ServerPlayer serverPlayer, ShrinkingDeviceConfiguration config) {
         switch (config.afterUseAction()) {
             case DAMAGE:
-                stack.hurtAndBreak(1, serverPlayer.serverLevel(), serverPlayer, item -> {
-                    // RIP, hope you have spare crafting materials nearby!
-                });
+                if(!serverPlayer.hasInfiniteMaterials()) {
+                    stack.hurtAndBreak(1, serverPlayer.serverLevel(), serverPlayer, item -> {
+                        // RIP, hope you have spare crafting materials nearby!
+                    });
+                }
                 break;
 
             case BREAK:
-                stack.consume(1, serverPlayer);
-                PlayerUtil.breakItemEffect(serverPlayer, stack);
+                if(!serverPlayer.hasInfiniteMaterials()) {
+                    stack.consume(1, serverPlayer);
+                    PlayerUtil.breakItemEffect(serverPlayer, stack);
+                }
                 break;
         }
     }