EndlessHostRoles/Gamemodes/CaptureTheFlag.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using AmongUs.GameOptions;
using EHR.Modules;
using EHR.Roles;
using Hazel;
using UnityEngine;

namespace EHR.Gamemodes;

public static class CaptureTheFlag
{
    private static OptionItem AlertTeamMembersOfFlagTaken;
    private static OptionItem ArrowToEnemyFlagCarrier;
    private static OptionItem AlertTeamMembersOfEnemyFlagTaken;
    private static OptionItem ArrowToOwnFlagCarrier;
    private static OptionItem TaggedPlayersGet;
    private static OptionItem BackTime;
    private static OptionItem WhenFlagCarrierGetsTagged;
    private static OptionItem SpeedReductionForFlagCarrier;
    private static OptionItem TagCooldown;
    private static OptionItem SpawnProtectionTime;
    private static OptionItem GameEndCriteria;
    private static OptionItem RoundsToPlay;
    private static OptionItem PointsToWin;
    private static OptionItem TimeLimit;
    private static OptionItem FlagPickupRange;

    private static readonly string[] TaggedPlayersGetOptions =
    [
        "CTF_TaggedPlayersGet.SentBackToBase",
        "CTF_TaggedPlayersGet.Killed",
        "CTF_TaggedPlayersGet.TemporarilyOut"
    ];

    private static readonly string[] WhenFlagCarrierGetsTaggedOptions =
    [
        "CTF_WhenFlagCarrierGetsTagged.FlagIsDropped",
        "CTF_WhenFlagCarrierGetsTagged.FlagIsReturned"
    ];

    private static readonly string[] GameEndCriteriaOptions =
    [
        "CTF_GameEndCriteria.PlayXRounds",
        "CTF_GameEndCriteria.FirstToXPoints",
        "CTF_GameEndCriteria.TimeLimit"
    ];

    public static (UnityEngine.Color Color, string Team) WinnerData = (Color.white, "No one wins");

    private static Dictionary<byte, CTFTeam> PlayerTeams = [];
    private static Dictionary<CTFTeam, CTFTeamData> TeamData = [];
    private static Dictionary<byte, CTFPlayerData> PlayerData = [];
    private static Dictionary<byte, NetworkedPlayerInfo.PlayerOutfit> DefaultOutfits = [];
    private static Dictionary<byte, long> TemporarilyOutPlayers = [];
    private static Dictionary<byte, long> SpawnProtectionTimes = [];
    private static bool ValidTag;
    private static long GameStartTS;
    private static int RoundsPlayed => TeamData.Values.Sum(x => x.RoundsWon);
    public static bool IsDeathPossible => TaggedPlayersGet.GetValue() == 1;
    public static float KCD => TagCooldown.GetFloat();
    public static int GameEndCriteriaType => GameEndCriteria.GetValue();
    public static int MaxGameLength => TimeLimit.GetInt();
    public static int TotalRoundsToPlay => GameEndCriteriaType == 0 ? RoundsToPlay.GetInt() : (int)Math.Round(PointsToWin.GetInt() * 1.5f);

    public static bool IsCarrier(byte id)
    {
        if (!ValidTag || !PlayerTeams.TryGetValue(id, out CTFTeam team)) return false;
        return TeamData[team.GetOppositeTeam()].FlagCarrier == id;
    }

    /*private static (Vector2 Position, string RoomName) BlueFlagBase => Main.CurrentMap switch
    {
        MapNames.Skeld => (, Translator.GetString(SystemTypes.Nav)),
        MapNames.MiraHQ => (, Translator.GetString(SystemTypes.Launchpad)),
        MapNames.Dleks => (new(-16.5f, -4.8f), Translator.GetString(SystemTypes.Nav)),
        MapNames.Polus => (, Translator.GetString(SystemTypes.Electrical)),
        MapNames.Airship => (, Translator.GetString(SystemTypes.Cockpit)),
        MapNames.Fungle => (, Translator.GetString(SystemTypes.Kitchen)),
        (MapNames)6 => (, Translator.GetString(SystemTypes.Engine)),
        _ => (Vector2.zero, string.Empty)
    };

    private static (Vector2 Position, string RoomName) YellowFlagBase => Main.CurrentMap switch
    {
        MapNames.Skeld => (, Translator.GetString(SystemTypes.Reactor)),
        MapNames.MiraHQ => (, Translator.GetString(SystemTypes.Greenhouse)),
        MapNames.Dleks => (new(20.5f, -5.5f), Translator.GetString(SystemTypes.Reactor)),
        MapNames.Polus => (, Translator.GetString(SystemTypes.Laboratory)),
        MapNames.Airship => (, Translator.GetString(SystemTypes.CargoBay)),
        MapNames.Fungle => (, Translator.GetString(SystemTypes.Comms)),
        (MapNames)6 => (, Translator.GetString(SystemTypes.Comms)),
        _ => (Vector2.zero, string.Empty)
    };*/

    private static readonly Dictionary<MapNames, Dictionary<CTFTeam, (Vector2 Position, SystemTypes Room)>> FlagBases = new()
    {
        [MapNames.Skeld] = new()
        {
            [CTFTeam.Blue] = (new(16.5f, -4.8f), SystemTypes.Nav),
            [CTFTeam.Yellow] = (new(-20.5f, -5.5f), SystemTypes.Reactor)
        },
        [MapNames.MiraHQ] = new()
        {
            [CTFTeam.Blue] = (new(-4.5f, 2.0f), SystemTypes.Launchpad),
            [CTFTeam.Yellow] = (new(17.8f, 23.0f), SystemTypes.Greenhouse)
        },
        [MapNames.Polus] = new()
        {
            [CTFTeam.Blue] = (new(9.5f, -12.5f), SystemTypes.Electrical),
            [CTFTeam.Yellow] = (new(36.5f, -7.5f), SystemTypes.Laboratory)
        },
        [MapNames.Airship] = new()
        {
            [CTFTeam.Blue] = (new(-23.5f, -1.6f), SystemTypes.Cockpit),
            [CTFTeam.Yellow] = (new(33.5f, -1.5f), SystemTypes.CargoBay)
        },
        [MapNames.Fungle] = new()
        {
            [CTFTeam.Blue] = (new(-15.5f, -7.5f), SystemTypes.Kitchen),
            [CTFTeam.Yellow] = (new(22.2f, 13.7f), SystemTypes.Comms)
        },
        [(MapNames)6] = new()
        {
            [CTFTeam.Blue] = (new(-13.31f, -34.56f), SystemTypes.Engine),
            [CTFTeam.Yellow] = (new(12.98f, -25.68f), SystemTypes.Comms)
        },
    };

    public static void SetupCustomOption()
    {
        const int id = 69_215_001;
        Color color = new Color32(0, 165, 255, 255);

        AlertTeamMembersOfFlagTaken = new BooleanOptionItem(id, "CTF_AlertTeamMembersOfFlagTaken", true, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetColor(color);

        ArrowToEnemyFlagCarrier = new BooleanOptionItem(id + 1, "CTF_ArrowToEnemyFlagCarrier", false, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetParent(AlertTeamMembersOfFlagTaken)
            .SetColor(color);

        AlertTeamMembersOfEnemyFlagTaken = new BooleanOptionItem(id + 2, "CTF_AlertTeamMembersOfEnemyFlagTaken", false, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetColor(color);

        ArrowToOwnFlagCarrier = new BooleanOptionItem(id + 3, "CTF_ArrowToOwnFlagCarrier", false, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetParent(AlertTeamMembersOfEnemyFlagTaken)
            .SetColor(color);

        TaggedPlayersGet = new StringOptionItem(id + 4, "CTF_TaggedPlayersGet", TaggedPlayersGetOptions, 2, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetColor(color);

        BackTime = new IntegerOptionItem(id + 20, "CTF_BackTime", new(1, 30, 1), 5, TabGroup.GameSettings)
            .SetParent(TaggedPlayersGet)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetValueFormat(OptionFormat.Seconds)
            .SetColor(color);

        WhenFlagCarrierGetsTagged = new StringOptionItem(id + 5, "CTF_WhenFlagCarrierGetsTagged", WhenFlagCarrierGetsTaggedOptions, 0, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetColor(color);

        SpeedReductionForFlagCarrier = new FloatOptionItem(id + 6, "CTF_SpeedReductionForFlagCarrier", new(0f, 1f, 0.05f), 0.25f, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetValueFormat(OptionFormat.Multiplier)
            .SetColor(color);

        TagCooldown = new FloatOptionItem(id + 7, "CTF_TagCooldown", new(0f, 30f, 0.5f), 4.5f, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetValueFormat(OptionFormat.Seconds)
            .SetColor(color);

        GameEndCriteria = new StringOptionItem(id + 8, "CTF_GameEndCriteria", GameEndCriteriaOptions, 0, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetColor(color)
            .RegisterUpdateValueEvent((_, _, currentValue) =>
            {
                RoundsToPlay.SetHidden(currentValue != 0);
                PointsToWin.SetHidden(currentValue != 1);
                TimeLimit.SetHidden(currentValue != 2);
            })
            .SetRunEventOnLoad(true);

        RoundsToPlay = new IntegerOptionItem(id + 9, "CTF_RoundsToPlay", new(1, 10, 1), 3, TabGroup.GameSettings)
            .SetParent(GameEndCriteria)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetColor(color);

        PointsToWin = new IntegerOptionItem(id + 10, "CTF_PointsToWin", new(1, 10, 1), 3, TabGroup.GameSettings)
            .SetParent(GameEndCriteria)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetColor(color);

        TimeLimit = new IntegerOptionItem(id + 11, "CTF_TimeLimit", new(10, 900, 10), 300, TabGroup.GameSettings)
            .SetParent(GameEndCriteria)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetValueFormat(OptionFormat.Seconds)
            .SetColor(color);

        FlagPickupRange = new FloatOptionItem(id + 12, "CTF_FlagPickupRange", new(0.25f, 5f, 0.25f), 1.5f, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetValueFormat(OptionFormat.Multiplier)
            .SetColor(color);

        SpawnProtectionTime = new IntegerOptionItem(id + 13, "CTF_SpawnProtectionTime", new(0, 30, 1), 5, TabGroup.GameSettings)
            .SetGameMode(CustomGameMode.CaptureTheFlag)
            .SetValueFormat(OptionFormat.Seconds)
            .SetColor(color);
    }

    public static bool KnowTargetRoleColor(PlayerControl target, ref string color)
    {
        if (!ValidTag || !PlayerTeams.TryGetValue(target.PlayerId, out CTFTeam team)) return false;

        color = team switch
        {
            CTFTeam.Blue => "#0000FF",
            CTFTeam.Yellow => "#FFFF00",
            _ => color
        };

        return true;
    }

    public static string GetSuffixText(PlayerControl seer, PlayerControl target)
    {
        if (!ValidTag || seer.PlayerId != target.PlayerId) return string.Empty;

        byte id = seer.PlayerId;

        var sb = new StringBuilder(128);

        string arrows = TargetArrow.GetAllArrows(id);
        if (arrows.Length > 0)
        {
            sb.Append(arrows);
            sb.Append('\n');
        }

        (int carriedFor, int tags) = GetStatistics(target.PlayerId);
        sb.Append("<size=1.4>");
        sb.Append(string.Format(Translator.GetString("CTF_PlayerStats_CarriedFor"), carriedFor));
        sb.Append('\n');
        sb.Append(string.Format(Translator.GetString("CTF_PlayerStats_Tags"), tags));
        sb.Append("</size>\n");

        if (GameEndCriteria.GetValue() == 2)
        {
            long timeLeft = TimeLimit.GetInt() - (Utils.TimeStamp - GameStartTS) + 1;

            if (timeLeft >= 0)
            {
                sb.Append("<size=1.8><#ffffff>");
                long minutes = timeLeft / 60;
                long seconds = timeLeft % 60;

                if (minutes < 10) sb.Append('0');
                sb.Append(minutes);
                sb.Append(':');
                if (seconds < 10) sb.Append('0');
                sb.Append(seconds);
            }
            else
            {
                sb.Append("<size=1.6><#ffffff>");
                sb.Append(Translator.GetString("CTF_TimeIsUp"));
            }

            sb.Append("</color></size>\n");
        }

        if (TemporarilyOutPlayers.TryGetValue(id, out long backTS))
        {
            sb.Append(string.Format(Translator.GetString("CTF_BackIn"), backTS - Utils.TimeStamp));
            sb.Append('\n');
        }
        else if (SpawnProtectionTimes.TryGetValue(id, out long protectionEndTS))
        {
            sb.Append(string.Format(Translator.GetString("CTF_Suffix.SpawnProtectionTime"), protectionEndTS - Utils.TimeStamp));
            sb.Append('\n');
        }

        sb.Append(Utils.ColorString(Color.blue, TeamData[CTFTeam.Blue].RoundsWon.ToString()));
        sb.Append("<#ffffff> | </color>");
        sb.Append(Utils.ColorString(Color.yellow, TeamData[CTFTeam.Yellow].RoundsWon.ToString()));

        return sb.ToString();
    }

    public static (int CarriedFor, int Tags) GetStatistics(byte id)
    {
        if (!PlayerData.TryGetValue(id, out CTFPlayerData stats)) return (0, 0);
        return ((int)Math.Round(stats.FlagTime, 1), stats.TagCount);
    }

    public static int GetFlagTime(byte id)
    {
        if (!PlayerData.TryGetValue(id, out CTFPlayerData data)) return 0;
        return (int)Math.Round(data.FlagTime);
    }

    public static int GetTagCount(byte id)
    {
        if (!PlayerData.TryGetValue(id, out CTFPlayerData data)) return 0;
        return data.TagCount;
    }

    public static bool CheckForGameEnd(out GameOverReason reason)
    {
        reason = GameOverReason.ImpostorsByKill;

        if (!ValidTag) return false;

        PlayerControl[] aapc = Main.EnumerateAlivePlayerControls().Concat(ExtendedPlayerControl.TempExiled.ToValidPlayers()).ToArray();

        switch (aapc.Length)
        {
            case 0:
                ResetSkins();
                CustomWinnerHolder.ResetAndSetWinner(CustomWinner.Draw);
                CustomWinnerHolder.WinnerIds = Main.PlayerStates.Keys.ToHashSet();
                reason = GameOverReason.CrewmateDisconnect;
                return true;
            case 1:
                ResetSkins();
                TeamData[PlayerTeams[aapc[0].PlayerId]].SetAsWinner();
                return true;
            default:
                // If WinnerData is already set, end the game
                if (WinnerData.Team != "No one wins")
                {
                    ResetSkins();
                    return true;
                }

                // If all players are on the same team, end the game
                if (aapc.All(x => PlayerTeams.TryGetValue(x.PlayerId, out CTFTeam team) && team == CTFTeam.Blue) || aapc.All(x => PlayerTeams.TryGetValue(x.PlayerId, out CTFTeam team) && team == CTFTeam.Yellow))
                {
                    ResetSkins();
                    TeamData[PlayerTeams[aapc[0].PlayerId]].SetAsWinner();
                    return true;
                }

                return false;
        }

        void ResetSkins()
        {
            foreach ((byte key, NetworkedPlayerInfo.PlayerOutfit outfit) in DefaultOutfits)
            {
                PlayerControl pc = key.GetPlayer();

                if (pc && outfit != null)
                    Utils.RpcChangeSkin(pc, outfit);
            }

            DefaultOutfits = [];
        }
    }

    public static void Init()
    {
        // Reset all data
        PlayerTeams = [];
        TeamData = [];
        WinnerData = (Color.white, "No one wins");
        PlayerData = Main.PlayerStates.Keys.ToDictionary(x => x, _ => new CTFPlayerData());
        DefaultOutfits = Main.EnumeratePlayerControls().ToDictionary(x => x.PlayerId, x => x.Data.DefaultOutfit);
        TemporarilyOutPlayers = [];
        SpawnProtectionTimes = [];
        ValidTag = false;
        SendRPC();
    }

    public static IEnumerator OnGameStart()
    {
        yield return new WaitForSecondsRealtime(0.2f);
        
        Main.AllPlayerKillCooldown.SetAllValues(TagCooldown.GetFloat());

        yield return new WaitForSecondsRealtime(3f);

        if (Main.LIMap)
        {
            PlainShipRoom[] rooms = ShipStatus.Instance.AllRooms.Where(x => x).DistinctBy(x => x.RoomId).ToArray();
            PlainShipRoom[] chosen = rooms.Select(plainShipRoom =>
            {
                KeyValuePair<PlainShipRoom, float> furthestRoom = rooms.Select(x => new KeyValuePair<PlainShipRoom, float>(x, Vector2.Distance(x.transform.position, plainShipRoom.transform.position))).MaxBy(x => x.Value);
                return new KeyValuePair<PlainShipRoom[], float>([furthestRoom.Key, plainShipRoom], furthestRoom.Value);
            }).MaxBy(x => x.Value).Key;
            FlagBases[(MapNames)7] = new()
            {
                [CTFTeam.Blue] = (chosen[0].transform.position, chosen[0].RoomId),
                [CTFTeam.Yellow] = (chosen[1].transform.position, chosen[1].RoomId)
            };
        }
        
        Stopwatch stopwatch = Stopwatch.StartNew();

        // Assign players to teams
        List<PlayerControl> players = Main.EnumerateAlivePlayerControls().Shuffle();
        if (Main.GM.Value) players.RemoveAll(x => x.AmOwner);
        if (ChatCommands.Spectators.Count > 0) players.RemoveAll(x => ChatCommands.Spectators.Contains(x.PlayerId));

        int blueCount = players.Count / 2;
        HashSet<byte> bluePlayers = [];
        HashSet<byte> yellowPlayers = [];

        for (var i = 0; i < blueCount; i++)
        {
            PlayerControl player = players.FirstOrDefault(p => p.Data.DefaultOutfit.ColorId == 1) ?? players.FirstOrDefault(x => x.Data.DefaultOutfit.ColorId != 5) ?? players.RandomElement();
            players.Remove(player);
            PlayerTeams[player.PlayerId] = CTFTeam.Blue;
            bluePlayers.Add(player.PlayerId);
            player.RpcChangeColor(1);
            yield return WaitFrameIfNecessary();
        }

        foreach (PlayerControl player in players)
        {
            PlayerTeams[player.PlayerId] = CTFTeam.Yellow;
            yellowPlayers.Add(player.PlayerId);
            player.RpcChangeColor(5);
            yield return WaitFrameIfNecessary();
        }

        // Create flags
        (Vector2 Position, SystemTypes Room) blueFlagBase = CTFTeam.Blue.GetFlagBase();
        (Vector2 Position, SystemTypes Room) yellowFlagBase = CTFTeam.Yellow.GetFlagBase();

        CustomNetObject blueFlag = new BlueFlag(blueFlagBase.Position);
        CustomNetObject yellowFlag = new YellowFlag(yellowFlagBase.Position);

        // Create team data
        TeamData[CTFTeam.Blue] = new(CTFTeam.Blue, blueFlag, bluePlayers, byte.MaxValue);
        TeamData[CTFTeam.Yellow] = new(CTFTeam.Yellow, yellowFlag, yellowPlayers, byte.MaxValue);

        // Teleport players to their respective bases
        foreach (PlayerControl pc in Main.EnumerateAlivePlayerControls())
        {
            if (PlayerTeams.TryGetValue(pc.PlayerId, out CTFTeam team))
            {
                switch (team)
                {
                    case CTFTeam.Blue:
                        pc.TP(blueFlagBase.Position);
                        pc.Notify(string.Format(Translator.GetString("CTF_Notify_EnemyTeamRoom"), Translator.GetString(yellowFlagBase.Room)));
                        break;
                    case CTFTeam.Yellow:
                        pc.TP(yellowFlagBase.Position);
                        pc.Notify(string.Format(Translator.GetString("CTF_Notify_EnemyTeamRoom"), Translator.GetString(blueFlagBase.Room)));
                        break;
                }
            }

            pc.RpcChangeRoleBasis(CustomRoles.CTFPlayer);
            pc.RpcResetAbilityCooldown();

            yield return WaitFrameIfNecessary();
        }
        
        var sender = CustomRpcSender.Create("CTF Set Teams");
        sender.StartPackedMessage();
        
        foreach (CTFTeamData data in TeamData.Values)
        {
            foreach (byte id1 in data.Players)
            {
                try
                {
                    var pc1 = id1.GetPlayer();
                    if (!pc1 || pc1.AmOwner || pc1.OwnerId < 0) continue;

                    sender.StartMessage(pc1.OwnerId);

                    foreach (byte id2 in data.Players)
                    {
                        try
                        {
                            if (id1 == id2) continue;

                            var pc2 = id2.GetPlayer();
                            if (!pc2) continue;

                            sender.StartRpc(pc2.NetId, RpcCalls.SetRole)
                                .Write((ushort)RoleTypes.Phantom)
                                .Write(true)
                                .EndRpc();
                        }
                        catch (Exception e) { Utils.ThrowException(e); }
                    }

                    sender.EndMessage();
                }
                catch (Exception e) { Utils.ThrowException(e); }
                
                yield return WaitFrameIfNecessary();
            }
        }
        
        sender.SendMessage(dispose: PlayerControl.AllPlayerControls.Count <= 1);

        ValidTag = true;
        GameStartTS = Utils.TimeStamp;
        
        yield break;

        IEnumerator WaitFrameIfNecessary()
        {
            if (stopwatch.ElapsedMilliseconds >= 5)
            {
                stopwatch.Reset();
                yield return null;
                stopwatch.Start();
            }
        }
    }

    private static void Restart()
    {
        Logger.Info("Restarting Capture The Flag game", "CTF");
        
        foreach ((CTFTeam team, CTFTeamData data) in TeamData)
        {
            Vector2 flagBase = team.GetFlagBase().Position;
            data.DropFlag();
            data.Flag.TP(flagBase);
            data.Players.ToValidPlayers().MassTP(flagBase);
        }
        
        Main.EnumeratePlayerControls().Do(x => TargetArrow.RemoveAllTarget(x.PlayerId));
    }

    public static void OnCheckMurder(PlayerControl killer, PlayerControl target)
    {
        if (!ValidTag || TemporarilyOutPlayers.ContainsKey(killer.PlayerId) || SpawnProtectionTimes.ContainsKey(target.PlayerId) || !PlayerTeams.TryGetValue(target.PlayerId, out CTFTeam targetTeam) || !PlayerTeams.TryGetValue(killer.PlayerId, out CTFTeam killerTeam) || killerTeam == targetTeam || TeamData.Values.Any(x => x.FlagCarrier == killer.PlayerId)) return;

        new[] { killer, target }.Do(x => x.SetKillCooldownNonSync(KCD));

        if (TeamData.FindFirst(x => x.Value.FlagCarrier == target.PlayerId, out KeyValuePair<CTFTeam, CTFTeamData> kvp))
        {
            kvp.Value.DropFlag();
            if (WhenFlagCarrierGetsTagged.GetValue() == 1) kvp.Value.Flag.TP(kvp.Key.GetFlagBase().Position);
        }

        switch (TaggedPlayersGet.GetValue())
        {
            case 0:
                target.TP(targetTeam.GetFlagBase().Position);
                Main.AllPlayerSpeed[target.PlayerId] = Main.RealOptionsData.GetFloat(FloatOptionNames.PlayerSpeedMod);
                target.MarkDirtySettings();
                SpawnProtectionTimes[target.PlayerId] = Utils.TimeStamp + SpawnProtectionTime.GetInt();
                break;
            case 1:
                target.Suicide(PlayerState.DeathReason.Kill);
                string notify = string.Format(Translator.GetString("CTF_TeamMemberFallen"), target.PlayerId.ColoredPlayerName());
                TeamData[targetTeam].Players.ToValidPlayers().NotifyPlayers(notify);
                if (Main.GM.Value && AmongUsClient.Instance.AmHost) PlayerControl.LocalPlayer.KillFlash();
                ChatCommands.Spectators.ToValidPlayers().Do(x => x.KillFlash());
                break;
            case 2:
                TemporarilyOutPlayers[target.PlayerId] = Utils.TimeStamp + BackTime.GetInt();
                target.ExileTemporarily();
                break;
        }

        if (PlayerData.TryGetValue(killer.PlayerId, out CTFPlayerData data))
        {
            data.TagCount++;
            Utils.NotifyRoles(SpecifySeer: killer, SpecifyTarget: killer, SendOption: SendOption.None);
        }
    }

    public static void TryPickUpFlag(PlayerControl pc)
    {
        if (!ValidTag) return;

        Logger.Info($"Received flag pickup request from {pc.GetRealName()}", "CTF");
        // If the player is near the enemy's flag, pick it up
        Vector2 pos = pc.Pos();
        CTFTeamData enemy = TeamData[PlayerTeams[pc.PlayerId].GetOppositeTeam()];
        if (enemy.IsNearFlag(pos)) enemy.PickUpFlag(pc.PlayerId);
    }

    public static void ApplyGameOptions()
    {
        AURoleOptions.PhantomCooldown = 5f;

        try { AURoleOptions.GuardianAngelCooldown = 900f; }
        catch (Exception e) { Utils.ThrowException(e); }
    }

    public static bool IsNotInLocalPlayersTeam(PlayerControl pc)
    {
        return pc.IsValidTargetForKillButton() && (!PlayerTeams.TryGetValue(pc.PlayerId, out CTFTeam team) || !PlayerTeams.TryGetValue(PlayerControl.LocalPlayer.PlayerId, out CTFTeam lpTeam) || team != lpTeam);
    }

    private static void SendRPC()
    {
        var w = Utils.CreateRPC(CustomRPC.CTFSync);

        w.Write(WinnerData.Color);
        w.Write(WinnerData.Team);

        w.Write(PlayerData.Count);

        foreach ((byte id, CTFPlayerData data) in PlayerData)
        {
            w.Write(id);
            w.Write(data.FlagTime);
            w.Write(data.TagCount);
        }

        Utils.EndRPC(w);
    }

    public static void ReceiveRPC(MessageReader reader)
    {
        WinnerData.Color = reader.ReadColor();
        WinnerData.Team = reader.ReadString();

        int count = reader.ReadInt32();

        for (int i = 0; i < count; i++)
        {
            byte id = reader.ReadByte();

            if (!PlayerData.TryGetValue(id, out CTFPlayerData data))
                PlayerData[id] = data = new CTFPlayerData();

            data.FlagTime = reader.ReadSingle();
            data.TagCount = reader.ReadInt32();
        }
    }

    extension(CTFTeam team)
    {
        private Color GetTeamColor()
        {
            return team switch
            {
                CTFTeam.Blue => Color.blue,
                CTFTeam.Yellow => Color.yellow,
                _ => Color.white
            };
        }

        private string GetTeamName()
        {
            return team switch
            {
                CTFTeam.Blue => Translator.GetString("CTF_BlueTeamWins"),
                CTFTeam.Yellow => Translator.GetString("CTF_YellowTeamWins"),
                _ => string.Empty
            };
        }

        private CTFTeam GetOppositeTeam()
        {
            return team switch
            {
                CTFTeam.Blue => CTFTeam.Yellow,
                _ => CTFTeam.Blue
            };
        }

        private (Vector2 Position, SystemTypes Room) GetFlagBase()
        {
            return FlagBases[Main.CurrentMap][team];
        }
    }

    private enum CTFTeam
    {
        Blue,
        Yellow
    }

    private class CTFTeamData(CTFTeam team, CustomNetObject flag, HashSet<byte> players, byte flagCarrier)
    {
        public CustomNetObject Flag { get; } = flag;
        public HashSet<byte> Players { get; } = players;
        public byte FlagCarrier { get; private set; } = flagCarrier;
        public int RoundsWon { get; private set; }

        public void SetAsWinner()
        {
            WinnerData = (team.GetTeamColor(), team.GetTeamName());
            CustomWinnerHolder.WinnerIds = Players;
            Logger.Info($"{team} team wins", "CTF");
            SendRPC();
        }

        public void Update()
        {
            try
            {
                if (FlagCarrier == byte.MaxValue) return;

                PlayerControl flagCarrierPc = FlagCarrier.GetPlayer();

                if (!flagCarrierPc || !flagCarrierPc.IsAlive())
                {
                    DropFlag();
                    return;
                }

                Flag.TP(flagCarrierPc.Pos());
                if (PlayerData.TryGetValue(FlagCarrier, out CTFPlayerData data)) data.FlagTime += Time.fixedDeltaTime;

                CTFTeam enemy = team.GetOppositeTeam();

                if (FastVector2.DistanceWithinRange(Flag.Position, enemy.GetFlagBase().Position, 2f))
                {
                    Main.EnumeratePlayerControls().NotifyPlayers(Translator.GetString($"CTF_{enemy}TeamWonThisRound"));
                    CTFTeamData enemyTeam = TeamData[enemy];

                    if (++enemyTeam.RoundsWon >= PointsToWin.GetInt() && GameEndCriteria.GetValue() == 1)
                    {
                        enemyTeam.SetAsWinner();
                        return;
                    }

                    if (GameEndCriteria.GetValue() == 0 && RoundsPlayed >= RoundsToPlay.GetInt() && RoundsWon != enemyTeam.RoundsWon)
                    {
                        CTFTeamData winner = TeamData.Values.MaxBy(x => x.RoundsWon);
                        winner.SetAsWinner();
                        return;
                    }

                    
                    Restart();
                }
            }
            catch { }
        }

        public void PickUpFlag(byte id)
        {
            if (FlagCarrier == id) return;

            FlagCarrier = id;
            Update();

            Logger.Info($"{id.ColoredPlayerName().RemoveHtmlTags()} picked up the {team} flag", "CTF");

            Main.AllPlayerSpeed[id] = Main.RealOptionsData.GetFloat(FloatOptionNames.PlayerSpeedMod) - SpeedReductionForFlagCarrier.GetFloat();
            PlayerGameOptionsSender.SetDirty(id);

            if (AlertTeamMembersOfFlagTaken.GetBool())
            {
                bool arrow = ArrowToEnemyFlagCarrier.GetBool();

                TeamData[team].Players
                    .ToValidPlayers()
                    .Do(x =>
                    {
                        if (arrow) TargetArrow.Add(x.PlayerId, id);
                        x.Notify(Utils.ColorString(Color.yellow, Translator.GetString("CTF_FlagTaken")));
                    });
            }

            if (AlertTeamMembersOfEnemyFlagTaken.GetBool())
            {
                bool arrow = ArrowToOwnFlagCarrier.GetBool();

                TeamData[team.GetOppositeTeam()].Players
                    .ToValidPlayers()
                    .Do(x =>
                    {
                        if (arrow) TargetArrow.Add(x.PlayerId, id);
                        x.Notify(Translator.GetString("CTF_EnemyFlagTaken"));
                    });
            }
        }

        public void DropFlag()
        {
            Logger.Info($"{FlagCarrier.ColoredPlayerName().RemoveHtmlTags()} dropped the {team} flag", "CTF");

            if (FlagCarrier != byte.MaxValue)
            {
                byte id = FlagCarrier;
                FlagCarrier = byte.MaxValue;

                Main.AllPlayerSpeed[id] = Main.RealOptionsData.GetFloat(FloatOptionNames.PlayerSpeedMod);
                PlayerGameOptionsSender.SetDirty(id);

                if (ArrowToEnemyFlagCarrier.GetBool() || ArrowToOwnFlagCarrier.GetBool())
                    TeamData.Values.SelectMany(x => x.Players).Do(x => TargetArrow.Remove(x, id));
                
                Utils.NotifyRoles(SendOption: SendOption.None);
            }
        }

        public bool IsNearFlag(Vector2 pos)
        {
            return FastVector2.DistanceWithinRange(Flag.Position, pos, FlagPickupRange.GetFloat());
        }
    }

    private class CTFPlayerData
    {
        public int TagCount { get; set; }
        public float FlagTime { get; set; }
    }

    //[HarmonyPatch(typeof(PlayerControl), nameof(PlayerControl.FixedUpdate))]
    public static class FixedUpdatePatch
    {
        private static long LastUpdate;
        
        public static void Postfix(PlayerControl __instance)
        {
            if (!AmongUsClient.Instance.AmHost || !GameStates.IsInTask || ExileController.Instance || Options.CurrentGameMode != CustomGameMode.CaptureTheFlag || !Main.IntroDestroyed || __instance.PlayerId >= 254 || WinnerData.Team != "No one wins") return;
            
            long now = Utils.TimeStamp;
            if (IntroCutsceneDestroyPatch.IntroDestroyTS + 5 > now) return;

            bool notified = false;
            
            if (__instance.AmOwner)
            {
                TeamData.Values.Do(x => x.Update());

                if (now != LastUpdate && GameEndCriteria.GetValue() == 2)
                {
                    LastUpdate = now;
                    long timeLeft = TimeLimit.GetInt() - (now - GameStartTS) + 1;

                    switch (timeLeft)
                    {
                        case <= 1 when TeamData.Count == 2 && TeamData[CTFTeam.Blue].RoundsWon != TeamData[CTFTeam.Yellow].RoundsWon:
                        {
                            CTFTeamData winner = TeamData.Values.MaxBy(x => x.RoundsWon);
                            winner.SetAsWinner();
                            return;
                        }
                        case >= -1:
                        {
                            Utils.NotifyRoles(SendOption: SendOption.None);
                            notified = true;
                            break;
                        }
                    }
                }
            }

            if (!PlayerTeams.TryGetValue(__instance.PlayerId, out CTFTeam team)) return;
            bool blue = team == CTFTeam.Blue;
            int colorId = blue ? 1 : 5;

            if (__instance.CurrentOutfit.ColorId != colorId)
            {
                __instance.SetColor(colorId);

                CustomRpcSender.Create("Color")
                    .AutoStartRpc(__instance.NetId, 8)
                    .Write(__instance.Data.NetId)
                    .Write((byte)colorId)
                    .EndRpc()
                    .SendMessage();
            }

            bool notify = false;

            if (TemporarilyOutPlayers.TryGetValue(__instance.PlayerId, out long endTS))
            {
                if (now >= endTS)
                {
                    Main.AllPlayerSpeed[__instance.PlayerId] = Main.RealOptionsData.GetFloat(FloatOptionNames.PlayerSpeedMod);
                    TemporarilyOutPlayers.Remove(__instance.PlayerId);
                    __instance.ReviveFromTemporaryExile();
                    __instance.TP(team.GetFlagBase().Position);
                    RPC.PlaySoundRPC(__instance.PlayerId, Sounds.SpawnSound);
                    SpawnProtectionTimes[__instance.PlayerId] = Utils.TimeStamp + SpawnProtectionTime.GetInt();
                }
                
                notify = true;
            }

            if (SpawnProtectionTimes.TryGetValue(__instance.PlayerId, out long protectionEndTS))
            {
                if (now >= protectionEndTS)
                    SpawnProtectionTimes.Remove(__instance.PlayerId);
                
                notify = true;
            }
            
            if (!notified && notify && PerSecondUpdateScheduler.ShouldRunUpdate(__instance.PlayerId))
                Utils.NotifyRoles(SpecifySeer: __instance, SpecifyTarget: __instance, SendOption: SendOption.None);
        }
    }
}

public class CTFPlayer : RoleBase
{
    public static bool On;

    public override bool IsEnable => On;

    public override void SetupCustomOption() { }

    public override void Init()
    {
        On = false;
    }

    public override void Add(byte playerId)
    {
        On = true;
    }

    public override bool OnVanish(PlayerControl pc)
    {
        CaptureTheFlag.TryPickUpFlag(pc);
        return false;
    }
}