nicks wip

Author: calcastor

platform/platform-modern/src/main/java/dev/pgm/community/platform/modern/PacketManipulations.java

@@ -8,6 +8,8 @@
 import com.github.retrooper.packetevents.protocol.player.TextureProperty;
 import com.github.retrooper.packetevents.protocol.player.UserProfile;
 import com.github.retrooper.packetevents.wrapper.play.server.WrapperPlayServerPlayerInfoUpdate;
+import com.github.retrooper.packetevents.wrapper.play.server.WrapperPlayServerTeams;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import net.kyori.adventure.text.serializer.legacy.LegacyComponentSerializer;
@@ -23,7 +25,11 @@ public class PacketManipulations {
   public PacketManipulations() {
     Packets.register(
         PacketListenerPriority.LOWEST,
-        Map.of(PacketType.Play.Server.PLAYER_INFO_UPDATE, this::handlePlayerInfo));
+        Map.of(
+            PacketType.Play.Server.PLAYER_INFO_UPDATE,
+            this::handlePlayerInfo,
+            PacketType.Play.Server.TEAMS,
+            this::handleScoreboardTeam));
   }
 
   private void handlePlayerInfo(@NonNull PacketSendEvent event) {
@@ -34,7 +40,7 @@ private void handlePlayerInfo(@NonNull PacketSendEvent event) {
     List<WrapperPlayServerPlayerInfoUpdate.PlayerInfo> entries = wrapper.getEntries();
     for (WrapperPlayServerPlayerInfoUpdate.PlayerInfo entry : entries) {
       Player player = Bukkit.getPlayer(entry.getProfileId());
-      if (player == null || player.equals(viewer) || !player.isOnline()) continue;
+      if (player == null || player.equals(viewer)) continue;
 
       String playerDisplayName = PLAYER_UTILS.getPlayerDisplayName(player, viewer);
       String playerName = PLAYER_UTILS.getPlayerName(player, viewer);
@@ -59,4 +65,30 @@ private void handlePlayerInfo(@NonNull PacketSendEvent event) {
 
     if (modified) event.markForReEncode(true);
   }
+
+  private void handleScoreboardTeam(@NonNull PacketSendEvent event) {
+    Player viewer = event.getPlayer();
+    WrapperPlayServerTeams wrapper = new WrapperPlayServerTeams(event);
+
+    List<String> players = new ArrayList<>(wrapper.getPlayers());
+    if (players.isEmpty()) return;
+
+    boolean modified = false;
+    for (String name : players) {
+      Player player = Bukkit.getPlayerExact(name);
+
+      if (player == null || player.equals(viewer)) continue;
+
+      String playerName = PLAYER_UTILS.getPlayerName(player, viewer);
+      if (StringUtils.isBlank(playerName) || playerName.equals(name)) continue;
+
+      players.set(players.indexOf(name), playerName);
+      modified = true;
+    }
+
+    if (modified) {
+      wrapper.setPlayers(players);
+      event.markForReEncode(true);
+    }
+  }
 }