Add sleeping CrafterBlockEntity

Author: 2No2Name

common/src/main/java/net/caffeinemc/mods/lithium/mixin/world/block_entity_ticking/sleeping/crafter/CrafterBlockEntityMixin.java

@@ -0,0 +1,88 @@
+package net.caffeinemc.mods.lithium.mixin.world.block_entity_ticking.sleeping.crafter;
+
+import com.llamalad7.mixinextras.sugar.Local;
+import net.caffeinemc.mods.lithium.common.block.entity.SetChangedHandlingBlockEntity;
+import net.caffeinemc.mods.lithium.common.block.entity.SleepingBlockEntity;
+import net.caffeinemc.mods.lithium.mixin.world.block_entity_ticking.sleeping.WrappedBlockEntityTickInvokerAccessor;
+import net.minecraft.core.BlockPos;
+import net.minecraft.world.level.block.entity.BlockEntity;
+import net.minecraft.world.level.block.entity.BlockEntityType;
+import net.minecraft.world.level.block.entity.CrafterBlockEntity;
+import net.minecraft.world.level.block.entity.TickingBlockEntity;
+import net.minecraft.world.level.block.state.BlockState;
+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;
+
+@Mixin(CrafterBlockEntity.class)
+public class CrafterBlockEntityMixin extends BlockEntity implements SleepingBlockEntity, SetChangedHandlingBlockEntity {
+
+    @Shadow
+    private int craftingTicksRemaining;
+    @Unique
+    private WrappedBlockEntityTickInvokerAccessor tickWrapper = null;
+    @Unique
+    private TickingBlockEntity sleepingTicker = null;
+
+    public CrafterBlockEntityMixin(BlockEntityType<?> blockEntityType, BlockPos blockPos, BlockState blockState) {
+        super(blockEntityType, blockPos, blockState);
+    }
+
+    @Override
+    public WrappedBlockEntityTickInvokerAccessor lithium$getTickWrapper() {
+        return this.tickWrapper;
+    }
+
+    @Override
+    public void lithium$setTickWrapper(WrappedBlockEntityTickInvokerAccessor tickWrapper) {
+        this.tickWrapper = tickWrapper;
+        this.lithium$setSleepingTicker(null);
+    }
+
+    @Override
+    public TickingBlockEntity lithium$getSleepingTicker() {
+        return this.sleepingTicker;
+    }
+
+    @Override
+    public void lithium$setSleepingTicker(TickingBlockEntity sleepingTicker) {
+        this.sleepingTicker = sleepingTicker;
+    }
+
+    @Inject(method = "serverTick", at = @At("RETURN"))
+    private static void checkSleep(CallbackInfo ci, @Local(argsOnly = true) CrafterBlockEntity crafterBlockEntity, @Local int remainingTicks) {
+        if (remainingTicks < 0) {
+            ((CrafterBlockEntityMixin) (Object) crafterBlockEntity).checkSleep();
+        }
+    }
+
+    @Unique
+    private void checkSleep() {
+        if (this.craftingTicksRemaining == 0) {
+            this.lithium$startSleeping();
+        }
+    }
+
+    @Inject(
+            method = {
+                    "loadAdditional(Lnet/minecraft/nbt/CompoundTag;Lnet/minecraft/core/HolderLookup$Provider;)V",
+                    "setCraftingTicksRemaining"
+            },
+            at = @At("RETURN")
+    )
+    private void wakeUpAfterRemainingTicksChanged(CallbackInfo ci) {
+        if (this.isSleeping() && this.level != null && !this.level.isClientSide) {
+            this.wakeUpNow();
+        }
+    }
+
+    @Override
+    public void lithium$handleSetChanged() {
+        if (this.isSleeping() && this.level != null && !this.level.isClientSide) {
+            this.wakeUpNow();
+        }
+    }
+}

common/src/main/java/net/caffeinemc/mods/lithium/mixin/world/block_entity_ticking/sleeping/crafter/package-info.java

@@ -0,0 +1,5 @@
+@MixinConfigOption(description = "BlockEntity sleeping for inactive crafters"
+)
+package net.caffeinemc.mods.lithium.mixin.world.block_entity_ticking.sleeping.crafter;
+
+import net.caffeinemc.gradle.MixinConfigOption;

common/src/main/resources/lithium.mixins.json

@@ -221,6 +221,7 @@
     "world.block_entity_ticking.sleeping.campfire.CampfireBlockEntityMixin",
     "world.block_entity_ticking.sleeping.campfire.lit.CampfireBlockEntityMixin",
     "world.block_entity_ticking.sleeping.campfire.unlit.CampfireBlockEntityMixin",
+    "world.block_entity_ticking.sleeping.crafter.CrafterBlockEntityMixin",
     "world.block_entity_ticking.sleeping.furnace.AbstractFurnaceBlockEntityMixin",
     "world.block_entity_ticking.sleeping.hopper.HopperBlockEntityMixin",
     "world.block_entity_ticking.sleeping.shulker_box.ShulkerBoxBlockEntityMixin",

lithium-fabric-mixin-config.md

@@ -592,6 +592,10 @@ BlockEntity sleeping for inactive lit campfires
 (default: `true`)  
 BlockEntity sleeping for inactive unlit campfires
 
+### `mixin.world.block_entity_ticking.sleeping.crafter`
+(default: `true`)  
+BlockEntity sleeping for inactive crafters
+  
 ### `mixin.world.block_entity_ticking.sleeping.furnace`
 (default: `true`)  
 BlockEntity sleeping for inactive furnaces

lithium-neoforge-mixin-config.md

@@ -573,6 +573,10 @@ BlockEntity sleeping for inactive lit campfires
 (default: `true`)  
 BlockEntity sleeping for inactive unlit campfires
 
+### `mixin.world.block_entity_ticking.sleeping.crafter`
+(default: `true`)  
+BlockEntity sleeping for inactive crafters
+  
 ### `mixin.world.block_entity_ticking.sleeping.furnace`
 (default: `true`)  
 BlockEntity sleeping for inactive furnaces