sebastienros.jint/Jint/Runtime/Interpreter/JintStatementList.cs

using Esprima.Ast;
using Jint.Native;
using Jint.Native.Error;
using Jint.Runtime.Environments;
using Jint.Runtime.Interpreter.Statements;

namespace Jint.Runtime.Interpreter
{
    internal sealed class JintStatementList
    {
        private sealed class Pair
        {
            internal JintStatement Statement = null!;
            internal Completion? Value;
        }

        private readonly Statement? _statement;
        private readonly NodeList<Statement> _statements;

        private Pair[]? _jintStatements;
        private bool _initialized;
        private readonly uint _index;
        private readonly bool _generator;

        public JintStatementList(IFunction function)
            : this((BlockStatement) function.Body)
        {
            _generator = function.Generator;
        }

        public JintStatementList(BlockStatement blockStatement)
            : this(blockStatement, blockStatement.Body)
        {
        }

        public JintStatementList(Program program)
            : this(null, program.Body)
        {
        }

        public JintStatementList(Statement? statement, in NodeList<Statement> statements)
        {
            _statement = statement;
            _statements = statements;
        }

        private void Initialize(EvaluationContext context)
        {
            var jintStatements = new Pair[_statements.Count];
            for (var i = 0; i < jintStatements.Length; i++)
            {
                var esprimaStatement = _statements[i];
                var statement = JintStatement.Build(esprimaStatement);
                // When in debug mode, don't do FastResolve: Stepping requires each statement to be actually executed.
                var value = context.DebugMode ? null : JintStatement.FastResolve(esprimaStatement);
                jintStatements[i] = new Pair
                {
                    Statement = statement,
                    Value = value
                };
            }

            _jintStatements = jintStatements;
        }

        public Completion Execute(EvaluationContext context)
        {
            if (!_initialized)
            {
                Initialize(context);
                _initialized = true;
            }

            if (_statement is not null)
            {
                context.LastSyntaxElement = _statement;
                context.RunBeforeExecuteStatementChecks(_statement);
            }

            JintStatement? s = null;
            Completion c = default;
            Completion sl = c;

            // The value of a StatementList is the value of the last value-producing item in the StatementList
            JsValue? lastValue = null;
            try
            {
                foreach (var pair in _jintStatements!)
                {
                    s = pair.Statement;
                    c = pair.Value.GetValueOrDefault();
                    if (c.Value is null)
                    {
                        c = s.Execute(context);
                        if (context.Engine._error is not null)
                        {
                            return HandleError(context.Engine, s);
                        }
                    }

                    if (c.Type != CompletionType.Normal)
                    {
                        return new Completion(c.Type, c.Value ?? sl.Value!, c._source);
                    }
                    sl = c;
                    if (c.Value is not null)
                    {
                        lastValue = c.Value;
                    }
                }
            }
            catch (Exception ex)
            {
                if (ex is JintException)
                {
                    return HandleException(context, ex, s);
                }

                throw;
            }

            return new Completion(c.Type, lastValue ?? JsValue.Undefined, c._source!);
        }

        private static Completion HandleException(EvaluationContext context, Exception exception, JintStatement? s)
        {
            if (exception is JavaScriptException javaScriptException)
            {
                return CreateThrowCompletion(s, javaScriptException);
            }
            if (exception is TypeErrorException typeErrorException)
            {
                var node = typeErrorException.Node ?? s!._statement;
                return CreateThrowCompletion(context.Engine.Realm.Intrinsics.TypeError, typeErrorException, node);
            }
            if (exception is RangeErrorException rangeErrorException)
            {
                return CreateThrowCompletion(context.Engine.Realm.Intrinsics.RangeError, rangeErrorException, s!._statement);
            }

            // should not happen unless there's problem in the engine
            throw exception;
        }

        private static Completion HandleError(Engine engine, JintStatement? s)
        {
            var error = engine._error!;
            engine._error = null;
            return CreateThrowCompletion(error.ErrorConstructor, error.Message, engine._lastSyntaxElement ?? s!._statement);
        }

        private static Completion CreateThrowCompletion(ErrorConstructor errorConstructor, string? message, SyntaxElement s)
        {
            var error = errorConstructor.Construct(message);
            return new Completion(CompletionType.Throw, error, s);
        }

        private static Completion CreateThrowCompletion(ErrorConstructor errorConstructor, Exception e, SyntaxElement s)
        {
            var error = errorConstructor.Construct(e.Message);
            return new Completion(CompletionType.Throw, error, s);
        }

        private static Completion CreateThrowCompletion(JintStatement? s, JavaScriptException v)
        {
            SyntaxElement source = s!._statement;
            if (v.Location != default)
            {
                source = EsprimaExtensions.CreateLocationNode(v.Location);
            }

            return new Completion(CompletionType.Throw, v.Error, source);
        }

        /// <summary>
        /// https://tc39.es/ecma262/#sec-blockdeclarationinstantiation
        /// </summary>
        internal static void BlockDeclarationInstantiation(
            Engine engine,
            EnvironmentRecord env,
            List<Declaration> declarations)
        {
            var privateEnv = env._engine.ExecutionContext.PrivateEnvironment;
            var boundNames = new List<string>();
            for (var i = 0; i < declarations.Count; i++)
            {
                var d = declarations[i];
                boundNames.Clear();
                d.GetBoundNames(boundNames);
                for (var j = 0; j < boundNames.Count; j++)
                {
                    var dn = boundNames[j];
                    if (d is VariableDeclaration { Kind: VariableDeclarationKind.Const })
                    {
                        env.CreateImmutableBinding(dn, strict: true);
                    }
                    else
                    {
                        env.CreateMutableBinding(dn, canBeDeleted: false);
                    }
                }

                if (d is FunctionDeclaration functionDeclaration)
                {
                    var definition = new JintFunctionDefinition(functionDeclaration);
                    var fn = definition.Name!;
                    var fo = env._engine.Realm.Intrinsics.Function.InstantiateFunctionObject(definition, env, privateEnv);
                    env.InitializeBinding(fn, fo);
                }
            }
        }
    }
}