mod buttons now disappear in a same way OW buttons do (#277)

* mod buttons now disappear in a same way OW buttons do
ow-mods · Jul 28, 2020 · af678bf · af678bf
1 parent 8a44981
commit af678bf
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/OWML.ModHelper.Menus/ModMainMenu.cs b/OWML.ModHelper.Menus/ModMainMenu.cs
@@ -3,6 +3,7 @@
 using OWML.Common.Menus;
 using OWML.ModHelper.Events;
 using UnityEngine;
+using UnityEngine.UI;
 
 namespace OWML.ModHelper.Menus
 {
@@ -18,6 +19,7 @@ public class ModMainMenu : ModMenu, IModMainMenu
         public IModButton QuitButton { get; private set; }
 
         private TitleAnimationController _anim;
+        private TitleScreenManager _titleManager;
 
         public ModMainMenu(IModConsole console) : base(console)
         {
@@ -26,6 +28,7 @@ public ModMainMenu(IModConsole console) : base(console)
 
         public void Initialize(TitleScreenManager titleScreenManager)
         {
+            _titleManager = titleScreenManager;
             _anim = titleScreenManager.GetComponent<TitleAnimationController>();
             var menu = titleScreenManager.GetValue<Menu>("_mainMenu");
             Initialize(menu);
@@ -50,6 +53,12 @@ public override IModButtonBase AddButton(IModButtonBase button, int index)
                 group = x.Button.GetComponent<CanvasGroup>()
             });
             _anim.SetValue("_buttonFadeControllers", fadeControllers.ToArray());
+            if (button is ModTitleButton titleButton)
+            {
+                var texts = _titleManager.GetValue<Text[]>("_mainMenuTextFields").ToList();
+                texts.Add(titleButton.Text);
+                _titleManager.SetValue("_mainMenuTextFields", texts.ToArray());
+            }
             return modButton;
         }
 

diff --git a/OWML.ModHelper.Menus/ModTitleButton.cs b/OWML.ModHelper.Menus/ModTitleButton.cs
@@ -6,20 +6,20 @@ namespace OWML.ModHelper.Menus
 {
     public class ModTitleButton : ModButtonBase, IModButton
     {
-        private readonly Text _text;
+        internal Text Text { get; }
         public virtual string Title
         {
-            get => _text != null ? _text.text : "";
+            get => Text != null ? Text.text : "";
             set
             {
                 Object.Destroy(Button.GetComponentInChildren<LocalizedText>());
-                _text.text = value;
+                Text.text = value;
             }
         }
 
         public ModTitleButton(Button button, IModMenu menu) : base(button, menu)
         {
-            _text = Button.GetComponentInChildren<Text>();
+            Text = Button.GetComponentInChildren<Text>();
         }
 
         public new IModButton Copy()