using NubLang.Tokenization;

namespace NubLang.Parsing.Syntax;

public abstract record DefinitionSyntax(IEnumerable<Token> Tokens) : SyntaxNode(Tokens);

public record FuncParameterSyntax(IEnumerable<Token> Tokens, string Name, TypeSyntax Type) : SyntaxNode(Tokens)
{
    public override IEnumerable<SyntaxNode> GetChildren()
    {
        yield return Type;
    }
}

public record FuncSignatureSyntax(IEnumerable<Token> Tokens, IReadOnlyList<FuncParameterSyntax> Parameters, TypeSyntax ReturnType) : SyntaxNode(Tokens)
{
    public override IEnumerable<SyntaxNode> GetChildren()
    {
        foreach (var parameter in Parameters)
        {
            yield return parameter;
        }

        yield return ReturnType;
    }
}

public record LocalFuncSyntax(IEnumerable<Token> Tokens, string Name, FuncSignatureSyntax Signature, BlockSyntax Body) : DefinitionSyntax(Tokens)
{
    public override IEnumerable<SyntaxNode> GetChildren()
    {
        yield return Signature;
        yield return Body;
    }
}

public record ExternFuncSyntax(IEnumerable<Token> Tokens, string Name, string CallName, FuncSignatureSyntax Signature) : DefinitionSyntax(Tokens)
{
    public override IEnumerable<SyntaxNode> GetChildren()
    {
        yield return Signature;
    }
}

public record StructFieldSyntax(IEnumerable<Token> Tokens, int Index, string Name, TypeSyntax Type, Optional<ExpressionSyntax> Value) : SyntaxNode(Tokens)
{
    public override IEnumerable<SyntaxNode> GetChildren()
    {
        yield return Type;
        if (Value.HasValue)
        {
            yield return Value.Value;
        }
    }
}

public record StructFuncSyntax(IEnumerable<Token> Tokens, string Name, FuncSignatureSyntax Signature, BlockSyntax Body) : SyntaxNode(Tokens)
{
    public override IEnumerable<SyntaxNode> GetChildren()
    {
        yield return Signature;
        yield return Body;
    }
}

public record StructSyntax(IEnumerable<Token> Tokens, string Name, IReadOnlyList<StructFieldSyntax> Fields, IReadOnlyList<StructFuncSyntax> Functions, IReadOnlyList<TypeSyntax> InterfaceImplementations) : DefinitionSyntax(Tokens)
{
    public override IEnumerable<SyntaxNode> GetChildren()
    {
        foreach (var field in Fields)
        {
            yield return field;
        }

        foreach (var func in Functions)
        {
            yield return func;
        }
    }
}

public record InterfaceFuncSyntax(IEnumerable<Token> Tokens, string Name, FuncSignatureSyntax Signature) : SyntaxNode(Tokens)
{
    public override IEnumerable<SyntaxNode> GetChildren()
    {
        yield return Signature;
    }
}

public record InterfaceSyntax(IEnumerable<Token> Tokens, string Name, IReadOnlyList<InterfaceFuncSyntax> Functions) : DefinitionSyntax(Tokens)
{
    public override IEnumerable<SyntaxNode> GetChildren()
    {
        foreach (var func in Functions)
        {
            yield return func;
        }
    }
}