From 7f493f682503f5186308de7b8f74b5b49233cfe4 Mon Sep 17 00:00:00 2001
From: chai <215380520@qq.com>
Date: Thu, 2 Nov 2023 11:51:31 +0800
Subject: +init

---
 GameCode/TFUIEnumMouseCatcher.cs | 51 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 GameCode/TFUIEnumMouseCatcher.cs

(limited to 'GameCode/TFUIEnumMouseCatcher.cs')

diff --git a/GameCode/TFUIEnumMouseCatcher.cs b/GameCode/TFUIEnumMouseCatcher.cs
new file mode 100644
index 0000000..e46eb62
--- /dev/null
+++ b/GameCode/TFUIEnumMouseCatcher.cs
@@ -0,0 +1,51 @@
+using System.Collections.Generic;
+using UnityEngine;
+
+public class TFUIEnumMouseCatcher : ThronefallUIElement
+{
+	public List<ThronefallUIElement> focusWhitelist = new List<ThronefallUIElement>();
+
+	public GameObject buttons;
+
+	private UIFrame targetFrame;
+
+	protected override void OnApply()
+	{
+	}
+
+	protected override void OnClear()
+	{
+	}
+
+	protected override void OnFocus()
+	{
+		buttons.SetActive(value: true);
+		targetFrame.onNewFocus.AddListener(ListenForUnfocus);
+	}
+
+	protected override void OnFocusAndSelect()
+	{
+	}
+
+	protected override void OnHardStateSet(SelectionState selectionState)
+	{
+	}
+
+	protected override void OnSelect()
+	{
+	}
+
+	private void Start()
+	{
+		targetFrame = GetComponentInParent<UIFrame>();
+	}
+
+	private void ListenForUnfocus()
+	{
+		if (!(targetFrame.CurrentFocus == this) && !focusWhitelist.Contains(targetFrame.CurrentFocus))
+		{
+			buttons.SetActive(value: false);
+			targetFrame.onNewFocus.RemoveListener(ListenForUnfocus);
+		}
+	}
+}
-- 
cgit v1.1-26-g67d0