GathererOfSouls/Assets/Scripts/GameUI.cs

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

public class GameUI : MonoBehaviour
{
    [Header("UI Documents")]
    public UIDocument mainUIDocument;

    private VisualElement root;
    private Label woodLabel;
    private Label stoneLabel;
    private Label foodLabel;
    private Label timeSpeedLabel;
    private Button pauseButton;
    private Button speed1xButton;
    private Button speed2xButton;
    private Button speed10xButton;

    // Villager Panel
    private VisualElement villagerPanel;
    private Label villagerInfoLabel;
    private VisualElement jobButtonsContainer;
    private List<Button> jobButtons = new List<Button>();
    private Villager currentSelectedVillager; // Store reference to avoid deselection issues

    // Selected Object Panel
    private VisualElement selectionPanel;
    private Label selectionInfoLabel;

    // Build Menu
    private VisualElement buildMenuPanel;
    private Button buildMenuToggle;
    private VisualElement buildCategoriesContainer;
    private VisualElement buildItemsContainer;
    private Button livingCategoryButton;
    private Button workingCategoryButton;
    private Button decorativeCategoryButton;
    private Label buildModeStatusLabel;
    private bool isBuildMenuOpen = false;
    private BuildingCategory currentBuildCategory;

    // Housing System
    private Label housingStatusLabel;
    private Label housingWarningLabel;
    private BuildingSystem buildingSystem;
    private HousingManager housingManager;

    void Start()
    {
        StartCoroutine(InitializeWhenReady());
    }

    System.Collections.IEnumerator InitializeWhenReady()
    {
        // Wait for GameManager to be initialized
        while (GameManager.Instance == null)
        {
            yield return null;
        }

        Debug.Log("GameManager found, initializing UI...");

        InitializeUI();
        SubscribeToEvents();

        // Update initial UI state after a short delay to ensure everything is ready
        yield return StartCoroutine(DelayedUIUpdate());
    }

    System.Collections.IEnumerator DelayedUIUpdate()
    {
        // Wait a few frames to ensure everything is initialized
        yield return new WaitForEndOfFrame();
        yield return new WaitForEndOfFrame();

        Debug.Log("Performing delayed UI update...");
        UpdateAllUI();

        // Force update resource displays with current values
        if (GameManager.Instance?.resourceManager != null)
        {
            var resources = GameManager.Instance.resourceManager.GetAllResources();
            Debug.Log($"Current resources: Wood={resources.GetValueOrDefault(ResourceType.Wood, 0)}, Stone={resources.GetValueOrDefault(ResourceType.Stone, 0)}, Food={resources.GetValueOrDefault(ResourceType.Food, 0)}");

            // Force manual UI updates
            foreach (var kvp in resources)
            {
                UpdateResourceDisplay(kvp.Key, kvp.Value);
            }
        }
    }
    void InitializeUI()
    {
        if (mainUIDocument == null)
        {
            mainUIDocument = GetComponent<UIDocument>();
        }

        if (mainUIDocument == null)
        {
            Debug.LogError("UI Document is required for GameUI! Please assign it in the inspector.");
            return;
        }

        Debug.Log($"UI Document found: {mainUIDocument.name}");
        Debug.Log($"UI Document's visual tree asset: {(mainUIDocument.visualTreeAsset != null ? mainUIDocument.visualTreeAsset.name : "NULL")}");

        if (mainUIDocument.visualTreeAsset == null)
        {
            Debug.LogWarning("UIDocument has no Visual Tree Asset assigned. Trying to auto-load...");

            // Try to load the UXML file from Resources folder
            VisualTreeAsset uxml = Resources.Load<VisualTreeAsset>("GameUI");
            if (uxml == null)
            {
                // Try alternative paths
                uxml = Resources.Load<VisualTreeAsset>("UI/GameUI");
            }
            if (uxml == null)
            {
                Debug.Log("Trying to load with .uxml extension...");
                // Some versions need explicit extension
                uxml = Resources.Load<VisualTreeAsset>("GameUI.uxml");
            }

            if (uxml != null)
            {
                mainUIDocument.visualTreeAsset = uxml;
                Debug.Log($"Auto-loaded UXML asset: {uxml.name}");
            }
            else
            {
                Debug.LogError("Could not auto-load GameUI.uxml! Please assign it manually or move it to Resources folder.");
                AttemptAutoFixUIDocument();
                return;
            }
        }

        root = mainUIDocument.rootVisualElement;
        Debug.Log($"Root element found: {root != null}, children count: {root?.childCount ?? 0}");

        if (root != null && root.childCount == 0)
        {
            Debug.LogError($"Root element has no children! The UXML file '{mainUIDocument.visualTreeAsset.name}' may be empty or not loading properly.");
            return;
        }

        // Get resource labels
        woodLabel = root.Q<Label>("wood-amount");
        stoneLabel = root.Q<Label>("stone-amount");
        foodLabel = root.Q<Label>("food-amount");
        Debug.Log($"Resource UI found - Wood: {woodLabel != null}, Stone: {stoneLabel != null}, Food: {foodLabel != null}");

        // Get time controls
        timeSpeedLabel = root.Q<Label>("time-speed");
        pauseButton = root.Q<Button>("pause-button");
        speed1xButton = root.Q<Button>("speed-1x");
        speed2xButton = root.Q<Button>("speed-2x");
        speed10xButton = root.Q<Button>("speed-10x");
        Debug.Log($"Time UI found - Speed: {timeSpeedLabel != null}, Pause: {pauseButton != null}, 1x: {speed1xButton != null}, 2x: {speed2xButton != null}, 10x: {speed10xButton != null}");

        // Get villager panel
        villagerPanel = root.Q<VisualElement>("villager-panel");
        villagerInfoLabel = root.Q<Label>("villager-info");
        jobButtonsContainer = root.Q<VisualElement>("job-buttons");
        Debug.Log($"Villager UI found - Panel: {villagerPanel != null}, Info: {villagerInfoLabel != null}, Jobs: {jobButtonsContainer != null}");

        // Get selection panel
        selectionPanel = root.Q<VisualElement>("selection-panel");
        selectionInfoLabel = root.Q<Label>("selection-info");
        Debug.Log($"Selection UI found - Panel: {selectionPanel != null}, Info: {selectionInfoLabel != null}");

        // Get build menu elements
        buildMenuPanel = root.Q<VisualElement>("build-menu-panel");
        buildMenuToggle = root.Q<Button>("build-menu-toggle");
        buildCategoriesContainer = root.Q<VisualElement>("build-categories");
        buildItemsContainer = root.Q<VisualElement>("build-items-container");
        livingCategoryButton = root.Q<Button>("living-category");
        workingCategoryButton = root.Q<Button>("working-category");
        decorativeCategoryButton = root.Q<Button>("decorative-category");
        buildModeStatusLabel = root.Q<Label>("build-mode-status");
        Debug.Log($"Build Menu UI found - Panel: {buildMenuPanel != null}, Toggle: {buildMenuToggle != null}");

        // Get housing elements
        housingStatusLabel = root.Q<Label>("housing-status");
        housingWarningLabel = root.Q<Label>("housing-warning");
        Debug.Log($"Housing UI found - Status: {housingStatusLabel != null}, Warning: {housingWarningLabel != null}");

        // Set up button callbacks
        SetupButtons();

        // Create job buttons
        CreateJobButtons();

        // Debug: List all found UI elements
        Debug.Log($"UI Initialization Summary:");
        Debug.Log($"  Resource Labels - Wood: {woodLabel != null}, Stone: {stoneLabel != null}, Food: {foodLabel != null}");
        Debug.Log($"  Time Controls - Speed Label: {timeSpeedLabel != null}, Buttons: {pauseButton != null}");
        Debug.Log($"  Villager Panel - Panel: {villagerPanel != null}, Info: {villagerInfoLabel != null}");
        Debug.Log($"  Selection Panel - Panel: {selectionPanel != null}, Info: {selectionInfoLabel != null}");

        // If no elements found, try to auto-fix by loading the UXML file
        if (woodLabel == null && stoneLabel == null && foodLabel == null)
        {
            AttemptAutoFixUIDocument();
        }
    }

    void AttemptAutoFixUIDocument()
    {
        Debug.LogError("=== UI SETUP ISSUE DETECTED ===");
        Debug.LogError("The UIDocument has no Visual Tree Asset assigned or the UXML file is empty.");
        Debug.LogError("To fix this:");
        Debug.LogError("1. Select the GameManager GameObject in the scene");
        Debug.LogError("2. Look at the UIDocument component in the inspector");
        Debug.LogError("3. Assign the GameUI.uxml file to the 'Visual Tree Asset' field");
        Debug.LogError("4. The GameUI.uxml file should be located at Assets/UI/GameUI.uxml");
        Debug.LogError("================================");
    }

    void SetupButtons()
    {
        // Time control buttons
        if (pauseButton != null)
        {
            pauseButton.RegisterCallback<ClickEvent>(evt =>
            {
                if (GameManager.Instance != null)
                {
                    GameManager.Instance.TogglePause();
                    Debug.Log("Pause button clicked");
                }
            });
        }

        if (speed1xButton != null)
        {
            speed1xButton.RegisterCallback<ClickEvent>(evt =>
            {
                if (GameManager.Instance?.timeManager != null)
                {
                    GameManager.Instance.timeManager.SetSpeed(0);
                    Debug.Log("1x speed button clicked");
                }
            });
        }

        if (speed2xButton != null)
        {
            speed2xButton.RegisterCallback<ClickEvent>(evt =>
            {
                if (GameManager.Instance?.timeManager != null)
                {
                    GameManager.Instance.timeManager.SetSpeed(1);
                    Debug.Log("2x speed button clicked");
                }
            });
        }

        if (speed10xButton != null)
        {
            speed10xButton.RegisterCallback<ClickEvent>(evt =>
            {
                if (GameManager.Instance?.timeManager != null)
                {
                    GameManager.Instance.timeManager.SetSpeed(2);
                    Debug.Log("10x speed button clicked");
                }
            });
        }

        // Build menu toggle button
        if (buildMenuToggle != null)
        {
            buildMenuToggle.RegisterCallback<ClickEvent>(evt =>
            {
                ToggleBuildMenu();
                Debug.Log("Build menu toggle clicked");
            });
        }

        // Build category buttons
        if (livingCategoryButton != null)
        {
            livingCategoryButton.RegisterCallback<ClickEvent>(evt =>
            {
                ShowBuildCategory(BuildingCategory.Living);
                Debug.Log("Living category selected");
            });
        }

        if (workingCategoryButton != null)
        {
            workingCategoryButton.RegisterCallback<ClickEvent>(evt =>
            {
                ShowBuildCategory(BuildingCategory.Working);
                Debug.Log("Working category selected");
            });
        }

        if (decorativeCategoryButton != null)
        {
            decorativeCategoryButton.RegisterCallback<ClickEvent>(evt =>
            {
                ShowBuildCategory(BuildingCategory.Decorative);
                Debug.Log("Decorative category selected");
            });
        }
    }

    void CreateJobButtons()
    {
        if (jobButtonsContainer == null)
        {
            Debug.LogWarning("Job buttons container not found!");
            return;
        }

        // Clear existing buttons
        jobButtonsContainer.Clear();
        jobButtons.Clear();

        // Create job buttons (skip None job type)
        foreach (JobType job in System.Enum.GetValues(typeof(JobType)))
        {
            if (job == JobType.None) continue; // Skip "None" job

            Button jobButton = new Button();
            jobButton.text = GetJobDisplayName(job);
            jobButton.AddToClassList("job-button");
            jobButton.RegisterCallback<ClickEvent>(evt => AssignJobToSelectedVillager(job));

            // Initially disable all job buttons
            jobButton.SetEnabled(false);

            jobButtonsContainer.Add(jobButton);
            jobButtons.Add(jobButton);
        }

        Debug.Log($"Created {jobButtons.Count} job buttons (initially disabled)");
    }

    string GetJobDisplayName(JobType job)
    {
        return job switch
        {
            JobType.Woodcutter => "Woodcutter",
            JobType.Stonecutter => "Stonecutter",
            JobType.Farmer => "Farmer",
            JobType.Builder => "Builder",
            _ => job.ToString()
        };
    }

    void AssignJobToSelectedVillager(JobType job)
    {
        // Use stored reference instead of real-time selection to avoid deselection issues
        if (currentSelectedVillager == null)
        {
            Debug.Log("No villager stored for job assignment - this shouldn't happen if buttons are disabled properly");
            return;
        }

        if (GameManager.Instance?.villagerManager == null)
        {
            Debug.LogWarning("VillagerManager not found!");
            return;
        }

        Debug.Log($"Assigning job {job} to stored villager: {currentSelectedVillager.name}");

        // Find the closest available workplace for this job
        GameObject workplace = GameManager.Instance.villagerManager.FindWorkplaceForJob(job, currentSelectedVillager.transform.position);

        if (workplace != null)
        {
            // Assign villager to specific workplace
            GameManager.Instance.villagerManager.AssignVillagerToSpecificWorkplace(currentSelectedVillager, job, workplace);
            Debug.Log($"Assigned {currentSelectedVillager.name} to {job} at {workplace.name}");
        }
        else
        {
            // Fallback to general job assignment if no specific workplace found
            GameManager.Instance.villagerManager.AssignVillagerToJob(currentSelectedVillager, job);
            Debug.Log($"Assigned {currentSelectedVillager.name} to {job} (no specific workplace found)");
        }

        // Update the villager panel to show new job
        UpdateVillagerPanel(currentSelectedVillager);
    }
    void SubscribeToEvents()
    {
        // Subscribe to resource changes
        ResourceManager.OnResourceChanged += UpdateResourceDisplay;

        // Subscribe to time changes
        TimeManager.OnTimeSpeedChanged += UpdateTimeDisplay;
        TimeManager.OnPauseStateChanged += UpdatePauseDisplay;

        // Subscribe to villager selection
        Villager.OnVillagerSelected += UpdateVillagerPanel;

        // Subscribe to building system events
        BuildingSystem.OnBuildModeEntered += OnBuildModeEntered;
        BuildingSystem.OnBuildModeExited += OnBuildModeExited;

        // Subscribe to housing events
        HousingManager.OnHousingStatusChanged += UpdateHousingStatus;
        HousingManager.OnHousingWarning += ShowHousingWarning;

        // Get references to building system components
        buildingSystem = FindFirstObjectByType<BuildingSystem>();
        housingManager = FindFirstObjectByType<HousingManager>();

        if (buildingSystem == null)
        {
            Debug.LogWarning("BuildingSystem not found! Build menu will not work properly.");
        }
        if (housingManager == null)
        {
            Debug.LogWarning("HousingManager not found! Housing warnings will not work.");
        }
    }

    void OnDestroy()
    {
        // Unsubscribe from events
        ResourceManager.OnResourceChanged -= UpdateResourceDisplay;
        TimeManager.OnTimeSpeedChanged -= UpdateTimeDisplay;
        TimeManager.OnPauseStateChanged -= UpdatePauseDisplay;
        Villager.OnVillagerSelected -= UpdateVillagerPanel;

        // Unsubscribe from building system events
        BuildingSystem.OnBuildModeEntered -= OnBuildModeEntered;
        BuildingSystem.OnBuildModeExited -= OnBuildModeExited;

        // Unsubscribe from housing events
        HousingManager.OnHousingStatusChanged -= UpdateHousingStatus;
        HousingManager.OnHousingWarning -= ShowHousingWarning;
    }

    void UpdateAllUI()
    {
        // Update resource displays
        if (GameManager.Instance?.resourceManager != null)
        {
            var resources = GameManager.Instance.resourceManager.GetAllResources();
            foreach (var kvp in resources)
            {
                UpdateResourceDisplay(kvp.Key, kvp.Value);
            }
        }

        // Update time display
        if (GameManager.Instance?.timeManager != null)
        {
            UpdateTimeDisplay(GameManager.Instance.timeManager.CurrentTimeScale);
            UpdatePauseDisplay(GameManager.Instance.timeManager.IsPaused);
        }

        // Update selection
        UpdateSelectionPanel();
    }


    void UpdateResourceDisplay(ResourceType type, int amount)
    {
        Label targetLabel = null;
        string elementName = "";

        switch (type)
        {
            case ResourceType.Wood:
                targetLabel = woodLabel;
                elementName = "wood-amount";
                break;
            case ResourceType.Stone:
                targetLabel = stoneLabel;
                elementName = "stone-amount";
                break;
            case ResourceType.Food:
                targetLabel = foodLabel;
                elementName = "food-amount";
                break;
        }

        if (targetLabel != null)
        {
            targetLabel.text = amount.ToString();
            Debug.Log($"Updated {type} to {amount} successfully");

            // Update building button states when resources change
            UpdateBuildingButtonStates();
        }
        else
        {
            Debug.LogWarning($"Could not find UI label for {type} (looking for '{elementName}')");

            // Try to re-find the element
            if (root != null)
            {
                targetLabel = root.Q<Label>(elementName);
                if (targetLabel != null)
                {
                    targetLabel.text = amount.ToString();
                    Debug.Log($"Re-found and updated {type} to {amount}");

                    // Update the cached reference
                    switch (type)
                    {
                        case ResourceType.Wood: woodLabel = targetLabel; break;
                        case ResourceType.Stone: stoneLabel = targetLabel; break;
                        case ResourceType.Food: foodLabel = targetLabel; break;
                    }
                }
                else
                {
                    Debug.LogError($"Element '{elementName}' not found in UXML! Check that the UXML file contains an element with name='{elementName}'");
                }
            }
        }
    }
    void UpdateTimeDisplay(float timeScale)
    {
        if (timeSpeedLabel != null)
        {
            if (timeScale == 0f)
            {
                timeSpeedLabel.text = "Paused";
            }
            else if (timeScale == 1f)
            {
                timeSpeedLabel.text = "1x";
            }
            else if (timeScale == 2f)
            {
                timeSpeedLabel.text = "2x";
            }
            else if (timeScale == 10f)
            {
                timeSpeedLabel.text = "10x";
            }
            else
            {
                timeSpeedLabel.text = $"{timeScale}x";
            }
            Debug.Log($"Updated time speed to {timeSpeedLabel.text}");
        }
        else
        {
            Debug.LogWarning("Time speed label not found!");
        }
    }

    void UpdatePauseDisplay(bool isPaused)
    {
        if (pauseButton != null)
        {
            pauseButton.text = isPaused ? "Resume" : "Pause";
            Debug.Log($"Updated pause button to: {pauseButton.text}");
        }

        if (timeSpeedLabel != null && isPaused)
        {
            timeSpeedLabel.text = "Paused";
        }
    }

    public void UpdateVillagerPanel(Villager villager)
    {
        Debug.Log($"UpdateVillagerPanel called for: {(villager != null ? villager.name : "null")}");

        // Store the villager reference for job assignment
        currentSelectedVillager = villager;
        Debug.Log($"Stored villager reference: {(currentSelectedVillager != null ? currentSelectedVillager.name : "null")}");

        if (villagerInfoLabel == null)
        {
            Debug.LogWarning("villagerInfoLabel is null - cannot show villager info");
            return;
        }

        // Update job button states
        UpdateJobButtonStates(villager);

        if (villager == null)
        {
            villagerInfoLabel.text = "No villager selected";
            Debug.Log("Updated villager panel: No villager selected");
        }
        else
        {
            string jobText = villager.currentJob == JobType.None ? "Unemployed" : GetJobDisplayName(villager.currentJob);
            string info = $"Name: {villager.name}\n" +
                         $"Job: {jobText}\n" +
                         $"Experience: {villager.experience.GetExperienceForJob(villager.currentJob):F1}\n" +
                         $"State: {villager.state}";

            villagerInfoLabel.text = info;
            Debug.Log($"Updated villager panel with info: {info}");
        }
    }

    void UpdateJobButtonStates(Villager selectedVillager)
    {
        bool hasSelection = selectedVillager != null;

        // Enable/disable all job buttons based on villager selection
        foreach (var button in jobButtons)
        {
            button.SetEnabled(hasSelection);

            // Optional: Change button appearance based on villager's current job
            if (hasSelection)
            {
                string buttonJobName = button.text;
                string currentJobName = GetJobDisplayName(selectedVillager.currentJob);

                if (buttonJobName == currentJobName)
                {
                    button.AddToClassList("current-job"); // You can style this in USS
                }
                else
                {
                    button.RemoveFromClassList("current-job");
                }
            }
        }

        Debug.Log($"Job buttons {(hasSelection ? "enabled" : "disabled")} - {jobButtons.Count} buttons updated");
    }

    // Public method to manually refresh all UI elements
    public void ManualUIRefresh()
    {
        Debug.Log("Manual UI refresh requested");

        // Re-find all UI elements
        if (root != null)
        {
            woodLabel = root.Q<Label>("wood-amount");
            stoneLabel = root.Q<Label>("stone-amount");
            foodLabel = root.Q<Label>("food-amount");
            villagerInfoLabel = root.Q<Label>("villager-info");
            selectionInfoLabel = root.Q<Label>("selection-info");

            Debug.Log($"Re-found UI elements - Wood: {woodLabel != null}, Stone: {stoneLabel != null}, Food: {foodLabel != null}, VillagerInfo: {villagerInfoLabel != null}");

            // Force update all displays
            UpdateAllUI();
        }
    }

    void Update()
    {
        // Check for debug key to manually refresh UI
        if (Keyboard.current != null && Keyboard.current.f5Key.wasPressedThisFrame)
        {
            Debug.Log("F5 pressed - Manual UI refresh");
            ManualUIRefresh();
        }
    }

    void UpdateSelectionPanel()
    {
        if (selectionInfoLabel == null) return;

        var selectedObject = GameManager.Instance?.selectionManager?.SelectedObject;
        var selectedVillager = GameManager.Instance?.selectionManager?.SelectedVillager;

        if (selectedObject == null && selectedVillager == null)
        {
            selectionInfoLabel.text = "Nothing selected";
            return;
        }

        string info = "";
        if (selectedVillager != null)
        {
            info = $"Selected: {selectedVillager.name}";
        }
        else if (selectedObject != null)
        {
            // Add specific info based on object type
            Building building = selectedObject.GetComponent<Building>();
            if (building != null)
            {
                info = building.GetBuildingInfo();
            }
            else
            {
                ResourceNode node = selectedObject.GetComponent<ResourceNode>();
                if (node != null)
                {
                    info = node.GetNodeInfo();
                }
                else
                {
                    info = "Selected: " + selectedObject.name;
                }
            }
        }

        selectionInfoLabel.text = info;
    }

    // Build Menu Methods
    void ToggleBuildMenu()
    {
        // Verify UI references first
        VerifyBuildMenuReferences();

        isBuildMenuOpen = !isBuildMenuOpen;

        Debug.Log($"ToggleBuildMenu called - Menu is now {(isBuildMenuOpen ? "OPEN" : "CLOSED")}");

        // Show/hide only the build menu categories, not the entire panel (to keep toggle visible)
        if (buildCategoriesContainer != null)
        {
            buildCategoriesContainer.style.display = isBuildMenuOpen ? DisplayStyle.Flex : DisplayStyle.None;
            Debug.Log($"Build menu categories visibility set to: {(isBuildMenuOpen ? "Flex" : "None")}");
        }
        else
        {
            Debug.LogWarning("buildCategoriesContainer is null!");
        }

        // Ensure the build menu panel itself is always visible (contains the toggle button)
        if (buildMenuPanel != null)
        {
            buildMenuPanel.style.display = DisplayStyle.Flex;
        }

        if (buildMenuToggle != null)
        {
            buildMenuToggle.text = isBuildMenuOpen ? "Close Build Menu" : "Open Build Menu";
            // Ensure the toggle button itself is always visible
            buildMenuToggle.style.display = DisplayStyle.Flex;
            Debug.Log($"Build menu toggle text updated to: {buildMenuToggle.text}");
        }
        else
        {
            Debug.LogWarning("buildMenuToggle is null!");
        }

        // Hide build items when closing menu
        if (!isBuildMenuOpen && buildItemsContainer != null)
        {
            buildItemsContainer.style.display = DisplayStyle.None;
        }

        Debug.Log($"Build menu {(isBuildMenuOpen ? "opened" : "closed")}");
    }

    void VerifyBuildMenuReferences()
    {
        if (root == null)
        {
            Debug.LogError("Root UI element is null!");
            return;
        }

        if (buildMenuPanel == null)
        {
            buildMenuPanel = root.Q<VisualElement>("build-menu-panel");
            Debug.Log($"Re-found buildMenuPanel: {buildMenuPanel != null}");
        }

        if (buildMenuToggle == null)
        {
            buildMenuToggle = root.Q<Button>("build-menu-toggle");
            Debug.Log($"Re-found buildMenuToggle: {buildMenuToggle != null}");
        }

        if (buildCategoriesContainer == null)
        {
            buildCategoriesContainer = root.Q<VisualElement>("build-categories");
            Debug.Log($"Re-found buildCategoriesContainer: {buildCategoriesContainer != null}");
        }
    }

    void ShowBuildCategory(BuildingCategory category)
    {
        if (buildingSystem == null)
        {
            Debug.LogWarning("BuildingSystem not found!");
            return;
        }

        currentBuildCategory = category;

        // Show build items container
        if (buildItemsContainer != null)
        {
            buildItemsContainer.style.display = DisplayStyle.Flex;

            // Clear existing items
            buildItemsContainer.Clear();

            // Get buildings for this category
            var buildings = buildingSystem.GetBuildingsByCategory(category);

            foreach (var building in buildings)
            {
                CreateBuildingButton(building);
            }

            Debug.Log($"Showing {buildings.Count} buildings for category: {category}");
        }
    }

    void CreateBuildingButton(BuildingTemplate template)
    {
        Button buildButton = new Button();
        buildButton.text = $"{template.name}\n{buildingSystem.GetResourceCostString(template)}";
        buildButton.AddToClassList("build-item-button");

        // Check if we can afford this building
        bool canAfford = CanAffordBuilding(template);
        buildButton.SetEnabled(canAfford);

        buildButton.RegisterCallback<ClickEvent>(evt =>
        {
            Debug.Log($"Selected building to build: {template.name}");
            buildingSystem.EnterBuildMode(template);
            // Don't close menu immediately - let it close when build mode exits
        });

        buildItemsContainer.Add(buildButton);
    }

    bool CanAffordBuilding(BuildingTemplate template)
    {
        if (GameManager.Instance?.resourceManager == null) return false;

        foreach (var cost in template.costs)
        {
            if (!GameManager.Instance.resourceManager.HasResource(cost.resourceType, cost.amount))
            {
                return false;
            }
        }
        return true;
    }

    void UpdateBuildingButtonStates()
    {
        if (buildItemsContainer == null || buildingSystem == null) return;

        var buttons = buildItemsContainer.Query<Button>().ToList();
        var templates = buildingSystem.GetBuildingTemplates();

        for (int i = 0; i < buttons.Count && i < templates.Count; i++)
        {
            bool canAfford = CanAffordBuilding(templates[i]);
            buttons[i].SetEnabled(canAfford);
        }
    }

    void OnBuildModeEntered()
    {
        if (buildModeStatusLabel != null)
        {
            buildModeStatusLabel.text = "Build Mode Active - Left click to place, Right click or ESC to cancel";
            buildModeStatusLabel.style.display = DisplayStyle.Flex;
        }
    }

    void OnBuildModeExited()
    {
        if (buildModeStatusLabel != null)
        {
            buildModeStatusLabel.text = "";
            buildModeStatusLabel.style.display = DisplayStyle.None;
        }

        // Keep build menu open if it was open
        if (isBuildMenuOpen)
        {
            if (buildCategoriesContainer != null)
            {
                buildCategoriesContainer.style.display = DisplayStyle.Flex;
                Debug.Log("Build menu categories visibility preserved as Flex");
            }
            if (buildMenuToggle != null)
            {
                buildMenuToggle.text = "Close Build Menu";
            }
        }

        // Ensure build menu panel is always visible (contains toggle button)
        if (buildMenuPanel != null)
        {
            buildMenuPanel.style.display = DisplayStyle.Flex;
        }

        Debug.Log($"Build mode exited - UI status cleared, build menu state: {(isBuildMenuOpen ? "OPEN" : "CLOSED")}");
    }

    // Public method to force build menu visibility check
    public void EnsureBuildMenuAccessible()
    {
        VerifyBuildMenuReferences();

        if (buildMenuToggle != null)
        {
            buildMenuToggle.style.display = DisplayStyle.Flex;
            Debug.Log("Ensured build menu toggle is visible");
        }

        // Ensure build menu panel is always visible (contains toggle button)
        if (buildMenuPanel != null)
        {
            buildMenuPanel.style.display = DisplayStyle.Flex;
            Debug.Log("Ensured build menu panel is visible");
        }

        // If menu was supposed to be open, make sure categories are visible
        if (isBuildMenuOpen && buildCategoriesContainer != null)
        {
            buildCategoriesContainer.style.display = DisplayStyle.Flex;
            Debug.Log("Ensured build menu categories are visible");
        }
    }

    // Debug method to check build menu state
    [System.Diagnostics.Conditional("UNITY_EDITOR")]
    public void DebugBuildMenuState()
    {
        Debug.Log("=== BUILD MENU DEBUG INFO ===");
        Debug.Log($"isBuildMenuOpen: {isBuildMenuOpen}");
        Debug.Log($"buildMenuPanel: {(buildMenuPanel != null ? "FOUND" : "NULL")}");
        Debug.Log($"buildMenuToggle: {(buildMenuToggle != null ? "FOUND" : "NULL")}");
        Debug.Log($"buildCategoriesContainer: {(buildCategoriesContainer != null ? "FOUND" : "NULL")}");
        Debug.Log($"buildItemsContainer: {(buildItemsContainer != null ? "FOUND" : "NULL")}");

        if (buildMenuPanel != null)
        {
            Debug.Log($"buildMenuPanel display: {buildMenuPanel.style.display.value}");
        }
        if (buildMenuToggle != null)
        {
            Debug.Log($"buildMenuToggle display: {buildMenuToggle.style.display.value}");
            Debug.Log($"buildMenuToggle text: {buildMenuToggle.text}");
        }
        Debug.Log("=============================");
    }

    // Housing System Methods
    void UpdateHousingStatus(int housedVillagers, int totalVillagers)
    {
        if (housingStatusLabel != null)
        {
            housingStatusLabel.text = $"Housing: {housedVillagers}/{totalVillagers} villagers housed";
        }

        // Hide warning if housing is adequate
        if (housedVillagers >= totalVillagers && housingWarningLabel != null)
        {
            housingWarningLabel.style.display = DisplayStyle.None;
        }
    }

    void ShowHousingWarning(string warningMessage)
    {
        if (housingWarningLabel != null)
        {
            housingWarningLabel.text = warningMessage;
            housingWarningLabel.style.display = DisplayStyle.Flex;
        }
    }
}