LuaCsForBarotraumaEP/Barotrauma/BarotraumaShared/SharedSource/Events/EventActions/WaitForItemUsedAction.cs

#nullable enable

using Barotrauma.Extensions;
using Barotrauma.Items.Components;
using System.Collections.Generic;
using System.Linq;

namespace Barotrauma
{
    /// <summary>
    /// Waits for some item(s) to be used before continuing the execution of the event.
    /// </summary>
    class WaitForItemUsedAction : EventAction
    {
        [Serialize("", IsPropertySaveable.Yes, description: "Tag of the item that must be used. Note that the item needs to have been tagged by the event - this does not refer to the tags that can be set per-item in the sub editor.")]
        public Identifier ItemTag { get; set; }

        [Serialize("", IsPropertySaveable.Yes, description: "Tag of the character that must use the item. If there's multiple matching characters, it's enough if any of them use the item. If empty, it doesn't matter who uses the item.")]
        public Identifier UserTag { get; set; }

        [Serialize("", IsPropertySaveable.Yes, description: "Name of the ItemComponent that the character must use. If empty, the character attempts to use all of them.")]
        public Identifier TargetItemComponent { get; set; }

        [Serialize("", IsPropertySaveable.Yes, description: "Optional tag to apply to the target item when it's used.")]
        public Identifier ApplyTagToItem { get; set; }

        [Serialize("", IsPropertySaveable.Yes, description: "Optional tag to apply to the user when the target item is used.")]
        public Identifier ApplyTagToUser{ get; set; }

        [Serialize("", IsPropertySaveable.Yes, description: "Optional tag to apply to the hull the target item is inside when the item is used.")]
        public Identifier ApplyTagToHull { get; set; }

        [Serialize("", IsPropertySaveable.Yes, description: "Optional tag to apply to the hull the target item is inside, and all the hulls it's linked to, when the item is used.")]
        public Identifier ApplyTagToLinkedHulls { get; set; }

        [Serialize(1, IsPropertySaveable.Yes, description: "How many times does the item need to be used. Defaults to 1.")]
        public int RequiredUseCount { get; set; }

        private bool isFinished;

        private readonly HashSet<Entity> targets = new HashSet<Entity>();
        private readonly HashSet<ItemComponent> targetComponents = new HashSet<ItemComponent>();

        private int useCount = 0;

        private Identifier onUseEventIdentifier;
        private Identifier OnUseEventIdentifier
        {
            get
            {
                if (onUseEventIdentifier.IsEmpty)
                {
                    onUseEventIdentifier = (ParentEvent.Prefab.Identifier + ParentEvent.Actions.IndexOf(this).ToString()).ToIdentifier();
                }
                return onUseEventIdentifier;
            }
        }

        public WaitForItemUsedAction(ScriptedEvent parentEvent, ContentXElement element) : base(parentEvent, element) 
        {
            if (ItemTag.IsEmpty)
            {
                DebugConsole.ThrowError($"Error in event \"{ParentEvent.Prefab.Identifier}\". {nameof(ItemTag)} not set in {nameof(WaitForItemUsedAction)}.");
            }
        }

        private void OnItemUsed(Item item, Character user)
        {
            if (!UserTag.IsEmpty)
            {
                if (!ParentEvent.GetTargets(UserTag).Contains(user)) { return; }
            }

            useCount++;
            if (useCount < RequiredUseCount) { return; }

            if (!ApplyTagToItem.IsEmpty)
            {
                ParentEvent.AddTarget(ApplyTagToItem, item);
            }
            if (!ApplyTagToUser.IsEmpty && user != null)
            {
                ParentEvent.AddTarget(ApplyTagToUser, user);
            }
            ApplyTagsToHulls(item, ApplyTagToHull, ApplyTagToLinkedHulls);
            DeregisterTargets();
            isFinished = true;
        }

        public override void Update(float deltaTime)
        {
            TryRegisterTargets();
        }

        private void TryRegisterTargets()
        {
            foreach (Entity target in ParentEvent.GetTargets(ItemTag))
            {
                //already registered, ignore
                if (targets.Contains(target)) { continue; }
                if (target is not Item item) { continue; }
                if (TargetItemComponent.IsEmpty)
                {
                    item.GetComponents<ItemComponent>().ForEach(ic => Register(ic));
                }
                else if (item.Components.FirstOrDefault(ic => ic.Name == TargetItemComponent) is ItemComponent targetItemComponent)
                {
                    Register(targetItemComponent);
                }
                else
                {
#if DEBUG
                    DebugConsole.ThrowError($"Failed to find the component {TargetItemComponent} on item {item.Prefab.Identifier}");
#endif
                }
            }
            void Register(ItemComponent ic)
            {
                targets.Add(ic.Item);
                targetComponents.Add(ic);
                ic.OnUsed.RegisterOverwriteExisting(
                    OnUseEventIdentifier,
                    i => { OnItemUsed(i.Item, i.User); });
            }
        }

        private void DeregisterTargets()
        {
            foreach (ItemComponent ic in targetComponents)
            {
                ic.OnUsed.Deregister(OnUseEventIdentifier);
            }
            targetComponents.Clear();
            targets.Clear();
        }


        public override bool IsFinished(ref string goTo)
        {
            return isFinished;
        }

        public override void Reset()
        {
            isFinished = false;
            useCount = 0;
            DeregisterTargets();
        }

        public override string ToDebugString()
        {
            return $"{ToolBox.GetDebugSymbol(isFinished)} {nameof(WaitForItemUsedAction)} -> ({ItemTag})";
        }
    }
}