From dd6d57343e096045e8c99af4898aa41e142855ad Mon Sep 17 00:00:00 2001
From: ang-xd <angxdxdd@gmail.com>
Date: Sat, 14 Sep 2024 15:47:02 -0400
Subject: [PATCH] Role singleton

---
 MiraAPI.Example/Roles/ChameleonRole.cs |  1 -
 MiraAPI/Roles/CustomRoleSingleton.cs   | 19 ++++++++++++
 MiraAPI/Roles/ICustomRole.cs           | 41 ++++++++++++++++++++------
 MiraAPI/Roles/RoleHintType.cs          |  2 +-
 4 files changed, 52 insertions(+), 11 deletions(-)
 create mode 100644 MiraAPI/Roles/CustomRoleSingleton.cs

diff --git a/MiraAPI.Example/Roles/ChameleonRole.cs b/MiraAPI.Example/Roles/ChameleonRole.cs
index 13db85a..e3eb6c4 100644
--- a/MiraAPI.Example/Roles/ChameleonRole.cs
+++ b/MiraAPI.Example/Roles/ChameleonRole.cs
@@ -15,7 +15,6 @@ public class ChameloenRole : CrewmateRole, ICustomRole
 
     public CustomRoleConfiguration Configuration => new CustomRoleConfiguration(this)
     {
-        MaxRoleCount = 0,
         OptionsScreenshot = ExampleAssets.Banner,
     };
 
diff --git a/MiraAPI/Roles/CustomRoleSingleton.cs b/MiraAPI/Roles/CustomRoleSingleton.cs
new file mode 100644
index 0000000..808202c
--- /dev/null
+++ b/MiraAPI/Roles/CustomRoleSingleton.cs
@@ -0,0 +1,19 @@
+using System.Linq;
+
+namespace MiraAPI.Roles;
+
+/// <summary>
+/// A utility class to get the instance of a custom role.
+/// </summary>
+/// <typeparam name="T">The role you are trying to access.</typeparam>
+public static class CustomRoleSingleton<T> where T : ICustomRole
+{
+    private static ICustomRole? _instance;
+
+    /// <summary>
+    /// Gets the instance of the role.
+    /// </summary>
+#pragma warning disable CA1000
+    public static ICustomRole Instance => _instance ??= CustomRoleManager.CustomRoles.Values.OfType<T>().Single();
+#pragma warning restore CA1000
+}
diff --git a/MiraAPI/Roles/ICustomRole.cs b/MiraAPI/Roles/ICustomRole.cs
index 53b7798..b7adf66 100644
--- a/MiraAPI/Roles/ICustomRole.cs
+++ b/MiraAPI/Roles/ICustomRole.cs
@@ -37,17 +37,10 @@ public interface ICustomRole
     /// </summary>
     ModdedRoleTeams Team { get; }
 
-    CustomRoleConfiguration Configuration { get; }
-
-    /// <summary>
-    /// Gets the BepInEx ConfigDefinition for the amount of players that can have this role.
-    /// </summary>
-    ConfigDefinition NumConfigDefinition => new("Roles", $"Num {GetType().FullName}");
-
     /// <summary>
-    /// Gets the BepInEx ConfigDefinition for the chance of this role being selected.
+    /// Configure advanced settings of the role.
     /// </summary>
-    ConfigDefinition ChanceConfigDefinition => new("Roles", $"Chance {GetType().FullName}");
+    CustomRoleConfiguration Configuration { get; }
 
     /// <summary>
     /// Gets the parent mod of this role.
@@ -62,6 +55,36 @@ void PlayerControlFixedUpdate(PlayerControl playerControl)
     {
     }
 
+    internal ConfigDefinition NumConfigDefinition => new("Roles", $"Num {GetType().FullName}");
+    internal ConfigDefinition ChanceConfigDefinition => new("Roles", $"Chance {GetType().FullName}");
+
+    /// <summary>
+    /// Get the role chance option.
+    /// </summary>
+    /// <returns>The role chance option.</returns>
+    public int? GetChance()
+    {
+        if (ParentMod.PluginConfig.TryGetEntry(ChanceConfigDefinition, out ConfigEntry<int> entry))
+        {
+            return Mathf.Clamp(entry.Value, 0, 100);
+        }
+
+        return null;
+    }
+
+    /// <summary>
+    /// Get the role count option.
+    /// </summary>
+    /// <returns>The role count option.</returns>
+    public int? GetCount()
+    {
+        if (ParentMod.PluginConfig.TryGetEntry(NumConfigDefinition, out ConfigEntry<int> entry))
+        {
+            return Mathf.Clamp(entry.Value, 0, Configuration.MaxRoleCount);
+        }
+
+        return null;
+    }
     /// <summary>
     /// This method runs on the HudManager.Update method ONLY when the LOCAL player has this role.
     /// </summary>
diff --git a/MiraAPI/Roles/RoleHintType.cs b/MiraAPI/Roles/RoleHintType.cs
index f3d5cc6..d00c5da 100644
--- a/MiraAPI/Roles/RoleHintType.cs
+++ b/MiraAPI/Roles/RoleHintType.cs
@@ -16,7 +16,7 @@ public enum RoleHintType
     TaskHint,
 
     /// <summary>
-    /// Use custom role tab.
+    /// Use Mira's custom role tab.
     /// </summary>
     RoleTab,
 }