src/Tests/OneScript.Language.Tests/TestAstNode.cs - develop - EvilBeaver/OneScript

src/Tests/OneScript.Language.Tests/TestAstNode.cs

61 строка2 KB

Andrei Ovsiankin

Разрешено использование комментариев в строке с директивой Использовать

17 июн 2025, 19:05

17 июн 2025, 19:056d0eb1a

/*----------------------------------------------------------
This Source Code Form is subject to the terms of the
Mozilla Public License, v.2.0. If a copy of the MPL
was not distributed with this file, You can obtain one
at http://mozilla.org/MPL/2.0/.
----------------------------------------------------------*/

using System;
using System.Collections.Generic;
using System.Linq;
using OneScript.Language.SyntaxAnalysis.AstNodes;

namespace OneScript.Language.Tests
{
    public class TestAstNode : BslSyntaxNode
    {
        private Lazy<IReadOnlyList<TestAstNode>> _childrenLazy;
        
        public BslSyntaxNode RealNode { get; }
        
        public TestAstNode(BslSyntaxNode node)
        {
            RealNode = node;
            Kind = node.Kind;
            if (node is NonTerminalNode nonTerm)
            {
                _childrenLazy = new Lazy<IReadOnlyList<TestAstNode>>(nonTerm.Children.Select(x => new TestAstNode(x)).ToArray());
                Value = nonTerm switch
                {
                    AnnotationNode anno => anno.Name,
                    BinaryOperationNode binary => binary.Operation.ToString(),
                    UnaryOperationNode unary => unary.Operation.ToString(),
                    PreprocessorDirectiveNode preproc => preproc.DirectiveName,
                    AnnotationParameterNode annParam => annParam.Value.Content,
                    _ => nonTerm.ToString()
                };
            }
            else if(node is TerminalNode term)
            {
                _childrenLazy = new Lazy<IReadOnlyList<TestAstNode>>(new TestAstNode[0]);
                Value = term.Lexem.Content;
            }
            else
            {
                _childrenLazy = new Lazy<IReadOnlyList<TestAstNode>>(new TestAstNode[0]);
                Value = node.ToString();
            }
        }
        
        public string Value { get; set; }

        public IReadOnlyList<TestAstNode> ChildrenList => _childrenLazy.Value;

        public override IReadOnlyList<BslSyntaxNode> Children => ChildrenList;
        
        public override string ToString()
        {
            return $"{Kind} ({Location.LineNumber},{Location.ColumnNumber})";
        }
    }
}