CelesteNet: display the lobby icon instead of the map icon in the player list when playing a collab

Author: maddie480

CollabUtils2.csproj

@@ -15,6 +15,17 @@
     </PackageReference>
   </ItemGroup>
 
+  <ItemGroup>
+    <Reference Include="CelesteNet.Client">
+      <HintPath>lib-stripped\CelesteNet.Client.dll</HintPath>
+      <Private>false</Private>
+    </Reference>
+    <Reference Include="CelesteNet.Shared">
+      <HintPath>lib-stripped\CelesteNet.Shared.dll</HintPath>
+      <Private>false</Private>
+    </Reference>
+  </ItemGroup>
+
   <Choose>
     <When Condition="Exists('..\..\Celeste.exe')">
       <ItemGroup>

LobbyHelper.cs

@@ -1,3 +1,4 @@
+using Celeste.Mod.CelesteNet.Client.Components;
 using Celeste.Mod.CollabUtils2.UI;
 using Celeste.Mod.UI;
 using Mono.Cecil;
@@ -35,6 +36,10 @@ internal static void OnInitialize() {
             ) {
                 LoadCollabIDFile(asset);
             }
+
+            if (Everest.Loader.DependencyLoaded(new EverestModuleMetadata() { Name = "CelesteNet.Client", Version = new Version(2, 0, 0) })) {
+                adjustCollabIcon();
+            }
         }
 
         public static void LoadCollabIDFile(ModAsset asset) {
@@ -169,6 +174,19 @@ internal static void Unload() {
             hookOnOuiFileSelectSlotRender?.Dispose();
         }
 
+        private static void adjustCollabIcon() {
+            CelesteNetPlayerListComponent.OnGetState += (blob, state) => {
+                // if we are in a collab map, change the icon displayed in the CelesteNet player list to the lobby icon.
+                AreaData data = AreaData.Get(state.SID);
+                if (data != null) {
+                    string lobbySID = GetLobbyForLevelSet(data.LevelSet);
+                    if (lobbySID != null) {
+                        blob.Location.Icon = AreaData.Get(lobbySID)?.Icon ?? blob.Location.Icon;
+                    }
+                }
+            };
+        }
+
         public static void OnSessionCreated() {
             Session session = SaveData.Instance.CurrentSession_Safe;
             string levelSet = GetLobbyLevelSet(session.Area.GetSID());

everest.yaml

@@ -1,5 +1,5 @@
 - Name: CollabUtils2
-  Version: 1.6.1
+  Version: 1.6.2
   DLL: bin/Debug/net452/CollabUtils2.dll
   Dependencies:
     - Name: Everest