H3VR-TNH-Quality-of-Life-Improvements/Assets/MeatKit/MeatKitPlugin.cs

#if H3VR_IMPORTED
using System.Collections;
using System;
using HarmonyLib;
using System.IO;
using System.Reflection;
using BepInEx;
using BepInEx.Bootstrap;
using BepInEx.Logging;
using BepInEx.Configuration;
using UnityEngine;
using UnityEngine.UI;
using UnityEngine.SceneManagement;

using FistVR;
using TNHQoLImprovements;
using Sodalite.Api;
using Sodalite;

/*
 * SUPER LARGE WARNING ABOUT THIS CLASS
 * This is the default and fallback class that MeatKit uses as a template to generate a BepInEx plugin
 * when building your mod. DO NOT MODIFY THIS FILE AT ALL, IN ANY WAY.
 *
 * If you want to add custom behavior to your mod, you should make a copy of this class, and put it inside
 * the main namespace of your mod (that namespace can be found by opening the 'Allowed Namespaces' list on your build
 * profile). MeatKit will then detect and use that class instead of this one, for that one specific profile.
 *
 * HOWEVER, YOU MUST KEEP ALL OF THE STUFF FROM THIS TEMPLATE, otherwise MeatKit may fail to correctly build
 * your plugin, or your mod may fail to correctly load.
 */

// DO NOT REMOVE OR CHANGE ANY OF THESE ATTRIBUTES
[BepInPlugin("MeatKit", "MeatKit Plugin", "1.0.0")]
[BepInProcess("h3vr.exe")]

// DO NOT CHANGE THE NAME OF THIS CLASS OR THE BASE CLASS. If you're making a custom plugin, make sure it extends BaseUnityPlugin.
public class MeatKitPlugin : BaseUnityPlugin
{
    // DO NOT CHANGE OR REMOVE THIS FIELD.
#pragma warning disable 414
    private static readonly string BasePath = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
    internal new static ManualLogSource Logger;
#pragma warning restore 414

public static AssetBundle bundle;
    public static Font fontAgencyFB;
    public static Font fontBombardier;

    public static GameObject playerCamera;

    // BepInEx configuration
    //--- Health Counter ---//
    public static ConfigEntry<bool> cfgHPHiddenWhenAiming;
    public static ConfigEntry<float> cfgHPAimOpacity;
    public static ConfigEntry<bool> cfgShowHPBackground;
    public static ConfigEntry<float> cfgHPBackgroundOpacity;
    public static ConfigEntry<HPTextType> cfgHPTextType;
    //--- Take and Hold Info ---//
    public static ConfigEntry<bool> cfgShowLPC;
    public static ConfigEntry<bool> cfgInfoFollowCamera;
    public static ConfigEntry<bool> cfgShowTokens;
    public static ConfigEntry<bool> cfgShowHolds;
    public static ConfigEntry<bool> cfgShowNumbersAtShop;
    public static ConfigEntry<bool> cfgShowInfoOnGameOver;
    public static ConfigEntry<bool> cfgShowWaves;
    //--- Misc. ---//
    public static ConfigEntry<HealthExpireIndicationType> cfgHealthCrystalIndicator;

    // Take and Hold modifications
    private static InPlay instance;

    // Searching for old leaderboards player count mod to disable
    private float lpcSearchTime;
    private bool lpcStopSearching = false;
    private float lpcModSearchTimeEnd;

    public static FVRHealthBar hpDisplay;
    
    // toggle HP visibility from wrist menu
    private bool hpDisplayEnabled = true;
    private WristMenuButton wmbHPToggle;

    private Harmony harmony;

    // Could soft-lock if we're not in TNH!
    public static IEnumerator WaitForTNHInit()
    {
        while (InPlay.tnhManager == null || FindObjectOfType<TNH_Manager>() == null)
        {
            yield return null;
        }
    }

    private void SceneChanged(Scene from, Scene to)
    {
        playerCamera = GameObject.FindGameObjectWithTag("MainCamera");
        hpDisplay = FindObjectOfType<FVRHealthBar>();
        GetFonts();

        // apply health counter tweaks globally
        if (hpDisplay != null)
        {
            HPReadability.ImproveHPTextReadability(hpDisplay.transform.GetChild(0).gameObject);

            if (cfgHPHiddenWhenAiming.Value)
                hpDisplay.gameObject.AddComponent<HPHideWhenAiming>();

            WristMenuAPI.Buttons.Add(wmbHPToggle);
        }
        else
        {
            WristMenuAPI.Buttons.Remove(wmbHPToggle);
        }

		StartCoroutine("TryTNH");
	}

    private IEnumerator TryTNH()
    {
        for (int i = 0; i < 11; ++i)
        {
			// TNH patches
			if (GameObject.Find("_GameManager") != null || FindObjectOfType<TNH_Manager>() != null)
			{
				Logger.LogInfo("We are in a TNH game!");
				instance = new GameObject().AddComponent<InPlay>();
                break;
			}
			else
			{
				Logger.LogInfo(String.Format("Couldn't find a TNH game. Trying again...({0}/10)", i));
				yield return new WaitForEndOfFrame();
			}
        }

        // destroy self only if we are FOR SURE not in a TNH game
        if (GameObject.Find("_GameManager") == null || FindObjectOfType<TNH_Manager>() == null)
        {
            Logger.LogInfo("We are NOT in a TNH game!");
            Destroy(instance);
        }
    }
    
    // called on scene change, find fonts from game if they're not set
    private void GetFonts()
    {
        // Agency FB
        if (fontAgencyFB == null)
        {
            if (hpDisplay != null)
            {
                fontAgencyFB = hpDisplay.transform.GetChild(0).GetChild(0).GetComponent<Text>().font;
            }
            else
            {
                var query = FindObjectsOfType<Text>();
                foreach (Text itm in query)
                {
                    if (itm.font.name == "AGENCYR")
                    {
                        fontAgencyFB = itm.font;
                        break;
                    }
                }
            }
        }
    }

    public MeatKitPlugin(): base()
    {
        harmony = new Harmony("muskit.TNHQualityOfLifeImprovements");
        lpcSearchTime = 30f + 30f * Mathf.Sin(System.DateTime.Today.DayOfYear / 365f); // lolz
    }

    // You are free to edit this method, however please ensure LoadAssets is still called somewhere inside it.
    private void Awake()
    {
        // ----- BEGIN MEATKIT REQ. CODE -----
        // This lets you use your BepInEx-provided logger from other scripts in your project
        Logger = base.Logger;
        
        // You may place code before/after this, but do not remove this call to LoadAssets
        LoadAssets();
        // ----- END MEATKIT REQ. CODE -----


        // get Agency FB from system (BAD IDEA, NOT EVERYONE WILL HAVE IT; MAY SET TO DEFAULT FONT)
        //fontAgencyFB = Font.CreateDynamicFontFromOSFont("Agency FB", 16);

        // load asset bundle
        bundle = AssetBundle.LoadFromFile(Path.Combine(BasePath, "tnh_qol_improvements"));
        SceneManager.activeSceneChanged += SceneChanged;

        fontBombardier = bundle.LoadAsset<Font>("Bombardier");

        // setup configuration
        //--- Health Counter ---//
        cfgHPHiddenWhenAiming = Config.Bind("Health Counter",
                                            "Hide HP Counter When Aiming",
                                            true,
                                            "While aiming around the health counter in view, hide it.");
        cfgHPAimOpacity = cfgHPBackgroundOpacity = Config.Bind("Health Counter",
                          "Aiming opacity",
                          0f,
                          "Opacity of Health Counter when aiming around it (if Hide HP is enabled).");
        cfgShowHPBackground = Config.Bind("Health Counter",
                                          "Background enabled",
                                          false,
                                          "Apply a background to the Health Counter.");
        cfgHPBackgroundOpacity = Config.Bind("Health Counter",
                                             "Background opacity",
                                             0.74f,
                                             "Set opacity of health text's background (if enabled).");
        cfgHPTextType = Config.Bind("Health Counter",
                                    "HP Text Type",
                                    HPTextType.Solidify,
                                    "Solidify: Set text to full opacity and give it a drop shadow\n" +
                                    "Untouched: Leave text untouched\n" +
                                    "Hidden: Hide health counter completely (will hide background if enabled)");
        //--- Take and Hold Info ---//
        cfgShowLPC = Config.Bind("Take and Hold Info",
                                 "Show Player Count in Online Leaderboards",
                                 true,
                                 "Shows the number of players in the currently selected TNH leaderboard.");
        cfgInfoFollowCamera = Config.Bind("Take and Hold Info",
                                          "Tilt Wrist Stats Towards Camera",
                                          true,
                                          "Tilt the extra wrist statistics from this mod towards the player's camera, allowing for easier readability.");
        cfgShowTokens = Config.Bind("Take and Hold Info",
                                    "Show Tokens",
                                    true,
                                    "Shows how many tokens the player has by their radar hand.");
        cfgShowHolds = Config.Bind("Take and Hold Info",
                                   "Show Holds",
                                   true,
                                   "Shows how many holds the player has completed by their radar hand.");
        cfgShowWaves = Config.Bind("Take and Hold Info",
                                   "Show Waves",
                                   true,
                                   "Shows how many waves the player has completed on the current hold by their radar hand.");
        cfgShowInfoOnGameOver = Config.Bind("Take and Hold Info",
                                            "Show Extra Info at Game Over",
                                            true,
                                            "Show enabled extra game information at the game over area.");
        cfgShowNumbersAtShop = Config.Bind("Take and Hold Info",
                                           "Show Numbers for Tokens at Item Station",
                                           true,
                                           "At the item station, add a numberical representation to costs and player's tokens.");
        //--- Misc. ---//
        cfgHealthCrystalIndicator = Config.Bind("Misc.",
                                                "Show expiration of Health Crystals",
                                                HealthExpireIndicationType.Flashing,
                                                "Add a visual indication on the Health Crystal's despawn timer.");

        // calculate end time to search for my deprecated Leaderboard
        lpcModSearchTimeEnd = Time.time + lpcSearchTime;

        wmbHPToggle = new WristMenuButton("Toggle HP Display", ToggleHPVisibility);

        RunPatches();
    }

    private void RunPatches()
    {
        if (harmony == null)
        {
            Logger.LogError("Could not run patches; Harmony didn't initialize correctly!");
            return;
        }

        // patch KillAll code (only acts w/ health crystals)
        if (cfgHealthCrystalIndicator.Value != HealthExpireIndicationType.None)
            TimedHealthCrystalPatch.Patch(harmony);

        // patch leaderboard code
        if (cfgShowLPC.Value)
            LeaderboardPlayerCountPatch.Patch(harmony);

        // for counting wins/loses for TNHInfo.holdCounter
        if (cfgShowHolds.Value)
            HoldCounterPatch.Patch(harmony);

        // stick stats to hand after game over
        if (cfgShowHolds.Value || cfgShowTokens.Value)
            TNHInfo.Patch(harmony);
        
        // show numerical representation of shop values
        if (cfgShowNumbersAtShop.Value)
        {
            // costs
            ShopCostPatch.Patch(harmony);
            // player tokens
            ShopTokenPatch.Patch(harmony);
        }

		Logger.LogInfo("Successfully ran patches!");
	}

    private void ToggleHPVisibility(object sender, ButtonClickEventArgs args)
    {
        hpDisplayEnabled = !hpDisplayEnabled;
        if (hpDisplay != null)
            hpDisplay.gameObject.SetActive(hpDisplayEnabled);
    }

    /// <summary>
    /// Its only purpose: to kill the deprecated TNH Leaderboard Player Count mod.
    /// </summary>
    private void Update()
    {
        if (lpcStopSearching)
            return;

        foreach (var plugin in Chainloader.PluginInfos)
        {
            if (plugin.Key == "me.muskit.tnhLeaderboardPlayerCount")
            {
                Logger.LogWarning("TNH Leaderboard Player Count mod detected. Destroying it to avoid interference.");
                Destroy(plugin.Value.Instance);
                lpcStopSearching = true;
            }
        }

        if (Time.realtimeSinceStartup >= lpcModSearchTimeEnd)
        {
            Logger.LogInfo(string.Format("Stopping search for TNH Leaderboard Player Count mod after {0} seconds.", lpcSearchTime));
            lpcStopSearching = true;
        }
    }

    // DO NOT CHANGE OR REMOVE THIS METHOD. It's contents will be overwritten when building your package.
    private void LoadAssets()
    {
        // Code to load your build items will be generated at build-time and inserted here
    }
}
#endif