implement ExtraTab rows and add friends only, highlight friends, and friend color settings

Author: beanbag44

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

@@ -17,15 +17,26 @@
 
 package com.lambda.mixin.render;
 
+import com.lambda.friend.FriendManager;
 import com.lambda.module.modules.render.ExtraTab;
+import com.lambda.util.text.TextBuilder;
+import com.lambda.util.text.TextDslKt;
+import com.llamalad7.mixinextras.injector.ModifyExpressionValue;
+import kotlin.Unit;
 import net.minecraft.client.MinecraftClient;
 import net.minecraft.client.gui.hud.PlayerListHud;
 import net.minecraft.client.network.PlayerListEntry;
+import net.minecraft.text.MutableText;
+import net.minecraft.text.Text;
+import org.jetbrains.annotations.Nullable;
 import org.spongepowered.asm.mixin.Final;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.Unique;
 import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Constant;
 import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.ModifyConstant;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
 
 import java.util.Comparator;
@@ -45,9 +56,35 @@ private void onCollectPlayerEntriesHead(CallbackInfoReturnable<List<PlayerListEn
                 client.player.networkHandler
                         .getListedPlayerListEntries()
                         .stream()
+                        .filter(entry -> !ExtraTab.getFriendsOnly() || FriendManager.INSTANCE.isFriend(entry.getProfile()))
                         .sorted(ENTRY_ORDERING)
                         .limit(ExtraTab.getTabEntries())
                         .toList()
         );
     }
+
+    @ModifyConstant(method = "render", constant = @Constant(intValue = 20))
+    private int modifyRowLimit(int original) {
+        return ExtraTab.INSTANCE.isEnabled() ? ExtraTab.getRows() : original;
+    }
+
+    @ModifyExpressionValue(method = "getPlayerName", at = @At(value = "INVOKE", target = "Lnet/minecraft/text/Text;copy()Lnet/minecraft/text/MutableText;"))
+    private MutableText modifyPlayerName(MutableText original) { return modifyName(original); }
+
+    @ModifyExpressionValue(method = "getPlayerName", at = @At(value = "INVOKE", target = "Lnet/minecraft/scoreboard/Team;decorateName(Lnet/minecraft/scoreboard/AbstractTeam;Lnet/minecraft/text/Text;)Lnet/minecraft/text/MutableText;"))
+    private MutableText modifyDecorateName(MutableText original) { return modifyName(original); }
+
+    @Unique
+    private @Nullable MutableText modifyName(Text original) {
+        if (ExtraTab.INSTANCE.isDisabled() ||
+                !ExtraTab.getHighlightFriends() ||
+                !FriendManager.INSTANCE.isFriend(original.getString())) return original.copy();
+        var newText = original.copy();
+        var textBuilder = new TextBuilder();
+        TextDslKt.color(textBuilder, ExtraTab.getFriendColor(), builder -> {
+            builder.styleAndAppend(newText);
+            return Unit.INSTANCE;
+        });
+        return textBuilder.getText();
+    }
 }

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

@@ -19,15 +19,16 @@ package com.lambda.module.modules.render
 
 import com.lambda.module.Module
 import com.lambda.module.tag.ModuleTag
+import java.awt.Color
 
 object ExtraTab : Module(
     name = "ExtraTab",
     description = "Adds more tabs to the main menu",
     tag = ModuleTag.RENDER,
 ) {
-    @JvmStatic
-    val tabEntries by setting("Tab Entries", 80L, 1L..500L, 1L)
-
-    @JvmStatic
-    val rows by setting("Rows", 20, 1..100, 1)
+    @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 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 }
 }