Upsilon/Upsilon/BaseTypes/ScriptTable.cs

using System.Collections.Generic;
using System.Linq;
using Upsilon.BaseTypes.ScriptTypeInterfaces;
using Upsilon.Binder;
using Upsilon.Evaluator;
using Upsilon.Text;

namespace Upsilon.BaseTypes
{
    internal class ScriptTable : ScriptType, IIndexable, IScopeOwner, IIterable
    {
        public EvaluationScope EvaluationScope { get; }

        public ScriptTable(EvaluationScope scope)
        {
            EvaluationScope = scope;
        }

        public override Type Type => Type.Table;
        public override object ToCSharpObject()
        {
            return EvaluationScope.Variables.ToDictionary(x => x.Key, x => x.Value.ToCSharpObject());
        }

        public override System.Type GetCSharpType()
        {
            return typeof(Dictionary<string, object>);
        }

        public ScriptType Get(Diagnostics diagnostics, TextSpan span, ScriptType index, EvaluationScope scope)
        {
            var s = index.ToString();
            if (EvaluationScope.TryGet(s, out var o))
            {
                return o;
            }
            diagnostics.LogError("Cannot find member 's' on Table", span);
            return new ScriptNull();
        }

        public void Set(Diagnostics diagnostics, TextSpan span, ScriptType index, ScriptType value)
        {
            var s = index.ToString();
            EvaluationScope.Set(new VariableSymbol(s, value.Type, false), value, true);
        }

        public IEnumerator<(string Key, ScriptType value)> GetEnumerator()
        {
            return Enumerator();
        }

        private IEnumerator<(string Key, ScriptType value)> Enumerator()
        {
            foreach (var variable in EvaluationScope.Variables)
            {
                yield return (variable.Key, variable.Value);
            }
        }
    }
}