MiraAPI/MiraAPI/GameModes/CustomGameModeManager.cs

using System;
using System.Collections.Generic;
using System.Linq;
using MiraAPI.PluginLoading;
using Reactor.Utilities;

namespace MiraAPI.GameModes;

/// <summary>
/// Manages custom gamemodes.
/// </summary>
public static class CustomGameModeManager
{
    internal static readonly Dictionary<uint, AbstractGameMode> IdToModeMap = [];

    private static uint GetNextId() => ++LastId;

    private static uint LastId { get; set; }

    /// <summary>
    /// Register gamemode from type.
    /// </summary>
    /// <param name="gameModeType">Type of gamemode class, should inherit from <see cref="AbstractGameMode"/>.</param>
    /// <param name="pluginInfo">The custom plugin info of the mod.</param>
    /// <returns>Whether the gamemode was successfully registered.</returns>
    internal static bool RegisterGameMode(Type gameModeType, MiraPluginInfo pluginInfo)
    {
        if (!typeof(AbstractGameMode).IsAssignableFrom(gameModeType))
        {
            Warning($"{gameModeType.Name} does not inherit CustomGameMode!");
            return false;
        }

        var instance = Activator.CreateInstance(gameModeType);

        if (instance is not AbstractGameMode mode)
        {
            Error($"Failed to create instance of {gameModeType.Name}");
            return false;
        }

        IdToModeMap.Add(GetNextId(), mode);
        pluginInfo.GameModes.Add(LastId, mode);
        mode.ID = LastId;
        GameModeOption.AddOption(mode);
        return true;
    }

    /// <summary>
    /// Checks to see if the current game mode is an instance of the classic mode.
    /// </summary>
    /// <returns>True if the classic mode is the current one.</returns>
    public static bool IsClassic() => ActiveMode != null && ActiveMode.GetType().IsAssignableTo(typeof(ClassicMode));

    /// <summary>
    /// Checks to see if the current game mode is an instance of the HNS mode.
    /// </summary>
    /// <returns>True if the Hide & Seek mode is the current one.</returns>
    public static bool IsHideNSeek() => ActiveMode != null && ActiveMode.GetType().IsAssignableTo(typeof(HideAndSeekMode));

    /// <summary>
    /// Checks if a provided GameMode is the current active one.
    /// </summary>
    /// <typeparam name="T">The AbstractGameMode subclass being checked.</typeparam>
    /// <returns>Whether the provided mode is the current active one.</returns>
    public static bool IsActiveGameMode<T>() where T : AbstractGameMode => ActiveMode is T;

    /// <summary>
    /// Gets the current gamemode.
    /// </summary>
    public static AbstractGameMode? ActiveMode { get; private set; }

    /// <summary>
    /// Gets a gamemode from an ID.
    /// </summary>
    /// <param name="id">The ID of the gamemode to fetch.</param>
    /// <returns>The gamemode matching that ID.</returns>
    public static AbstractGameMode GetMode(uint id) => IdToModeMap[id];

    /// <summary>
    /// Finds the parent mod of a custom gamemode.
    /// </summary>
    /// <param name="gameMode">The <see cref="AbstractGameMode"/> object.</param>
    /// <returns>A <see cref="MiraPluginInfo"/> object representing the parent mod of the gamemode.</returns>
    public static MiraPluginInfo? FindParentMod(AbstractGameMode gameMode)
    {
        return MiraPluginManager.Instance.RegisteredPlugins.FirstOrDefault(plugin => plugin.GameModes.ContainsValue(gameMode));
    }

    internal static void RegisterDefaultMode()
    {
        var defaultMode = new ClassicMode();
        IdToModeMap.Add(0, defaultMode);
        // no need to add to game mode option as it already contains it
        // because we cannot have the option be created with no values
        defaultMode.ID = 0;
        var hnsMode = new HideAndSeekMode();
        IdToModeMap.Add(1, hnsMode);
        hnsMode.ID = 1;
        GameModeOption.AddOption(hnsMode);
        LastId++;
    }

    internal static void GetAndSetGameMode()
    {
        var id = (uint)GameModeOption.Value;

        if (IdToModeMap.TryGetValue(id, out var mode))
        {
            ActiveMode = mode;
            return;
        }

        ActiveMode = IdToModeMap[0];
        Logger<MiraApiPlugin>.Warning($"Unable to find game mode of id {id}!");
    }
}