Plugin.cs

using BepInEx;
using System.IO;
using UnityEngine;
using HarmonyLib;
using GameConsole;
using USTManager.Data;
using USTManager.Patches;
using USTManager.Commands;
using System.Collections.Generic;
using TMPro;
using UnityEngine.UI;
using USTManager.Misc;
using USTManager.Utility;
using System.Linq;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using System.Threading.Tasks;
using BepInEx.Configuration;

namespace USTManager
{
    [BepInPlugin("dev.zeddevstuff.ustmanager", "USTManager", "1.5.4")]
    public class Plugin : BaseUnityPlugin
    {
        public static string UKPath, USTDir, LastUSTs;
        public ConfigEntry<bool> UpdateBasedSwappingEnabled;
        public static GameObject MenuEntryPrefab, SelectionScreenPrefab, SelectionScreenEntryPrefab, ConflictEntryPrefab, ConflictResolutionScreenPrefab, ToastPrefab;
        private void Awake()
        {
            UpdateBasedSwappingEnabled = Config.Bind(
                "Experimental", 
                "UpdateBasedSwappingEnabled", 
                false, 
                "Enable this to use Update based swapping on top of method patching. This should allow most sounds to be swapped. This method will probably hurt performance.");
            instance = this;
            Harmony.CreateAndPatchAll(typeof(AudioSourcePatches));
            Harmony.CreateAndPatchAll(typeof(MainMenuPatches));
            UKPath = new DirectoryInfo(Application.dataPath).Parent.FullName;
            LastUSTs = Path.Combine(UKPath, "USTs","lastUSTs.json");
            if(File.Exists(LastUSTs))
            {
                try
                {
                    Manager.USTSave? data = JsonConvert.DeserializeObject<Manager.USTSave>(File.ReadAllText(LastUSTs));
                    if(data != null)
                    {
                        USTSelectionScreen.InternalConfirm(data.Selected, data.UST);
                    }
                }
                catch(System.Exception e)
                {
                    Debug.LogError(e);
                }
            }
            DirectoryInfo ustDir = new(Path.Combine(UKPath, "USTs"));
            USTDir = ustDir.FullName;
            if(!ustDir.Exists) ustDir.Create();

            AssetBundle bundle = AssetBundle.LoadFromMemory(Resources.Resource1.ust);
            MenuEntryPrefab = bundle.LoadAsset<GameObject>("MenuEntry");
            //Debug.Log("MenuEntryprefab " + (MenuEntryPrefab == null ? "null" : "not null"));
            MenuEntryPrefab.AddComponent<HudOpenEffect>();
            SelectionScreenEntryPrefab = bundle.LoadAsset<GameObject>("USTEntry");
            //Debug.Log("SelectionScreenEntryPrefab " + (SelectionScreenEntryPrefab == null ? "null" : "not null"));
            USTEntry entry = SelectionScreenEntryPrefab.AddComponent<USTEntry>();
            entry.Image = SelectionScreenEntryPrefab.transform.GetChild(0).GetComponent<Image>();
            entry.Name = SelectionScreenEntryPrefab.transform.GetChild(1).GetComponent<TMP_Text>();
            entry.Author = SelectionScreenEntryPrefab.transform.GetChild(2).GetComponent<TMP_Text>();
            entry.IconButton = SelectionScreenEntryPrefab.transform.GetChild(0).GetComponent<Button>();
            entry.Status = SelectionScreenEntryPrefab.transform.GetChild(0).GetChild(0).GetComponent<TMP_Text>();
            SelectionScreenPrefab = bundle.LoadAsset<GameObject>("OptionMenu");
            //Debug.Log("SelectionScreenPrefab " + (SelectionScreenPrefab == null ? "null" : "not null"));
            SelectionScreenPrefab?.AddComponent<USTSelectionScreen>();
            ConflictEntryPrefab = bundle.LoadAsset<GameObject>("Conflict");
            //Debug.Log("ConflictEntryPrefab " + (ConflictEntryPrefab == null ? "null" : "not null"));
            ConflictResolutionScreenPrefab = bundle.LoadAsset<GameObject>("ConflictResolutionScreen");
            //Debug.Log("ConflictResolutionScreenPrefab " + (ConflictResolutionScreenPrefab == null ? "null" : "not null"));
            ConflictResolutionScreenPrefab?.AddComponent<ConflictResolutionScreen>();
            //ToastPrefab = bundle.LoadAsset<GameObject>("Popup");
            //ToastPrefab.AddComponent<Toast>();
            
            CreateTemplate();
            Logger.LogInfo($"Plugin {PluginInfo.PLUGIN_GUID} is loaded!");
        }
        System.Diagnostics.Stopwatch sw = new();
        public void Update()
        {
            //sw.Restart();
            if(!UpdateBasedSwappingEnabled.Value) return;
            AudioSource[] sources = /*UnityEngine.Resources.*/FindObjectsOfType<AudioSource>();
            foreach(AudioSource source in sources)
            {
                if(source.GetComponent<USTTarget>()) continue;
                else source.gameObject.AddComponent<USTTarget>();
            }
            //sw.Stop();
            //Logging.Log($"Finding {sources.Length} audio sources took {sw.Elapsed}", Color.cyan);
        }
        public static USTSelectionScreen Screen;
        public static void OpenMenu(Transform transform)
        {
            if(Screen == null)
            {
                Screen = Instantiate(SelectionScreenPrefab, transform).GetComponent<USTSelectionScreen>();
                Screen.GetComponent<RectTransform>().anchoredPosition = new Vector2(0, 0);
            }
            else
            {
                Screen.gameObject.SetActive(true);
            }
        }
        private static Plugin instance;
        public static void RunCoroutine(System.Collections.IEnumerator routine)
        {
            instance.StartCoroutine(routine);
        }
        public void CreateTemplate()
        {
            Directory.CreateDirectory(Path.Combine(UKPath, "USTs", "Template"));
            File.WriteAllText(Path.Combine(UKPath, "USTs", "Template", "template.ust"), CustomUST.GetTemplateJson());
            File.WriteAllText(Path.Combine(UKPath, "USTs", "Template", "readme.md"), StaticResources.Readme);
        }

        private void OnGUI()
        {
            if(!Manager.IsDebug) return;

            AudioSource[] sources = FindObjectsOfType<AudioSource>()
                .Where(s => s.isPlaying)
                .Where(s => Manager.IsExtendedDebug ? true : s.spatialBlend < 1.0)
                //.OrderByDescending(s => s.clip.length)
                .ToArray();

            void DrawColumn(System.Action<AudioSource> action)
            {
                using var _ = new GUILayout.VerticalScope();
                foreach(AudioSource s in sources)
                {
                    if(s is null) continue;
                    action(s);
                }
            }

            GUIStyle style = GUI.skin.label;
            style.fontSize = 24;

            using var _ = new GUILayout.HorizontalScope();

            style.normal.textColor = Color.green;
            DrawColumn(s => GUILayout.Label(s.name ?? "null"));

            GUILayout.Space(20);
            style.normal.textColor = Color.white;
            DrawColumn(s => GUILayout.Label(s.clip ? s.clip.name ?? "null" : "null"));

            GUILayout.Space(20);
            style.normal.textColor = Color.red;
            DrawColumn(s =>
            {
                using var _ = new GUILayout.HorizontalScope();
                GUILayout.Label(s.volume * 100 + "%");
                GUILayout.HorizontalSlider(s.time, 0, s.clip ? s.clip.length : 0, GUILayout.Width(150));
            });
        }
    }
}