add custom lobby map rendering + make dash count methods public (will require update to dashcountmod)

Author: aonkeeper4

UI/JournalHelper.cs

@@ -6,7 +6,7 @@
 
 namespace Celeste.Mod.CollabUtils2.UI {
     public static class JournalHelper {
-        private static Dictionary<string, Action<OuiJournal, string, bool>> JournalEditors = new Dictionary<string, Action<OuiJournal, string, bool>>();
+        private static readonly Dictionary<string, Action<OuiJournal, string, bool>> JournalEditors = new Dictionary<string, Action<OuiJournal, string, bool>>();
 
         internal static bool VanillaJournal = true; // default to vanilla journal
         internal static bool ShowOnlyDiscovered = false;
@@ -70,7 +70,6 @@ private static void OnJournalEnter(OuiJournal journal, Oui from) {
         }
 
         // ModInterop exports
-
         [ModExportName("CollabUtils2.JournalHelper")]
         private static class ModExports {
             public static void AddJournalEditor(string collabID, Action<OuiJournal, string, bool> editor) {

UI/LobbyMapUI.cs

@@ -3,6 +3,7 @@
 using Microsoft.Xna.Framework;
 using Microsoft.Xna.Framework.Graphics;
 using Monocle;
+using MonoMod.ModInterop;
 using System;
 using System.Collections;
 using System.Collections.Generic;
@@ -81,6 +82,19 @@ public class LobbyMapUI : Entity {
         private int lastSelectedWarpIndex = -1;
         private float scaleMultiplier = 1f;
         private float finalScale => actualScale * scaleMultiplier;
+        
+        // mod interop
+        private static readonly Dictionary<string, Action<Entity, List<Component>>> CustomRenderActions = new Dictionary<string, Action<Entity, List<Component>>>();
+        public static void AddCustomRenderAction(string collabID, Action<Entity, List<Component>> action) {
+            if (CustomRenderActions.TryGetValue(collabID, out _))
+                CustomRenderActions[collabID] = action;
+            else
+                CustomRenderActions.Add(collabID, action);
+        }
+        public static void RemoveCustomRenderAction(string collabID) {
+            if (CustomRenderActions.TryGetValue(collabID, out _))
+                CustomRenderActions.Remove(collabID);
+        }
 
         private Rectangle windowBounds;
         private Rectangle mapBounds;
@@ -701,6 +715,9 @@ public override void Render() {
             if (CollabModule.Instance.SaveData.ShowVisitedPoints) {
                 drawVisitedPoints();
             }
+            
+            if (CustomRenderActions.TryGetValue(LobbyHelper.GetCollabNameForSID(lobbyMapInfo.LevelSet), out Action<Entity, List<Component>> customRenderAction))
+                customRenderAction(this, markerComponents);
 
             drawForeground();
         }
@@ -1204,5 +1221,20 @@ public LobbySelection(EntityData data, MapData map) {
         }
 
         #endregion
+        
+        #region ModInterop
+        
+        // ModInterop exports
+        [ModExportName("CollabUtils2.JournalHelper")]
+        private static class ModExports {
+            public static void AddCustomRenderAction(string collabID, Action<Entity, List<Component>> editor) {
+                LobbyMapUI.AddCustomRenderAction(collabID, editor);
+            }
+            public static void RemoveCustomRenderAction(string collabID) {
+                LobbyMapUI.RemoveCustomRenderAction(collabID);
+            }
+        }
+        
+        #endregion
     }
 }

UI/OuiJournalCollabProgressDashCountMod.cs

@@ -8,19 +8,19 @@ namespace Celeste.Mod.CollabUtils2.UI {
     /// </summary>
     public static class OuiJournalCollabProgressDashCountMod {
         [MethodImpl(MethodImplOptions.NoInlining)]
-        internal static bool IsDashCountEnabled() {
+        public static bool IsDashCountEnabled() {
             return false;
         }
 
         // those depend on Dash Count Mod settings / save data, and will be implemented by Dash Count Mod itself.
 
         [MethodImpl(MethodImplOptions.NoInlining)]
-        internal static bool DisplaysTotalDashes() {
+        public static bool DisplaysTotalDashes() {
             return false;
         }
 
         [MethodImpl(MethodImplOptions.NoInlining)]
-        internal static int GetLevelDashesForJournalProgress(AreaStats stats) {
+        public static int GetLevelDashesForJournalProgress(AreaStats stats) {
             return stats.BestTotalDashes;
         }
     }