Fix balance message being used for priority kick

Signed-off-by: Pablo Herrera <[email protected]>
PGMDev · Nov 24, 2024 · 4e80b34 · 4e80b34
1 parent b932174
commit 4e80b34
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/core/src/main/java/tc/oc/pgm/join/JoinMatchModule.java b/core/src/main/java/tc/oc/pgm/join/JoinMatchModule.java
@@ -80,6 +80,11 @@ public boolean priorityKickAllowed() {
     return getConfig().canPriorityKick() && !match.isRunning();
   }
 
+  public boolean canPriorityKick(MatchPlayer player) {
+    JoinRequest request = requests.get(player.getBukkit());
+    return request != null && canPriorityKick(request);
+  }
+
   public boolean canPriorityKick(JoinRequest request) {
     return priorityKickAllowed() && request.isForcedOr(JoinRequest.Flag.JOIN_FULL);
   }

diff --git a/core/src/main/java/tc/oc/pgm/teams/TeamMatchModule.java b/core/src/main/java/tc/oc/pgm/teams/TeamMatchModule.java
@@ -526,7 +526,7 @@ public boolean kickPlayerOffTeam(Team kickFrom, boolean forBalance) {
     }
 
     // Give them the bad news
-    if (jmm.canBePriorityKicked(kickMe)) {
+    if (jmm.canPriorityKick(kickMe)) {
       kickMe.sendMessage(translatable("join.ok.moved", kickTo.getName()));
       kickMe.sendMessage(translatable("join.ok.moved.explanation"));
     } else {