BadTableExpression.cs

Go to the documentation of this file.

using BadScript2.Common;
using BadScript2.Optimizations.Folding;
using BadScript2.Reader.Token;
using BadScript2.Runtime;
using BadScript2.Runtime.Objects;
 
namespace BadScript2.Parser.Expressions.Constant;
 
public class BadTableExpression : BadExpression
{
    private readonly Dictionary<BadWordToken, BadExpression> m_Table;
 
    public BadTableExpression(Dictionary<BadWordToken, BadExpression> table, BadSourcePosition position) : base(
        false,
        position
    )
    {
        m_Table = table;
    }
 
    public int Length => m_Table.Count;
 
    public IDictionary<BadWordToken, BadExpression> Table => m_Table;
 
    public override IEnumerable<BadExpression> GetDescendants()
    {
        return m_Table.SelectMany(kvp => kvp.Value.GetDescendantsAndSelf());
    }
 
    public override void Optimize()
    {
        KeyValuePair<BadWordToken, BadExpression>[] branches = m_Table.ToArray();
        m_Table.Clear();
 
        foreach (KeyValuePair<BadWordToken, BadExpression> branch in branches)
        {
            m_Table[branch.Key] = BadConstantFoldingOptimizer.Optimize(branch.Value);
        }
    }
 
    protected override IEnumerable<BadObject> InnerExecute(BadExecutionContext context)
    {
        Dictionary<string, BadObject> table = new Dictionary<string, BadObject>();
 
        foreach (KeyValuePair<BadWordToken, BadExpression> entry in m_Table)
        {
            BadObject value = BadObject.Null;
 
            foreach (BadObject o in entry.Value.Execute(context))
            {
                value = o;
 
                yield return o;
            }
 
            value = value.Dereference();
 
            table[entry.Key.Text] = value;
        }
 
        yield return new BadTable(table);
    }
}