src/ScriptEngine/Machine/Contexts/ScriptedEnumeratorWrapper.cs - develop - EvilBeaver/OneScript

src/ScriptEngine/Machine/Contexts/ScriptedEnumeratorWrapper.cs

83 строки3 KB

EvilBeaver

Удален устаревший метод OnAttach и массово исправлены неиспользуемые using

02 дек 2025, 11:20

02 дек 2025, 11:200301fed

/*----------------------------------------------------------
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;
using System.Collections.Generic;
using OneScript.Contexts;
using OneScript.Exceptions;
using OneScript.Execution;
using OneScript.Localization;
using OneScript.Values;
using ScriptEngine.Machine.Interfaces;

namespace ScriptEngine.Machine.Contexts
{
    public sealed class ScriptedEnumeratorWrapper : IEnumerator<BslValue>
    {
        private readonly UserScriptContextInstance _userObject;
        private readonly IBslProcess _process;

        private BslScriptMethodInfo _moveNextMethod;
        private BslScriptMethodInfo _getCurrentMethod;
        private BslScriptMethodInfo _onDisposeMethod;
        
        internal ScriptedEnumeratorWrapper(UserScriptContextInstance userObject, IBslProcess process)
        {
            _userObject = userObject;
            _process = process;
            CheckAndSetMethods();
        }

        private void CheckAndSetMethods()
        {
            var bslInterface = _userObject.Module.GetInterface<IteratorBslInterface>() 
                               ?? IteratorBslInterfaceChecker.CheckModule(_userObject.Module);

            _moveNextMethod = bslInterface.MoveNextMethod;
            _getCurrentMethod = bslInterface.GetCurrentMethod;
            _onDisposeMethod = bslInterface.OnDisposeMethod;
        }

        public bool MoveNext()
        {
            _userObject.CallAsFunction(_moveNextMethod.DispatchId, Array.Empty<IValue>(), out var result, _process);
            return result.AsBoolean();
        }

        public void Reset()
        {
            throw new System.NotSupportedException();
        }

        public BslValue Current
        {
            get
            {
                _userObject.CallAsFunction(_getCurrentMethod.DispatchId, Array.Empty<IValue>(), out var result, _process);
                return (BslValue)result;
            }
        }

        object IEnumerator.Current => Current;

        public void Dispose()
        {
            if (_onDisposeMethod != null)
                _userObject.CallAsProcedure(_onDisposeMethod.DispatchId, Array.Empty<IValue>(), _process);
        }

        public static RuntimeException IncompatibleInterfaceError()
        {
            var error = new BilingualString(
                "Итератор не соответствует интерфейсу итератора",
                "Iterator doesn't match Iterator interface");

            return new RuntimeException(error);
        }
    }
}