sebastienros.jint/Jint/Native/Map/MapInstance.cs

using System.Runtime.CompilerServices;
using Jint.Native.Object;
using Jint.Runtime;
using Jint.Runtime.Descriptors;

namespace Jint.Native.Map
{
    public class MapInstance : ObjectInstance
    {
        private readonly Realm _realm;
        internal readonly OrderedDictionary<JsValue, JsValue> _map;

        public MapInstance(Engine engine, Realm realm)
            : base(engine, objectClass: ObjectClass.Map)
        {
            _realm = realm;
            _map = new OrderedDictionary<JsValue, JsValue>();
        }

        public override PropertyDescriptor GetOwnProperty(JsValue property)
        {
            if (property == CommonProperties.Size)
            {
                return new PropertyDescriptor(_map.Count, PropertyFlag.None);
            }

            return base.GetOwnProperty(property);
        }

        protected override bool TryGetProperty(JsValue property, out PropertyDescriptor descriptor)
        {
            if (property == CommonProperties.Size)
            {
                descriptor = new PropertyDescriptor(_map.Count, PropertyFlag.None);
                return true;
            }

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

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

        internal bool Has(JsValue key)
        {
            return _map.ContainsKey(key);
        }

        internal bool MapDelete(JsValue key)
        {
            return _map.Remove(key);
        }

        internal void MapSet(JsValue key, JsValue value)
        {
            _map[key] = value;
        }

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

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

            _engine._jsValueArrayPool.ReturnArray(args);
        }

        internal JsValue MapGet(JsValue key)
        {
            if (!_map.TryGetValue(key, out var value))
            {
                return Undefined;
            }

            return value;
        }

        internal ObjectInstance Iterator()
        {
            return _realm.Intrinsics.Iterator.Construct(this);
        }

        internal ObjectInstance Keys()
        {
            return _realm.Intrinsics.Iterator.Construct(_map.Keys);
        }

        internal ObjectInstance Values()
        {
            return _realm.Intrinsics.Iterator.Construct(_map.Values);
        }

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        internal uint GetSize()
        {
            return (uint) _map.Count;
        }
    }
}