BABA_YAGA/Assets/Scripts/UI/LobbyController.cs

using UnityEngine;
using UnityEngine.UIElements;

namespace UI
{
    public class LobbyController : MonoBehaviour
    {
        private VisualElement _joinView;
        private VisualElement _createView;
        
        private float _lastInteractionTime;
        private bool _isCreateMode = false;
        private const float AutoReturnDelay = 5f;

        private void OnEnable()
        {
            var root = GetComponent<UIDocument>().rootVisualElement;

            _joinView = root.Q<VisualElement>("join-view");
            _createView = root.Q<VisualElement>("create-view");

            // Back button
            root.Q<Button>("btn-back")?.RegisterCallback<ClickEvent>(evt => UIManager.Instance.GoBack());
            root.Q<Button>("btn-settings")?.RegisterCallback<ClickEvent>(evt => UIManager.Instance.ToggleSettings());
            
            // Create confirm -> Lounge
            root.Q<Button>("btn-create-confirm")?.RegisterCallback<ClickEvent>(evt => UIManager.Instance.ShowScreen("Lounge"));

            // Register Interaction Resetters
            var textFields = root.Query<TextField>().ToList();
            foreach (var field in textFields)
                field.RegisterValueChangedCallback(evt => ResetInteractionTimer());

            var toggles = root.Query<Toggle>().ToList();
            foreach (var t in toggles)
                t.RegisterValueChangedCallback(evt => ResetInteractionTimer());

            // Password Toggle Logic
            var passToggle = root.Q<Toggle>("toggle-password");
            var passField = root.Q<TextField>("field-password");
            passToggle?.RegisterValueChangedCallback(evt => {
                if(passField != null) passField.style.display = evt.newValue ? DisplayStyle.Flex : DisplayStyle.None;
            });

            ResetInteractionTimer();
        }

        private void Update()
        {
            if (_isCreateMode)
            {
                if (Time.time - _lastInteractionTime > AutoReturnDelay)
                {
                    SetMode(false); // Auto return to Stage 1
                }
            }
        }

        public void SetMode(bool isCreate)
        {
            _isCreateMode = isCreate;
            if (_joinView == null) return;
            
            _joinView.style.display = isCreate ? DisplayStyle.None : DisplayStyle.Flex;
            _createView.style.display = isCreate ? DisplayStyle.Flex : DisplayStyle.None;
            
            if (isCreate) ResetInteractionTimer();
        }

        private void ResetInteractionTimer()
        {
            _lastInteractionTime = Time.time;
        }
    }
}