Pixel Crushers Forum

Code: Select all

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using ORKFramework;
using PixelCrushers.QuestMachine;

namespace PixelCrushers.QuestMachine.ORKSupport
{
    public class QuestJournalForORKWithVerify : QuestJournalForORK
    {
        private DataObject pendingData;

        public override void LoadGame(DataObject data)
        {
            if (data == null)
            {
                Debug.Log("No saved data. Exiting.");
                return;
            }
            if (questList == null)
            {
                Debug.LogError("questList is null.");
                return;
            }

            // Store the data and start the coroutine.
            pendingData = data;
            StartCoroutine(DeferredLoadGame());
        }

        private IEnumerator DeferredLoadGame()
        {
            // First frame: Just wait.
            yield return null;

            // Second frame: call the base method.
            // This will run Quest Machine's ApplyData(), which triggers VerifyTrueNodeChildrenAreActive().
            base.LoadGame(pendingData);

            // Wait a couple more frames to ensure Quest Machine finishes any enumerations,
            // auto-verifications, or UI updates it might still be doing.
            yield return null;
            yield return null;

            // Now, in the third frame after calling base.LoadGame, 
            // do your custom quest modifications:
            DoQuestCleanupAndFixes();

            // Done! You shouldn't get the enumeration error now.
        }

        private void DoQuestCleanupAndFixes()
        {
            // 1) Remove duplicates
            var firstQuestInstances = new List<Quest>();
            var duplicateQuestInstances = new List<Quest>();
            var allQuests = new List<Quest>(questList);

            foreach (var quest in allQuests)
            {
                if (quest == null) continue;

                bool alreadyHasQuest = firstQuestInstances.Find(x => StringField.Equals(x.id, quest.id)) != null;
                if (!alreadyHasQuest)
                {
                    firstQuestInstances.Add(quest);
                }
                else
                {
                    duplicateQuestInstances.Add(quest);
                }
            }
            foreach (var duplicate in duplicateQuestInstances)
            {
                Debug.Log($"Removing duplicate quest: {duplicate.id}");
                DeleteQuest(duplicate);
            }

            // 2) Identify completed/bad quests
            var badQuests = new List<Quest>();
            var completedQuests = new List<Quest>();
            allQuests = new List<Quest>(questList);

            foreach (var quest in allQuests)
            {
                if (quest == null) continue;
                if (quest.GetState() == QuestState.Successful)
                {
                    completedQuests.Add(quest);
                }
                else if (quest.startNode == null || quest.startNode.GetState() != QuestNodeState.True)
                {
                    badQuests.Add(quest);
                }
            }

            // 3) Remove completed quests
            foreach (var completedQuest in completedQuests)
            {
                Debug.Log($"Removing completed quest {completedQuest.id}");
                DeleteQuest(completedQuest);
            }

            // 4) Replace bad quests with fresh copies
            foreach (var badQuest in badQuests)
            {
                if (badQuest.originalAsset == null)
                {
                    Debug.LogWarning($"Quest {badQuest.id} has a null original asset.");
                    continue;
                }
                Debug.Log($"Replacing bad quest {badQuest.id} with a fresh copy.");
                var questAsset = badQuest.originalAsset;
                DeleteQuest(badQuest);

                var questInstance = questAsset.Clone();
                if (questInstance == null)
                {
                    Debug.LogWarning($"Failed to clone quest asset for quest {badQuest.id}.");
                    continue;
                }

                var questerTextInfo = new QuestParticipantTextInfo(id, displayName, image, null);
                questInstance.AssignQuester(questerTextInfo);
                questInstance.timesAccepted = 1;

                deletedStaticQuests.Remove(StringField.GetStringValue(questInstance.id));
                AddQuest(questInstance);
                questInstance.SetState(QuestState.Active);
            }

            // 5) Refresh UI
            QuestMachineMessages.RefreshUIs(this);
        }
    }
}