BadFormattedStringExpression.cs

Go to the documentation of this file.

using BadScript2.Common;
using BadScript2.Optimizations.Folding;
using BadScript2.Parser.Expressions.Constant;
using BadScript2.Runtime;
using BadScript2.Runtime.Objects;
 
namespace BadScript2.Parser.Expressions.Variables;
 
public class BadFormattedStringExpression : BadStringExpression
{
    private readonly BadExpression[] m_Expressions;
 
    public BadFormattedStringExpression(BadExpression[] exprs, string str, BadSourcePosition position) : base(
        str,
        position
    )
    {
        m_Expressions = exprs;
    }
 
    public IEnumerable<BadExpression> Expressions => m_Expressions;
 
    public int ExpressionCount => m_Expressions.Length;
 
    public override void Optimize()
    {
        for (int i = 0; i < m_Expressions.Length; i++)
        {
            m_Expressions[i] = BadConstantFoldingOptimizer.Optimize(m_Expressions[i]);
        }
    }
 
    protected override IEnumerable<BadObject> InnerExecute(BadExecutionContext context)
    {
        if (m_Expressions.Length == 0)
        {
            yield return Value;
 
            yield break;
        }
 
        List<BadObject> objs = new List<BadObject>();
 
        foreach (BadExpression expr in m_Expressions)
        {
            BadObject obj = BadObject.Null;
 
            foreach (BadObject o in expr.Execute(context))
            {
 
                obj = o;
 
                yield return o;
            }
 
            objs.Add(obj.Dereference());
        }
 
        yield return string.Format(Value, objs.Cast<object?>().ToArray());
    }
 
    public override IEnumerable<BadExpression> GetDescendants()
    {
        foreach (BadExpression expr in m_Expressions)
        {
            foreach (BadExpression e in expr.GetDescendantsAndSelf())
            {
                yield return e;
            }
        }
 
        foreach (BadExpression e in base.GetDescendants())
        {
            yield return e;
        }
    }
}