ScoreMod.cs

using System;
using System.Collections.Generic;
using SpinCore.Utility;
using UnityEngine;

namespace SRXDScoreMod; 

/// <summary>
/// Enables the introduction of custom score systems and score modifiers
/// </summary>
public static class ScoreMod {
    /// <summary>
    /// Invoke when the current score system is changed
    /// </summary>
    public static event Action<IScoreSystem> OnScoreSystemChanged;

    /// <summary>
    /// The currently visible score system
    /// </summary>
    public static IScoreSystem CurrentScoreSystem => CurrentScoreSystemInternal;

    /// <summary>
    /// The currently used modifier set
    /// </summary>
    public static ScoreModifierSet CurrentModifierSet { get; private set; }

    internal static bool AnyModifiersEnabled { get; private set; }
    internal static IScoreSystemInternal CurrentScoreSystemInternal { get; private set; }
    internal static List<IScoreSystemInternal> ScoreSystems { get; } = new();
    internal static List<CustomScoreSystem> CustomScoreSystems { get; } = new();

    private static int scoreSystemIndex;
    private static float modifierMultiplier = 1f;

    /// <summary>
    /// Adds a new custom score system with a given profile
    /// </summary>
    /// <param name="profile">The profile to use</param>
    public static void AddCustomScoreSystem(ScoreSystemProfile profile) {
        foreach (var system in ScoreSystems) {
            if (profile.Id != system.Id)
                continue;
            
            Plugin.Logger.LogWarning($"WARNING: Score system with ID {profile.Id} already exists");
                
            return;
        }
        
        var scoreSystem = new CustomScoreSystem(profile);

        ScoreSystems.Add(scoreSystem);
        CustomScoreSystems.Add(scoreSystem);
    }

    /// <summary>
    /// Sets the modifier set to use
    /// </summary>
    /// <param name="modifierSet">The modifier set to use</param>
    public static void SetModifierSet(ScoreModifierSet modifierSet) {
        if (modifierSet == CurrentModifierSet)
            return;

        if (CurrentModifierSet != null) {
            CurrentModifierSet.ModifierChanged -= OnModifierChanged;
            CurrentModifierSet.DisableAll();
        }

        CurrentModifierSet = modifierSet;

        if (modifierSet != null) {
            modifierSet.DisableAll();
            modifierSet.ModifierChanged += OnModifierChanged;
            HighScoresContainer.RemoveInvalidHighScoresForModifierSet(modifierSet);
        }

        OnModifierChanged();
    }

    internal static void Init() {
        ScoreSystems.Add(new BaseScoreSystemWrapper());
        AddCustomScoreSystem(DefaultScoreSystemProfiles.StandardPPM16);
        // AddCustomScoreSystem(DefaultScoreSystemProfiles.StandardPPM32);
        CurrentScoreSystemInternal = ScoreSystems[0];
    }

    internal static void LateInit() => Plugin.CurrentSystem.BindAndInvoke(OnCurrentSystemChanged);

    internal static int GetModifiedScore(int score) {
        if (AnyModifiersEnabled)
            return Mathf.CeilToInt(modifierMultiplier * score);

        return score;
    }

    private static void OnCurrentSystemChanged(int value) {
        if (value == scoreSystemIndex)
            return;

        scoreSystemIndex = value;
        CurrentScoreSystemInternal = ScoreSystems[value];
        GameplayUI.UpdateUI();
        CompleteScreenUI.UpdateUI(true);
        LevelSelectUI.UpdateUI();
        OnScoreSystemChanged?.Invoke(CurrentScoreSystem);
    }

    private static void OnModifierChanged() {
        if (CurrentModifierSet == null) {
            AnyModifiersEnabled = false;
            modifierMultiplier = 1f;
            ScoreSubmissionUtility.EnableScoreSubmission(Plugin.Instance);
        }
        else {
            AnyModifiersEnabled = CurrentModifierSet.GetAnyEnabled();
            modifierMultiplier = 0.01f * CurrentModifierSet.GetOverallMultiplier();
            
            if (AnyModifiersEnabled)
                ScoreSubmissionUtility.DisableScoreSubmission(Plugin.Instance);
            else
                ScoreSubmissionUtility.EnableScoreSubmission(Plugin.Instance);
        }
    }
}