modifier didwins

MiraAPI/Modifiers/Types/GameModifier.cs

@@ -23,4 +23,10 @@ public abstract class GameModifier : BaseModifier
     /// <param name="role">The role to be checked.</param>
     /// <returns>True if the modifier is valid on the role, otherwise false.</returns>
     public virtual bool IsModifierValidOn(RoleBehaviour role) => true;
+
+    /// <summary>
+    /// Determines whether the player won the game with this modifier.
+    /// </summary>
+    /// <returns>True if the player won, false if they lost. Return null to use the player's default win condition.</returns>
+    public virtual bool? DidWin(GameOverReason reason) => null;
 }

MiraAPI/Patches/Modifiers/EndGameDidWinPatch.cs

@@ -0,0 +1,40 @@
+using System.Linq;
+using HarmonyLib;
+using MiraAPI.Modifiers.Types;
+using MiraAPI.Utilities;
+
+namespace MiraAPI.Patches.Modifiers;
+
+[HarmonyPatch(typeof(AmongUsClient), nameof(AmongUsClient.CoEndGame))]
+public static class EndGameDidWinPatch
+{
+    public static void Prefix()
+    {
+        var gameOverReason = EndGameResult.CachedGameOverReason;
+        EndGameResult.CachedWinners.Clear();
+
+        var players = GameData.Instance.AllPlayers.ToArray();
+        for (var i = 0; i < GameData.Instance.PlayerCount; i++)
+        {
+            var networkedPlayerInfo = players[i];
+            if (networkedPlayerInfo == null)
+            {
+                continue;
+            }
+
+            var didWin = networkedPlayerInfo.Role.DidWin(gameOverReason);
+
+            var modifierWin = networkedPlayerInfo.Object?.GetModifierComponent()?.ActiveModifiers.OfType<GameModifier>()
+                .FirstOrDefault(x => x.DidWin(gameOverReason) != null)?.DidWin(gameOverReason);
+            if (modifierWin != null)
+            {
+                didWin = modifierWin.Value;
+            }
+
+            if (didWin)
+            {
+                EndGameResult.CachedWinners.Add(new CachedPlayerData(networkedPlayerInfo));
+            }
+        }
+    }
+}