From f270db813244539e7930c4ca2a47b441893b913c Mon Sep 17 00:00:00 2001
From: Ray  Neiheiser <ray.neiheiser@gmail.com>
Date: Mon, 4 Nov 2024 19:29:53 +0100
Subject: [PATCH] use static util to get team without needing the colony.

---
 src/main/java/com/minecolonies/api/colony/IColony.java   | 6 +++---
 .../api/entity/citizen/AbstractEntityCitizen.java        | 4 ++--
 .../citizen/citizenhandlers/ICitizenColonyHandler.java   | 8 ++++++++
 src/main/java/com/minecolonies/core/colony/Colony.java   | 4 ++--
 .../citizen/citizenhandlers/CitizenColonyHandler.java    | 9 +++++++++
 5 files changed, 24 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/minecolonies/api/colony/IColony.java b/src/main/java/com/minecolonies/api/colony/IColony.java
index 4212bcd6795..8b49e882c6b 100755
--- a/src/main/java/com/minecolonies/api/colony/IColony.java
+++ b/src/main/java/com/minecolonies/api/colony/IColony.java
@@ -124,10 +124,10 @@ public interface IColony
      *
      * @return The team name
      */
-    default String getTeamName()
+    static String getTeamName(final Level level, final int id)
     {
-        final String dim = getDimension().location().getPath();
-        return TEAM_COLONY_NAME + "_" + (dim.length() > 10 ? dim.hashCode() : dim) + "_" + getID();
+        final String dim = level.dimension().location().getPath();
+        return TEAM_COLONY_NAME + "_" + (dim.length() > 10 ? dim.hashCode() : dim) + "_" + id;
     }
 
     /**
diff --git a/src/main/java/com/minecolonies/api/entity/citizen/AbstractEntityCitizen.java b/src/main/java/com/minecolonies/api/entity/citizen/AbstractEntityCitizen.java
index 6d631267eb1..8b6ffe83a0a 100755
--- a/src/main/java/com/minecolonies/api/entity/citizen/AbstractEntityCitizen.java
+++ b/src/main/java/com/minecolonies/api/entity/citizen/AbstractEntityCitizen.java
@@ -240,11 +240,11 @@ public boolean isNoAi()
     protected PlayerTeam getAssignedTeam()
     {
         final ICitizenColonyHandler citizenColonyHandler = getCitizenColonyHandler();
-        if (citizenColonyHandler == null || citizenColonyHandler.getColony() == null)
+        if (citizenColonyHandler == null)
         {
             return null;
         }
-        return citizenColonyHandler.getColony().getTeam();
+        return citizenColonyHandler.getTeam(level);
     }
 
     /**
diff --git a/src/main/java/com/minecolonies/api/entity/citizen/citizenhandlers/ICitizenColonyHandler.java b/src/main/java/com/minecolonies/api/entity/citizen/citizenhandlers/ICitizenColonyHandler.java
index 9894600f529..122a60d4711 100755
--- a/src/main/java/com/minecolonies/api/entity/citizen/citizenhandlers/ICitizenColonyHandler.java
+++ b/src/main/java/com/minecolonies/api/entity/citizen/citizenhandlers/ICitizenColonyHandler.java
@@ -3,6 +3,8 @@
 import com.minecolonies.api.colony.IColony;
 import com.minecolonies.api.colony.buildings.IBuilding;
 import net.minecraft.network.syncher.EntityDataAccessor;
+import net.minecraft.world.level.Level;
+import net.minecraft.world.scores.PlayerTeam;
 import org.jetbrains.annotations.Nullable;
 
 public interface ICitizenColonyHandler
@@ -73,4 +75,10 @@ public interface ICitizenColonyHandler
     void onSyncDataUpdate(EntityDataAccessor<?> dataAccessor);
 
     boolean registered();
+
+    /**
+     * Get the citizen team.
+     * @return the team.
+     */
+    PlayerTeam getTeam(final Level level);
 }
diff --git a/src/main/java/com/minecolonies/core/colony/Colony.java b/src/main/java/com/minecolonies/core/colony/Colony.java
index 885c6ee1ba1..12bf606ccbb 100644
--- a/src/main/java/com/minecolonies/core/colony/Colony.java
+++ b/src/main/java/com/minecolonies/core/colony/Colony.java
@@ -363,7 +363,7 @@ protected Colony(final int id, @Nullable final Level world)
         {
             this.dimensionId = world.dimension();
             onWorldLoad(world);
-            checkOrCreateTeam(world, getTeamName(), false);
+            checkOrCreateTeam(world, IColony.getTeamName(world, id), false);
         }
         this.permissions = new Permissions(this);
         researchManager = new ResearchManager(this);
@@ -632,7 +632,7 @@ public void updateAttackingPlayers()
     public PlayerTeam getTeam()
     {
         // This getter will create the team if it doesn't exist. Could do something different though in the future.
-        return checkOrCreateTeam(world, getTeamName(), false);
+        return checkOrCreateTeam(world, IColony.getTeamName(world, id), false);
     }
 
     /**
diff --git a/src/main/java/com/minecolonies/core/entity/citizen/citizenhandlers/CitizenColonyHandler.java b/src/main/java/com/minecolonies/core/entity/citizen/citizenhandlers/CitizenColonyHandler.java
index 0f8725b14dd..62662369ee3 100755
--- a/src/main/java/com/minecolonies/core/entity/citizen/citizenhandlers/CitizenColonyHandler.java
+++ b/src/main/java/com/minecolonies/core/entity/citizen/citizenhandlers/CitizenColonyHandler.java
@@ -9,10 +9,13 @@
 import net.minecraft.network.syncher.EntityDataAccessor;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.world.entity.Entity;
+import net.minecraft.world.level.Level;
+import net.minecraft.world.scores.PlayerTeam;
 import org.jetbrains.annotations.Nullable;
 
 import static com.minecolonies.api.entity.citizen.AbstractEntityCitizen.*;
 import static com.minecolonies.api.util.constant.CitizenConstants.SATURATION_DECREASE_FACTOR;
+import static com.minecolonies.core.util.TeamUtils.checkOrCreateTeam;
 
 /**
  * Handles all colony related methods for the citizen.
@@ -233,4 +236,10 @@ public void onCitizenRemoved()
             citizen.getCitizenData().setLastPosition(citizen.blockPosition());
         }
     }
+
+    @Override
+    public PlayerTeam getTeam(final Level level)
+    {
+        return checkOrCreateTeam(level, IColony.getTeamName(level, colonyId));
+    }
 }