diff --git a/UI/QuickMenu/ReCategoryPage.cs b/UI/QuickMenu/ReCategoryPage.cs index d00d3d0..32a01cd 100644 --- a/UI/QuickMenu/ReCategoryPage.cs +++ b/UI/QuickMenu/ReCategoryPage.cs @@ -29,6 +29,7 @@ private static GameObject MenuPrefab private static int SiblingIndex => QuickMenuEx.Instance.field_Public_Transform_0.Find("Window/QMParent/Modal_AddMessage").GetSiblingIndex(); public event Action OnOpen; + public event Action OnClose; private readonly bool _isRoot; private readonly Transform _container; @@ -98,7 +99,9 @@ public ReCategoryPage(string text, bool isRoot = false) : base(MenuPrefab, Quick } EnableDisableListener.RegisterSafe(); - GameObject.AddComponent().OnEnableEvent += () => OnOpen?.Invoke(); + var listener = GameObject.AddComponent(); + listener.OnEnableEvent += () => OnOpen?.Invoke(); + listener.OnDisableEvent += () => OnClose?.Invoke(); } public ReCategoryPage(Transform transform) : base(transform) diff --git a/UI/QuickMenu/ReMenuPage.cs b/UI/QuickMenu/ReMenuPage.cs index 5ff1a60..b06652c 100644 --- a/UI/QuickMenu/ReMenuPage.cs +++ b/UI/QuickMenu/ReMenuPage.cs @@ -30,6 +30,7 @@ private static GameObject MenuPrefab private static int SiblingIndex => QuickMenuEx.Instance.field_Public_Transform_0.Find("Window/QMParent/Modal_AddMessage").GetSiblingIndex(); public event Action OnOpen; + public event Action OnClose; private readonly bool _isRoot; private readonly Transform _container; @@ -119,7 +120,9 @@ public ReMenuPage(string text, bool isRoot = false) : base(MenuPrefab, QuickMenu } EnableDisableListener.RegisterSafe(); - GameObject.AddComponent().OnEnableEvent += () => OnOpen?.Invoke(); + var listener = GameObject.AddComponent(); + listener.OnEnableEvent += () => OnOpen?.Invoke(); + listener.OnDisableEvent += () => OnClose?.Invoke(); } public ReMenuPage(Transform transform) : base(transform)