SacrificeHelperMod.cs

using Il2CppAssets.Scripts.Models;
using Il2CppAssets.Scripts.Simulation.Towers.Behaviors;
using Il2CppAssets.Scripts.Unity.UI_New.InGame.TowerSelectionMenu;
using BTD_Mod_Helper;
using BTD_Mod_Helper.Api.Components;
using BTD_Mod_Helper.Api.Enums;
using BTD_Mod_Helper.Api.ModOptions;
using BTD_Mod_Helper.Extensions;
using HarmonyLib;
using Il2Cpp;
using Il2CppAssets.Scripts.Unity.UI_New.InGame;
using Il2CppAssets.Scripts.Unity.UI_New.InGame.TowerSelectionMenu.TowerSelectionMenuThemes;
using Il2CppAssets.Scripts.Unity.UI_New.Popups;
using Il2CppAssets.Scripts.Unity.UI_New.Utils;
using MelonLoader;
using SacrificeHelper;
using UnityEngine;

[assembly: MelonInfo(typeof(SacrificeHelperMod), ModHelperData.Name, ModHelperData.Version, ModHelperData.RepoOwner)]
[assembly: MelonGame("Ninja Kiwi", "BloonsTD6")]

namespace SacrificeHelper;

public class SacrificeHelperMod : BloonsTD6Mod
{
    public static readonly ModSettingDouble SliderContributionPenalty = new(0.05f)
    {
        description = "The popup added in BTD6 v39 comes with a default 5% penalty to manually invested cash.\n" +
                      "Setting this to 0 would stop it, or negative would counteract it.",
        stepSize = .01f,
        min = -.99,
        max = 1,
        icon = VanillaSprites.UpgradeContainerParagonUnlocked
    };

    public static readonly ModSettingCategory ParagonPowerMaximums = new("Paragon Power Maximums");

    public static readonly ModSettingInt MaxPowerFromPops = new(90000)
    {
        displayName = "Max Paragon Power From Pops\n(-1 for unlimited)",
        min = -1,
        max = 200000,
        icon = VanillaSprites.PopIcon,
        category = ParagonPowerMaximums,
    };

    public static readonly ModSettingInt MaxPowerFromCash = new(60000)
    {
        displayName = "Max Paragon Power From Cash\n(-1 for unlimited)",
        min = -1,
        max = 200000,
        icon = VanillaSprites.CoinIcon,
        category = ParagonPowerMaximums
    };

    public static readonly ModSettingInt MaxPowerFromNonTier5s = new(10000)
    {
        displayName = "Max Paragon Power From Non Tier 5s\n(-1 for unlimited)",
        min = -1,
        max = 200000,
        icon = VanillaSprites.UpgradeContainerGrey,
        modifyOption = option => option.Icon.AddText(new Info("Text", InfoPreset.FillParent), "<5", 100),
        category = ParagonPowerMaximums
    };

    public static readonly ModSettingInt MaxPowerFromTier5s = new(50000)
    {
        displayName = "Max Paragon Power From Tier 5s\n(-1 for unlimited)",
        min = -1,
        max = 200000,
        icon = VanillaSprites.UpgradeContainerTier5,
        modifyOption = option => option.Icon.AddText(new Info("Text", InfoPreset.FillParent), "5", 100),
        category = ParagonPowerMaximums
    };

    public static readonly ModSettingCategory ParagonPowerWeights = new("Paragon Power Weights");

    private static readonly ModSettingInt PopsScaleFactor = new(180)
    {
        displayName = "Pops per Point of Paragon Power",
        min = 1,
        icon = VanillaSprites.PopIcon,
        category = ParagonPowerWeights
    };

    private static readonly ModSettingInt CashScaleFactor = new(20000)
    {
        displayName = "Paragon Power Scale Factor for Cash",
        min = 1,
        icon = VanillaSprites.CoinIcon,
        category = ParagonPowerWeights,
        description = "As of v39, the Paragon Upgrade Price is divided by this to get the final value"
    };

    private static readonly ModSettingInt NonTier5ScaleFactor = new(100)
    {
        displayName = "Paragon Power Scale Factor for Non Tier 5s",
        min = 0,
        icon = VanillaSprites.UpgradeContainerGrey,
        modifyOption = option => option.Icon.AddText(new Info("Text", InfoPreset.FillParent), "<5", 100),
        category = ParagonPowerWeights
    };

    private static readonly ModSettingInt Tier5ScaleFactor = new(6000)
    {
        displayName = "Paragon Power Scale Factor for Tier 5s",
        min = 0,
        icon = VanillaSprites.UpgradeContainerTier5,
        modifyOption = option => option.Icon.AddText(new Info("Text", InfoPreset.FillParent), "5", 100),
        category = ParagonPowerWeights
    };

    public static readonly ModSettingCategory TempleAlternateCosts = new("Template Alternate Costs");

    public static readonly ModSettingDouble TempleAlternateCostMod = new(.5)
    {
        displayName = "Alternate Sun Temple Cost",
        min = 0,
        max = 1,
        stepSize = .01f,
        icon = VanillaSprites.SunTempleUpgradeIcon,
        description = "What portion the cost should be if you decide to get a Sun Temple without doing sacrifices",
        category = TempleAlternateCosts
    };

    public static readonly ModSettingDouble GodAlternateCostMod = new(.2)
    {
        displayName = "Alternate True Sun God Cost",
        min = 0,
        max = 1,
        stepSize = .01f,
        icon = VanillaSprites.TrueSonGodUpgradeIcon,
        description = "What portion the cost should be if you decide to get a True Sun God without doing sacrifices",
        category = TempleAlternateCosts
    };

    public static bool templeSacrificesOff;

    public override void OnNewGameModel(GameModel result)
    {
        var degreeData = result.paragonDegreeDataModel;

        degreeData.maxPowerFromPops = MaxPowerFromPops;
        if (degreeData.maxPowerFromPops < 0)
        {
            degreeData.maxPowerFromPops = degreeData.MaxInvestment;
        }

        degreeData.maxPowerFromMoneySpent = MaxPowerFromCash;
        if (degreeData.maxPowerFromMoneySpent < 0)
        {
            degreeData.maxPowerFromMoneySpent = degreeData.MaxInvestment;
        }

        degreeData.maxPowerFromNonTier5Count = MaxPowerFromNonTier5s;
        if (degreeData.maxPowerFromNonTier5Count < 0)
        {
            degreeData.maxPowerFromNonTier5Count = degreeData.MaxInvestment;
        }

        degreeData.maxPowerFromTier5Count = MaxPowerFromTier5s;
        if (degreeData.maxPowerFromTier5Count < 0)
        {
            degreeData.maxPowerFromTier5Count = degreeData.MaxInvestment;
        }

        degreeData.popsOverX = PopsScaleFactor;
        degreeData.moneySpentOverX = CashScaleFactor;
        degreeData.nonTier5TowersMultByX = NonTier5ScaleFactor;
        degreeData.tier5TowersMultByX = Tier5ScaleFactor;
        
        degreeData.paidContributionPenalty = SliderContributionPenalty;
        
        templeSacrificesOff = false;
    }

    [HarmonyPatch(typeof(MonkeyTemple), nameof(MonkeyTemple.StartSacrifice))]
    public class MonkeyTemple_StartSacrifice
    {
        [HarmonyPrefix]
        public static bool Prefix() => !templeSacrificesOff;
    }

    [HarmonyPatch(typeof(TowerSelectionMenu), nameof(TowerSelectionMenu.UpdateTower))]
    internal static class TowerSelectionMenu_UpdateTower
    {
        [HarmonyPostfix]
        private static void Postfix(TowerSelectionMenu __instance)
        {
            var themeManager = __instance.themeManager;
            var currentTheme = themeManager.CurrentTheme;

            if (currentTheme == null) return;

            var ui = currentTheme.GetComponent<SacrificeHelperUI>();
            if (ui != null)
            {
                ui.TowerInfoChanged();
            }
        }
    }


    [HarmonyPatch(typeof(MenuThemeManager), nameof(MenuThemeManager.SetTheme))]
    internal static class MenuThemeManager_SetTheme
    {
        [HarmonyPostfix]
        private static void Postfix(MenuThemeManager __instance, BaseTSMTheme newTheme)
        {
            if (!__instance.selectionMenu.Is(out TowerSelectionMenu menu)) return;

            var ui = newTheme.GetComponent<SacrificeHelperUI>();
            if (ui == null)
            {
                ui = newTheme.gameObject.AddComponent<SacrificeHelperUI>();
                ui.Initialise(menu);
                ui.TowerInfoChanged();
            }
        }
    }

    [HarmonyPatch(typeof(ParagonConfirmationPopup), nameof(ParagonConfirmationPopup.UpdateCurrentInvestment))]
    internal static class ParagonConfirmationPopup_UpdateCurrentInvestment
    {
        [HarmonyPostfix]
        private static void Postfix(ParagonConfirmationPopup __instance, float current)
        {
            var tower = TowerSelectionMenu.instance.selectedTower;

            var degree = Utils.GetParagonDegree(tower, out _, current);

            var text = __instance.transform.GetComponentFromChildrenByName<NK_TextMeshProUGUI>("DegreeText");
            text.SetText(degree.ToString());
            text.color = degree >= 100 ? Color.green : Color.white;
        }
    }

    [HarmonyPatch(typeof(ParagonConfirmationPopup), nameof(ParagonConfirmationPopup.Init),
        typeof(Il2CppSystem.Action<double>), typeof(Il2CppSystem.Action), typeof(int), typeof(int), typeof(int),
        typeof(PopupScreen.BackGround), typeof(Popup.TransitionAnim))]
    internal static class ParagonConfirmationPopup_Init
    {
        [HarmonyPostfix]
        private static void Postfix(ParagonConfirmationPopup __instance)
        {
            var tower = TowerSelectionMenu.instance.selectedTower;
            var degree = Utils.GetParagonDegree(tower, out _);

            var mainObject = __instance.animator.gameObject;

            var indicator = ModHelperImage.Create(new Info("DegreeIndicator", 0, -450, 250, 250),
                VanillaSprites.UpgradeContainerParagon);
            mainObject.AddModHelperComponent(indicator);
            indicator.AddText(new Info("DegreeText", InfoPreset.FillParent), degree.ToString(), 100f);
        }
    }
}