sebastienros.jint/Jint/Native/Set/JsSet.cs

using System.Diagnostics.CodeAnalysis;
using Jint.Native.Object;
using Jint.Runtime;
using Jint.Runtime.Descriptors;

namespace Jint.Native.Set;

internal sealed class JsSet : ObjectInstance
{
    internal readonly OrderedSet<JsValue> _set;

    public JsSet(Engine engine) : this(engine, new OrderedSet<JsValue>(SameValueZeroComparer.Instance))
    {
    }

    public JsSet(Engine engine, OrderedSet<JsValue> set) : base(engine)
    {
        _set = set;
    }

    public override PropertyDescriptor GetOwnProperty(JsValue property)
    {
        if (CommonProperties.Size.Equals(property))
        {
            return new PropertyDescriptor(_set.Count, PropertyFlag.AllForbidden);
        }

        return base.GetOwnProperty(property);
    }

    protected override bool TryGetProperty(JsValue property, [NotNullWhen(true)] out PropertyDescriptor? descriptor)
    {
        if (CommonProperties.Size.Equals(property))
        {
            descriptor = new PropertyDescriptor(_set.Count, PropertyFlag.AllForbidden);
            return true;
        }

        return base.TryGetProperty(property, out descriptor);
    }

    internal void Add(JsValue value)
    {
        _set.Add(value);
    }

    internal void Clear()
    {
        _set.Clear();
    }

    internal bool Has(JsValue key)
    {
        return _set.Contains(key);
    }

    internal bool SetDelete(JsValue key)
    {
        return _set.Remove(key);
    }

    internal void ForEach(ICallable callable, JsValue thisArg)
    {
        var args = _engine._jsValueArrayPool.RentArray(3);
        args[2] = this;

        for (var i = 0; i < _set._list.Count; i++)
        {
            var value = _set._list[i];
            args[0] = value;
            args[1] = value;
            callable.Call(thisArg, args);
        }

        _engine._jsValueArrayPool.ReturnArray(args);
    }

    internal ObjectInstance Entries()
    {
        return _engine.Realm.Intrinsics.SetIteratorPrototype.ConstructEntryIterator(this);
    }

    internal ObjectInstance Values()
    {
        return _engine.Realm.Intrinsics.SetIteratorPrototype.ConstructValueIterator(this);
    }
}