org.truffleruby.language.arguments.ShouldDestructureNode Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ruby-language Show documentation
Core module of Ruby on Truffle
The newest version!
/*
 * Copyright (c) 2013, 2024 Oracle and/or its affiliates. All rights reserved. This
 * code is released under a tri EPL/GPL/LGPL license. You can use it,
 * redistribute it and/or modify it under the terms of the:
 *
 * Eclipse Public License version 2.0, or
 * GNU General Public License version 2, or
 * GNU Lesser General Public License version 2.1.
 */
package org.truffleruby.language.arguments;

import org.truffleruby.language.RubyContextSourceNode;
import org.truffleruby.language.RubyGuards;

import com.oracle.truffle.api.CompilerDirectives;
import com.oracle.truffle.api.frame.VirtualFrame;
import com.oracle.truffle.api.profiles.BranchProfile;
import org.truffleruby.language.RubyNode;
import org.truffleruby.language.dispatch.DispatchNode;

import static org.truffleruby.language.dispatch.DispatchConfiguration.PRIVATE_RETURN_MISSING;

public final class ShouldDestructureNode extends RubyContextSourceNode {

    @Child private DispatchNode respondToToAry;

    private final boolean keywordArguments;
    private final BranchProfile checkIsArrayProfile = BranchProfile.create();

    public ShouldDestructureNode(boolean keywordArguments) {
        this.keywordArguments = keywordArguments;
    }

    @Override
    public Object execute(VirtualFrame frame) {
        if (RubyArguments.getDescriptor(frame) instanceof KeywordArgumentsDescriptor) {
            return false;
        }

        if (RubyArguments.getPositionalArgumentsCount(frame, keywordArguments) != 1) {
            return false;
        }

        checkIsArrayProfile.enter();

        final Object singleArgument = RubyArguments.getArgument(frame, 0);

        if (RubyGuards.isRubyArray(singleArgument)) {
            return true;
        }

        if (respondToToAry == null) {
            CompilerDirectives.transferToInterpreterAndInvalidate();
            respondToToAry = insert(DispatchNode.create());
        }

        var respondToCallResult = respondToToAry.call(PRIVATE_RETURN_MISSING, singleArgument, "respond_to?",
                getLanguage().coreSymbols.TO_ARY);
        // the object may not have the #respond_to? method (e.g. an instance of BasicObject class)
        if (respondToCallResult == DispatchNode.MISSING) {
            return false;
        }

        assert respondToCallResult instanceof Boolean;
        return (boolean) respondToCallResult;
    }

    @Override
    public RubyNode cloneUninitialized() {
        var copy = new ShouldDestructureNode(keywordArguments);
        return copy.copyFlags(this);
    }

}