com.intuit.karate.graal.JsFunction Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of karate-core Show documentation
The newest version!
/*
 * The MIT License
 *
 * Copyright 2023 Karate Labs Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package com.intuit.karate.graal;

import com.intuit.karate.core.ScenarioEngine;
import org.graalvm.polyglot.Value;
import org.graalvm.polyglot.proxy.ProxyExecutable;
import org.graalvm.polyglot.proxy.ProxyInstantiable;
import org.graalvm.polyglot.proxy.ProxyObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 *
 * @author peter
 */
public abstract class JsFunction implements ProxyObject {

    protected static final Logger logger = LoggerFactory.getLogger(JsFunction.class);

    public static final Object LOCK = new Object();

    protected final Value value;

    protected JsFunction(Value v) {
        this.value = v;
    }

    public static ProxyExecutable wrap(Value value) {
        return new Executable(value, true);
    }

    public Value getValue() {
        return value;
    }

    @Override
    public void putMember(String key, Value value) {
        this.value.putMember(key, new JsValue(value).value);
    }

    @Override
    public boolean hasMember(String key) {
        return value.hasMember(key);
    }

    @Override
    public Object getMemberKeys() {
        return value.getMemberKeys().toArray(new String[0]);
    }

    @Override
    public Object getMember(String key) {
        return new JsValue(value.getMember(key)).value;
    }

    @Override
    public boolean removeMember(String key) {
        return value.removeMember(key);
    }

    public static class Executable extends JsFunction implements ProxyExecutable {

        private final boolean lock;
        private final String source;

        protected Executable(Value value) {
            this(value, false);
        }

        protected Executable(Value value, boolean lock) {
            super(value);
            this.lock = lock;
            source = "(" + value.getSourceLocation().getCharacters() + ")";
        }

        public Object execute(JsEngine je, Object... args) {
            Object[] newArgs = new Object[args.length];
            for (int i = 0; i < newArgs.length; i++) {
                newArgs[i] = JsValue.fromJava(args[i]);
            }            
            Value attached = je.evalForValue(source);
            return new JsValue(attached.execute(newArgs)).value;
        }

        @Override
        public Object execute(Value... args) {
            Object[] newArgs = new Object[args.length];
            for (int i = 0; i < newArgs.length; i++) {
                newArgs[i] = JsValue.fromJava(args[i]);
            }
            if (lock) {
                synchronized (LOCK) {
                    return new JsValue(value.execute(newArgs)).value;
                }
            }
            ScenarioEngine se = ScenarioEngine.get();
            JsEngine je = se == null ? null : se.getJsEngine();
            if (je == null || je.context.equals(value.getContext())) {
                return new JsValue(value.execute(newArgs)).value;
            }
            Value attached = je.evalForValue(source);
            return new JsValue(attached.execute(newArgs)).value;
        }

    }

    protected static class Instantiable extends Executable implements ProxyInstantiable {

        protected Instantiable(Value value) {
            super(value);
        }

        @Override
        public Object newInstance(Value... args) {
            Object[] newArgs = new Object[args.length];
            for (int i = 0; i < newArgs.length; i++) {
                newArgs[i] = JsValue.fromJava(args[i]);
            }
            return new JsValue(value.execute(newArgs)).value;
        }

    }

}