EndlessHostRoles/Modules/DoorsReset.cs

namespace EHR.Modules;

public static class DoorsReset
{
    public enum ResetMode
    {
        AllOpen,
        AllClosed,
        RandomByDoor
    }

    private static bool IsEnabled;
    private static ResetMode Mode;
    private static DoorsSystemType DoorsSystem => ShipStatus.Instance.Systems.TryGetValue(SystemTypes.Doors, out ISystemType system) ? system.CastFast<DoorsSystemType>() : null;

    public static void Initialize()
    {
        // Not supported except for Airship, Polus and Fungle
        if (Main.CurrentMap is not (MapNames.Airship or MapNames.Polus or MapNames.Fungle) && !(SubmergedCompatibility.Loaded && Main.NormalOptions.MapId == 6))
        {
            IsEnabled = false;
            return;
        }

        IsEnabled = Options.ResetDoorsEveryTurns.GetBool();
        Mode = (ResetMode)Options.DoorsResetMode.GetValue();
        Logger.Info($"Initalization: [ {IsEnabled}, {Mode} ]", "DoorsReset");
    }

    /// <summary>Reset door status according to settings</summary>
    public static void ResetDoors()
    {
        if (!IsEnabled || DoorsSystem == null) return;

        Logger.Info("Reset Completed", "DoorsReset");

        switch (Mode)
        {
            case ResetMode.AllOpen:
                OpenAllDoors();
                break;
            case ResetMode.AllClosed:
                CloseAllDoors();
                break;
            case ResetMode.RandomByDoor:
                OpenOrCloseAllDoorsRandomly();
                break;
            default:
                Logger.Warn($"Invalid Reset Doors Mode: {Mode}", "DoorsReset");
                break;
        }
    }

    /// <summary>Open all doors on the map</summary>
    public static void OpenAllDoors()
    {
        if (!ShipStatus.Instance) return;

        foreach (OpenableDoor door in ShipStatus.Instance.AllDoors)
        {
            if (door == null) continue;
            SetDoorOpenState(door, true);
        }

        DoorsSystemType doorsSystem = DoorsSystem;
        if (doorsSystem != null) doorsSystem.IsDirty = true;
    }

    /// <summary>Close all doors on the map</summary>
    public static void CloseAllDoors()
    {
        if (!ShipStatus.Instance) return;

        foreach (OpenableDoor door in ShipStatus.Instance.AllDoors)
        {
            if (door == null) continue;

            SetDoorOpenState(door, false);
        }

        DoorsSystemType doorsSystem = DoorsSystem;
        if (doorsSystem != null) doorsSystem.IsDirty = true;
    }

    /// <summary>Randomly opens and closes all doors on the map</summary>
    public static void OpenOrCloseAllDoorsRandomly()
    {
        if (!ShipStatus.Instance) return;

        foreach (OpenableDoor door in ShipStatus.Instance.AllDoors)
        {
            bool isOpen = IRandom.Instance.Next(2) > 0;
            SetDoorOpenState(door, isOpen);
        }

        DoorsSystemType doorsSystem = DoorsSystem;
        if (doorsSystem != null) doorsSystem.IsDirty = true;
    }

    /// <summary>Sets the open/close status of the door. Do nothing for doors that cannot be closed by sabotage</summary>
    /// <param name="door">Target door</param>
    /// <param name="isOpen">true for open, false for close</param>
    private static void SetDoorOpenState(OpenableDoor door, bool isOpen)
    {
        if (IsValidDoor(door)) door.SetDoorway(isOpen);
    }

    /// <summary>Determine if the door is subject to reset</summary>
    /// <returns>true if it is subject to reset</returns>
    private static bool IsValidDoor(OpenableDoor door)
    {
        // Airship lounge toilets and Polus decontamination room doors are not closed
        return door.Room is not (SystemTypes.Lounge or SystemTypes.Decontamination);
    }
}