allow server resource packs to disable chat heads via "assets/chat_heads/disable" file

Author: Fourmisain

common/src/main/java/dzwdz/chat_heads/ChatHeads.java

@@ -57,6 +57,7 @@
 public class ChatHeads {
     public static final String MOD_ID = "chat_heads";
     public static final String NON_NAME_REGEX = "(§.)|[^\\w]";
+    public static final ResourceLocation DISABLE_RESOURCE = new ResourceLocation("chat_heads", "disable");
 
     public static ChatHeadsConfig CONFIG = new ChatHeadsConfigDefaults();
 
@@ -75,7 +76,8 @@ public class ChatHeads {
     public static int lastY = 0;
     public static float lastOpacity = 0.0f;
     public static int lastChatOffset;
-    public static boolean serverSentUuid = false;
+    public static volatile boolean serverSentUuid = false;
+    public static volatile boolean serverDisabledChatHeads = false;
 
     public static final Set<ResourceLocation> blendedHeadTextures = new HashSet<>();
 
@@ -92,6 +94,11 @@ public static void resetLineOwner() {
     }
 
     public static void handleAddedMessage(Component message, @Nullable ChatType.Bound bound, @Nullable PlayerInfo playerInfo) {
+        if (ChatHeads.serverDisabledChatHeads) {
+            ChatHeads.lastSender = null;
+            return;
+        }
+
         if (ChatHeads.CONFIG.senderDetection() != HEURISTIC_ONLY) {
             if (playerInfo != null) {
                 ChatHeads.lastSender = playerInfo;
@@ -117,7 +124,7 @@ public static int getChatOffset(@NotNull GuiMessage.Line guiMessage) {
     }
 
     public static int getChatOffset(@Nullable PlayerInfo owner) {
-        if (owner != null || ChatHeads.CONFIG.offsetNonPlayerText()) {
+        if (owner != null || (ChatHeads.CONFIG.offsetNonPlayerText() && !ChatHeads.serverDisabledChatHeads)) {
             return 10;
         } else {
             return 0;

common/src/main/java/dzwdz/chat_heads/mixin/ConnectionMixin.java

@@ -0,0 +1,20 @@
+package dzwdz.chat_heads.mixin;
+
+import dzwdz.chat_heads.ChatHeads;
+import net.minecraft.network.Connection;
+import net.minecraft.network.protocol.login.ClientLoginPacketListener;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+@Mixin(Connection.class)
+public abstract class ConnectionMixin {
+    // note: can run on different threads
+    @Inject(method = "initiateServerboundPlayConnection", at = @At("HEAD"))
+    public void chatheads$resetServerKnowledge(String string, int i, ClientLoginPacketListener clientLoginPacketListener, CallbackInfo ci) {
+        // reset every time we build a connection, be it singleplayer or multiplayer
+        ChatHeads.serverSentUuid = false;
+        ChatHeads.serverDisabledChatHeads = false;
+    }
+}

common/src/main/java/dzwdz/chat_heads/mixin/DownloadedPackSourceMixin.java

@@ -0,0 +1,35 @@
+package dzwdz.chat_heads.mixin;
+
+import dzwdz.chat_heads.ChatHeads;
+import net.minecraft.client.resources.DownloadedPackSource;
+import net.minecraft.server.packs.PackResources;
+import net.minecraft.server.packs.PackType;
+import net.minecraft.server.packs.repository.Pack;
+import net.minecraft.server.packs.repository.PackSource;
+import org.jetbrains.annotations.Nullable;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
+
+import java.io.File;
+import java.util.concurrent.CompletableFuture;
+
+@Mixin(DownloadedPackSource.class)
+public abstract class DownloadedPackSourceMixin {
+	@Shadow @Nullable
+	private Pack serverPack;
+
+	// note: runs on Netty Client IO thread
+    @Inject(method = "setServerPack", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/Minecraft;delayTextureReload()Ljava/util/concurrent/CompletableFuture;"))
+    public void f(File file, PackSource packSource, CallbackInfoReturnable<CompletableFuture<Void>> cir) {
+		if (serverPack == null) return; // should never be null
+
+		try (PackResources resources = serverPack.open()) {
+			if (resources.getResource(PackType.CLIENT_RESOURCES, ChatHeads.DISABLE_RESOURCE) != null) {
+				ChatHeads.serverDisabledChatHeads = true;
+			}
+		}
+    }
+}

common/src/main/java/dzwdz/chat_heads/mixin/MinecraftMixin.java

@@ -9,10 +9,11 @@
     "ChatListenerMixin",
     "ClientPacketListenerMixin",
     "CommandSuggestionSuggestionsListMixin",
+    "ConnectionMixin",
+    "DownloadedPackSourceMixin",
     "GuiMessageLineMixin",
     "GuiMessageMixin",
     "HttpTextureMixin",
-    "MinecraftMixin",
     "PlayerChatMessageMixin",
     "SkinManagerMixin"
   ],