BadModulusAssignExpression.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 BadModulusAssignExpression : BadBinaryExpression
{
    public BadModulusAssignExpression(BadExpression left, BadExpression right, BadSourcePosition position) : base(
        left,
        right,
        position
    ) { }
 
    public static BadObject Modulus(
        BadObjectReference leftRef,
        BadObject left,
        BadObject right,
        BadSourcePosition position,
        string symbol)
    {
        if (left is not IBadNumber lNum || right is not IBadNumber rNum)
        {
            throw new BadRuntimeException($"Can not apply operator '{symbol}' to {left} and {right}", position);
        }
 
        BadObject r = BadObject.Wrap(lNum.Value % rNum.Value);
        leftRef.Set(r);
 
        return r;
    }
 
    public static IEnumerable<BadObject> ModulusWithOverride(
        BadExecutionContext? context,
        BadObjectReference leftRef,
        BadObject right,
        BadSourcePosition position,
        string symbol)
    {
        BadObject left = leftRef.Dereference();
 
        if (left.HasProperty(BadStaticKeys.MODULO_ASSIGN_OPERATOR_NAME, context?.Scope))
        {
            foreach (BadObject o in ExecuteOperatorOverride(
                         left,
                         right,
                         context!,
                         BadStaticKeys.MODULO_ASSIGN_OPERATOR_NAME,
                         position
                     ))
            {
                yield return o;
            }
        }
        else
        {
            yield return Modulus(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();
 
        foreach (BadObject o in ModulusWithOverride(context, leftRef, right, Position, GetSymbol()))
        {
            yield return o;
        }
    }
 
    protected override string GetSymbol()
    {
        return "%=";
    }
}