package com.hyteamselector.manager; import com.andrei1058.bedwars.api.arena.IArena; import com.andrei1058.bedwars.api.arena.team.ITeam; import com.hyteamselector.HyTeamSelector; import org.bukkit.entity.Player; import java.util.*; import java.util.concurrent.ConcurrentHashMap; public class TeamManager { private final HyTeamSelector plugin; private final Map playerSelections; private final Map> arenaSelections; public TeamManager(HyTeamSelector plugin) { this.plugin = plugin; this.playerSelections = new ConcurrentHashMap<>(); this.arenaSelections = new ConcurrentHashMap<>(); } public void selectTeam(Player player, IArena arena, ITeam team) { playerSelections.put(player.getUniqueId(), team.getName()); arenaSelections.computeIfAbsent(arena.getArenaName(), k -> new HashSet<>()).add(player.getUniqueId()); } public boolean hasSelected(Player player) { return playerSelections.containsKey(player.getUniqueId()); } public String getSelectedTeam(Player player) { return playerSelections.get(player.getUniqueId()); } public void clearSelection(Player player) { playerSelections.remove(player.getUniqueId()); } public void clearArena(String arenaName) { Set players = arenaSelections.remove(arenaName); if (players != null) { players.forEach(playerSelections::remove); } } public void autoAssignPlayers(IArena arena) { List unassignedPlayers = new ArrayList<>(); // Only auto-assign solo players (not in a party) for (Player player : arena.getPlayers()) { if (!hasSelected(player)) { // Check if player is in a party - skip auto-assign if they are if (arena.getPlayer(player).getParty() == null) { unassignedPlayers.add(player); } } } if (unassignedPlayers.isEmpty()) { return; } boolean balanceTeams = plugin.getConfig().getBoolean("auto-assign.balance-teams", true); if (balanceTeams) { assignBalanced(arena, unassignedPlayers); } else { assignRandom(arena, unassignedPlayers); } } private void assignBalanced(IArena arena, List players) { List availableTeams = new ArrayList<>(arena.getTeams()); availableTeams.sort(Comparator.comparingInt(ITeam::getSize)); for (Player player : players) { ITeam smallestTeam = null; for (ITeam team : availableTeams) { if (team.getSize() < team.getMaxPlayers()) { smallestTeam = team; break; } } if (smallestTeam != null) { smallestTeam.addPlayers(player); selectTeam(player, arena, smallestTeam); String message = plugin.getConfig().getString("messages.auto-assigned", "&eYou were auto-assigned to {team}&e!") .replace("{team}", smallestTeam.getDisplayName(plugin.getBedwarsAPI().getArenaUtil().getArenaByPlayer(player).getPlayer(player).getLanguage())); player.sendMessage(plugin.colorize(plugin.getPrefix() + message)); availableTeams.sort(Comparator.comparingInt(ITeam::getSize)); } } } private void assignRandom(IArena arena, List players) { List availableTeams = new ArrayList<>(arena.getTeams()); Collections.shuffle(players); for (Player player : players) { Collections.shuffle(availableTeams); for (ITeam team : availableTeams) { if (team.getSize() < team.getMaxPlayers()) { team.addPlayers(player); selectTeam(player, arena, team); String message = plugin.getConfig().getString("messages.auto-assigned", "&eYou were auto-assigned to {team}&e!") .replace("{team}", team.getDisplayName(plugin.getBedwarsAPI().getArenaUtil().getArenaByPlayer(player).getLanguage())); player.sendMessage(plugin.colorize(plugin.getPrefix() + message)); break; } } } } public void cleanup() { playerSelections.clear(); arenaSelections.clear(); } }