From 75f26f4ed9011e9dd12f5b49b26b94f8f4dacac1 Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Fri, 15 Jan 2021 00:42:54 +0800
Subject: Version 1.4.0: supports reintroducing llama item duplication. Add
 configuration file.

---
 .../java/com/keuin/ohmyvanillamc/mixin/DisableFishSchooling.java    | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'src/main/java/com/keuin/ohmyvanillamc/mixin/DisableFishSchooling.java')

diff --git a/src/main/java/com/keuin/ohmyvanillamc/mixin/DisableFishSchooling.java b/src/main/java/com/keuin/ohmyvanillamc/mixin/DisableFishSchooling.java
index c30ab02..c2de0d5 100644
--- a/src/main/java/com/keuin/ohmyvanillamc/mixin/DisableFishSchooling.java
+++ b/src/main/java/com/keuin/ohmyvanillamc/mixin/DisableFishSchooling.java
@@ -2,6 +2,7 @@ package com.keuin.ohmyvanillamc.mixin;
 
 import com.keuin.ohmyvanillamc.OhMyVanillaMinecraft;
 import net.minecraft.entity.EntityType;
+import net.minecraft.entity.ai.goal.FollowGroupLeaderGoal;
 import net.minecraft.entity.passive.FishEntity;
 import net.minecraft.entity.passive.SchoolingFishEntity;
 import net.minecraft.world.World;
@@ -25,7 +26,7 @@ public abstract class DisableFishSchooling extends FishEntity {
      */
     @Overwrite
     public void moveTowardLeader() {
-        if (!OhMyVanillaMinecraft.disableFishSchooling) {
+        if (!OhMyVanillaMinecraft.getConfiguration().isDisableFishSchooling()) {
             if (this.hasLeader()) {
                 this.getNavigation().startMovingTo(this.leader, 1.0D);
             }
@@ -39,5 +40,8 @@ public abstract class DisableFishSchooling extends FishEntity {
     @Overwrite
     public void initGoals() {
         super.initGoals();
+        if (!OhMyVanillaMinecraft.getConfiguration().isDisableFishSchooling()) {
+            this.goalSelector.add(5, new FollowGroupLeaderGoal((SchoolingFishEntity) (Object) this));
+        }
     }
 }
-- 
cgit v1.2.3