Canon/Canon.Core/SyntaxNodes/StatementList.cs

using Canon.Core.Enums;

namespace Canon.Core.SyntaxNodes;

public class StatementList : NonTerminatedSyntaxNode
{
    public override NonTerminatorType Type => NonTerminatorType.StatementList;

    public bool IsRecursive { get; private init; }

    public IEnumerable<Statement> Statements => GetStatements();

    public static StatementList Create(List<SyntaxNodeBase> children)
    {
        bool isRecursive;

        if (children.Count == 1)
        {
            isRecursive = false;
        }
        else if (children.Count == 3)
        {
            isRecursive = true;
        }
        else
        {
            throw new InvalidOperationException();
        }

        return new StatementList { Children = children, IsRecursive = isRecursive };
    }

    private IEnumerable<Statement> GetStatements()
    {
        StatementList list = this;

        while (true)
        {
            if (list.IsRecursive)
            {
                yield return list.Children[2].Convert<Statement>();
                list = list.Children[0].Convert<StatementList>();
            }
            else
            {
                yield return list.Children[0].Convert<Statement>();
                break;
            }
        }
    }
}