ShadowsOfDoubtMods/DialogUIRework.cs at master · piepieonline/ShadowsOfDoubtMods · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
    [HarmonyPatch(typeof(InteractionController), nameof(InteractionController.RefreshDialogOptions))]
    public class InteractionController_RefreshDialogOptions
    {
        public static CategorizedDialogUI CategorizedUI;
        private static bool _inRefreshDialogOptions;

        public static void Postfix()
        {
            if (_inRefreshDialogOptions)
                return;

            _inRefreshDialogOptions = true;
            try
            {
                DebugModPlugin.PluginLogger.LogInfo("----------------- Refreshing dialog options -----------------");
                // PrefabControls.Instance.dialogOption, PrefabControls.Instance.dialogOptionContainer

    private GameObject _lastDialogContainer;
    private int _lastChildCount = -1;
    private bool _isBuilding;

                if (CategorizedUI == null)
                {
                    CategorizedUI = new CategorizedDialogUI();
                    PrefabControls.Instance.dialogOptionContainer.parent.Find("Header").localPosition += new Vector3(0, 45, 0);
                    PrefabControls.Instance.dialogOptionContainer.parent.Find("Border").gameObject.SetActive(false);
                    Il2CppInterop.Runtime.Injection.ClassInjector.RegisterTypeInIl2Cpp<TabNavigationComponent>();
                    var nav = PrefabControls.Instance.dialogOptionContainer.gameObject.AddComponent<TabNavigationComponent>();
                    nav.dialogUI = CategorizedUI;
                }
                CategorizedUI.ModifyDialogContainer(PrefabControls.Instance.dialogOptionContainer.gameObject);
            }
            finally
            {
                _inRefreshDialogOptions = false;
            }
        }
    }

    public void EnsureFreshOptions(GameObject dialogContainer)
    {
        if (_isBuilding)
            return;

        if (dialogContainer == null)
            return;

        // If container changed or child count changed, our cached references are likely stale.
        if (!ReferenceEquals(_lastDialogContainer, dialogContainer) ||
            _lastChildCount != dialogContainer.transform.childCount ||
            categorizedOptions == null ||
            HasNullOptionReference())
        {
            ModifyDialogContainer(dialogContainer);

            // If current category vanished after rebuild, fall back safely.
            if (currentCategory == null || categoryOrder == null || !categoryOrder.Contains(currentCategory))
                SwitchToTab("All");
        }
    }

    private bool HasNullOptionReference()
    {
        if (categorizedOptions == null)
            return true;

        foreach (var kvp in categorizedOptions)
        {
            var list = kvp.Value;
            if (list == null)
                return true;

            for (int i = 0; i < list.Count; i++)
            {
                if (list[i] == null)
                    return true;
            }
        }
        return false;
    }