src/OneScript.StandardLibrary/Native/CompiledBlock.cs - develop - smwwiki/OneScript_fork

src/OneScript.StandardLibrary/Native/CompiledBlock.cs
218 строк8 KB
EvilBeaver
Выделена отдельная папка для инфраструктуры исключений
24 июл 2023, 22:46
24 июл 2023, 22:46d1c2e94
/*----------------------------------------------------------
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 System.Linq.Expressions;
using System.Text;
using OneScript.Commons;
using OneScript.Compilation.Binding;
using OneScript.Contexts;
using OneScript.DependencyInjection;
using OneScript.Exceptions;
using OneScript.Language;
using OneScript.Language.LexicalAnalysis;
using OneScript.Language.SyntaxAnalysis;
using OneScript.Language.SyntaxAnalysis.AstNodes;
using OneScript.Native.Compiler;
using OneScript.Native.Runtime;
using OneScript.Sources;
using OneScript.StandardLibrary.Collections;
using OneScript.Types;
using OneScript.Values;
using ScriptEngine.Machine;
using ScriptEngine.Machine.Contexts;

namespace OneScript.StandardLibrary.Native
{
    [ContextClass("СкомпилированныйФрагмент", "CompiledCodeBlock")]
    public class CompiledBlock : AutoContext<CompiledBlock>
    {
        private readonly IServiceContainer _services;
        private string _codeBlock;
        private BslSyntaxNode _ast;
        private IErrorSink _errors;
        private SourceCode _codeLinesReferences;

        public CompiledBlock(IServiceContainer services)
        {
            _services = services;
        }
        
        public SymbolTable Symbols { get; set; }
        
        [ContextProperty("Параметры", "Parameters")]
        public StructureImpl Parameters { get; set; } = new StructureImpl();
        
        [ContextProperty("ФрагментКода", "CodeFragment")]
        public string CodeBlock
        {
            get => _codeBlock ?? string.Empty;
            set
            {
                _codeBlock = value;
                ParseCode();
            }
        }

        private void ParseCode()
        {
            var lexer = new DefaultLexer();
            var source = SourceCodeBuilder.Create()
                .FromString(CodeBlock)
                .WithName($"Compiled source {CodeBlock.GetHashCode():X8}")
                .Build();
            
            lexer.Iterator = source.CreateIterator();

            _codeLinesReferences = source;
            _errors = new ListErrorSink();
            var parser = new DefaultBslParser(lexer, _errors, new PreprocessorHandlers());

            try
            {
                _ast = parser.ParseCodeBatch(true);
            }
            catch (ScriptException e)
            {
                _errors.AddError(new CodeError
                {
                    Description = e.Message,
                    Position = e.GetPosition()
                });
            }
            
            if (_errors.HasErrors)
            {
                var prefix = Locale.NStr("ru = 'Ошибка комиляции модуля'; en = 'Module compilation error'");
                var text = string.Join('\n', (new[] {prefix}).Concat(_errors.Errors.Select(x => x.ToString(CodeError.ErrorDetails.Position))));
                throw new RuntimeException(text);
            }
        }

        [ContextMethod("Скомпилировать", "Compile")]
        public DelegateAction Compile()
        {
            var method = CreateDelegate();
            return new DelegateAction(method);
        }

        public Func<BslValue[], BslValue> CreateDelegate()
        {
            var l = MakeExpression();

            var arrayOfValuesParam = Expression.Parameter(typeof(BslValue[]));
            var convertedAccessList = new List<Expression>();

            int index = 0;
            foreach (var parameter in Parameters)
            {
                var targetType = parameter.Value as BslTypeValue;
                var arrayAccess = Expression.ArrayIndex(arrayOfValuesParam, Expression.Constant(index));
                var convertedParam = ExpressionHelpers.ConvertToType(arrayAccess, ConvertTypeToClrType(targetType));
                convertedAccessList.Add(convertedParam);
                ++index;
            }
            
            var lambdaInvocation = Expression.Invoke(l, convertedAccessList);
            var func = Expression.Lambda<Func<BslValue[], BslValue>>(lambdaInvocation, arrayOfValuesParam);

            return func.Compile();
        }

        public T CreateDelegate<T>() where T:class
        {
            var l = MakeExpression();
            var call = Expression.Invoke(l, l.Parameters);
            var func = Expression.Lambda<T>(call, l.Parameters);

            return func.Compile();
        }

        public LambdaExpression MakeExpression()
        {
            if(_ast == default)
                ParseCode();

            var expression = ReduceAst(_ast);
            
            if (_errors.HasErrors)
            {
                var prefix = Locale.NStr("ru = 'Ошибка комиляции модуля'; en = 'Module compilation error'");
                var sb = new StringBuilder();
                sb.AppendLine(prefix);
                foreach (var error in _errors.Errors)
                {
                    sb.AppendLine(error.ToString(CodeError.ErrorDetails.Position));
                }

                throw new RuntimeException(sb.ToString());
            }

            return expression;
        }

        private LambdaExpression ReduceAst(BslSyntaxNode ast)
        {
            // в параметрах лежат соответствия имени переменной и ее типа
            // блок кода надо скомпилировтаь в лямбду с параметрами по количеству в коллекции Parameters и с типами параметров, как там
            // пробежать по аст 1С и превратить в BlockExpression<IValue>

            if (Symbols == null)
                Symbols = new SymbolTable();
            
            Symbols.PushObject(new StandardGlobalContext());
            
            var methodInfo = CreateMethodInfo();
            var methodCompiler = new MethodCompiler(new BslWalkerContext
            {
                Errors = _errors,
                Source = _codeLinesReferences,
                Symbols = Symbols,
                Services = _services
            }, methodInfo);
            
            methodCompiler.CompileModuleBody(_ast.Children.FirstOrDefault(x => x.Kind == NodeKind.ModuleBody));
            return methodInfo.Implementation;
        }

        private BslNativeMethodInfo CreateMethodInfo()
        {
            var methodInfo = new BslMethodInfoFactory<BslNativeMethodInfo>(()=>new BslNativeMethodInfo())
                .NewMethod()
                .Name("$__compiled")
                .ReturnType(typeof(IValue));

            foreach (var parameter in Parameters)
            {
                methodInfo.NewParameter()
                    .Name(parameter.Key.AsString())
                    .ParameterType(ConvertTypeToClrType(parameter.Value as BslTypeValue));
            }

            return methodInfo.Build();
        }

        private static Type ConvertTypeToClrType(BslTypeValue typeVal)
        {
            var type = typeVal.TypeValue;
            return GetClrType(type);
        }
        
        private static Type GetClrType(TypeDescriptor type)
        {
            return CompilerHelpers.GetClrType(type);
        }

        [ScriptConstructor]
        public static CompiledBlock Create(TypeActivationContext context)
        {
            return new CompiledBlock(context.Services);
        }
    }
}