ch.obermuhlner.scriptengine.java.execution.DefaultExecutionStrategy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-scriptengine Show documentation

Java script engine for Java as a scripting language.

There is a newer version: 2.0.0

package ch.obermuhlner.scriptengine.java.execution;

import javax.script.ScriptException;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Supplier;

public class DefaultExecutionStrategy implements ExecutionStrategy {

    private final Method method;

    public DefaultExecutionStrategy(Class clazz) {
        method = findCallableMethod(clazz);
    }

    @Override
    public Object execute(Object instance) throws ScriptException {
        if (instance == null) {
            return null;
        }

        if (instance instanceof Supplier) {
            Supplier supplier = (Supplier) instance;
            return supplier.get();
        }

        if (instance instanceof Runnable) {
            Runnable runnable = (Runnable) instance;
            runnable.run();
            return null;
        }

        if (method != null) {
            try {
                return method.invoke(instance);
            } catch (IllegalAccessException | InvocationTargetException e) {
                throw new ScriptException(e);
            }
        }

        throw new ScriptException("No method found to execute instance of type: " + instance.getClass());
    }

    private static Method findCallableMethod(Class clazz) {
        List callableMethods = new ArrayList<>();
        for (Method method : clazz.getDeclaredMethods()) {
            int modifiers = method.getModifiers();
            if (method.getParameterCount() == 0 && Modifier.isPublic(modifiers)) {
                callableMethods.add(method);
            }
        }

        if (callableMethods.size() == 1) {
            return callableMethods.get(0);
        }

        return null;
    }
}