BTKUILib.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using BTKUILib.UIObjects;
using BTKUILib.UIObjects.Objects;
using MelonLoader;

namespace BTKUILib
{
    internal static class BuildInfo
    {
        public const string Name = "BTKUILib";
        public const string Author = "BTK Development Team";
        public const string Company = "BTK Development";
        public const string Version = "2.3.1";
    }
    
    internal class BTKUILib : MelonMod
    {
        internal static MelonLogger.Instance Log;
        internal static BTKUILib Instance;
        internal static Page UISettingsPage;

        internal UserInterface UI;
        internal Queue<Action> MainThreadQueue = new();
        internal Dictionary<string, Page> MLPrefsPages = new();
        internal MelonPreferences_Entry<PlayerListStyleEnum> PlayerListStyle;
        internal MelonPreferences_Entry<bool> QMPlayerSelectorRedirect;

        private MelonPreferences_Entry<bool> _displayPrefsTab;
        private MelonPreferences_Entry<bool> _miscTabFirst;

        private Thread _mainThread;
        private Page _mlPrefsPage;
        private MultiSelection _playerListButtonStyle;
        private string[] _playerListStyleNames;

        public override void OnInitializeMelon()
        {
            Log = LoggerInstance;
            Instance = this;
            _mainThread = Thread.CurrentThread;
            
            Log.Msg("BTKUILib is starting up!");

            MelonPreferences.CreateCategory("BTKUILib", "BTKUILib");
            _displayPrefsTab = MelonPreferences.CreateEntry("BTKUILib", "DisplayPrefsTab", false, "Display MelonPrefs Tab", "Sets if the MelonLoader Prefs tab should be displayed");
            _displayPrefsTab.OnEntryValueChanged.Subscribe((b1, _) =>
            {
                if (_mlPrefsPage != null)
                    _mlPrefsPage.HideTab = b1;
            });

            PlayerListStyle = MelonPreferences.CreateEntry("BTKUILib", "PlayerListStyleNew", PlayerListStyleEnum.TabBar, "PlayerList Button Style", "Sets where the playerlist button will appear");

            _miscTabFirst = MelonPreferences.CreateEntry("BTKUILib", "MiscTabFirst", false, "Misc Tab Always First", "Makes sure the misc tab is always first in the tab row");

            QMPlayerSelectorRedirect = MelonPreferences.CreateEntry("BTKUILib", "PlayerSelectorRedirect", true, "Redirect Player QM Selector", "Chooses if the player selector should open the QM player details or always go to the big menu");
            
            Patches.Initialize(HarmonyInstance);

            UI = new UserInterface();
            UI.SetupUI();

            ColourPicker.SetupColourPicker();
            PlayerList.SetupPlayerList();

            if (!_miscTabFirst.Value) return;

            QuickMenuAPI.MiscTabPage = Page.GetOrCreatePage("Misc", "Misc", true, "MiscIcon");
            QuickMenuAPI.MiscTabPage.Protected = true;
            QuickMenuAPI.MiscTabPage.MenuTitle = "Misc";
            QuickMenuAPI.MiscTabPage.MenuSubtitle = "Miscellaneous mod elements be found here!";
            QuickMenuAPI.MiscTabPage.HideTab = true;
        }

        internal void GenerateSettingsPage()
        {
            if (UISettingsPage != null) return;

            UISettingsPage = new Page("btkUI-SettingsPage");
            var mainCat = UISettingsPage.AddCategory("Main", false);

            var prefsTabDisplay = mainCat.AddToggle("Show ML Prefs Tab", "Displays the MelonLoader prefs tab", _displayPrefsTab.Value);
            prefsTabDisplay.OnValueUpdated += b =>
            {
                _displayPrefsTab.Value = b;
                MelonPreferences.Save();
            };

            var openListStyle = mainCat.AddButton("Playerlist Button Position", "BTKList", "Change the position of the playerlist button");
            openListStyle.OnPress += () =>
            {
                QuickMenuAPI.OpenMultiSelect(_playerListButtonStyle);
            };

            var miscTabFirst = mainCat.AddToggle("Misc Tab First", "Sets if the Misc Tab should be first in the tab list (requires restart)", _miscTabFirst.Value);
            miscTabFirst.OnValueUpdated += b =>
            {
                _miscTabFirst.Value = b;
                MelonPreferences.Save();

                QuickMenuAPI.ShowNotice("Restart Required!", "To change the Misc tab first setting you must restart your game! This setting will be applied on the next startup!");
            };

            var playerSelector = mainCat.AddToggle("Player Selector QM Redirect", "Sets if the player selector should open the QM Player Details page or the big menu", QMPlayerSelectorRedirect.Value);
            playerSelector.OnValueUpdated += b =>
            {
                QMPlayerSelectorRedirect.Value = b;
                MelonPreferences.Save();
            };

            _playerListStyleNames = Enum.GetNames(typeof(PlayerListStyleEnum));

            _playerListButtonStyle = new MultiSelection("PlayerList Button Position", _playerListStyleNames, (int)PlayerListStyle.Value);
            _playerListButtonStyle.OnOptionUpdated += i =>
            {
                if(!Enum.IsDefined(typeof(PlayerListStyleEnum), i)) return;

                QuickMenuAPI.ShowAlertToast("You must restart ChilloutVR for this change to apply!");
                PlayerListStyle.Value = (PlayerListStyleEnum)Enum.Parse(typeof(PlayerListStyleEnum), Enum.GetNames(typeof(PlayerListStyleEnum))[i]);
                MelonPreferences.Save();
            };
        }

        internal void GenerateMlPrefsTab()
        {
            if(_mlPrefsPage != null) return;

            _mlPrefsPage = Page.GetOrCreatePage("MelonLoader", "Prefs", true, "Settings");
            _mlPrefsPage.MenuTitle = "MelonLoader Preferences";
            _mlPrefsPage.MenuSubtitle = "Control your MelonLoader Preferences from other mods!";
            _mlPrefsPage.Protected = true;
            _mlPrefsPage.HideTab = !_displayPrefsTab.Value;

            var prefCat = _mlPrefsPage.AddCategory("Categories");

            MLPrefsPages.Clear();

            foreach (var category in MelonPreferences.Categories.OrderBy(x => x.DisplayName))
            {
                var page = prefCat.AddPage(category.DisplayName, "Star", $"Opens the preferences category for {category.DisplayName}", "MelonLoader");
                MLPrefsPages.Add(category.Identifier, page);
                var pageCat = page.AddCategory("Preferences");

                foreach (var pref in category.Entries)
                {
                    if (pref.GetReflectedType() == typeof(bool))
                    {
                        var toggle = pageCat.AddToggle(pref.DisplayName, pref.Description, (bool)pref.BoxedValue);
                        toggle.OnValueUpdated += b =>
                        {
                            pref.BoxedValue = b;
                        };

                        if (pref.GetReflectedType() == typeof(string))
                        {
                            var button = pageCat.AddButton($"Edit {pref.DisplayName}", "Pencil", pref.Description);
                            button.OnPress += () =>
                            {
                                QuickMenuAPI.OpenKeyboard((string)pref.BoxedValue, s =>
                                {
                                    pref.BoxedValue = s;
                                });
                            };
                        }
                    }
                }
            }
        }

        internal bool IsOnMainThread(Thread thread = null)
        {
            thread ??= Thread.CurrentThread;

            return thread.Equals(_mainThread);
        }
        
        public override void OnUpdate()
        {
            if (MainThreadQueue.Count == 0) return;

            //If the queue has any amount of objects dequeue and invoke all of them
            while (MainThreadQueue.Count > 0)
            {
                MainThreadQueue.Dequeue()?.Invoke();
            }
        }
    }

    /// <summary>
    /// Enum containing the usable styles of playerlist button
    /// </summary>
    public enum PlayerListStyleEnum
    {
        /// <summary>
        /// Default style, button appears on the tab bar
        /// </summary>
        TabBar,
        /// <summary>
        /// Button replaces the existing TTS button on the right sidebar
        /// </summary>
        ReplaceTTS,
        /// <summary>
        /// Button replaces the unused events button on the QM
        /// </summary>
        ReplaceEvents,
    }
}