new: util.chunk_access

src/main/java/me/jellysquid/mods/lithium/common/world/ChunkView.java

@@ -0,0 +1,10 @@
+package me.jellysquid.mods.lithium.common.world;
+
+import net.minecraft.world.chunk.Chunk;
+import org.jetbrains.annotations.Nullable;
+
+public interface ChunkView {
+
+    @Nullable
+    Chunk getLoadedChunk(int chunkX, int chunkZ);
+}

src/main/java/me/jellysquid/mods/lithium/mixin/util/chunk_access/ChunkCacheMixin.java

@@ -0,0 +1,20 @@
+package me.jellysquid.mods.lithium.mixin.util.chunk_access;
+
+import me.jellysquid.mods.lithium.common.world.ChunkView;
+import net.minecraft.world.chunk.Chunk;
+import net.minecraft.world.chunk.ChunkCache;
+import org.jetbrains.annotations.Nullable;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+
+@Mixin(ChunkCache.class)
+public abstract class ChunkCacheMixin implements ChunkView {
+
+    @Shadow
+    protected abstract Chunk getChunk(int chunkX, int chunkZ);
+
+    @Override
+    public @Nullable Chunk getLoadedChunk(int chunkX, int chunkZ) {
+        return this.getChunk(chunkX, chunkZ);
+    }
+}

src/main/java/me/jellysquid/mods/lithium/mixin/util/chunk_access/WorldViewMixin.java

@@ -0,0 +1,21 @@
+package me.jellysquid.mods.lithium.mixin.util.chunk_access;
+
+import me.jellysquid.mods.lithium.common.world.ChunkView;
+import net.minecraft.world.WorldView;
+import net.minecraft.world.chunk.Chunk;
+import net.minecraft.world.chunk.ChunkStatus;
+import org.jetbrains.annotations.Nullable;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+
+@Mixin(WorldView.class)
+public interface WorldViewMixin extends ChunkView {
+
+    @Shadow
+    @Nullable Chunk getChunk(int var1, int var2, ChunkStatus var3, boolean var4);
+
+    @Override
+    default @Nullable Chunk getLoadedChunk(int chunkX, int chunkZ) {
+        return this.getChunk(chunkX, chunkZ, ChunkStatus.FULL, false);
+    }
+}

src/main/java/me/jellysquid/mods/lithium/mixin/util/chunk_access/package-info.java

@@ -0,0 +1,6 @@
+@MixinConfigOption(
+        description = "Access chunks of worlds, chunk caches and chunk regions directly."
+)
+package me.jellysquid.mods.lithium.mixin.util.chunk_access;
+
+import net.caffeinemc.gradle.MixinConfigOption;

src/main/resources/lithium.mixins.json

@@ -157,6 +157,8 @@
     "util.block_entity_retrieval.WorldMixin",
     "util.block_tracking.AbstractBlockStateMixin",
     "util.block_tracking.ChunkSectionMixin",
+    "util.chunk_access.ChunkCacheMixin",
+    "util.chunk_access.WorldViewMixin",
     "util.entity_movement_tracking.EntityTrackingSectionMixin",
     "util.entity_movement_tracking.SectionedEntityCacheMixin",
     "util.entity_movement_tracking.ServerEntityManagerAccessor",