BABA_YAGA/Assets/Scripts/UI/MainMenuController.cs

using UnityEngine;
using UnityEngine.UIElements;
using PrimeTween;
using System.Threading.Tasks;

namespace Hallucinate.UI
{
    public class MainMenuController : BaseUIController
    {
        public enum MenuState { Idle, Ribbon }
        private MenuState _currentState = MenuState.Idle;

        private VisualElement _logo;
        private VisualElement _ribbon;
        private VisualElement _logoSpace;
        
        private float _lastInteractionTime;
        private const float IDLE_TIMEOUT = 5.0f;
        private bool _isFirstLoad = true; // Biến cờ để nhận biết lần đầu vào game

        private Tween _pulseTween;
        private Tween _rotationTween;
        private Texture2D _currentIcon;

        public override void Initialize(VisualElement uxmlRoot, UIManager manager)
        {
            base.Initialize(uxmlRoot, manager);

            _logo = root.Q<VisualElement>("Logo");
            _ribbon = root.Q<VisualElement>("Ribbon");
            _logoSpace = root.Q<VisualElement>("LogoSpace");

            if (_logo == null)
            {
                Debug.LogError($"[MainMenuController] Element 'Logo' not found in UXML!");
                return;
            }

            _logo.RegisterCallback<ClickEvent>(OnLogoClicked);

            // Bind Buttons
            root.Q<Button>("SettingsBtn").clicked += () => uiManager.Push<SettingsController>();
            root.Q<Button>("JoinBtn").clicked += () => uiManager.Push<LobbyController>();
            root.Q<Button>("CreateBtn").clicked += () => uiManager.Push<LobbyController>();
            root.Q<Button>("ProfileBtn").clicked += () => uiManager.Push<ProfileController>();
            root.Q<Button>("ExitBtn").clicked += () => Application.Quit();

            ResetLogoPosition();
            StartPulse();
            _lastInteractionTime = Time.time;
        }

        private void ResetLogoPosition()
        {
            if (_logo == null) return;
            _logo.style.translate = new StyleTranslate(new Translate(0, 0));
            _logo.style.left = (Screen.width / 2f) - 100;
            _logo.style.top = (Screen.height / 2f) - 100;
            _logo.style.width = 200;
            _logo.style.height = 200;
        }

        public void SetGameIcon(Texture2D icon)
        {
            if (icon == null || _logo == null) return;
            _currentIcon = icon;
            _logo.style.backgroundImage = icon;
            
            var radius = new StyleLength(new Length(50, LengthUnit.Percent));
            _logo.style.borderTopLeftRadius = radius;
            _logo.style.borderTopRightRadius = radius;
            _logo.style.borderBottomLeftRadius = radius;
            _logo.style.borderBottomRightRadius = radius;
            _logo.style.overflow = Overflow.Hidden;

            var label = _logo.Q<Label>();
            if (label != null) label.style.display = DisplayStyle.None;

            StartRotation();
        }

        private void StartRotation()
        {
            if (_currentIcon == null) return;
            if (_rotationTween.isAlive) _rotationTween.Stop();
            
            _rotationTween = Tween.Custom(0f, 360f, duration: 4f, cycles: -1, ease: Ease.Linear, 
                onValueChange: val => _logo.style.rotate = new StyleRotate(new Rotate(Angle.Degrees(val))));
        }

        public override async Task PlayTransitionIn()
        {
            _lastInteractionTime = Time.time;
            _currentState = MenuState.Idle;
            ResetLogoPosition();
            
            if (_ribbon != null) 
            {
                _ribbon.style.display = DisplayStyle.None;
                _ribbon.style.opacity = 0;
            }

            // Khởi động lại rotation nếu có icon
            StartRotation();

            await base.PlayTransitionIn();

            // Nếu không phải lần đầu load (tức là quay lại bằng nút Back), tự động bung Ribbon
            if (!_isFirstLoad)
            {
                TransitionToRibbon();
            }
            else
            {
                _isFirstLoad = false; // Đã xong lần đầu, các lần sau sẽ tự động bung
            }
        }

        public override async Task PlayTransitionOut()
        {
            if (_rotationTween.isAlive) _rotationTween.Stop();
            await base.PlayTransitionOut();
        }

        private void OnLogoClicked(ClickEvent evt)
        {
            _lastInteractionTime = Time.time;
            if (_currentState == MenuState.Idle) TransitionToRibbon();
            else _ = uiManager.Push<LobbyController>();
        }

        private void TransitionToRibbon()
        {
            if (_currentState == MenuState.Ribbon && _ribbon.resolvedStyle.display == DisplayStyle.Flex) return;
            _currentState = MenuState.Ribbon;
            _lastInteractionTime = Time.time;

            _ribbon.style.display = DisplayStyle.Flex;
            Tween.Custom(0f, 1f, duration: 0.3f, onValueChange: val => _ribbon.style.opacity = val);

            _logoSpace.RegisterCallback<GeometryChangedEvent>(OnLogoSpaceReady);
        }

        private void OnLogoSpaceReady(GeometryChangedEvent evt)
        {
            _logoSpace.UnregisterCallback<GeometryChangedEvent>(OnLogoSpaceReady);
            Rect targetBounds = _logoSpace.worldBound;
            if (targetBounds.width <= 0) return;

            Tween.Custom(_logo.resolvedStyle.left, targetBounds.x, duration: 0.5f, ease: Ease.OutQuad, 
                onValueChange: val => _logo.style.left = val);
            
            Tween.Custom(_logo.resolvedStyle.top, targetBounds.y - 35, duration: 0.5f, ease: Ease.OutQuad, 
                onValueChange: val => _logo.style.top = val);

            Tween.Custom(_logo.resolvedStyle.width, 120f, duration: 0.5f, ease: Ease.OutQuad, 
                onValueChange: val => _logo.style.width = val);

            Tween.Custom(_logo.resolvedStyle.height, 120f, duration: 0.5f, ease: Ease.OutQuad, 
                onValueChange: val => _logo.style.height = val);
            
            _lastInteractionTime = Time.time;
        }

        private void TransitionToIdle()
        {
            if (_currentState == MenuState.Idle) return;
            _currentState = MenuState.Idle;

            float targetX = (Screen.width / 2f) - 100;
            float targetY = (Screen.height / 2f) - 100;

            Tween.Custom(_logo.resolvedStyle.left, targetX, duration: 0.5f, ease: Ease.OutQuad, 
                onValueChange: val => _logo.style.left = val);
            Tween.Custom(_logo.resolvedStyle.top, targetY, duration: 0.5f, ease: Ease.OutQuad, 
                onValueChange: val => _logo.style.top = val);
            Tween.Custom(_logo.resolvedStyle.width, 200f, duration: 0.5f, ease: Ease.OutQuad, 
                onValueChange: val => _logo.style.width = val);
            Tween.Custom(_logo.resolvedStyle.height, 200f, duration: 0.5f, ease: Ease.OutQuad, 
                onValueChange: val => _logo.style.height = val);

            Tween.Custom(1f, 0f, duration: 0.5f, onValueChange: val => _ribbon.style.opacity = val)
                .OnComplete(() => _ribbon.style.display = DisplayStyle.None);
        }

        public void Update()
        {
            if (Input.GetAxis("Mouse X") != 0 || Input.GetAxis("Mouse Y") != 0 || Input.anyKey)
            {
                _lastInteractionTime = Time.time;
            }

            if (_currentState == MenuState.Ribbon && Time.time - _lastInteractionTime > IDLE_TIMEOUT)
            {
                TransitionToIdle();
            }
        }

        private void StartPulse()
        {
            _pulseTween = Tween.Scale(_logo.transform, Vector3.one * 1.1f, duration: 0.8f, cycles: -1, cycleMode: CycleMode.Yoyo, ease: Ease.InOutSine);
        }
    }
}