org.ssssssss.script.MagicResourceLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of magic-script Show documentation
magic script
The newest version!
package org.ssssssss.script;

import org.ssssssss.script.exception.ResourceNotFoundException;
import org.ssssssss.script.functions.DynamicModuleImport;

import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.BiFunction;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 资源加载器
 */
public class MagicResourceLoader {

	/**
	 * 保存自动导入的包路径
	 */
	private static final Set PACKAGES = new HashSet<>();
	/**
	 * 函数加载器
	 */
	private static final List> FUNCTION_LOADERS = new ArrayList<>();
	/**
	 * JSR223 脚本函数加载器
	 */
	private static final List, String, Object>>> SCRIPT_LANGUAGE_LOADERS = new ArrayList<>();
	/**
	 * 保存已注册的模块
	 */
	private static Map MODULES = new ConcurrentHashMap<>();
	/**
	 * 默认的类加载器
	 */
	private static Function classLoader = (className) -> {
		try {
			return Class.forName(className);
		} catch (Exception e) {
			return null;
		}
	};

	static {
		// 默认导入 java.util.* 、java.lang.*
		addPackage("java.util.*");
		addPackage("java.lang.*");
	}

	/**
	 * 获取已注册的模块信息，此方法主要用于代码提示
	 */
	public static Map getModules() {
		return MODULES.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> {
			ScriptClass scriptClass;
			if (entry.getValue() instanceof Class) {
				Class clazz = (Class) entry.getValue();
				scriptClass = MagicScriptEngine.getScriptClassFromClass(clazz);
			} else if (entry.getValue() instanceof DynamicModuleImport) {
				DynamicModuleImport dmi = (DynamicModuleImport) entry.getValue();
				scriptClass = MagicScriptEngine.getScriptClassFromClass(dmi.getTargetClass());
			} else {
				scriptClass = MagicScriptEngine.getScriptClassFromClass(entry.getValue().getClass());
			}
			scriptClass.setModule(true);
			return scriptClass;
		}));
	}

	/**
	 * 添加函数加载器
	 */
	public static void addFunctionLoader(BiFunction functionLoader) {
		FUNCTION_LOADERS.add(functionLoader);
	}

	/**
	 * 设置类加载器
	 */
	public static void setClassLoader(Function classLoader) {
		MagicResourceLoader.classLoader = classLoader;
	}

	/**
	 * 添加模块
	 *
	 * @param moduleName 模块名称
	 * @param target     模块，可以是对象实例，也可以是Class类型的，此时只能使用类中的静态方法
	 */
	public static void addModule(String moduleName, Object target) {
		MODULES.put(moduleName, target);
	}

	/**
	 * 加载模块
	 *
	 * @param moduleName 模块名称
	 */
	public static Object loadModule(MagicScriptContext context, String moduleName) {
		return loadModule(moduleName);
	}

	/**
	 * 加载模块
	 *
	 * @param moduleName 模块名称
	 */
	public static Object loadModule(String moduleName) {
		Object module = MODULES.get(moduleName);
		if (module == null){
			throw new ResourceNotFoundException("找不到模块：" + moduleName);
		}
		return module;
	}

	/**
	 * 加载类
	 *
	 * @param className 类全限定名
	 */
	public static Object loadClass(MagicScriptContext context, String className) {
		return classLoader.apply(className);
	}

	/**
	 * 获取可用的模块列表
	 */
	public static Set getModuleNames() {
		return MODULES.keySet();
	}

	/**
	 * 添加自动导包
	 *
	 * @param prefix 包前缀，如java.lang.*， 不支持 java.lang.**.*
	 */
	public static void addPackage(String prefix) {
		PACKAGES.add(prefix.replace("*", ""));
	}

	/**
	 * 加载类
	 *
	 * @param simpleName 类缩写，如HashMap、ArrayList
	 */
	public static Class findClass(String simpleName) {
		for (String prefix : PACKAGES) {
			try {
				return Class.forName(prefix + simpleName);
			} catch (Exception ignored) {
			}
		}
		return null;
	}

	/**
	 * 添加JSR223 脚本函数加载器
	 */
	public static void addScriptLanguageLoader(Function, String, Object>> loader) {
		SCRIPT_LANGUAGE_LOADERS.add(loader);
	}

	/**
	 * 加载脚本函数加载器
	 *
	 * @param name 脚本名称
	 */
	public static BiFunction, String, Object> loadScriptLanguage(String name) {
		for (Function, String, Object>> languageLoader : SCRIPT_LANGUAGE_LOADERS) {
			try {
				BiFunction, String, Object> function = languageLoader.apply(name);
				if (function != null) {
					return function;
				}
			} catch (Exception ignored) {
			}
		}
		throw new ResourceNotFoundException("找不到语言：" + name);
	}

	/**
	 * 加载函数加载器
	 *
	 * @param name 函数名称
	 */
	public static Object loadFunction(MagicScriptContext context, String name) {
		for (BiFunction loader : FUNCTION_LOADERS) {
			try {
				Object value = loader.apply(context, name);
				if (value != null) {
					return value;
				}
			} catch (Exception ignored) {
			}
		}
		throw new ResourceNotFoundException("找不到函数：" + name);
	}
}