EvolutionDojoHandler.cs

using Il2Cpp;
using UnityEngine;

namespace DigimonNOAccess
{
    public class EvolutionDojoHandler : IAccessibilityHandler
    {
        private const string LogTag = "[EvoDojo]";
        public int Priority => 70;

        private uEvolutionDojoPanel _panel;
        private bool _wasOpen;
        private int _lastCursorX = -1;
        private int _lastCursorY = -1;
        private int _lastSelectPartner = -1;
        private uGenelogyUI.Type _lastType = uGenelogyUI.Type.None;
        private bool _lastEvoAfter = false;
        private string _lastAnnouncedName = "";

        private static readonly string[] GrowthNames =
            { "Baby 1", "Baby 2", "Rookie", "Champion", "Ultimate", "Mega", "Ultra" };

        public bool IsOpen()
        {
            try
            {
                _panel = uEvolutionDojoPanel.Ref;
                if (_panel == null) return false;
                return _panel.m_State == uEvolutionDojoPanel.State.Normal;
            }
            catch
            {
                _panel = null;
                return false;
            }
        }

        public void Update()
        {
            bool isOpen = IsOpen();

            if (isOpen && !_wasOpen)
                OnOpen();
            else if (!isOpen && _wasOpen)
                OnClose();
            else if (isOpen)
                OnUpdate();

            _wasOpen = isOpen;
        }

        private void OnOpen()
        {
            _lastCursorX = -1;
            _lastCursorY = -1;
            _lastSelectPartner = -1;
            _lastType = uGenelogyUI.Type.None;
            _lastEvoAfter = false;
            _lastAnnouncedName = "";

            var genelogy = GetGenelogy();
            if (genelogy == null)
            {
                ScreenReader.Say(GetPanelTitle());
                return;
            }

            _lastSelectPartner = _panel.m_SelectPartner;
            _lastType = genelogy.m_Type;

            string generation = GetGenerationText();
            string announcement = GetPanelTitle();
            if (!string.IsNullOrEmpty(generation))
                announcement += ", " + generation;

            ScreenReader.Say(announcement);
            DebugLogger.Log($"{LogTag} Opened: {announcement}");

            // Announce cursor position (name, stage, probability, index, description)
            _lastCursorX = genelogy.m_CursorX;
            _lastCursorY = genelogy.m_CursorY;
            AnnounceCursorItem(genelogy, queued: true);
        }

        private void OnClose()
        {
            _panel = null;
            _lastCursorX = -1;
            _lastCursorY = -1;
            _lastSelectPartner = -1;
            _lastAnnouncedName = "";
            DebugLogger.Log($"{LogTag} Closed");
        }

        private void OnUpdate()
        {
            if (_panel == null) return;

            var genelogy = GetGenelogy();
            if (genelogy == null) return;

            try
            {
                // Check partner switch
                int currentPartner = _panel.m_SelectPartner;
                if (currentPartner != _lastSelectPartner && _lastSelectPartner >= 0)
                {
                    _lastSelectPartner = currentPartner;
                    _lastAnnouncedName = "";
                    _lastCursorX = genelogy.m_CursorX;
                    _lastCursorY = genelogy.m_CursorY;
                    AnnounceCursorItem(genelogy);
                    return;
                }
                _lastSelectPartner = currentPartner;

                // Check type change (Dojo vs History)
                var currentType = genelogy.m_Type;
                if (currentType != _lastType)
                {
                    DebugLogger.Log($"{LogTag} Type changed: {_lastType} -> {currentType}");
                    _lastType = currentType;
                }

                // Check evo after state change (entering/exiting future evolution view)
                bool evoAfter = genelogy.m_SelectEvoAfter;
                if (evoAfter != _lastEvoAfter)
                {
                    _lastEvoAfter = evoAfter;
                    _lastAnnouncedName = "";
                }

                // Check cursor movement
                int cx = genelogy.m_CursorX;
                int cy = genelogy.m_CursorY;
                if (cx != _lastCursorX || cy != _lastCursorY)
                {
                    _lastCursorX = cx;
                    _lastCursorY = cy;
                    AnnounceCursorItem(genelogy);
                }
            }
            catch (System.Exception ex)
            {
                DebugLogger.Error($"{LogTag} Update error: {ex.Message}");
            }
        }

        private void AnnounceCursorItem(uGenelogyUI genelogy, bool queued = false)
        {
            bool evoAfter = genelogy.m_SelectEvoAfter;

            // When viewing future evolutions, read evo name; otherwise read current name
            string name;
            if (evoAfter)
            {
                name = GetInfoPanelEvoName();
                if (string.IsNullOrEmpty(name))
                    name = GetInfoPanelName();
            }
            else
            {
                name = GetInfoPanelName();
                if (string.IsNullOrEmpty(name))
                    name = GetInfoPanelEvoName();
            }

            // Avoid repeating the exact same announcement
            if (name == _lastAnnouncedName && !string.IsNullOrEmpty(name))
                return;
            _lastAnnouncedName = name ?? "";

            if (string.IsNullOrEmpty(name))
            {
                DebugLogger.Log($"{LogTag} Cursor ({_lastCursorX},{_lastCursorY}): empty cell");
                return;
            }

            string announcement = name;

            // Add growth stage
            string stage = GetGrowthStageForName(name);
            if (!string.IsNullOrEmpty(stage))
                announcement += ", " + stage;

            // Add probability
            string probability = GetProbability(genelogy);
            if (!string.IsNullOrEmpty(probability))
                announcement += ", " + probability;

            // Add locked status
            if (IsCurrentLocked(genelogy))
                announcement += ", Locked";

            // Add grid position context
            int maxX = genelogy.m_CursorX_Max;
            int maxY = genelogy.m_CursorY_Max;
            if (maxX > 0 || maxY > 0)
                announcement += $", {_lastCursorX + 1} of {maxX + 1}";

            // Add description after index (only for past/current forms - not for future evos)
            if (!evoAfter)
            {
                string description = GetInfoPanelText("m_Description");
                if (!string.IsNullOrEmpty(description))
                    announcement += ", " + description;
            }

            // For future evolutions, add nature/attribute and conditions
            if (evoAfter)
            {
                string nature = GetInfoPanelText("m_Nature_Evo");
                if (!string.IsNullOrEmpty(nature))
                    announcement += $", {GetInfoLabel(true, true)}: {nature}";

                string property = GetInfoPanelText("m_Property_Evo");
                if (!string.IsNullOrEmpty(property))
                    announcement += $", {GetInfoLabel(false, true)}: {property}";

                string conditions = GetConditionsText();
                if (!string.IsNullOrEmpty(conditions))
                    announcement += ", Conditions: " + conditions;
            }

            if (queued)
                ScreenReader.SayQueued(announcement);
            else
                ScreenReader.Say(announcement);
            DebugLogger.Log($"{LogTag} Cursor ({_lastCursorX},{_lastCursorY}): {announcement}");
        }

        public void AnnounceStatus()
        {
            if (!IsOpen()) return;

            var genelogy = GetGenelogy();
            if (genelogy == null)
            {
                ScreenReader.Say(GetPanelTitle());
                return;
            }

            string announcement = GetPanelTitle();

            // Partner
            string partnerName = GetCurrentPartnerName();
            if (!string.IsNullOrEmpty(partnerName))
                announcement += ", " + partnerName;

            // Current digimon name
            string name = GetInfoPanelName();
            if (!string.IsNullOrEmpty(name))
                announcement += ", " + name;

            // Nature and property
            string nature = GetInfoPanelText("m_Nature");
            if (!string.IsNullOrEmpty(nature))
                announcement += $", {GetInfoLabel(true, false)}: {nature}";

            string property = GetInfoPanelText("m_Property");
            if (!string.IsNullOrEmpty(property))
                announcement += $", {GetInfoLabel(false, false)}: {property}";

            // Description
            string description = GetInfoPanelText("m_Description");
            if (!string.IsNullOrEmpty(description))
                announcement += ", " + description;

            // Growth stage
            string stage = GetGrowthStageForName(name);
            if (!string.IsNullOrEmpty(stage))
                announcement += ", Stage: " + stage;

            // Probability
            string probability = GetProbability(genelogy);
            if (!string.IsNullOrEmpty(probability))
                announcement += ", " + probability;

            // Locked
            if (IsCurrentLocked(genelogy))
                announcement += ", Locked";

            // Evolution conditions
            string conditions = GetConditionsText();
            if (!string.IsNullOrEmpty(conditions))
                announcement += ", Conditions: " + conditions;

            // Evolution target
            string evoName = GetInfoPanelEvoName();
            if (!string.IsNullOrEmpty(evoName) && evoName != name)
                announcement += ", Evolves to: " + evoName;

            // Generation
            string generation = GetGenerationText();
            if (!string.IsNullOrEmpty(generation))
                announcement += ", " + generation;

            // Grid position
            announcement += $", Position {_lastCursorX + 1} of {genelogy.m_CursorX_Max + 1}";

            ScreenReader.Say(announcement);
            DebugLogger.Log($"{LogTag} Status: {announcement}");
        }

        // --- Helpers ---

        private string GetPanelTitle()
        {
            const string fallback = "Evolution Dojo";
            try
            {
                if (_panel == null)
                {
                    DebugLogger.Log($"{LogTag} Title: m_panel was null");
                    return fallback;
                }

                var headLine = _panel.m_HeadLine;
                if (headLine == null)
                {
                    DebugLogger.Log($"{LogTag} Title: m_HeadLine was null");
                    return fallback;
                }

                return GetRenderedText(headLine.m_ScreenNameText, "m_HeadLine.m_ScreenNameText", fallback);
            }
            catch (System.Exception ex)
            {
                DebugLogger.Log($"{LogTag} Title read failed: {ex.Message}");
                return fallback;
            }
        }

        private string GetInfoLabel(bool nature, bool evo)
        {
            string fallback = nature ? "Nature" : "Attribute";
            try
            {
                if (_panel == null)
                {
                    DebugLogger.Log($"{LogTag} {fallback} label: m_panel was null");
                    return fallback;
                }

                var genelogy = _panel.m_Genelogy;
                if (genelogy == null)
                {
                    DebugLogger.Log($"{LogTag} {fallback} label: m_Genelogy was null");
                    return fallback;
                }

                var info = genelogy.m_GenelogyInfo;
                if (info == null)
                {
                    DebugLogger.Log($"{LogTag} {fallback} label: m_GenelogyInfo was null");
                    return fallback;
                }

                UnityEngine.UI.Text label;
                string fieldName;
                if (nature)
                {
                    label = evo ? info.m_NatureTitle_Evo : info.m_NatureTitle;
                    fieldName = evo ? "m_NatureTitle_Evo" : "m_NatureTitle";
                }
                else
                {
                    label = evo ? info.m_PropertyTitle_Evo : info.m_PropertyTitle;
                    fieldName = evo ? "m_PropertyTitle_Evo" : "m_PropertyTitle";
                }

                return GetRenderedText(label, $"m_GenelogyInfo.{fieldName}", fallback)
                    .TrimEnd(':', '：')
                    .Trim();
            }
            catch (System.Exception ex)
            {
                DebugLogger.Log($"{LogTag} {fallback} label read failed: {ex.Message}");
                return fallback;
            }
        }

        private string GetRenderedText(UnityEngine.UI.Text textComponent, string fieldName, string fallback)
        {
            if (textComponent == null)
            {
                DebugLogger.Log($"{LogTag} {fieldName} was null");
                return fallback;
            }

            try
            {
                string cleaned = TextUtilities.StripRichTextTags(ButtonHintCache.Filter(textComponent.text))?.Trim();
                if (string.IsNullOrWhiteSpace(cleaned) || TextUtilities.IsPlaceholderText(cleaned))
                {
                    DebugLogger.Log($"{LogTag} {fieldName}.text unusable: {TextUtilities.DescribeUnusable(cleaned)}");
                    return fallback;
                }

                return cleaned;
            }
            catch (System.Exception ex)
            {
                DebugLogger.Log($"{LogTag} {fieldName} read failed: {ex.Message}");
                return fallback;
            }
        }

        private uGenelogyUI GetGenelogy()
        {
            try
            {
                return _panel?.m_Genelogy;
            }
            catch { return null; }
        }

        private uGenelogyInformationUI GetInfoPanel()
        {
            try
            {
                return GetGenelogy()?.m_GenelogyInfo;
            }
            catch { return null; }
        }

        private string GetInfoPanelName()
        {
            try
            {
                var info = GetInfoPanel();
                if (info == null) return null;
                var nameText = info.m_DigimonName;
                if (nameText == null) return null;
                string text = nameText.text;
                if (string.IsNullOrEmpty(text)) return null;
                return TextUtilities.StripRichTextTags(text);
            }
            catch { return null; }
        }

        private string GetInfoPanelEvoName()
        {
            try
            {
                var info = GetInfoPanel();
                if (info == null) return null;
                var nameText = info.m_DigimonName_Evo;
                if (nameText == null) return null;
                string text = nameText.text;
                if (string.IsNullOrEmpty(text)) return null;
                return TextUtilities.StripRichTextTags(text);
            }
            catch { return null; }
        }

        private string GetInfoPanelText(string fieldName)
        {
            try
            {
                var info = GetInfoPanel();
                if (info == null) return null;

                UnityEngine.UI.Text textComp = fieldName switch
                {
                    "m_Nature" => info.m_Nature,
                    "m_Property" => info.m_Property,
                    "m_Description" => info.m_Description,
                    "m_Nature_Evo" => info.m_Nature_Evo,
                    "m_Property_Evo" => info.m_Property_Evo,
                    _ => null
                };

                if (textComp == null) return null;
                string text = textComp.text;
                if (string.IsNullOrEmpty(text)) return null;
                return TextUtilities.StripRichTextTags(text);
            }
            catch { return null; }
        }

        private string GetProbability(uGenelogyUI genelogy)
        {
            try
            {
                // The game shows one of these GameObjects depending on probability
                if (genelogy.m_Fix != null && genelogy.m_Fix.activeSelf)
                    return "Guaranteed";
                if (genelogy.m_High != null && genelogy.m_High.activeSelf)
                    return "High chance";
                if (genelogy.m_Mid != null && genelogy.m_Mid.activeSelf)
                    return "Medium chance";
                if (genelogy.m_Low != null && genelogy.m_Low.activeSelf)
                    return "Low chance";

                // Check the Possibility parent - if not active, no probability shown
                if (genelogy.m_Possibility != null && !genelogy.m_Possibility.activeSelf)
                    return null;

                return null;
            }
            catch { return null; }
        }

        private bool IsCurrentLocked(uGenelogyUI genelogy)
        {
            try
            {
                if (genelogy.m_Rock != null && genelogy.m_Rock.activeSelf)
                    return true;
                return false;
            }
            catch { return false; }
        }

        private string GetGrowthStageForName(string digimonName)
        {
            if (string.IsNullOrEmpty(digimonName)) return null;

            try
            {
                var paramManager = AppMainScript.parameterManager;
                if (paramManager == null) return null;
                var digimonData = paramManager.digimonData;
                if (digimonData == null) return null;

                int count = digimonData.GetRecordMax();
                for (int i = 0; i < count; i++)
                {
                    try
                    {
                        var param = digimonData.GetParams(i);
                        if (param == null) continue;
                        string name = param.GetDefaultName();
                        if (name == digimonName)
                        {
                            int growth = param.m_growth;
                            if (growth >= 0 && growth < GrowthNames.Length)
                                return GrowthNames[growth];
                            return null;
                        }
                    }
                    catch { }
                }
                return null;
            }
            catch { return null; }
        }

        private string GetConditionsText()
        {
            try
            {
                var info = GetInfoPanel();
                if (info == null) return null;

                var titles = info.m_ConditionsTitleText;
                var values = info.m_ConditionsValueText;
                if (titles == null || values == null) return null;

                var parts = new System.Collections.Generic.List<string>();
                int count = System.Math.Min(titles.Length, values.Length);

                for (int i = 0; i < count; i++)
                {
                    try
                    {
                        if (titles[i] == null || values[i] == null) continue;
                        string title = titles[i].text;
                        string value = values[i].text;
                        if (string.IsNullOrEmpty(title) || string.IsNullOrEmpty(value)) continue;

                        title = TextUtilities.StripRichTextTags(title);
                        value = TextUtilities.StripRichTextTags(value);

                        // Replace universal visual symbols with speech-friendly text
                        if (value == "???" || value == "？？？")
                            value = "not met";
                        else if (value == "—" || value == "－" || value == "-" || value == "―")
                            value = "met";

                        parts.Add($"{title} {value}");
                    }
                    catch { }
                }

                return parts.Count > 0 ? string.Join(", ", parts) : null;
            }
            catch { return null; }
        }

        private string GetGenerationText()
        {
            try
            {
                if (_panel == null)
                {
                    DebugLogger.Log($"{LogTag} Generation: m_panel was null");
                    return null;
                }

                var genelogy = _panel.m_Genelogy;
                if (genelogy == null)
                {
                    DebugLogger.Log($"{LogTag} Generation: m_Genelogy was null");
                    return null;
                }

                var text = genelogy.m_GenerationText;
                if (text == null)
                {
                    DebugLogger.Log($"{LogTag} Generation: m_GenerationText was null");
                    return null;
                }

                string value = TextUtilities.StripRichTextTags(ButtonHintCache.Filter(text.text))?.Trim();
                if (string.IsNullOrWhiteSpace(value) || TextUtilities.IsPlaceholderText(value))
                {
                    DebugLogger.Log($"{LogTag} Generation: m_GenerationText.text unusable: {TextUtilities.DescribeUnusable(value)}");
                    return null;
                }

                string unit = GetRenderedText(genelogy.m_GenerationUnit, "m_GenerationUnit", "Generation");
                return $"{unit} {value}";
            }
            catch (System.Exception ex)
            {
                DebugLogger.Log($"{LogTag} Generation read failed: {ex.Message}");
                return null;
            }
        }

        private string GetCurrentPartnerName()
        {
            try
            {
                if (_panel == null) return null;
                int partner = _panel.m_SelectPartner;

                // Dojo panel uses m_SelectPartner directly (not inverted like uPartnerPanel)
                var partnerCtrl = MainGameManager.GetPartnerCtrl(partner);
                if (partnerCtrl != null)
                {
                    var commonData = partnerCtrl.gameData?.m_commonData;
                    if (commonData != null)
                    {
                        string name = commonData.m_name;
                        if (!string.IsNullOrEmpty(name) && !name.Contains("ランゲージ"))
                            return name;
                    }
                }

                return PartnerUtilities.GetPartnerLabel(partner);
            }
            catch
            {
                return _panel != null ? PartnerUtilities.GetPartnerLabel(_panel.m_SelectPartner) : "Partner";
            }
        }
    }
}