From 618d6d48b0fbf3009e0122c550b7ec4c668e4398 Mon Sep 17 00:00:00 2001
From: legoandmars <legoandmars@gmail.com>
Date: Sun, 18 Feb 2024 19:04:43 -0800
Subject: [PATCH] Remove modded indicator from displayed lobby names, while
 retaining as much data as possible

---
 LobbyCompatibility/Enums/LobbyMetadata.cs     |  5 ++++
 .../LoadLobbyListAndFilterTranspiler.cs       | 28 +++++++++++++++++--
 .../SteamMatchmakingOnLobbyCreatedPostfix.cs  |  2 +-
 3 files changed, 32 insertions(+), 3 deletions(-)

diff --git a/LobbyCompatibility/Enums/LobbyMetadata.cs b/LobbyCompatibility/Enums/LobbyMetadata.cs
index b3136ab..640a8e6 100644
--- a/LobbyCompatibility/Enums/LobbyMetadata.cs
+++ b/LobbyCompatibility/Enums/LobbyMetadata.cs
@@ -51,4 +51,9 @@ public static class LobbyMetadata
     ///     The required plugins checksum to filter against.
     /// </summary>
     public const string RequiredChecksum = "checksum";
+
+    /// <summary>
+    ///     The prefix added before modded lobbies.
+    /// </summary>
+    public const string ModdedLobbyPrefix = "[MOD]";
 }
\ No newline at end of file
diff --git a/LobbyCompatibility/Patches/LoadLobbyListAndFilterTranspiler.cs b/LobbyCompatibility/Patches/LoadLobbyListAndFilterTranspiler.cs
index 6c9c78d..1cc4204 100644
--- a/LobbyCompatibility/Patches/LoadLobbyListAndFilterTranspiler.cs
+++ b/LobbyCompatibility/Patches/LoadLobbyListAndFilterTranspiler.cs
@@ -2,6 +2,7 @@
 using System.Reflection.Emit;
 using HarmonyLib;
 using LobbyCompatibility.Behaviours;
+using LobbyCompatibility.Enums;
 using Steamworks.Data;
 using UnityEngine;
 
@@ -30,14 +31,27 @@ private static IEnumerable<CodeInstruction> Transpiler(IEnumerable<CodeInstructi
         
         var levelListContainerField =
             AccessTools.Field(typeof(SteamLobbyManager), nameof(SteamLobbyManager.levelListContainer));
-        var initializeLobbySlotMethod = 
+        var initializeLobbySlotMethod =
             AccessTools.Method(typeof(LoadLobbyListAndFilterTranspiler), nameof(InitializeLobbySlot));
-        
+
+        var lobbyGetDataMethod =
+            AccessTools.Method(typeof(Lobby), nameof(Lobby.GetData));
+        var replaceLobbyNameMethod =
+            AccessTools.Method(typeof(LoadLobbyListAndFilterTranspiler), nameof(ReplaceLobbyName));
+
         // Does the following:
+        // - Calls ReplaceLobbyName(lobbyName) the line before lobbyName local variable is set
         // - Adds dup before last componentInChildren line to keep componentInChildren value on the stack
         // - Loads SteamLobbyManager.levelListContainer onto the stack
         // - Calls InitializeLobbySlot(lobbySlot, levelListContainer)
         return new CodeMatcher(instructions)
+            .MatchForward(false, new[] {
+                new CodeMatch(OpCodes.Ldstr, "name"),
+                new CodeMatch(OpCodes.Call, lobbyGetDataMethod) })
+            .ThrowIfNotMatch("Unable to find Lobby.GetData(name) line.")
+            .Advance(2)
+            .InsertAndAdvance(new[] {
+                new CodeInstruction(OpCodes.Call, replaceLobbyNameMethod)})
             .MatchForward(false, new [] {
                 new CodeMatch(OpCodes.Ldloc_1),
                 new CodeMatch(OpCodes.Ldfld, currentLobbyListField),
@@ -66,4 +80,14 @@ private static void InitializeLobbySlot(LobbySlot lobbySlot, Transform levelList
         // Set container parent for hover tooltip position math
         moddedLobbySlot.SetParentContainer(levelListContainer.parent);
     }
+
+    // Replace any modded text indicators as they're redundant
+    // This is run before the string is truncated to 40 characters
+    private static string ReplaceLobbyName(string lobbyName)
+    {
+        if (lobbyName.Length == 0)
+            return lobbyName;
+
+        return lobbyName.Replace(LobbyMetadata.ModdedLobbyPrefix, "");
+    }
 }
\ No newline at end of file
diff --git a/LobbyCompatibility/Patches/SteamMatchmakingOnLobbyCreatedPostfix.cs b/LobbyCompatibility/Patches/SteamMatchmakingOnLobbyCreatedPostfix.cs
index 736c138..5bf6bcc 100644
--- a/LobbyCompatibility/Patches/SteamMatchmakingOnLobbyCreatedPostfix.cs
+++ b/LobbyCompatibility/Patches/SteamMatchmakingOnLobbyCreatedPostfix.cs
@@ -45,7 +45,7 @@ private static void Postfix(Result result, ref Lobby lobby)
         if (pluginInfo.Any(plugin => plugin.CompatibilityLevel is CompatibilityLevel.ServerOnly
                 or CompatibilityLevel.Everyone or CompatibilityLevel.ClientOptional) &&
             !lobby.GetData(LobbyMetadata.Name).ToLower().Contains("modded"))
-            lobby.SetData(LobbyMetadata.Name, "modded // " + lobby.GetData(LobbyMetadata.Name));
+            lobby.SetData(LobbyMetadata.Name, LobbyMetadata.ModdedLobbyPrefix + lobby.GetData(LobbyMetadata.Name));
 
         // Check if there are any required plugins in the lobby
         if (pluginInfo.Any(plugin => plugin.CompatibilityLevel == CompatibilityLevel.Everyone))