EndlessHostRoles/Roles/Standard/Neutral/Evil/Workaholic.cs

using System.Collections.Generic;
using System.Linq;
using AmongUs.GameOptions;
using EHR.Modules;
using static EHR.Options;

// From: TOH_Y
namespace EHR.Roles;

internal class Workaholic : RoleBase
{
    public static bool On;

    public static List<byte> WorkaholicAlive = [];

    public static OptionItem WorkaholicVentCooldown;
    public static OptionItem WorkaholicCannotWinAtDeath;
    public static OptionItem WorkaholicVisibleToEveryone;
    public static OptionItem WorkaholicGiveAdviceAlive;
    public static OptionItem WorkaholicCanGuess;
    public static OptionItem WorkaholicSpeed;
    public override bool IsEnable => On;

    public override void SetupCustomOption()
    {
        SetupRoleOptions(11700, TabGroup.NeutralRoles, CustomRoles.Workaholic);

        WorkaholicCannotWinAtDeath = new BooleanOptionItem(11710, "WorkaholicCannotWinAtDeath", true, TabGroup.NeutralRoles)
            .SetParent(CustomRoleSpawnChances[CustomRoles.Workaholic]);

        WorkaholicVentCooldown = new FloatOptionItem(11711, "VentCooldown", new(0f, 180f, 0.5f), 30f, TabGroup.NeutralRoles)
            .SetParent(CustomRoleSpawnChances[CustomRoles.Workaholic])
            .SetValueFormat(OptionFormat.Seconds);

        WorkaholicVisibleToEveryone = new BooleanOptionItem(11712, "WorkaholicVisibleToEveryone", false, TabGroup.NeutralRoles)
            .SetParent(CustomRoleSpawnChances[CustomRoles.Workaholic]);

        WorkaholicGiveAdviceAlive = new BooleanOptionItem(11713, "WorkaholicGiveAdviceAlive", false, TabGroup.NeutralRoles)
            .SetParent(WorkaholicVisibleToEveryone);

        OverrideTasksData.Create(11714, TabGroup.NeutralRoles, CustomRoles.Workaholic);

        WorkaholicCanGuess = new BooleanOptionItem(11725, "CanGuess", true, TabGroup.NeutralRoles)
            .SetParent(CustomRoleSpawnChances[CustomRoles.Workaholic]);

        WorkaholicSpeed = new FloatOptionItem(11726, "WorkaholicSpeed", new(0.1f, 3f, 0.1f), 1.5f, TabGroup.NeutralRoles)
            .SetParent(CustomRoleSpawnChances[CustomRoles.Workaholic])
            .SetValueFormat(OptionFormat.Multiplier);
    }

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

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

    public override void Remove(byte playerId)
    {
        WorkaholicAlive.Remove(playerId);
    }

    public override void ApplyGameOptions(IGameOptions opt, byte playerId)
    {
        AURoleOptions.EngineerCooldown = WorkaholicVentCooldown.GetFloat();
        AURoleOptions.EngineerInVentMaxTime = 0f;
        Main.AllPlayerSpeed[playerId] = WorkaholicSpeed.GetFloat();
    }

    public override void OnTaskComplete(PlayerControl player, int CompletedTasksCount, int AllTasksCount)
    {
        if (CompletedTasksCount + 1 >= AllTasksCount && (!WorkaholicCannotWinAtDeath.GetBool() || player.IsAlive()))
        {
            Logger.Info("Workaholic Tasks Finished", "Workaholic");
            RPC.PlaySoundRPC(player.PlayerId, Sounds.KillSound);

            foreach (PlayerControl pc in Main.EnumerateAlivePlayerControls())
                pc.Suicide(pc.PlayerId == player.PlayerId ? PlayerState.DeathReason.Overtired : PlayerState.DeathReason.Ashamed, player);

            CustomWinnerHolder.ShiftWinnerAndSetWinner(CustomWinner.Workaholic);
            CustomWinnerHolder.WinnerIds.Add(player.PlayerId);
        }
    }
}