Pixel Crushers Forum

Code: Select all

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

public class QuestTrackingLimiter : MonoBehaviour, IMessageHandler
{
    public int maxTrackedQuests = 5; // Define your limit here
    public List<string> trackedQuests = new List<string>();

    private void OnEnable()
    {
        // Listen for Quest Track Toggle Changed messages:
        MessageSystem.AddListener(this, QuestMachineMessages.QuestStateChangedMessage, string.Empty);
        MessageSystem.AddListener(this, QuestMachineMessages.QuestTrackToggleChangedMessage, string.Empty);
        Debug.Log("QuestTrackingLimiter: Listening for quest track changes.");
    }

    private void OnDisable()
    {
        // Stop listening:
        MessageSystem.RemoveListener(this);
        Debug.Log("QuestTrackingLimiter: Stopped listening for quest track changes.");
    }

    public void OnMessage(MessageArgs messageArgs)
    {
        string questID = messageArgs.parameter;
        if (QuestMachine.GetQuestState(questID) == QuestState.WaitingToStart) return;
        bool isTracked = false;
        if (messageArgs.message == QuestMachineMessages.QuestStateChangedMessage)
        {
            bool isMainQuestState = (messageArgs.values[0] == null);
            if (!isMainQuestState) return; // Only act if the main quest state has changed.
            QuestState questState = (QuestState)messageArgs.values[1];
            if (questState == QuestState.Disabled) return; // Quest giver disables its instance when given.
            if (questState == QuestState.Active) // Was just activated
            {
                var questInstance = QuestMachine.GetQuestInstance(questID);
                isTracked = questInstance.showInTrackHUD;
            }
        }
        else if (messageArgs.message == QuestMachineMessages.QuestTrackToggleChangedMessage)
        {
            isTracked = (bool)messageArgs.values[0]; // This will tell if the quest is being tracked or untracked
        }
        else
        {
            return; // Neither Quest State Changed nor Quest Track Toggle Changed, so exit.
        }

        if (isTracked)
        {
            Debug.Log($"Quest '{questID}' is now being tracked.");

            // Check if max tracked quests limit is reached
            if (trackedQuests.Count >= maxTrackedQuests)
            {
                // Prevent tracking by immediately untracking the new quest
                Debug.Log($"Max tracked quests limit reached: {maxTrackedQuests}. Untracking quest '{questID}'.");
                UntrackQuest(questID);
                return;
            }

            // Add the new tracked quest if below the limit
            trackedQuests.Add(questID);
            Debug.Log($"Quest '{questID}' added to the tracked list. Current number of tracked quests: {trackedQuests.Count}/{maxTrackedQuests}.");
        }
        else
        {
            // Remove the untracked quest
            if (trackedQuests.Contains(questID))
            {
                trackedQuests.Remove(questID);
                Debug.Log($"Quest '{questID}' was untracked. Current number of tracked quests: {trackedQuests.Count}/{maxTrackedQuests}.");
            }
        }

        // Log the current list of tracked quests
        LogTrackedQuests();
    }

    private void UntrackQuest(string questID)
    {
        // Send a message to untrack the quest using the messaging system
        // Not this: MessageSystem.SendMessage(this, QuestMachineMessages.QuestTrackToggleChangedMessage, questID, false);
        // Set the quest's showInTrackHUD property instead:
        QuestMachine.GetQuestInstance(questID).showInTrackHUD = false;
        Debug.Log($"Quest '{questID}' untracked.");
    }

    private void LogTrackedQuests()
    {
        if (trackedQuests.Count == 0)
        {
            Debug.Log("No quests are currently being tracked.");
        }
        else
        {
            Debug.Log("Currently tracked quests:");
            foreach (var quest in trackedQuests)
            {
                Debug.Log($"- {quest}");
            }
        }
    }
}