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;

        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);

            var settingsBtn = root.Q<Button>("SettingsBtn");
            if (settingsBtn != null) settingsBtn.clicked += () => uiManager.ToggleSettings();
            
            root.Q<Button>("JoinBtn").clicked += async () => await uiManager.Push<LobbyController>();
            root.Q<Button>("CreateBtn").clicked += async () => await uiManager.Push<LobbyController>();
            root.Q<Button>("ProfileBtn").clicked += async () => await 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, 
                onValueChange: val => _logo.style.rotate = new StyleRotate(new Rotate(Angle.Degrees(val))),
                cycles: -1, 
                ease: Ease.Linear);
        }

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

            StartRotation();
            await base.PlayTransitionIn();

            if (!_isFirstLoad) TransitionToRibbon();
            else _isFirstLoad = false;
        }

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

        private async void OnLogoClicked(ClickEvent evt)
        {
            _lastInteractionTime = Time.time;
            if (_currentState == MenuState.Idle) TransitionToRibbon();
            else await 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;

            // Center logo in LogoSpace (within the centered Ribbon)
            float targetX = targetBounds.x + (targetBounds.width / 2f) - 50f; 
            float targetY = targetBounds.y + (targetBounds.height / 2f) - 50f;

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

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

            Tween.Custom(_logo.resolvedStyle.height, 100f, duration: 0.5f, 
                onValueChange: val => _logo.style.height = val,
                ease: Ease.OutQuad);
            
            _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, onValueChange: val => _logo.style.left = val, ease: Ease.OutQuad);
            Tween.Custom(_logo.resolvedStyle.top, targetY, duration: 0.5f, onValueChange: val => _logo.style.top = val, ease: Ease.OutQuad);
            Tween.Custom(_logo.resolvedStyle.width, 200f, duration: 0.5f, onValueChange: val => _logo.style.width = val, ease: Ease.OutQuad);
            Tween.Custom(_logo.resolvedStyle.height, 200f, duration: 0.5f, onValueChange: val => _logo.style.height = val, ease: Ease.OutQuad);

            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()
        {
            if (_pulseTween.isAlive) _pulseTween.Stop();
            _pulseTween = Tween.Custom(Vector3.one, Vector3.one * 1.1f, duration: 0.8f, 
                onValueChange: val => _logo.style.scale = new StyleScale(new Scale(val)),
                cycles: -1, 
                cycleMode: CycleMode.Yoyo,
                ease: Ease.InOutSine);
        }
    }
}