ModMain.cs

using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using BandTogether.Debug;
using BandTogether.Quantum;
using BandTogether.Util;
using HarmonyLib;
using NewHorizons.External;
using OWML.Common;
using OWML.ModHelper;
using OWML.Utils;
using UnityEngine;
using static BandTogether.Quantum.QuantumGroup;
using static BandTogether.Quantum.QuantumTarget;

namespace BandTogether;
public class ModMain : ModBehaviour
{
    private static readonly IDictionary<string, (QuantumGroup[] groups, QuantumTarget destination)> GroupDialogueConditions =
        new Dictionary<string, (QuantumGroup[], QuantumTarget)>
        {
            { "BT_NOMAI_VILLAGE_A_TO_DOOR", (new[] { NomaiA }, Door) },
            { "BT_NOMAI_VILLAGE_B_TO_DOOR", (new[] { NomaiB }, Door) },
            { "BT_GHIRD_VILLAGE_A_TO_DOOR", (new[] { GhirdA }, Door) },
            { "BT_GHIRD_VILLAGE_B_TO_DOOR", (new[] { GhirdB }, Door) },

            { "BT_SEARCHED_GREAT_DOOR", (new[] { Captial }, Door) },
            { "BT_CLANS_LEAVE_DOOR", (new[] { NomaiA, NomaiB, GhirdA, GhirdB }, Away) },

            { "BT_DOORKEEPER_TO_FIRE", (new[] { Captial }, Away) },
            { "NOMAI_TO_FIRE", (new[] { NomaiA, NomaiB }, Fire) },
            { "GHIRD_TO_FIRE", (new[] { GhirdA, GhirdB }, Fire) },
        };

    private static readonly IDictionary<string, QuantumGroup> ShardConditions = new Dictionary<string, QuantumGroup>
    {
        { "BT_GOT_NOMAI_SHARD_A", NomaiA },
        { "BT_GOT_NOMAI_SHARD_B", NomaiB },
        { "BT_GOT_GHIRD_SHARD_A", GhirdA },
        { "BT_GOT_GHIRD_SHARD_B", GhirdB },
    };

    private static readonly List<string> FifthShardFactNames = new()
    {
        "BT_FIFTH_SHARD_POS",
        "BT_FIFTH_SHARD_QUANTUM",
        "BT_FIFTH_SHARD_PILLAR",
        "BT_FIFTH_SHARD_LIGHT"
    };

    public static ModMain Instance;
    public delegate void MoveNpcEvent(QuantumGroup targetGroup, QuantumTarget targetType, bool ignoreVisibility);
    public event MoveNpcEvent OnMoveGroup;
    public delegate void ModStartEvent();
    public event ModStartEvent OnMainQuest;
    public delegate void MoveGatekeeperEvent();
    public event MoveGatekeeperEvent OnCompleteAccidentalCode;

    public delegate void ShardFoundEvent(QuantumGroup shardGroup);
    public event ShardFoundEvent OnShardFound;

    public delegate void DialogueConditionChanged(string condition, bool value);

    public INewHorizons nhAPI;
    public bool inEndSequence = false;
    public bool fadeEndMusic = false;
    public bool startedEndSequence = false;
    public ItemType shrubberyType { get; private set; }
    public GameObject Planet { get; private set; }
    public List<SingleLightSensor> ghostLightSensors { get; private set; } = [];

    private bool _debugEnabled = true;
    //private bool _setFlashlightListDirty = false;
    private int _numClansConvinced;
    private bool _allFifthShardFacts;
    private List<GameObject> factsToEnable = new();
    private List<ScrollItem> scrollsToEnable = new();
    private List<GameObject> textToEnable = new();
    private DebugMenu _debugMenu;
    private SignalFrequency _planetFrequency;
    private readonly IDictionary<string, List<DialogueConditionChanged>> _dialogueConditionListeners =
        new Dictionary<string, List<DialogueConditionChanged>>();

    private readonly IDictionary<QuantumGroup, QuantumTarget> _groupCurrentLocation = GroupDialogueConditions
        .Values
        .SelectMany(value => value.groups)
        .Distinct()
        .ToDictionary(key => key, _ => QuantumTarget.Start);

    private void Awake()
    {
        Instance = this;
        Harmony.CreateAndPatchAll(Assembly.GetExecutingAssembly());
    }

    private void Start()
    {
        nhAPI = ModHelper.Interaction.TryGetModApi<INewHorizons>("xen.NewHorizons");
        nhAPI.LoadConfigs(this);

        if (!EnumUtils.IsDefined<ItemType>("Shrubbery"))
        {
            shrubberyType = EnumUtils.Create<ItemType>("Shrubbery");
        }

        LoadManager.OnCompleteSceneLoad += (scene, loadScene) =>
        {
            if (loadScene != OWScene.SolarSystem || nhAPI.GetCurrentStarSystem() != "Jam3")
            {
                return;
            }

            ModHelper.Console.WriteLine("Loaded into solar system!", MessageType.Success);
            //WriteDebugMessage("Lantern: " + ReferenceLocator.GetDreamLanternItem());

            nhAPI.GetBodyLoadedEvent().AddListener(OnBodyLoaded);

            GlobalMessenger<string, bool>.AddListener("DialogueConditionChanged", OnDialogueConditionChanged);
            AddDialogueConditionListener(OnMainQuestStart, "BT_MAIN_QUEST_START");
            AddDialogueConditionListener(OnShardCondition, ShardConditions.Keys.ToArray());
            AddDialogueConditionListener(OnGroupMoveCondition, GroupDialogueConditions.Keys.ToArray());
            AddDialogueConditionListener(OnGatekeeperToDoor, "BT_SUNPOST_PUZZLE_SOLVED");

            ModHelper.Events.Unity.FireInNUpdates(() =>
            {
                FifthShardFactNames
                    .Select(factName => Locator.GetShipLogManager().GetFact(factName))
                    .WhereNotNull()
                    .ForEach(fact => fact.OnFactRevealed += OnFifthShardFactRevealed);
                OnFifthShardFactRevealed();

                var relativeLocation = new RelativeLocationData(Vector3.up * 2 + Vector3.forward * 2, Quaternion.identity, Vector3.zero);

                //nessesary for owlks, lets them work somehow????????
                var location = DreamArrivalPoint.Location.Zone3;
                var arrivalPoint = Locator.GetDreamArrivalPoint(location);
                var dreamCampfire = Locator.GetDreamCampfire(location);
                if (Locator.GetToolModeSwapper().GetItemCarryTool().GetHeldItemType() != ItemType.DreamLantern)
                {
                    DreamLanternItem lantern = ReferenceLocator.GetDreamLanternItem();
                    lantern.OnEnterDreamWorld();
                    Locator.GetDreamWorldController()._playerLantern = lantern;
                };

                //Puts player into a semi-dreamworld state
                Locator.GetDreamWorldController().EnterDreamWorld(dreamCampfire, arrivalPoint, relativeLocation);

                ghostLightSensors = [];

                ReferenceLocator.GetWorshipGhosts().ForEach((brain) =>
                {
                    brain.enabled = true;
                    brain.GetComponentsInChildren<SingleLightSensor>().ForEach((sensor) => ghostLightSensors.Add(sensor));
                    brain.WakeUp();
                    brain.OnEnterDreamWorld();
                });
            }, 15);
        };

        LoadManager.OnStartSceneLoad += (scene, loadScene) =>
        {
            if (scene == OWScene.SolarSystem)
            {
                nhAPI.GetBodyLoadedEvent().RemoveListener(OnBodyLoaded);
                FifthShardFactNames
                    .Select(factName => Locator.GetShipLogManager().GetFact(factName))
                    .WhereNotNull()
                    .ForEach(fact => fact.OnFactRevealed -= OnFifthShardFactRevealed);
                inEndSequence = false;
                fadeEndMusic = false;
                startedEndSequence = false;
                factsToEnable.Clear();
                scrollsToEnable.Clear();
                textToEnable.Clear();
                ghostLightSensors.Clear();
            }
        };
    }

    private void OnBodyLoaded(string bodyName)
    {
        if (bodyName == "Fractured Harmony")
        {
            _debugEnabled = PlayerData.GetPersistentCondition("BAND_TOGETHER_DEBUG_ENABLED");

            Planet = nhAPI.GetPlanet(bodyName);

            if (IsDebugEnabled()) InitDebugMenu();

            ReferenceLocator.GetSacredEntryway().LoadWaterObject(Planet);

            ReferenceLocator.GetCreditsSong().transform.parent = null;

            ModHelper.Events.Unity.FireOnNextUpdate(() =>
            {
                InitializeConditions();
            });
        }
    }

    public void InitDebugMenu()
    {
        _debugMenu = DebugMenu.InitMenu(Planet);
    }

    private void InitializeConditions()
    {
        WriteDebugMessage("init conditions");

        if (!GetPersistentCondition("BT_MAIN_QUEST_START"))
        {
            Planet
                .GetComponentsInChildren<ShipLogFactTriggerVolume>()
                .ForEach(factTrigger =>
                {
                    factsToEnable.Add(factTrigger.gameObject);
                    factTrigger.gameObject.SetActive(false);
                });

            Planet.GetComponentsInChildren<ScrollItem>().ForEach(scrollItem =>
            {
                scrollsToEnable.Add(scrollItem);
                scrollItem.EnableInteraction(false);
            });

            Planet.GetComponentsInChildren<NomaiWallText>().ForEach(wallText =>
            {
                textToEnable.Add(wallText.gameObject);
                wallText.gameObject.SetActive(false);
            });
        }

        ShardConditions
            .Where(condition => GetPersistentCondition(condition.Key))
            .ForEach(condition =>
            {
                OnMoveGroup?.Invoke(condition.Value, Door, true);
                OnShardFound?.Invoke(condition.Value);
                _groupCurrentLocation[condition.Value] = Door;
                _numClansConvinced++;
            });

        if (GetPersistentCondition("BT_SHRUB_GIVEN_TO_NOMAI"))
        {
            ShrubberySocketNomai socketNomai = ReferenceLocator.GetShrubSocketNomai();
            socketNomai.PlaceIntoSocket(ReferenceLocator.GetShrubbery());
            socketNomai.EnableInteraction(false);
        }
        else if (GetPersistentCondition("BT_FINISH_SHRUB_QUEST"))
        {
            TheDivineThrone socketThrone = ReferenceLocator.GetShrubSocketThrone();
            socketThrone.PlaceIntoSocket(ReferenceLocator.GetShrubbery());
            if (!GetPersistentCondition("BT_START_STEAL_QUEST"))
            {
                socketThrone.EnableInteraction(false);
            }
        }

        if (GetPersistentCondition("BT_OPEN_SUNPOST_DOOR"))
        {
            ReferenceLocator.GetSunpostDetector().OpenDoor();
        }

        NewHorizonsData.LearnFrequency("Wide-Band Transmission");

        ReferenceLocator.GetLightsOutController().InitializeGhirds();
    }

    private void OnDialogueConditionChanged(string condition, bool value)
    {
        WriteDebugMessage($"condition: {condition}");
        if (_dialogueConditionListeners.TryGetValue(condition, out var listeners))
        {
            listeners.ForEach(listener => listener(condition, value));
        }
    }

    private void OnMainQuestStart(string condition, bool value)
    {
        if (condition != "BT_MAIN_QUEST_START") return;

        OnMainQuest?.Invoke();
        if (factsToEnable.Count > 0)
        {
            factsToEnable.ForEach(factTrigger => factTrigger.SetActive(true));
        }
        if (scrollsToEnable.Count > 0)
        {
            scrollsToEnable.ForEach(scrollItem => scrollItem.EnableInteraction(true));
        }
        if (textToEnable.Count > 0)
        {
            textToEnable.ForEach(wallText => wallText.gameObject.SetActive(true));
        }
    }

    private void OnShardCondition(string condition, bool value)
    {
        // Instance.ModHelper.Console.WriteLine($"condition changed: {condition}");
        if (!ShardConditions.TryGetValue(condition, out var shardGroup)) return;

        OnShardFound?.Invoke(shardGroup);

        _numClansConvinced += 1;
        if (_numClansConvinced == 3 && !GetPersistentCondition("BT_LAST_CLAN_TO_AGREE"))
        {
            SetPersistentCondition("BT_LAST_CLAN_TO_AGREE", true);
        }
        else if (_numClansConvinced == 4 && !GetPersistentCondition("BT_ALL_CLANS_AGREED"))
        {
            Locator.GetShipLogManager().RevealFact("BT_GREAT_DOOR_CLANS_AGREED");
            SetPersistentCondition("BT_ALL_CLANS_AGREED", true);
        }
    }

    private void OnGatekeeperToDoor(string condition, bool value)
    {
        if (condition != "BT_SUNPOST_PUZZLE_SOLVED" || !GetPersistentCondition("BT_ACCIDENTAL_SOLVE")) return;

        OnCompleteAccidentalCode?.Invoke();
    }

    private void OnFifthShardFactRevealed()
    {
        _allFifthShardFacts = FifthShardFactNames
            .All(shardLog => Locator.GetShipLogManager().IsFactRevealed(shardLog));
    }

    private void OnGroupMoveCondition(string condition, bool value)
    {
        if (!GroupDialogueConditions.ContainsKey(condition) || !value) return;

        var destination = GroupDialogueConditions[condition];
        WriteDebugMessage($"move condition: {destination}");

        destination.groups.ForEach(group => OnMoveGroup?.Invoke(group, destination.destination, false));
    }

    public static void AddDialogueConditionListener(DialogueConditionChanged listener, params string[] conditions) =>
        conditions.ForEach(condition =>
            Instance._dialogueConditionListeners
                .GetOrInit(condition, new List<DialogueConditionChanged>())
                .Add(listener)
        );

    public static void RemoveDialogueConditionListener(DialogueConditionChanged listener, params string[] conditions) =>
        conditions.ForEach(condition =>
            Instance._dialogueConditionListeners
                .GetOrInit(condition, new List<DialogueConditionChanged>())
                .Remove(listener)
        );

    public static void SetCondition(string condition, bool value) =>
        DialogueConditionManager.SharedInstance.SetConditionState(condition, value);

    public static bool GetCondition(string condition) =>
        DialogueConditionManager.SharedInstance.GetConditionState(condition);

    public static void SetPersistentCondition(string condition, bool value, bool includeTransient = true)
    {
        PlayerData.SetPersistentCondition(condition, value);
        if (includeTransient) SetCondition(condition, value);
    }

    public static bool GetPersistentCondition(string condition) =>
        PlayerData.GetPersistentCondition(condition);

    public static bool AllFifthShardRumors() =>
        Instance._allFifthShardFacts;

    public static void TriggerEnd()
    {
        Instance.ModHelper.Events.Unity.FireOnNextUpdate(() =>
        {
            Instance.startedEndSequence = true;
            Instance.fadeEndMusic = true;
            Locator.GetDeathManager().BeginEscapedTimeLoopSequence(TimeloopEscapeType.Ship);
        });
    }

    public void OnTriggerCampfireEnd()
    {
        FindObjectOfType<GameOverController>()._deathText.text = "Despite there being nothing of value behind the Great Door, you managed to\nreunite the clans and bring harmony back to the planet.";
        FindObjectOfType<GameOverController>().SetupGameOverScreen(10f);
    }

    public static void WriteDebugMessage(object msg)
    {
        if (!IsDebugEnabled()) return;
        Instance.ModHelper.Console.WriteLine(msg.ToString());
    }


    private void OnDestroy()
    {
        nhAPI.GetBodyLoadedEvent().RemoveListener(OnBodyLoaded);
    }

    public static bool IsDebugEnabled() => Instance._debugEnabled;
}