Rasagar/Library/PackageCache/com.unity.visualscripting/Runtime/VisualScripting.Flow/Connections/InvalidConnection.cs

using System;
using System.Linq;

namespace Unity.VisualScripting
{
    public sealed class InvalidConnection : UnitConnection<IUnitOutputPort, IUnitInputPort>, IUnitConnection
    {
        [Obsolete(Serialization.ConstructorWarning)]
        public InvalidConnection() : base() { }

        public InvalidConnection(IUnitOutputPort source, IUnitInputPort destination) : base(source, destination) { }

        public override void AfterRemove()
        {
            base.AfterRemove();
            source.unit.RemoveUnconnectedInvalidPorts();
            destination.unit.RemoveUnconnectedInvalidPorts();
        }

        #region Ports

        public override IUnitOutputPort source => sourceUnit.outputs.Single(p => p.key == sourceKey);

        public override IUnitInputPort destination => destinationUnit.inputs.Single(p => p.key == destinationKey);

        public IUnitOutputPort validSource => sourceUnit.validOutputs.Single(p => p.key == sourceKey);

        public IUnitInputPort validDestination => destinationUnit.validInputs.Single(p => p.key == destinationKey);

        #endregion

        #region Dependencies

        public override bool sourceExists => sourceUnit.outputs.Any(p => p.key == sourceKey);

        public override bool destinationExists => destinationUnit.inputs.Any(p => p.key == destinationKey);

        public bool validSourceExists => sourceUnit.validOutputs.Any(p => p.key == sourceKey);

        public bool validDestinationExists => destinationUnit.validInputs.Any(p => p.key == destinationKey);

        public override bool HandleDependencies()
        {
            // Replace the invalid connection with a valid connection if it can be created instead.
            if (validSourceExists && validDestinationExists && validSource.CanValidlyConnectTo(validDestination))
            {
                validSource.ValidlyConnectTo(validDestination);

                return false;
            }

            // Add the invalid ports to the nodes if need be
            if (!sourceExists)
            {
                sourceUnit.invalidOutputs.Add(new InvalidOutput(sourceKey));
            }

            if (!destinationExists)
            {
                destinationUnit.invalidInputs.Add(new InvalidInput(destinationKey));
            }

            return true;
        }

        #endregion
    }
}