BadAddAssignExpression.cs

Go to the documentation of this file.

using BadScript2.Common;
using BadScript2.Runtime;
using BadScript2.Runtime.Error;
using BadScript2.Runtime.Objects;
using BadScript2.Runtime.Objects.Native;
 
namespace BadScript2.Parser.Expressions.Binary.Math.Assign;
 
public class BadAddAssignExpression : BadBinaryExpression
{
    public BadAddAssignExpression(BadExpression left, BadExpression right, BadSourcePosition position) : base(left,
                                                                                                              right,
                                                                                                              position
                                                                                                             ) { }
 
 
    public static BadObject Add(BadObjectReference leftRef,
                                BadObject left,
                                BadObject right,
                                BadSourcePosition position,
                                string symbol)
    {
        switch (left)
        {
            case IBadString lStr:
            {
                if (right is not IBadNative rNative)
                {
                    throw new BadRuntimeException($"Can not apply operator '{symbol}' to {left} and {right}", position);
                }
 
                BadObject r = BadObject.Wrap(lStr.Value + rNative.Value);
                leftRef.Set(r, position);
 
                return r;
            }
            case IBadNumber lNum when right is IBadString rStr:
            {
                BadObject r = BadObject.Wrap(lNum.Value + rStr.Value);
                leftRef.Set(r, position);
 
                return r;
            }
            case IBadNumber lNum when right is IBadNumber rNum:
            {
                BadObject r = BadObject.Wrap(lNum.Value + rNum.Value);
                leftRef.Set(r, position);
 
                return r;
            }
            case IBadBoolean lBool when right is IBadString rStr:
            {
                BadObject r = BadObject.Wrap(lBool.Value + rStr.Value);
                leftRef.Set(r, position);
 
                return r;
            }
            default:
                throw new BadRuntimeException($"Can not apply operator '{symbol}' to {left} and {right}", position);
        }
    }
 
    public static IEnumerable<BadObject> AddWithOverride(BadExecutionContext? context,
                                                         BadObjectReference leftRef,
                                                         BadObject right,
                                                         BadSourcePosition position,
                                                         string symbol)
    {
        BadObject left = leftRef.Dereference(position);
 
        if (left.HasProperty(BadStaticKeys.ADD_ASSIGN_OPERATOR_NAME, context?.Scope))
        {
            foreach (BadObject o in ExecuteOperatorOverride(left,
                                                            right,
                                                            context!,
                                                            BadStaticKeys.ADD_ASSIGN_OPERATOR_NAME,
                                                            position
                                                           ))
            {
                yield return o;
            }
        }
        else
        {
            yield return Add(leftRef, left, right, position, symbol);
        }
    }
 
    protected override IEnumerable<BadObject> InnerExecute(BadExecutionContext context)
    {
        BadObject left = BadObject.Null;
 
        foreach (BadObject o in Left.Execute(context))
        {
            left = o;
 
            yield return o;
        }
 
        if (left is not BadObjectReference leftRef)
        {
            throw new BadRuntimeException($"Left side of {GetSymbol()} must be a reference", Position);
        }
 
        BadObject right = BadObject.Null;
 
        foreach (BadObject o in Right.Execute(context))
        {
            right = o;
 
            yield return o;
        }
 
        right = right.Dereference(Position);
 
        foreach (BadObject o in AddWithOverride(context, leftRef, right, Position, GetSymbol()))
        {
            yield return o;
        }
    }
 
    protected override string GetSymbol()
    {
        return "+=";
    }
}