Merge pull request #3 from rettichlp/feature/wanted-color-nametags

feature/wanted-color-nametags

Author: rettichlp

src/client/java/de/rettichlp/common/manager/MessageManager.java

@@ -1,18 +1,65 @@
 package de.rettichlp.common.manager;
 
 import com.mojang.brigadier.Message;
+import de.rettichlp.common.storage.schema.WantedEntry;
 import lombok.NoArgsConstructor;
 import net.minecraft.client.MinecraftClient;
 import net.minecraft.client.network.ClientPlayNetworkHandler;
 
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import static de.rettichlp.PKUtilsClient.storage;
+import static java.lang.Integer.parseInt;
+import static java.lang.System.currentTimeMillis;
+import static java.util.Optional.ofNullable;
+import static java.util.regex.Pattern.compile;
+
 @NoArgsConstructor
 public class MessageManager {
 
-    private ClientPlayNetworkHandler networkHandler;
+    private final static Pattern WELCOME_BACK_PATTERN = compile("Willkommen zurück!");
+    private final static Pattern ONLINE_WANTED_PLAYERS_HEADER_PATTERN = compile("Online Spieler mit WantedPunkten:");
+    private final static Pattern ONLINE_WANTED_PLAYERS_ENTRY_PATTERN = compile("- (?<playerName>[A-Za-z0-9_]+) \\| (?<wantedPointAmount>\\d+) \\| (?<reason>.+)(?<afk> \\| AFK|)");
+
     private String rawMessage;
+    private ClientPlayNetworkHandler networkHandler;
+    private long activeWantedCheck = 0;
 
     public void process(Message message) {
-        this.networkHandler = MinecraftClient.getInstance().player.networkHandler;
         this.rawMessage = message.getString();
+        this.networkHandler = ofNullable(MinecraftClient.getInstance().player)
+                .map(clientPlayerEntity -> clientPlayerEntity.networkHandler)
+                .orElseThrow();
+
+        executeJoinCommands();
+        readWantedPoints();
+    }
+
+    private void executeJoinCommands() {
+        Matcher welcomeBackMatcher = WELCOME_BACK_PATTERN.matcher(this.rawMessage);
+        if (welcomeBackMatcher.find()) {
+
+            this.networkHandler.sendChatCommand("wanteds");
+        }
+    }
+
+    public void readWantedPoints() {
+        Matcher onlineWantedPlayersHeaderMatcher = ONLINE_WANTED_PLAYERS_HEADER_PATTERN.matcher(this.rawMessage);
+        if (onlineWantedPlayersHeaderMatcher.find()) {
+            this.activeWantedCheck = currentTimeMillis();
+            storage.resetWantedEntries();
+            return;
+        }
+
+        Matcher onlineWantedPlayersEntryMatcher = ONLINE_WANTED_PLAYERS_ENTRY_PATTERN.matcher(this.rawMessage);
+        if (onlineWantedPlayersEntryMatcher.find() && (currentTimeMillis() - this.activeWantedCheck < 2000)) {
+            String playerName = onlineWantedPlayersEntryMatcher.group("playerName");
+            int wantedPointAmount = parseInt(onlineWantedPlayersEntryMatcher.group("wantedPointAmount"));
+            String reason = onlineWantedPlayersEntryMatcher.group("reason");
+
+            WantedEntry wantedEntry = new WantedEntry(playerName, wantedPointAmount, reason);
+            storage.addWantedEntry(wantedEntry);
+        }
     }
 }

src/client/java/de/rettichlp/mixin/client/ExampleClientMixin.java

@@ -0,0 +1,74 @@
+package de.rettichlp.mixin.client;
+
+import de.rettichlp.common.storage.schema.WantedEntry;
+import net.minecraft.client.render.entity.EntityRenderer;
+import net.minecraft.client.render.entity.state.EntityRenderState;
+import net.minecraft.client.render.entity.state.PlayerEntityRenderState;
+import net.minecraft.entity.Entity;
+import net.minecraft.text.Text;
+import net.minecraft.util.Formatting;
+import org.jetbrains.annotations.Contract;
+import org.jetbrains.annotations.NotNull;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Unique;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.ModifyVariable;
+
+import java.util.Optional;
+
+import static de.rettichlp.PKUtilsClient.storage;
+import static java.util.Objects.nonNull;
+import static net.minecraft.util.Formatting.*;
+import static net.minecraft.util.Formatting.DARK_RED;
+import static net.minecraft.util.Formatting.RED;
+
+@Mixin(EntityRenderer.class)
+public abstract class NametagColorMixin<S extends Entity, T extends EntityRenderState> {
+
+    @ModifyVariable(
+            method = "renderLabelIfPresent(Lnet/minecraft/client/render/entity/state/EntityRenderState;Lnet/minecraft/text/Text;Lnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;I)V",
+            at = @At("HEAD"),
+            argsOnly = true
+    )
+    private Text changeNameColor(Text original, EntityRenderState state) {
+        if (state instanceof PlayerEntityRenderState playerState && nonNull(playerState.displayName)) {
+            Text displayName = playerState.displayName;
+            String displayNameString = displayName.getString();
+
+            Optional<WantedEntry> optionalWantedEntry = storage.getWantedEntries().stream()
+                    .filter(wantedEntry -> wantedEntry.getPlayerName().equals(displayNameString))
+                    .findAny();
+
+            if (optionalWantedEntry.isEmpty()) {
+                return original;
+            }
+
+            Formatting color = getWantedPointColor(optionalWantedEntry.get());
+            return displayName.copy().styled(style -> style.withColor(color));
+        }
+
+        return original;
+    }
+
+    @Unique
+    @Contract(pure = true)
+    private @NotNull Formatting getWantedPointColor(@NotNull WantedEntry wantedEntry) {
+        int wantedPointAmount = wantedEntry.getWantedPointAmount();
+        Formatting color;
+
+        if (wantedPointAmount >= 60) {
+            color = DARK_RED;
+        } else if (wantedPointAmount >= 50) {
+            color = RED;
+        } else if (wantedPointAmount >= 25) {
+            color = GOLD;
+        } else if (wantedPointAmount >= 15) {
+            color = YELLOW;
+        } else if (wantedPointAmount >= 2) {
+            color = GREEN;
+        } else {
+            color = DARK_GREEN;
+        }
+        return color;
+    }
+}

src/client/java/de/rettichlp/mixin/client/NametagColorMixin.java

@@ -3,7 +3,7 @@
     "package": "de.rettichlp.mixin.client",
     "compatibilityLevel": "JAVA_21",
     "client": [
-        "ExampleClientMixin"
+        "NametagColorMixin"
     ],
     "injectors": {
         "defaultRequire": 1