CheckConnectionAction.cs

using Barotrauma.Extensions;
using Barotrauma.Items.Components;
using System.Linq;
 
namespace Barotrauma;
 
class CheckConnectionAction : BinaryOptionAction
{
    [Serialize("", IsPropertySaveable.Yes, description: "Tag of the item to check.")]
    public Identifier ItemTag { get; set; }
 
    [Serialize("", IsPropertySaveable.Yes, description: "The name of the connection to check on the target item.")]
    public Identifier ConnectionName { get; set; }
 
    [Serialize("", IsPropertySaveable.Yes, description: "Tag of the item the connection must be wired to. If omitted, it doesn't matter what the connection is wired to.")]
    public Identifier ConnectedItemTag { get; set; }
 
    [Serialize("", IsPropertySaveable.Yes, description: "The name of the other connection the connection must be wired to. If omitted, it doesn't matter what the connection is wired to.")]
    public Identifier OtherConnectionName { get; set; }
 
    [Serialize(1, IsPropertySaveable.Yes, description: "Minimum number of matching connections for the check to succeed.")]
    public int MinAmount { get; set; }
 
    public CheckConnectionAction(ScriptedEvent parentEvent, ContentXElement element) : base(parentEvent, element) { }
 
    protected override bool? DetermineSuccess()
    {
        int amount = 0;
        var connectTargets = !ConnectedItemTag.IsEmpty ? ParentEvent.GetTargets(ConnectedItemTag) : Enumerable.Empty<Entity>();
        foreach (var target in ParentEvent.GetTargets(ItemTag))
        {
            if (target is not Item targetItem) { continue; }
            if (targetItem.GetComponent<ConnectionPanel>() is not ConnectionPanel panel) { continue; }
            if (panel.Connections == null || panel.Connections.None()) { continue; }
            foreach (var connection in panel.Connections)
            {
                if (!IsCorrectConnection(connection, ConnectionName)) { continue; }
                if (ConnectedItemTag.IsEmpty && OtherConnectionName.IsEmpty)
                {
                    amount += connection.Wires.Count;
                    if (amount >= MinAmount) { return true; }
                    continue;
                }
                foreach (var wire in connection.Wires)
                {
                    if (wire.OtherConnection(connection) is not Connection otherConnection) { continue; }
                    if (!ConnectedItemTag.IsEmpty && !IsCorrectConnection(otherConnection, OtherConnectionName)) { continue; }
                    if (!ConnectedItemTag.IsEmpty && !IsCorrectItem()) { continue; }
                    amount++;
                    if (amount >= MinAmount) { return true; }
                    bool IsCorrectItem() => connectTargets.Contains(otherConnection.Item);
                }
 
                bool IsCorrectConnection(Connection connection, Identifier id) => connection.Name.ToIdentifier() == id;
            }
        }
        return false;
    }
}