BadInteropEnumerator.cs

Go to the documentation of this file.

using System.Collections;
 
using BadScript2.Runtime.Error;
using BadScript2.Runtime.Interop.Functions;
using BadScript2.Runtime.Objects;
using BadScript2.Runtime.Objects.Types;
using BadScript2.Runtime.Objects.Types.Interface;
 
namespace BadScript2.Runtime.Interop;
 
public class BadInteropEnumerator : BadObject, IBadEnumerator
{
    private static readonly BadClassPrototype s_Prototype = new BadNativeClassPrototype<BadInteropEnumerator>(
        "Enumerator",
        (_, _) => throw new BadRuntimeException("Cannot call method on enumerator"),
        () => new []{(BadInterfacePrototype)BadNativeClassBuilder.Enumerator.CreateGeneric(new []{BadAnyPrototype.Instance})}
    );
 
    private readonly BadObjectReference m_Current;
 
    private readonly IEnumerator<BadObject> m_Enumerator;
 
    private readonly BadObjectReference m_Next;
 
    public BadInteropEnumerator(IEnumerator<BadObject> enumerator)
    {
        m_Enumerator = enumerator;
        BadDynamicInteropFunction next = new BadDynamicInteropFunction(
            "MoveNext",
            _ => m_Enumerator.MoveNext(),
            BadNativeClassBuilder.GetNative("bool")
        );
 
        BadDynamicInteropFunction current = new BadDynamicInteropFunction(
            "GetCurrent",
            _ => m_Enumerator.Current,
            BadAnyPrototype.Instance
        );
 
        m_Next = BadObjectReference.Make("BadArrayEnumerator.MoveNext", () => next);
        m_Current = BadObjectReference.Make("BadArrayEnumerator.GetCurrent", () => current);
    }
 
    public bool MoveNext()
    {
        return m_Enumerator.MoveNext();
    }
 
    public void Reset()
    {
        m_Enumerator.Reset();
    }
 
    public BadObject Current => m_Enumerator.Current!;
 
    object IEnumerator.Current => m_Enumerator.Current!;
 
    public void Dispose()
    {
        m_Enumerator.Dispose();
    }
 
    public override BadClassPrototype GetPrototype()
    {
        return s_Prototype;
    }
 
    public override bool HasProperty(string propName, BadScope? caller = null)
    {
        return propName is "MoveNext" or "GetCurrent" ||
               base.HasProperty(propName, caller);
    }
 
    public override BadObjectReference GetProperty(string propName, BadScope? caller = null)
    {
        switch (propName)
        {
            case "MoveNext":
                return m_Next;
            case "GetCurrent":
                return m_Current;
        }
 
        return base.GetProperty(propName, caller);
    }
 
    public override string ToSafeString(List<BadObject> done)
    {
        return "InteropEnumerator";
    }
}