sort friends first setting in ExtraTab

Author: beanbag44

src/main/java/com/lambda/mixin/render/PlayerListHudMixin.java

@@ -26,8 +26,11 @@
 import net.minecraft.client.MinecraftClient;
 import net.minecraft.client.gui.hud.PlayerListHud;
 import net.minecraft.client.network.PlayerListEntry;
+import net.minecraft.scoreboard.Team;
 import net.minecraft.text.MutableText;
 import net.minecraft.text.Text;
+import net.minecraft.util.Nullables;
+import net.minecraft.world.GameMode;
 import org.jetbrains.annotations.Nullable;
 import org.spongepowered.asm.mixin.Final;
 import org.spongepowered.asm.mixin.Mixin;
@@ -45,6 +48,12 @@
 @Mixin(PlayerListHud.class)
 public class PlayerListHudMixin {
     @Shadow @Final private static Comparator<PlayerListEntry> ENTRY_ORDERING;
+    @Unique @Final private static Comparator<PlayerListEntry> FRIENDS_FIRST_ENTRY_ORDERING = Comparator
+            .comparingInt((PlayerListEntry entry) -> FriendManager.INSTANCE.isFriend(entry.getProfile().name()) ? 0 : 1)
+            .thenComparingInt(entry -> -entry.getListOrder())
+            .thenComparingInt((entry) -> entry.getGameMode() == GameMode.SPECTATOR ? 1 : 0)
+            .thenComparing((entry) -> Nullables.mapOrElse(entry.getScoreboardTeam(), Team::getName, ""))
+            .thenComparing((entry) -> entry.getProfile().name(), String::compareToIgnoreCase);
 
     @Shadow @Final private MinecraftClient client;
 
@@ -57,7 +66,7 @@ private void onCollectPlayerEntriesHead(CallbackInfoReturnable<List<PlayerListEn
                         .getListedPlayerListEntries()
                         .stream()
                         .filter(entry -> !ExtraTab.getFriendsOnly() || FriendManager.INSTANCE.isFriend(entry.getProfile()))
-                        .sorted(ENTRY_ORDERING)
+                        .sorted(ExtraTab.getSortFriendsFirst() ? FRIENDS_FIRST_ENTRY_ORDERING : ENTRY_ORDERING)
                         .limit(ExtraTab.getTabEntries())
                         .toList()
         );

src/main/kotlin/com/lambda/module/modules/render/ExtraTab.kt

@@ -29,6 +29,7 @@ object ExtraTab : Module(
     @JvmStatic val tabEntries by setting("Tab Entries", 80L, 1L..500L, 1L)
     @JvmStatic val rows by setting("Rows", 20, 1..100, 1)
     @JvmStatic val friendsOnly by setting("Friends Only", false)
+    @JvmStatic val sortFriendsFirst by setting("Sort Friends First", true)
     @JvmStatic val highlightFriends by setting("Highlight Friends", true)
     @JvmStatic val friendColor by setting("Friend Color", Color(120, 120, 255, 255), "The color friends will be highlighted") { highlightFriends }
 }