RPG-RougeLiteBattler/Assets/Scripts/BattleScene/BattleItemSelector.cs

using UnityEngine;
using UnityEngine.UIElements;
using System.Collections.Generic;
using System;

/// <summary>
/// UI Toolkit-based item selection UI for battle
/// </summary>
public class BattleItemSelector : MonoBehaviour
{
    [Header("UI References")]
    public UIDocument uiDocument;

    [Header("Settings")]
    public int maxItemsToShow = 10;

    private Character currentCharacter;
    private VisualElement rootElement;
    private VisualElement container;
    private VisualElement itemList;
    private Label titleLabel;
    private Label characterLabel;
    private VisualElement noItemsContainer;
    private Button cancelButton;
    private Button closeButton;

    // Events
    public event Action<string, int> OnItemSelected;
    public event Action OnSelectionCancelled;

    void Awake()
    {
        if (uiDocument == null)
            uiDocument = GetComponent<UIDocument>();

        if (uiDocument == null)
        {
            // Create UIDocument if it doesn't exist
            uiDocument = gameObject.AddComponent<UIDocument>();
        }
    }

    void Start()
    {
        InitializeUI();
        SetVisible(false);
    }

    void Update()
    {
        if (container != null && container.style.display == DisplayStyle.Flex)
        {
            if (Input.GetKeyDown(KeyCode.Escape))
            {
                OnSelectionCancelled?.Invoke();
                HideSelection();
            }
        }
    }

    private void InitializeUI()
    {
        // Load UXML and USS
        var visualTreeAsset = Resources.Load<VisualTreeAsset>("UI/ItemSelectionUI");
        var styleSheet = Resources.Load<StyleSheet>("UI/ItemSelectionUI");

        Debug.Log($"🔍 Loading UXML: {(visualTreeAsset != null ? "✅ Found" : "❌ Not Found")}");
        Debug.Log($"🔍 Loading USS: {(styleSheet != null ? "✅ Found" : "❌ Not Found")}");

        // Try simple version if main version fails
        if (visualTreeAsset == null)
        {
            Debug.LogWarning("Main UXML failed, trying simple version...");
            visualTreeAsset = Resources.Load<VisualTreeAsset>("UI/ItemSelectionUI_Simple");
            Debug.Log($"🔍 Loading Simple UXML: {(visualTreeAsset != null ? "✅ Found" : "❌ Not Found")}");
        }

        if (visualTreeAsset == null)
        {
            Debug.LogError("Could not load any ItemSelectionUI.uxml from Resources/UI/");
            CreateFallbackUI();
            return;
        }

        rootElement = uiDocument.rootVisualElement;
        rootElement.Clear();

        // Clone the visual tree
        var clonedTree = visualTreeAsset.CloneTree();
        rootElement.Add(clonedTree);

        // Apply styles
        if (styleSheet != null)
        {
            rootElement.styleSheets.Add(styleSheet);
        }

        // Get UI elements
        container = rootElement.Q<VisualElement>("ItemSelectionContainer");
        itemList = rootElement.Q<VisualElement>("ItemList");
        titleLabel = rootElement.Q<Label>("TitleLabel");
        characterLabel = rootElement.Q<Label>("CharacterLabel");
        noItemsContainer = rootElement.Q<VisualElement>("NoItemsContainer");
        cancelButton = rootElement.Q<Button>("CancelButton");
        closeButton = rootElement.Q<Button>("CloseButton");

        // Debug element finding
        Debug.Log($"🔍 Elements found:");
        Debug.Log($"  - container: {(container != null ? "✅" : "❌")}");
        Debug.Log($"  - itemList: {(itemList != null ? "✅" : "❌")}");
        Debug.Log($"  - titleLabel: {(titleLabel != null ? "✅" : "❌")}");
        Debug.Log($"  - characterLabel: {(characterLabel != null ? "✅" : "❌")}");
        Debug.Log($"  - noItemsContainer: {(noItemsContainer != null ? "✅" : "❌")}");
        Debug.Log($"  - cancelButton: {(cancelButton != null ? "✅" : "❌")}");
        Debug.Log($"  - closeButton: {(closeButton != null ? "✅" : "❌")}");

        // Setup event handlers
        if (cancelButton != null)
        {
            cancelButton.clicked += () =>
            {
                OnSelectionCancelled?.Invoke();
                HideSelection();
            };
        }

        if (closeButton != null)
        {
            closeButton.clicked += () =>
            {
                OnSelectionCancelled?.Invoke();
                HideSelection();
            };
        }

        Debug.Log("✅ BattleItemSelector initialized with UI Toolkit");
    }

    private void CreateFallbackUI()
    {
        Debug.LogWarning("Creating fallback BattleItemSelector UI");

        rootElement = uiDocument.rootVisualElement;
        rootElement.Clear();

        // Create basic container
        container = new VisualElement();
        container.name = "ItemSelectionContainer";
        container.style.position = Position.Absolute;
        container.style.top = 0;
        container.style.left = 0;
        container.style.right = 0;
        container.style.bottom = 0;
        container.style.backgroundColor = new Color(0, 0, 0, 0.7f);
        container.style.alignItems = Align.Center;
        container.style.justifyContent = Justify.Center;

        // Create modal
        var modal = new VisualElement();
        modal.style.backgroundColor = new Color(0.18f, 0.18f, 0.18f, 1f);
        modal.style.borderTopWidth = modal.style.borderBottomWidth = modal.style.borderLeftWidth = modal.style.borderRightWidth = 2;
        modal.style.borderTopColor = modal.style.borderBottomColor = modal.style.borderLeftColor = modal.style.borderRightColor = Color.gray;
        modal.style.borderTopLeftRadius = modal.style.borderTopRightRadius = modal.style.borderBottomLeftRadius = modal.style.borderBottomRightRadius = 8;
        modal.style.width = 400;
        modal.style.height = 500;
        modal.style.paddingTop = modal.style.paddingBottom = modal.style.paddingLeft = modal.style.paddingRight = 15;

        // Title
        titleLabel = new Label("Select Item");
        titleLabel.style.fontSize = 18;
        titleLabel.style.color = Color.white;
        titleLabel.style.unityFontStyleAndWeight = FontStyle.Bold;
        titleLabel.style.marginBottom = 10;
        titleLabel.style.unityTextAlign = TextAnchor.MiddleCenter;

        // Character label
        characterLabel = new Label();
        characterLabel.style.fontSize = 16;
        characterLabel.style.color = new Color(1f, 0.86f, 0.39f);
        characterLabel.style.unityFontStyleAndWeight = FontStyle.Bold;
        characterLabel.style.marginBottom = 15;
        characterLabel.style.unityTextAlign = TextAnchor.MiddleCenter;

        // Item list
        var scrollView = new ScrollView();
        scrollView.style.flexGrow = 1;
        scrollView.style.marginBottom = 15;

        itemList = new VisualElement();
        scrollView.Add(itemList);
        Debug.Log("🔧 Fallback UI: itemList created successfully");

        // No items message
        noItemsContainer = new VisualElement();
        noItemsContainer.style.alignItems = Align.Center;
        noItemsContainer.style.justifyContent = Justify.Center;
        noItemsContainer.style.flexGrow = 1;
        noItemsContainer.style.display = DisplayStyle.None;

        var noItemsLabel = new Label("No usable items available");
        noItemsLabel.style.color = new Color(0.6f, 0.6f, 0.6f);
        noItemsContainer.Add(noItemsLabel);

        // Cancel button
        cancelButton = new Button(() =>
        {
            OnSelectionCancelled?.Invoke();
            HideSelection();
        });
        cancelButton.text = "Cancel";
        cancelButton.style.paddingTop = cancelButton.style.paddingBottom = 8;
        cancelButton.style.paddingLeft = cancelButton.style.paddingRight = 16;

        // Assemble UI
        modal.Add(titleLabel);
        modal.Add(characterLabel);
        modal.Add(scrollView);
        modal.Add(noItemsContainer);
        modal.Add(cancelButton);

        container.Add(modal);
        rootElement.Add(container);
    }

    public void ShowItemSelection(Character character)
    {
        Debug.Log($"📦 ShowItemSelection called for {character.CharacterName}");
        Debug.Log($"📦 UI State - itemList: {(itemList != null ? "✅" : "❌")}, container: {(container != null ? "✅" : "❌")}");

        if (itemList == null)
        {
            Debug.LogError("❌ itemList is null - reinitializing UI");
            InitializeUI();
        }

        currentCharacter = character;

        if (characterLabel != null)
            characterLabel.text = $"{character.CharacterName}";

        if (titleLabel != null)
            titleLabel.text = "Select Item";

        PopulateItemList();
        SetVisible(true);

        Debug.Log($"📦 Item selection shown for {character.CharacterName}");
    }

    public void HideSelection()
    {
        SetVisible(false);
        ClearItemList();
        currentCharacter = null;

        Debug.Log("📦 Item selection hidden");
    }

    private void SetVisible(bool visible)
    {
        if (container != null)
        {
            container.style.display = visible ? DisplayStyle.Flex : DisplayStyle.None;
        }
    }

    private void PopulateItemList()
    {
        Debug.Log($"📦 PopulateItemList called - itemList null: {itemList == null}");

        ClearItemList();

        var items = GetCharacterItems();
        Debug.Log($"📦 Found {items.Count} items to display");

        if (items.Count == 0)
        {
            if (noItemsContainer != null)
                noItemsContainer.style.display = DisplayStyle.Flex;
            return;
        }

        if (noItemsContainer != null)
            noItemsContainer.style.display = DisplayStyle.None;

        for (int i = 0; i < items.Count && i < maxItemsToShow; i++)
        {
            CreateItemButton(items[i], i);
        }
    }

    private void CreateItemButton(ItemData item, int index)
    {
        // Safety check to prevent NullReferenceException
        if (itemList == null)
        {
            Debug.LogError("❌ itemList is null! UI initialization may have failed.");
            return;
        }

        var button = new Button();
        button.AddToClassList("item-button");

        // Item name
        var nameLabel = new Label(item.name);
        nameLabel.AddToClassList("item-name");

        // Item description
        var descLabel = new Label(item.description);
        descLabel.AddToClassList("item-description");

        button.Add(nameLabel);
        button.Add(descLabel);

        button.clicked += () =>
        {
            OnItemSelected?.Invoke(item.name, index);
            HideSelection();
        };

        itemList.Add(button);
    }

    private void ClearItemList()
    {
        if (itemList != null)
        {
            itemList.Clear();
        }
    }

    private List<ItemData> GetCharacterItems()
    {
        // Placeholder implementation - return sample items
        // TODO: Get actual items from character's inventory

        return new List<ItemData>
        {
            new ItemData { name = "Health Potion", description = "Restores 50 HP" },
            new ItemData { name = "Mana Potion", description = "Restores 30 MP" },
            new ItemData { name = "Antidote", description = "Cures poison" },
            new ItemData { name = "Strength Elixir", description = "+3 Str for 5 rounds" },
            new ItemData { name = "Magic Scroll", description = "Casts Magic Missile" }
        };
    }

    [System.Serializable]
    public class ItemData
    {
        public string name;
        public string description;
        public int quantity = 1;
        public bool isUsable = true;
    }

    void OnDestroy()
    {
        OnItemSelected = null;
        OnSelectionCancelled = null;
    }
}