src/QueryCat.Backend/Ast/Nodes/Function/FunctionSignatureNode.cs - develop - krasninja/querycat

src/QueryCat.Backend/Ast/Nodes/Function/FunctionSignatureNode.cs
93 строки3 KB
Ivan Kozhin
Add experimental AST cache, fix AST nodes clone issues
28 апр 2024, 13:17
28 апр 2024, 13:17145905e
using QueryCat.Backend.Core.Types;

namespace QueryCat.Backend.Ast.Nodes.Function;

/// <summary>
/// Function signature contains function name, argument and
/// return type.
/// </summary>
internal sealed class FunctionSignatureNode : AstNode, IEquatable<FunctionSignatureNode>
{
    /// <summary>
    /// Function name.
    /// </summary>
    public string Name { get; }

    /// <summary>
    /// Function return type. VOID - no return.
    /// </summary>
    public FunctionTypeNode ReturnTypeNode { get; }

    public DataType ReturnType => ReturnTypeNode.Type;

    /// <summary>
    /// Argument nodes.
    /// </summary>
    public FunctionSignatureArgumentNode[] ArgumentNodes { get; }

    /// <inheritdoc />
    public override string Code => "func_sig";

    /// <inheritdoc />
    public FunctionSignatureNode(
        string name,
        FunctionTypeNode returnTypeNode,
        IEnumerable<FunctionSignatureArgumentNode>? argumentNodes = null)
    {
        Name = name;
        ReturnTypeNode = returnTypeNode;
        ArgumentNodes = argumentNodes != null ? argumentNodes.ToArray() : [];
    }

    public FunctionSignatureNode(FunctionSignatureNode node) :
        this(
            node.Name,
            (FunctionTypeNode)node.ReturnTypeNode.Clone(),
            node.ArgumentNodes.Select(an => (FunctionSignatureArgumentNode)an.Clone()))
    {
        node.CopyTo(this);
    }

    /// <inheritdoc />
    public override IEnumerable<IAstNode> GetChildren() => ArgumentNodes;

    /// <inheritdoc />
    public override object Clone() => new FunctionSignatureNode(this);

    /// <inheritdoc />
    public override void Accept(AstVisitor visitor) => visitor.Visit(this);

    /// <inheritdoc />
    public override string ToString()
        => $"{Name}({string.Join(", ", ArgumentNodes.AsEnumerable())}): {ReturnTypeNode}";

    /// <inheritdoc />
    public bool Equals(FunctionSignatureNode? other)
    {
        if (ReferenceEquals(null, other))
        {
            return false;
        }
        if (ReferenceEquals(this, other))
        {
            return true;
        }
        return Name == other.Name
            && ReturnTypeNode == other.ReturnTypeNode
            && ArrayUtils.EqualsAll(ArgumentNodes, other.ArgumentNodes);
    }

    /// <inheritdoc />
    public override bool Equals(object? obj)
        => ReferenceEquals(this, obj) || (obj is FunctionSignatureNode other && Equals(other));

    /// <inheritdoc />
    public override int GetHashCode()
        => HashCode.Combine(Name, (int)ReturnType, ArgumentNodes);

    public static bool operator ==(FunctionSignatureNode? left, FunctionSignatureNode? right)
        => Equals(left, right);

    public static bool operator !=(FunctionSignatureNode? left, FunctionSignatureNode? right)
        => !Equals(left, right);
}