org.jruby.ir.instructions.calladapter.AttrAssignCallAdapter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jruby

There is a newer version: 9.4.9.0

package org.jruby.ir.instructions.calladapter;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.jruby.RubyArray;
import org.jruby.ir.operands.Operand;
import org.jruby.ir.operands.Splat;
import org.jruby.javasupport.util.RuntimeHelpers;
import org.jruby.runtime.Block;
import org.jruby.runtime.CallSite;
import org.jruby.runtime.CallType;
import org.jruby.runtime.DynamicScope;
import org.jruby.runtime.ThreadContext;
import org.jruby.runtime.builtin.IRubyObject;

/**
 * AttrAssign is weird.  self[i] = 1 is treated as a functional call and 
 * it also returns no result. 
 */
public class AttrAssignCallAdapter extends CallAdapter {
    private String name;
    private Operand[] args;
    
    public AttrAssignCallAdapter(CallSite callSite, String name, Operand[] args) {
        super(callSite);
        
        this.args = args;
        this.name = name;
    }

    @Override
    public Object call(ThreadContext context, IRubyObject self, IRubyObject receiver, DynamicScope currDynScope, Object[] temp) {
        IRubyObject[] values = prepareArguments(context, self, args, currDynScope, temp);
        
        if (callSite == null) {
            CallType callType = self == receiver ? CallType.FUNCTIONAL : CallType.NORMAL;
            RuntimeHelpers.invoke(context, receiver, name, values, callType, Block.NULL_BLOCK);
        } else {
            callSite.call(context, self, receiver, values);
        }
        
        return null;
    }
    
    protected IRubyObject[] prepareArguments(ThreadContext context, IRubyObject self, Operand[] args, DynamicScope currDynScope, Object[] temp) {
        List argList = new ArrayList();
        for (int i = 0; i < args.length; i++) {
            IRubyObject rArg = (IRubyObject) args[i].retrieve(context, self, currDynScope, temp);
            if (args[i] instanceof Splat) {
                argList.addAll(Arrays.asList(((RubyArray) rArg).toJavaArray()));
            } else {
                argList.add(rArg);
            }
        }

        return argList.toArray(new IRubyObject[argList.size()]);
    } 
}