fix: Use 1.16.1 behavior for pathfinding near lava

Fixes the half of lithium 185 that wasn't addressed by e3de204
mrmangohands · Jan 5, 2021 · 339111b · 339111b
1 parent b5e8ff4
commit 339111b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/me/jellysquid/mods/lithium/common/ai/pathing/PathNodeDefaults.java b/src/main/java/me/jellysquid/mods/lithium/common/ai/pathing/PathNodeDefaults.java
@@ -22,6 +22,8 @@ public static PathNodeType getNeighborNodeType(BlockState state) {
             return PathNodeType.DANGER_FIRE;
         } else if (state.getFluidState().isIn(FluidTags.WATER)) {
             return PathNodeType.WATER_BORDER;
+        } else if (state.getFluidState().isIn(FluidTags.LAVA)) {
+            return PathNodeType.LAVA;
         } else {
             return PathNodeType.OPEN;
         }
@@ -96,6 +98,6 @@ public static PathNodeType getNodeType(BlockState state) {
     }
 
     private static boolean isFireDangerSource(BlockState blockState) {
-        return blockState.isIn(BlockTags.FIRE) || blockState.isOf(Blocks.LAVA) || blockState.isOf(Blocks.MAGMA_BLOCK) || CampfireBlock.isLitCampfire(blockState);
+        return blockState.isIn(BlockTags.FIRE) || blockState.isOf(Blocks.MAGMA_BLOCK) || CampfireBlock.isLitCampfire(blockState);
     }
 }